source: trunk/imports/G2phase_GPX.py @ 2049

# -*- coding: utf-8 -*-
########### SVN repository information ###################
# $Date: 2015-04-27 18:22:06 +0000 (Mon, 27 Apr 2015) $
# $Author: toby $
# $Revision: 1812 $
# $URL: trunk/imports/G2phase_GPX.py $
# $Id: G2phase_GPX.py 1812 2015-04-27 18:22:06Z toby $
########### SVN repository information ###################
'''
*Module G2phase_GPX: Import phase from GSAS-II project*
--------------------------------------------------------

Copies a phase from another GSAS-II project file into the
current project.

'''
import sys
import cPickle
import random as ran
import GSASIIIO as G2IO
import GSASIIstrIO as G2stIO
import GSASIIpath
GSASIIpath.SetVersionNumber("$Revision: 1812 $")

class PhaseReaderClass(G2IO.ImportPhase):
    'Opens a .GPX file and pulls out a selected phase'
    def __init__(self):
        super(self.__class__,self).__init__( # fancy way to say ImportPhase.__init__
            extensionlist=('.gpx',),
            strictExtension=True,
            formatName = 'GSAS-II gpx',
            longFormatName = 'GSAS-II project (.gpx file) import'
            )
        
    def ContentsValidator(self, filepointer):
        "Test if the 1st section can be read as a cPickle block, if not it can't be .GPX!"
        try: 
            cPickle.load(filepointer)
        except:
            self.errors = 'This is not a valid .GPX file. Not recognized by cPickle'
            return False
        return True

    def Reader(self,filename,filepointer, ParentFrame=None, **unused):
        '''Read a phase from a .GPX file. Does not (yet?) support selecting and reading
        more than one phase at a time.'''
        try:
            phasenames = G2stIO.GetPhaseNames(filename)
        except:
            self.errors = 'Reading of phase names failed'
            return False
        if not phasenames:
            self.errors = 'No phases found in '+str(filename)
            return False            # no blocks with coordinates
        elif len(phasenames) == 1: # one block, no choices
            selblk = 0
        else:                       # choose from options                
            selblk = self.PhaseSelector(
                phasenames,
                ParentFrame=ParentFrame,
                title= 'Select a phase from the list below',
                )
            if selblk is None:
                self.errors = 'No phase selected'
                return False # User pressed cancel
        try:
            self.Phase = G2stIO.GetAllPhaseData(filename,phasenames[selblk])
            self.Phase['Histograms'] = {}       #remove any histograms
            self.Phase['Pawley ref'] = []       # & any Pawley refl.
            self.Phase['ranId'] = ran.randint(0,sys.maxint)
            return True
        except Exception as detail:
            self.errors = 'Error reading selected phase'
            self.errors += '\n  '+str(detail)
            print self.formatName+' error:',detail # for testing
            print sys.exc_info()[0] # for testing
            import traceback
            print traceback.format_exc()