def raw(run,name,start=50,end=100,size=512): p = PelFile(basedir+"%04i/" % run + name+".pel") mon = MonFile(basedir+"%04i/" % run + name+ ".pel.txt",False) val = np.sum(p.make3d(size)[:,:,start:end],axis=2) print val.shape spectrum_total = np.sum(mon.spec) return val/spectrum_total,np.sqrt(val)/spectrum_total
def get_2d_int(run,name): p = PelFile(basedir+"SESAME_%i/" % run + name+"_neutron_event.dat") mon = MonFile(basedir+"SESAME_%i/" % run + name+"_bmon_histo.dat",False) return np.sum(p.make3d(),axis=2) / np.sum(mon.spec)
import __future__ from reader import PelFile import glob import numpy as np import sys files = glob.glob(sys.argv[1]) print(files) files = [x for x in files if x[-3:]=="pel"] for file in files: data = PelFile(file) hist = data.make3d() np.save(file+".npy",hist) print(file)