60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
|
import logging as log
|
||
|
|
log.basicConfig(level=log.INFO)
|
||
|
|
|
||
|
|
import os
|
||
|
|
import collections
|
||
|
|
import numpy as np
|
||
|
|
from .xray import azav
|
||
|
|
|
||
|
|
def _conv(x):
|
||
|
|
try:
|
||
|
|
x = float(x)
|
||
|
|
except:
|
||
|
|
x = np.nan
|
||
|
|
return x
|
||
|
|
|
||
|
|
def getFiles(folder,basename="*.edf*"):
|
||
|
|
files = glob.glob(folder + "/" + basename)
|
||
|
|
files.sort()
|
||
|
|
return files
|
||
|
|
|
||
|
|
def getEdfFiles(folder):
|
||
|
|
return getFiles(folder,basename="*.edf*")
|
||
|
|
|
||
|
|
def removeExt(fname):
|
||
|
|
""" special remove extension meant to work with compressed files.edf and .edf.gz files """
|
||
|
|
if fname[-3:] == ".gz": fname = fname[-3:]
|
||
|
|
return os.path.splitext(fname)[0]
|
||
|
|
|
||
|
|
|
||
|
|
def readDelayFromDiagnostic(fname):
|
||
|
|
""" return an ordered dict dictionary of filename; for each key a rounded
|
||
|
|
value of delay is associated """
|
||
|
|
if os.path.isdir(fname): fname += "/diagnostics.log"
|
||
|
|
data = np.genfromtxt(fname,usecols=(2,3),\
|
||
|
|
dtype=None,converters={3: lambda x: _conv(x)},
|
||
|
|
names = ['fname','delay'])
|
||
|
|
files = data['fname'].astype(np.str); # to avoid encoding problems
|
||
|
|
delays = data['delay']
|
||
|
|
# skip lines that cannot be interpreted as float (like done, etc)
|
||
|
|
idx_ok = np.isfinite( delays )
|
||
|
|
files = files[idx_ok]
|
||
|
|
delays = delays[idx_ok]
|
||
|
|
delays = np.round(delays.astype(float),12)
|
||
|
|
return collections.OrderedDict( zip(files,delays) )
|
||
|
|
|
||
|
|
|
||
|
|
def doFolder_azav(folder,nQ=1500,force=False,mask=None,saveChi=True,
|
||
|
|
poni='auto',h5File='auto'):
|
||
|
|
""" very small wrapper around azav.doFolder, essentially just reading
|
||
|
|
the diagnostics.log """
|
||
|
|
|
||
|
|
if h5File == 'auto': n5File = folder + "/" + "pyfai_1d.h5"
|
||
|
|
|
||
|
|
diag = dict( delays = readDelayFromDiagnostic(folder) )
|
||
|
|
|
||
|
|
return azav.doFolder(folder,files="*.edf*",nQ=nQ,force=force,mask=mask,
|
||
|
|
saveChi=saveChi,poni=poni,h5File=h5File,diagnostic=diag)
|
||
|
|
|
||
|
|
|