def get_results(loadfile, savefile, v):
"""
Get all the important numbers for each event.
(basic info:)
Etot
Edep
Etrack
alpha_true
beta_true
errorcode
file
ind
(ridge following result:)
alpha_rf
(moments result:)
alpha_m
(general rejection parameters:)
min_end_energy
max_end_energy
n_ends
(moments rejection parameters:)
phi
edge_pixels
edge_segments
(Monte Carlo rejection parameters:)
overlap flag
wrong_end flag
early_scatter flag
total_scatter_angle
"""
progressflag = not file_vars()[0]
pn = 'pix10_5noise15'
HTname = 'python HT v1.52'
MTname = 'moments v1.0'
vprint(
v, 1, 'Starting {} at {} with {}% mem usage'.format(
loadfile, time.ctime(),
psutil.virtual_memory().percent))
datalen = 1000
# basic info
energy_tot_kev = np.nan * np.ones(datalen)
energy_dep_kev = np.nan * np.ones(datalen)
energy_track_kev = np.nan * np.ones(datalen)
alpha_true_deg = np.nan * np.ones(datalen)
beta_true_deg = np.nan * np.ones(datalen)
trk_errorcode = np.nan * np.ones(datalen)
cl_errorcode = np.nan * np.ones(datalen)
mom_errorcode = np.nan * np.ones(datalen)
filename = ['' for _ in xrange(datalen)]
fileind = np.nan * np.ones(datalen)
# ridge following
alpha_ridge_deg = np.nan * np.ones(datalen)
# moments
alpha_moments_deg = np.nan * np.ones(datalen)
# rejection
min_end_energy_kev = np.nan * np.ones(datalen)
max_end_energy_kev = np.nan * np.ones(datalen)
n_ends = np.nan * np.ones(datalen)
phi_deg = np.nan * np.ones(datalen)
edge_pixels = np.nan * np.ones(datalen)
edge_segments = np.nan * np.ones(datalen)
# monte carlo
overlap_flag = np.nan * np.ones(datalen)
wrong_end_flag = np.nan * np.ones(datalen)
early_scatter_flag = np.nan * np.ones(datalen)
total_scatter_angle_deg = np.nan * np.ones(datalen)
try:
with h5py.File(loadfile, 'r',
driver='core') as f, h5py.File(savefile,
'a',
driver='core') as h5save:
if progressflag:
pbar = progressbar.ProgressBar(widgets=[
progressbar.Percentage(), ' ',
progressbar.Bar(), ' ',
progressbar.ETA()
],
maxval=datalen)
pbar.start()
for ind in xrange(datalen):
indstr = '{:05d}'.format(ind)
trkstr = indstr
clstr = 'cl_' + indstr
momstr = 'mom_' + indstr
this_trk_errorcode = 0
this_cl_errorcode = 0
this_mom_errorcode = 0
filename[ind] = os.path.split(loadfile)[-1]
fileind[ind] = ind
# track and algorithm alpha's
try:
this_trk = Track.from_hdf5(f[trkstr][pn])
except KeyError:
# could be an MException from MATLAB
this_trk_errorcode = 10
trk_errorcode[ind] = this_trk_errorcode
continue
except trackio.InterfaceError:
read_errorcode = f[trkstr][pn].attrs['errorcode']
if read_errorcode > 0:
# multiplicity event
this_trk_errorcode = read_errorcode
trk_errorcode[ind] = this_trk_errorcode
continue
else:
raise
energy_tot_kev[ind] = this_trk.g4track.energy_tot_kev
energy_dep_kev[ind] = this_trk.g4track.energy_dep_kev
energy_track_kev[ind] = this_trk.energy_kev
alpha_true_deg[ind] = this_trk.g4track.alpha_deg
beta_true_deg[ind] = this_trk.g4track.beta_deg
alpha_ridge_deg[ind] = this_trk[HTname].alpha_deg
alpha_moments_deg[ind] = this_trk[MTname].alpha_deg
# classifier
this_cl = cl.Classifier.from_hdf5(f[clstr][pn])
min_end_energy_kev[ind] = this_cl.min_end_energy
max_end_energy_kev[ind] = this_cl.max_end_energy
n_ends[ind] = this_cl.n_ends
overlap_flag[ind] = this_cl.overlap
wrong_end_flag[ind] = this_cl.wrong_end
if this_cl.error == 'TrackTooShortError':
this_cl_errorcode = 8
else:
early_scatter_flag[ind] = this_cl.early_scatter
total_scatter_angle_deg[ind] = (
this_cl.total_scatter_angle / np.pi * 180)
# moments info
try:
this_mom = tm.MomentsReconstruction.from_hdf5(
f[momstr][pn])
except trackio.InterfaceError:
if f[momstr][pn].attrs['errorcode'] == 4:
# no ends found
this_mom_errorcode = 4
else:
raise
else:
if this_mom.error == 'CheckSegmentBoxError':
this_mom_errorcode = 9
elif this_mom.error == 'what the heck happened?':
this_mom.errorcode = 11
elif this_mom.error == 'Rotation angle conditions not met':
this_mom.errorcode = 12
else:
phi_deg[ind] = this_mom.phi / np.pi * 180
edge_pixels[ind] = this_mom.edge_pixel_count
edge_segments[ind] = this_mom.edge_pixel_segments
trk_errorcode[ind] = this_trk_errorcode
cl_errorcode[ind] = this_cl_errorcode
mom_errorcode[ind] = this_mom_errorcode
if progressflag:
pbar.update(ind)
# save to file
# basic info
h5save.create_dataset('energy_tot_kev',
shape=(datalen, ),
data=energy_tot_kev)
h5save.create_dataset('energy_dep_kev',
shape=(datalen, ),
data=energy_dep_kev)
h5save.create_dataset('energy_track_kev',
shape=(datalen, ),
data=energy_track_kev)
h5save.create_dataset('alpha_true_deg',
shape=(datalen, ),
data=alpha_true_deg)
h5save.create_dataset('beta_true_deg',
shape=(datalen, ),
data=beta_true_deg)
h5save.create_dataset('trk_errorcode',
shape=(datalen, ),
data=trk_errorcode,
dtype='f2')
h5save.create_dataset('cl_errorcode',
shape=(datalen, ),
data=cl_errorcode,
dtype='f2')
h5save.create_dataset('mom_errorcode',
shape=(datalen, ),
data=mom_errorcode,
dtype='f2')
h5save.create_dataset('filename',
shape=(datalen, ),
data=filename,
dtype=h5py.special_dtype(vlen=str))
h5save.create_dataset('fileind',
shape=(datalen, ),
data=fileind,
dtype='f4')
# ridge following
h5save.create_dataset('alpha_ridge_deg',
shape=(datalen, ),
data=alpha_ridge_deg)
# moments
h5save.create_dataset('alpha_moments_deg',
shape=(datalen, ),
data=alpha_moments_deg)
# rejection parameters
h5save.create_dataset('min_end_energy_kev',
shape=(datalen, ),
data=min_end_energy_kev)
h5save.create_dataset('max_end_energy_kev',
shape=(datalen, ),
data=max_end_energy_kev)
h5save.create_dataset('n_ends',
shape=(datalen, ),
data=n_ends,
dtype='f4')
h5save.create_dataset('phi_deg', shape=(datalen, ), data=phi_deg)
h5save.create_dataset('edge_pixels',
shape=(datalen, ),
data=edge_pixels,
dtype='f4')
h5save.create_dataset('edge_segments',
shape=(datalen, ),
data=edge_segments,
dtype='f2')
# Monte Carlo
h5save.create_dataset('overlap_flag',
shape=(datalen, ),
data=overlap_flag,
dtype='f2')
h5save.create_dataset('wrong_end_flag',
shape=(datalen, ),
data=wrong_end_flag,
dtype='f2')
h5save.create_dataset('early_scatter_flag',
shape=(datalen, ),
data=early_scatter_flag,
dtype='f2')
h5save.create_dataset('total_scatter_angle_deg',
shape=(datalen, ),
data=total_scatter_angle_deg)
if progressflag:
pbar.finish()
except NotImplementedError:
pass
def get_results(loadfile, savefile, v):
"""
Get all the important numbers for each event.
(basic info:)
Etot
Edep
Etrack
alpha_true
beta_true
errorcode
file
ind
(ridge following result:)
alpha_rf
(moments result:)
alpha_m
(general rejection parameters:)
min_end_energy
max_end_energy
n_ends
(moments rejection parameters:)
phi
edge_pixels
edge_segments
(Monte Carlo rejection parameters:)
overlap flag
wrong_end flag
early_scatter flag
total_scatter_angle
"""
progressflag = not file_vars()[0]
pn = 'pix10_5noise15'
HTname = 'python HT v1.52'
MTname = 'moments v1.0'
vprint(v, 1, 'Starting {} at {} with {}% mem usage'.format(
loadfile, time.ctime(), psutil.virtual_memory().percent))
datalen = 1000
# basic info
energy_tot_kev = np.nan * np.ones(datalen)
energy_dep_kev = np.nan * np.ones(datalen)
energy_track_kev = np.nan * np.ones(datalen)
alpha_true_deg = np.nan * np.ones(datalen)
beta_true_deg = np.nan * np.ones(datalen)
trk_errorcode = np.nan * np.ones(datalen)
cl_errorcode = np.nan * np.ones(datalen)
mom_errorcode = np.nan * np.ones(datalen)
filename = ['' for _ in xrange(datalen)]
fileind = np.nan * np.ones(datalen)
# ridge following
alpha_ridge_deg = np.nan * np.ones(datalen)
# moments
alpha_moments_deg = np.nan * np.ones(datalen)
# rejection
min_end_energy_kev = np.nan * np.ones(datalen)
max_end_energy_kev = np.nan * np.ones(datalen)
n_ends = np.nan * np.ones(datalen)
phi_deg = np.nan * np.ones(datalen)
edge_pixels = np.nan * np.ones(datalen)
edge_segments = np.nan * np.ones(datalen)
# monte carlo
overlap_flag = np.nan * np.ones(datalen)
wrong_end_flag = np.nan * np.ones(datalen)
early_scatter_flag = np.nan * np.ones(datalen)
total_scatter_angle_deg = np.nan * np.ones(datalen)
try:
with h5py.File(loadfile, 'r', driver='core') as f, h5py.File(
savefile, 'a', driver='core') as h5save:
if progressflag:
pbar = progressbar.ProgressBar(
widgets=[progressbar.Percentage(), ' ',
progressbar.Bar(), ' ',
progressbar.ETA()], maxval=datalen)
pbar.start()
for ind in xrange(datalen):
indstr = '{:05d}'.format(ind)
trkstr = indstr
clstr = 'cl_' + indstr
momstr = 'mom_' + indstr
this_trk_errorcode = 0
this_cl_errorcode = 0
this_mom_errorcode = 0
filename[ind] = os.path.split(loadfile)[-1]
fileind[ind] = ind
# track and algorithm alpha's
try:
this_trk = Track.from_hdf5(f[trkstr][pn])
except KeyError:
# could be an MException from MATLAB
this_trk_errorcode = 10
trk_errorcode[ind] = this_trk_errorcode
continue
except trackio.InterfaceError:
read_errorcode = f[trkstr][pn].attrs['errorcode']
if read_errorcode > 0:
# multiplicity event
this_trk_errorcode = read_errorcode
trk_errorcode[ind] = this_trk_errorcode
continue
else:
raise
energy_tot_kev[ind] = this_trk.g4track.energy_tot_kev
energy_dep_kev[ind] = this_trk.g4track.energy_dep_kev
energy_track_kev[ind] = this_trk.energy_kev
alpha_true_deg[ind] = this_trk.g4track.alpha_deg
beta_true_deg[ind] = this_trk.g4track.beta_deg
alpha_ridge_deg[ind] = this_trk[HTname].alpha_deg
alpha_moments_deg[ind] = this_trk[MTname].alpha_deg
# classifier
this_cl = cl.Classifier.from_hdf5(f[clstr][pn])
min_end_energy_kev[ind] = this_cl.min_end_energy
max_end_energy_kev[ind] = this_cl.max_end_energy
n_ends[ind] = this_cl.n_ends
overlap_flag[ind] = this_cl.overlap
wrong_end_flag[ind] = this_cl.wrong_end
if this_cl.error == 'TrackTooShortError':
this_cl_errorcode = 8
else:
early_scatter_flag[ind] = this_cl.early_scatter
total_scatter_angle_deg[ind] = (
this_cl.total_scatter_angle / np.pi * 180)
# moments info
try:
this_mom = tm.MomentsReconstruction.from_hdf5(
f[momstr][pn])
except trackio.InterfaceError:
if f[momstr][pn].attrs['errorcode'] == 4:
# no ends found
this_mom_errorcode = 4
else:
raise
else:
if this_mom.error == 'CheckSegmentBoxError':
this_mom_errorcode = 9
elif this_mom.error == 'what the heck happened?':
this_mom.errorcode = 11
elif this_mom.error == 'Rotation angle conditions not met':
this_mom.errorcode = 12
else:
phi_deg[ind] = this_mom.phi / np.pi * 180
edge_pixels[ind] = this_mom.edge_pixel_count
edge_segments[ind] = this_mom.edge_pixel_segments
trk_errorcode[ind] = this_trk_errorcode
cl_errorcode[ind] = this_cl_errorcode
mom_errorcode[ind] = this_mom_errorcode
if progressflag:
pbar.update(ind)
# save to file
# basic info
h5save.create_dataset(
'energy_tot_kev', shape=(datalen,), data=energy_tot_kev)
h5save.create_dataset(
'energy_dep_kev', shape=(datalen,), data=energy_dep_kev)
h5save.create_dataset(
'energy_track_kev', shape=(datalen,), data=energy_track_kev)
h5save.create_dataset(
'alpha_true_deg', shape=(datalen,), data=alpha_true_deg)
h5save.create_dataset(
'beta_true_deg', shape=(datalen,), data=beta_true_deg)
h5save.create_dataset(
'trk_errorcode', shape=(datalen,), data=trk_errorcode,
dtype='f2')
h5save.create_dataset(
'cl_errorcode', shape=(datalen,), data=cl_errorcode,
dtype='f2')
h5save.create_dataset(
'mom_errorcode', shape=(datalen,), data=mom_errorcode,
dtype='f2')
h5save.create_dataset(
'filename', shape=(datalen,), data=filename,
dtype=h5py.special_dtype(vlen=str))
h5save.create_dataset(
'fileind', shape=(datalen,), data=fileind,
dtype='f4')
# ridge following
h5save.create_dataset(
'alpha_ridge_deg', shape=(datalen,), data=alpha_ridge_deg)
# moments
h5save.create_dataset(
'alpha_moments_deg', shape=(datalen,), data=alpha_moments_deg)
# rejection parameters
h5save.create_dataset(
'min_end_energy_kev', shape=(datalen,),
data=min_end_energy_kev)
h5save.create_dataset(
'max_end_energy_kev', shape=(datalen,),
data=max_end_energy_kev)
h5save.create_dataset(
'n_ends', shape=(datalen,), data=n_ends,
dtype='f4')
h5save.create_dataset(
'phi_deg', shape=(datalen,), data=phi_deg)
h5save.create_dataset(
'edge_pixels', shape=(datalen,), data=edge_pixels,
dtype='f4')
h5save.create_dataset(
'edge_segments', shape=(datalen,), data=edge_segments,
dtype='f2')
# Monte Carlo
h5save.create_dataset(
'overlap_flag', shape=(datalen,), data=overlap_flag,
dtype='f2')
h5save.create_dataset(
'wrong_end_flag', shape=(datalen,), data=wrong_end_flag,
dtype='f2')
h5save.create_dataset(
'early_scatter_flag', shape=(datalen,),
data=early_scatter_flag, dtype='f2')
h5save.create_dataset(
'total_scatter_angle_deg', shape=(datalen,),
data=total_scatter_angle_deg)
if progressflag:
pbar.finish()
except NotImplementedError:
pass
