Changeset 2297 for trunk/imports/G2img_GE.py

Timestamp:

May 27, 2016 11:25:21 AM (6 years ago)

Author:

vondreele

Message:

implement creation of summed file from multi-frame ge image files; creates a G2img file which is put on tree in place of multi frame file. Subsequent accesses get teg G2img file.
Insert skipPageChange = True to prevent plot tab change causing tree change from Strain plots & Image plots

File:

• trunk/imports/G2img_GE.py (modified) (6 diffs)

trunk/imports/G2img_GE.py

@@ -60 +60 @@
         if imagenum is None: imagenum = 1
         self.Comments,self.Data,self.Npix,self.Image,more = \
-            GetGEsumData(filename,imagenum=imagenum)
+            GetGEsumData(self,filename,imagenum=imagenum)
         if self.Npix == 0 or not self.Comments:
             return False
@@ -82 +82 @@
             extensionlist=('.ge1','.ge2','.ge3','.ge4'),
             strictExtension=True,
-            formatName = 'GE multi-image summed',
+            formatName = 'sum GE multi-image',
             longFormatName = 'sum of GE multi-image file'
             )
@@ -104 +104 @@
         if imagenum is None: imagenum = 1
         self.Comments,self.Data,self.Npix,self.Image,more = \
-            GetGEsumData(filename,imagenum=imagenum,sum=True)
+            GetGEsumData(self,filename,imagenum=imagenum,sum=True)
         if self.Npix == 0 or not self.Comments:
             return False
@@ -112 +112 @@
         return True
 
-def GetGEsumData(filename,imagenum=1,sum=False):
+def GetGEsumData(self,filename,imagenum=1,sum=False):
     '''Read G.E. detector images from various files as produced at 1-ID and
     with Detector Pool detector. Also sums multiple image files if desired
@@ -118 +118 @@
     import struct as st
     import array as ar
+    import cPickle
     more = False
     File = open(filename,'rb')
@@ -154 +155 @@
         head += ['file: '+filename+' image #'+str(imagenum),]
         if sum:    #will ignore imagenum
+            print 'Frames to read %d'%(nframes),
             while nframes > 1: #OK, this will sum the frames.
-                image += np.array(ar.array('H',File.read(2*Npix)),dtype=np.int32)
+                try:
+                    image += np.array(ar.array('H',File.read(2*Npix)),dtype=np.int32)
+                except ValueError:
+                    break
                 nframes -= 1
+                print '%d'%(nframes),
+            print ''   
             more = False
+            filename = os.path.splitext(filename)[0]+'.G2img'
+            File = open(filename,'wb')
+            Data = {'pixelSize':[200,200],'wavelength':0.15,'distance':250.0,'center':[204.8,204.8],'size':sizexy}
+            image = np.reshape(image,(sizexy[1],sizexy[0]))
+            cPickle.dump([head,Data,Npix,image],File,1)
+            File.close()
+            self.sumfile = filename
+            self.formatName = 'Summed GSAS-II image'
+            sum = False
     image = np.reshape(image,(sizexy[1],sizexy[0]))
     data = {'pixelSize':[200,200],'wavelength':0.15,'distance':250.0,'center':[204.8,204.8],'size':sizexy}