from pyFAI.azimuthalIntegrator import AzimuthalIntegrator
if __name__ == "__main__":
paramFile = None
processFile = []
for param in sys.argv[1:]:
if param.startswith("-p="):
paramFile = param.split("=", 1)[1]
elif os.path.isfile(param):
processFile.append(param)
if paramFile and processFile:
integrator = AzimuthalIntegrator()
integrator.setChiDiscAtZero()
integrator.load(paramFile)
print integrator
for oneFile in processFile:
sys.stdout.write("Integrating %s --> " % oneFile)
outFile = os.path.splitext(oneFile)[0] + ".xy"
azimFile = os.path.splitext(oneFile)[0] + ".azim"
data = fabio.open(oneFile).data.astype("float32")
t0 = time.time()
tth, I = integrator.xrpd_halfSplitPixel(data=data, nbPt=1495, filename=outFile, correctSolidAngle=False)
t1 = time.time()
# integrator.xrpd2(data, 1000, 360, azimFile)
print "%s\t 1D took %.3fs, 2D took %.3fs" % (outFile, t1 - t0, time.time() - t1)
print "raw int: %s ; integrated: %s " % (data.sum() / data.size, I.sum())
else:
print ("Usage:")
print (" $ python integrate.py -p=param.poni file.edf file2.edf file3.edf")
