def testResidualsRunner(self): x = map(lambda x: x + gauss(0,0.005 + 0.3*x*x), arange(-1,1,0.05)) y = map(lambda x: x + gauss(0,0.005 + 0.3*x*x), arange(-1,1,0.05)) z = map(lambda x: x + gauss(0,0.005 + 0.3*x*x), arange(-1,1,0.05)) line = array(zip(x,y,z)) lpc = LPCImpl(h = 0.2, convergence_at = 0.0001, it = 100, mult = 5) lpc_curve = lpc.lpc(X=line) residuals_calc = LPCResiduals(line, tube_radius = 0.15) residuals_runner = LPCResidualsRunner(lpc.getCurve(), residuals_calc) residuals_runner.setTauRange([0.05, 0.07]) residuals = residuals_runner.calculateResiduals() pprint(residuals)
def LPCEfficiencyCalculator(curves, data_range, particle_hits, tau):
#first rescale the curves then calculate the proton and muon efficiency
if data_range is None:
data_range = 1.0
muon_hits = array([[h[0][0], h[0][1], h[0][2]] for h in particle_hits[13]]) / data_range
proton_hits = array([[h[0][0], h[0][1], h[0][2]] for h in particle_hits[2212]]) / data_range
residuals_calc = LPCResiduals(muon_hits, tube_radius = 0.12)
residuals_runner = LPCResidualsRunner(curves, residuals_calc)
residuals_runner.setTauRange([tau])
muon_residuals = residuals_runner.calculateResiduals(True, False, False)
muon_coverage = max([len(muon_residuals['curve_residuals'][i]['coverage_indices'][tau]) for i in range(len(curves))])
muon_efficiency = float(muon_coverage) / len(muon_hits)
residuals_calc.setDataPoints(proton_hits)
proton_residuals = residuals_runner.calculateResiduals(True, False, False)
proton_coverage = max([len(proton_residuals['curve_residuals'][i]['coverage_indices'][tau]) for i in range(len(curves))])
proton_efficiency = float(proton_coverage) / len(proton_hits)
print 'residuals'
if len(sys.argv) != 2:
raise ValueError, 'Must supply the name of a configuration file'
proc = lpcProcessor(sys.argv[1])
proc.runProcessor()
metadata_filename = proc.getMetadataFilename()
lpc_loader = lpcAnalysisPickleReader(metadata_filename)
#toyLPCPlotter(evt_curves['Xi'], evt_curves['lpc_curve'])
out_data = []
while 1:
try:
evt = lpc_loader.getEvent()
#pprint(truth_evt.getParticlesInVoxelDict())
#now calcualte the residuals
residuals_calc = LPCResiduals(evt[0]['Xi'], tube_radius = 3.0)
residuals_runner = LPCResidualsRunner(evt[0]['lpc_curve'], residuals_calc)
residuals_runner.setTauRange([2.0])
pruner = lpcCurvePruner(residuals_runner, closeness_threshold = 5.0, path_length_threshold = 10.0)
remaining_curves = pruner.pruneCurves()
tau = 2.0
#muon_proton_hits = truth_evt.getParticleHits([13, 2212])
#eff = LPCEfficiencyCalculator(remaining_curves, evt['data_range'], muon_proton_hits, tau)
voxel_to_pdg_dictionary = evt[1].getParticlesInVoxelDict()
pur = LPCPurityCalculator(remaining_curves, evt[0]['data_range'], voxel_to_pdg_dictionary, tau)
out_data.append({'voxel_dict': voxel_to_pdg_dictionary, 'pur': pur})
print 'breakpoint'
except EOFError:
break
outfile = open('/tmp/purity_data.pkl', 'w')
