Changeset 2817 for trunk/imports/G2img_MAR.py

Timestamp:

May 2, 2017 10:03:41 AM (6 years ago)

Author:

vondreele

Message:

major revision - move all importers to GSASIIobj & make them independent of wx so they can be used in a scripting environment.
Still to move are PhaseSelector?, and 3 BlockSelector? dialogs

File:

• trunk/imports/G2img_MAR.py (modified) (2 diffs)

trunk/imports/G2img_MAR.py

@@ -10 +10 @@
 *Module G2img_MAR: MAR image files*
 --------------------------------------
-
 '''
 
-import sys
-import os
-import GSASIIIO as G2IO
+import GSASIIobj as G2obj
 import GSASIIpath
+import numpy as np
 GSASIIpath.SetVersionNumber("$Revision$")
-class MAR_ReaderClass(G2IO.ImportImage):
+class MAR_ReaderClass(G2obj.ImportImage):
     '''Routine to read several MAR formats, .mar3450,.mar2300,.mar2560
     '''
@@ -35 +33 @@
         
     def Reader(self,filename,filepointer, ParentFrame=None, **unused):
-        '''Read using Bob's routine :func:`GSASIIIO.GetMAR345Data`
-        (to be moved to this file, eventually)
-        '''
-        self.Comments,self.Data,self.Npix,self.Image = G2IO.GetMAR345Data(filename)
+        self.Comments,self.Data,self.Npix,self.Image = GetMAR345Data(filename)
         if self.Npix == 0 or not self.Comments:
             return False
         self.LoadImage(ParentFrame,filename)
         return True
+
+def GetMAR345Data(filename,imageOnly=False):
+    'Read a MAR-345 image plate image'
+    try:
+        import pack_f as pf
+    except:
+        print '**** ERROR - Unable to load the GSAS MAR image decompression, pack_f'
+        return None,None,None,None
+
+    if not imageOnly:
+        print 'Read Mar345 file: ',filename
+    File = open(filename,'rb')
+    head = File.read(4095)
+    lines = head[128:].split('\n')
+    head = []
+    for line in lines:
+        line = line.strip()
+        if 'PIXEL' in line:
+            values = line.split()
+            pixel = (int(values[2]),int(values[4]))     #in microns
+        elif 'WAVELENGTH' in line:
+            wave = float(line.split()[1])
+        elif 'DISTANCE' in line:
+            distance = float(line.split()[1])           #in mm
+            if not distance:
+                distance = 500.
+        elif 'CENTER' in line:
+            values = line.split()
+            center = [float(values[2])/10.,float(values[4])/10.]    #make in mm from pixels
+        if line: 
+            head.append(line)
+    data = {'pixelSize':pixel,'wavelength':wave,'distance':distance,'center':center}
+    for line in head:
+        if 'FORMAT' in line[0:6]:
+            items = line.split()
+            sizex = int(items[1])
+            Npix = int(items[3])
+            sizey = int(Npix/sizex)
+    pos = 4096
+    data['size'] = [sizex,sizey]
+    File.seek(pos)
+    line = File.read(8)
+    while 'CCP4' not in line:       #get past overflow list for now
+        line = File.read(8)
+        pos += 8
+    pos += 37
+    File.seek(pos)
+    raw = File.read()
+    File.close()
+    image = np.zeros(shape=(sizex,sizey),dtype=np.int32)
+    
+    image = np.flipud(pf.pack_f(len(raw),raw,sizex,sizey,image).T)  #transpose to get it right way around & flip
+    if imageOnly:
+        return image
+    else:
+        return head,data,Npix,image
+