print("Wire thickness :",
shipPatRec_prev.ship_geo.strawtubes.WireThickness)
print("Straw pitch :",
shipPatRec_prev.ship_geo.strawtubes.StrawPitch)
print("z-offset between layers:",
shipPatRec_prev.ship_geo.strawtubes.DeltazLayer)
print("z-offset between planes:",
shipPatRec_prev.ship_geo.strawtubes.DeltazPlane)
print("Straws per layer :",
shipPatRec_prev.ship_geo.strawtubes.StrawsPerLayer)
print("z-offset between planes:",
shipPatRec_prev.ship_geo.strawtubes.DeltazPlane)
print("z-offset between views :",
shipPatRec_prev.ship_geo.strawtubes.DeltazView)
shipPatRec_prev.initialize(fgeo)
def EventLoop(SmearedHits):
#loop over events
for n in range(nEvents):
fittedtrackids = []
SmearedHits.Delete()
fGenFitArray_PR.Delete()
fitTrack2MC_PR.clear()
fPartArray_PR.Delete()
rc = sTree.GetEvent(n)
if shipPatRec_prev.monitor == True:
shipPatRec_prev.ReconstructibleMCTracks = shipPatRec_prev.getReconstructibleTracks(
if shipPatRec_prev.debug==1:
print "Straw tracker geometry parameters (cm)"
print "--------------------------------------"
print "Strawlength :",2*shipPatRec_prev.ship_geo.strawtubes.StrawLength
print "Inner straw diameter :",shipPatRec_prev.ship_geo.strawtubes.InnerStrawDiameter
print "Straw wall thickness :",shipPatRec_prev.ship_geo.strawtubes.WallThickness
print "Outer straw diameter :",shipPatRec_prev.ship_geo.strawtubes.OuterStrawDiameter
print "Wire thickness :",shipPatRec_prev.ship_geo.strawtubes.WireThickness
print "Straw pitch :",shipPatRec_prev.ship_geo.strawtubes.StrawPitch
print "z-offset between layers:",shipPatRec_prev.ship_geo.strawtubes.DeltazLayer
print "z-offset between planes:",shipPatRec_prev.ship_geo.strawtubes.DeltazPlane
print "Straws per layer :",shipPatRec_prev.ship_geo.strawtubes.StrawsPerLayer
print "z-offset between planes:",shipPatRec_prev.ship_geo.strawtubes.DeltazPlane
print "z-offset between views :",shipPatRec_prev.ship_geo.strawtubes.DeltazView
shipPatRec_prev.initialize(fgeo)
def EventLoop(SmearedHits):
#loop over events
for n in range(nEvents):
fittedtrackids=[]
SmearedHits.Delete()
fGenFitArray_PR.Delete()
fitTrack2MC_PR.clear()
fPartArray_PR.Delete()
rc = sTree.GetEvent(n)
if shipPatRec_prev.monitor==True:
shipPatRec_prev.ReconstructibleMCTracks=shipPatRec_prev.getReconstructibleTracks(n,sTree,sGeo)
if len(shipPatRec_prev.ReconstructibleMCTracks)!=shipPatRec_prev.reconstructiblerequired :
def __init__(self,fout,fgeo):
self.fn = ROOT.TFile.Open(fout,'update')
self.sTree = self.fn.cbmsim
if self.sTree.GetBranch("FitTracks"):
print "remove RECO branches and rerun reconstruction"
self.fn.Close()
# make a new file without reco branches
f = ROOT.TFile(fout)
sTree = f.cbmsim
if sTree.GetBranch("FitTracks"): sTree.SetBranchStatus("FitTracks",0)
if sTree.GetBranch("goodTracks"): sTree.SetBranchStatus("goodTracks",0)
if sTree.GetBranch("VetoHitOnTrack"): sTree.SetBranchStatus("VetoHitOnTrack",0)
if sTree.GetBranch("Particles"): sTree.SetBranchStatus("Particles",0)
if sTree.GetBranch("fitTrack2MC"): sTree.SetBranchStatus("fitTrack2MC",0)
if sTree.GetBranch("EcalClusters"): sTree.SetBranchStatus("EcalClusters",0)
if sTree.GetBranch("EcalReconstructed"): sTree.SetBranchStatus("EcalReconstructed",0)
if sTree.GetBranch("Pid"): sTree.SetBranchStatus("Pid",0)
if sTree.GetBranch("Digi_StrawtubesHits"): sTree.SetBranchStatus("Digi_StrawtubesHits",0)
if sTree.GetBranch("Digi_SBTHits"): sTree.SetBranchStatus("Digi_SBTHits",0)
if sTree.GetBranch("digiSBT2MC"): sTree.SetBranchStatus("digiSBT2MC",0)
if sTree.GetBranch("Digi_TimeDetHits"): sTree.SetBranchStatus("Digi_TimeDetHits",0)
if sTree.GetBranch("Digi_MuonHits"): sTree.SetBranchStatus("Digi_MuonHits",0)
rawFile = fout.replace("_rec.root","_raw.root")
recf = ROOT.TFile(rawFile,"recreate")
newTree = sTree.CloneTree(0)
for n in range(sTree.GetEntries()):
sTree.GetEntry(n)
rc = newTree.Fill()
sTree.Clear()
newTree.AutoSave()
f.Close()
recf.Close()
os.system('cp '+rawFile +' '+fout)
self.fn = ROOT.TFile(fout,'update')
self.sTree = self.fn.cbmsim
# check that all containers are present, otherwise create dummy version
self.dummyContainers={}
branch_class = {"vetoPoint":"vetoPoint","ShipRpcPoint":"ShipRpcPoint","TargetPoint":"TargetPoint",\
"strawtubesPoint":"strawtubesPoint","EcalPointLite":"ecalPoint","HcalPointLite":"hcalPoint",\
"splitcalPoint":"splitcalPoint","TimeDetPoint":"TimeDetPoint","muonPoint":"muonPoint"}
for x in branch_class:
if not self.sTree.GetBranch(x):
self.dummyContainers[x+"_array"] = ROOT.TClonesArray(branch_class[x])
self.dummyContainers[x] = self.sTree.Branch(x,self.dummyContainers[x+"_array"],32000,-1)
setattr(self.sTree,x,self.dummyContainers[x+"_array"])
self.dummyContainers[x].Fill()
#
if self.sTree.GetBranch("GeoTracks"): self.sTree.SetBranchStatus("GeoTracks",0)
# prepare for output
# event header
self.header = ROOT.FairEventHeader()
self.eventHeader = self.sTree.Branch("ShipEventHeader",self.header,32000,-1)
# fitted tracks
self.fGenFitArray = ROOT.TClonesArray("genfit::Track")
self.fGenFitArray.BypassStreamer(ROOT.kFALSE)
self.fitTrack2MC = ROOT.std.vector('int')()
self.goodTracksVect = ROOT.std.vector('int')()
self.mcLink = self.sTree.Branch("fitTrack2MC",self.fitTrack2MC,32000,-1)
self.fitTracks = self.sTree.Branch("FitTracks", self.fGenFitArray,32000,-1)
self.goodTracksBranch = self.sTree.Branch("goodTracks",self.goodTracksVect,32000,-1)
self.fTrackletsArray = ROOT.TClonesArray("Tracklet")
self.Tracklets = self.sTree.Branch("Tracklets", self.fTrackletsArray,32000,-1)
#
self.digiStraw = ROOT.TClonesArray("strawtubesHit")
self.digiStrawBranch = self.sTree.Branch("Digi_StrawtubesHits",self.digiStraw,32000,-1)
self.digiSBT = ROOT.TClonesArray("vetoHit")
self.digiSBTBranch=self.sTree.Branch("Digi_SBTHits",self.digiSBT,32000,-1)
self.vetoHitOnTrackArray = ROOT.TClonesArray("vetoHitOnTrack")
self.vetoHitOnTrackBranch=self.sTree.Branch("VetoHitOnTrack",self.vetoHitOnTrackArray,32000,-1)
self.digiSBT2MC = ROOT.std.vector('std::vector< int >')()
self.mcLinkSBT = self.sTree.Branch("digiSBT2MC",self.digiSBT2MC,32000,-1)
self.digiTimeDet = ROOT.TClonesArray("TimeDetHit")
self.digiTimeDetBranch=self.sTree.Branch("Digi_TimeDetHits",self.digiTimeDet,32000,-1)
self.digiMuon = ROOT.TClonesArray("muonHit")
self.digiMuonBranch=self.sTree.Branch("Digi_muonHits",self.digiMuon,32000,-1)
# for the digitizing step
self.v_drift = modules["Strawtubes"].StrawVdrift()
self.sigma_spatial = modules["Strawtubes"].StrawSigmaSpatial()
# setup ecal reconstruction
self.caloTasks = []
if self.sTree.GetBranch("EcalPoint"):
# Creates. exports and fills calorimeter structure
dflag = 0
if debug: dflag = 10
ecalGeo = ecalGeoFile+'z'+str(ShipGeo.ecal.z)+".geo"
if not ecalGeo in os.listdir(os.environ["FAIRSHIP"]+"/geometry"): shipDet_conf.makeEcalGeoFile(ShipGeo.ecal.z,ShipGeo.ecal.File)
ecalFiller=ROOT.ecalStructureFiller("ecalFiller", dflag,ecalGeo)
ecalFiller.SetUseMCPoints(ROOT.kTRUE)
ecalFiller.StoreTrackInformation()
self.caloTasks.append(ecalFiller)
#GeV -> ADC conversion
ecalDigi=ROOT.ecalDigi("ecalDigi",0)
self.caloTasks.append(ecalDigi)
#ADC -> GeV conversion
ecalPrepare=ROOT.ecalPrepare("ecalPrepare",0)
self.caloTasks.append(ecalPrepare)
# Maximums locator
ecalMaximumFind=ROOT.ecalMaximumLocator("maximumFinder",dflag)
self.caloTasks.append(ecalMaximumFind)
# Cluster calibration
ecalClusterCalib=ROOT.ecalClusterCalibration("ecalClusterCalibration", 0)
#4x4 cm cells
ecalCl3PhS=ROOT.TFormula("ecalCl3PhS", "[0]+x*([1]+x*([2]+x*[3]))")
ecalCl3PhS.SetParameters(6.77797e-04, 5.75385e+00, 3.42690e-03, -1.16383e-04)
ecalClusterCalib.SetStraightCalibration(3, ecalCl3PhS)
ecalCl3Ph=ROOT.TFormula("ecalCl3Ph", "[0]+x*([1]+x*([2]+x*[3]))+[4]*x*y+[5]*x*y*y")
ecalCl3Ph.SetParameters(0.000750975, 5.7552, 0.00282783, -8.0025e-05, -0.000823651, 0.000111561)
ecalClusterCalib.SetCalibration(3, ecalCl3Ph)
#6x6 cm cells
ecalCl2PhS=ROOT.TFormula("ecalCl2PhS", "[0]+x*([1]+x*([2]+x*[3]))")
ecalCl2PhS.SetParameters(8.14724e-04, 5.67428e+00, 3.39030e-03, -1.28388e-04)
ecalClusterCalib.SetStraightCalibration(2, ecalCl2PhS)
ecalCl2Ph=ROOT.TFormula("ecalCl2Ph", "[0]+x*([1]+x*([2]+x*[3]))+[4]*x*y+[5]*x*y*y")
ecalCl2Ph.SetParameters(0.000948095, 5.67471, 0.00339177, -0.000122629, -0.000169109, 8.33448e-06)
ecalClusterCalib.SetCalibration(2, ecalCl2Ph)
self.caloTasks.append(ecalClusterCalib)
# Cluster finder
ecalClusterFind=ROOT.ecalClusterFinder("clusterFinder",dflag)
self.caloTasks.append(ecalClusterFind)
# Calorimeter reconstruction
ecalReco=ROOT.ecalReco('ecalReco',0)
self.caloTasks.append(ecalReco)
# Match reco to MC
ecalMatch=ROOT.ecalMatch('ecalMatch',0)
self.caloTasks.append(ecalMatch)
if EcalDebugDraw:
# ecal drawer: Draws calorimeter structure, incoming particles, clusters, maximums
ecalDrawer=ROOT.ecalDrawer("clusterFinder",10)
self.caloTasks.append(ecalDrawer)
# add pid reco
import shipPid
self.caloTasks.append(shipPid.Task(self))
# prepare vertexing
self.Vertexing = shipVertex.Task(h,self.sTree)
# setup random number generator
self.random = ROOT.TRandom()
ROOT.gRandom.SetSeed(13)
self.PDG = ROOT.TDatabasePDG.Instance()
# access ShipTree
self.sTree.GetEvent(0)
if len(self.caloTasks)>0:
print "** initialize Calo reconstruction **"
self.ecalStructure = ecalFiller.InitPython(self.sTree.EcalPointLite)
ecalDigi.InitPython(self.ecalStructure)
ecalPrepare.InitPython(self.ecalStructure)
self.ecalMaximums = ecalMaximumFind.InitPython(self.ecalStructure)
self.ecalCalib = ecalClusterCalib.InitPython()
self.ecalClusters = ecalClusterFind.InitPython(self.ecalStructure, self.ecalMaximums, self.ecalCalib)
self.EcalClusters = self.sTree.Branch("EcalClusters",self.ecalClusters,32000,-1)
self.ecalReconstructed = ecalReco.InitPython(self.sTree.EcalClusters, self.ecalStructure, self.ecalCalib)
self.EcalReconstructed = self.sTree.Branch("EcalReconstructed",self.ecalReconstructed,32000,-1)
ecalMatch.InitPython(self.ecalStructure, self.ecalReconstructed, self.sTree.MCTrack)
if EcalDebugDraw: ecalDrawer.InitPython(self.sTree.MCTrack, self.sTree.EcalPoint, self.ecalStructure, self.ecalClusters)
else:
ecalClusters = ROOT.TClonesArray("ecalCluster")
ecalReconstructed = ROOT.TClonesArray("ecalReconstructed")
self.EcalClusters = self.sTree.Branch("EcalClusters",ecalClusters,32000,-1)
self.EcalReconstructed = self.sTree.Branch("EcalReconstructed",ecalReconstructed,32000,-1)
#
# init geometry and mag. field
gMan = ROOT.gGeoManager
self.geoMat = ROOT.genfit.TGeoMaterialInterface()
#
self.bfield = ROOT.genfit.FairShipFields()
self.fM = ROOT.genfit.FieldManager.getInstance()
self.fM.init(self.bfield)
ROOT.genfit.MaterialEffects.getInstance().init(self.geoMat)
# init fitter, to be done before importing shipPatRec
#fitter = ROOT.genfit.KalmanFitter()
#fitter = ROOT.genfit.KalmanFitterRefTrack()
self.fitter = ROOT.genfit.DAF()
self.fitter.setMaxIterations(50)
if debug: self.fitter.setDebugLvl(1) # produces lot of printout
#set to True if "real" pattern recognition is required also
if debug == True: shipPatRec.debug = 1
# for 'real' PatRec
shipPatRec.initialize(fgeo)
def __init__(self,fout,fgeo):
self.fn = ROOT.TFile.Open(fout,'update')
self.sTree = self.fn.cbmsim
if self.sTree.GetBranch("FitTracks"):
print "remove RECO branches and rerun reconstruction"
self.fn.Close()
# make a new file without reco branches
f = ROOT.TFile(fout)
sTree = f.cbmsim
if sTree.GetBranch("FitTracks"): sTree.SetBranchStatus("FitTracks",0)
if sTree.GetBranch("goodTracks"): sTree.SetBranchStatus("goodTracks",0)
if sTree.GetBranch("VetoHitOnTrack"): sTree.SetBranchStatus("VetoHitOnTrack",0)
if sTree.GetBranch("Particles"): sTree.SetBranchStatus("Particles",0)
if sTree.GetBranch("fitTrack2MC"): sTree.SetBranchStatus("fitTrack2MC",0)
if sTree.GetBranch("EcalClusters"): sTree.SetBranchStatus("EcalClusters",0)
if sTree.GetBranch("EcalReconstructed"): sTree.SetBranchStatus("EcalReconstructed",0)
if sTree.GetBranch("Pid"): sTree.SetBranchStatus("Pid",0)
if sTree.GetBranch("Digi_StrawtubesHits"): sTree.SetBranchStatus("Digi_StrawtubesHits",0)
if sTree.GetBranch("Digi_SBTHits"): sTree.SetBranchStatus("Digi_SBTHits",0)
if sTree.GetBranch("digiSBT2MC"): sTree.SetBranchStatus("digiSBT2MC",0)
if sTree.GetBranch("Digi_TimeDetHits"): sTree.SetBranchStatus("Digi_TimeDetHits",0)
if sTree.GetBranch("Digi_MuonHits"): sTree.SetBranchStatus("Digi_MuonHits",0)
rawFile = fout.replace("_rec.root","_raw.root")
recf = ROOT.TFile(rawFile,"recreate")
newTree = sTree.CloneTree(0)
for n in range(sTree.GetEntries()):
sTree.GetEntry(n)
rc = newTree.Fill()
sTree.Clear()
newTree.AutoSave()
f.Close()
recf.Close()
os.system('cp '+rawFile +' '+fout)
self.fn = ROOT.TFile(fout,'update')
self.sTree = self.fn.cbmsim
# check that all containers are present, otherwise create dummy version
self.dummyContainers={}
branch_class = {"vetoPoint":"vetoPoint","ShipRpcPoint":"ShipRpcPoint","TargetPoint":"TargetPoint",\
"strawtubesPoint":"strawtubesPoint","EcalPointLite":"ecalPoint",\
"TimeDetPoint":"TimeDetPoint","muonPoint":"muonPoint"}
for x in branch_class:
if not self.sTree.GetBranch(x):
self.dummyContainers[x+"_array"] = ROOT.TClonesArray(branch_class[x])
self.dummyContainers[x] = self.sTree.Branch(x,self.dummyContainers[x+"_array"],32000,-1)
setattr(self.sTree,x,self.dummyContainers[x+"_array"])
self.dummyContainers[x].Fill()
#
if self.sTree.GetBranch("GeoTracks"): self.sTree.SetBranchStatus("GeoTracks",0)
# prepare for output
# event header
self.header = ROOT.FairEventHeader()
self.eventHeader = self.sTree.Branch("ShipEventHeader",self.header,32000,-1)
# fitted tracks
self.fGenFitArray = ROOT.TClonesArray("genfit::Track")
self.fGenFitArray.BypassStreamer(ROOT.kFALSE)
self.fitTrack2MC = ROOT.std.vector('int')()
self.goodTracksVect = ROOT.std.vector('int')()
self.mcLink = self.sTree.Branch("fitTrack2MC",self.fitTrack2MC,32000,-1)
self.fitTracks = self.sTree.Branch("FitTracks", self.fGenFitArray,32000,-1)
self.goodTracksBranch = self.sTree.Branch("goodTracks",self.goodTracksVect,32000,-1)
self.fTrackletsArray = ROOT.TClonesArray("Tracklet")
self.Tracklets = self.sTree.Branch("Tracklets", self.fTrackletsArray,32000,-1)
#
self.digiStraw = ROOT.TClonesArray("strawtubesHit")
self.digiStrawBranch = self.sTree.Branch("Digi_StrawtubesHits",self.digiStraw,32000,-1)
self.digiSBT = ROOT.TClonesArray("vetoHit")
self.digiSBTBranch=self.sTree.Branch("Digi_SBTHits",self.digiSBT,32000,-1)
self.vetoHitOnTrackArray = ROOT.TClonesArray("vetoHitOnTrack")
self.vetoHitOnTrackBranch=self.sTree.Branch("VetoHitOnTrack",self.vetoHitOnTrackArray,32000,-1)
self.digiSBT2MC = ROOT.std.vector('std::vector< int >')()
self.mcLinkSBT = self.sTree.Branch("digiSBT2MC",self.digiSBT2MC,32000,-1)
self.digiTimeDet = ROOT.TClonesArray("TimeDetHit")
self.digiTimeDetBranch=self.sTree.Branch("Digi_TimeDetHits",self.digiTimeDet,32000,-1)
self.digiMuon = ROOT.TClonesArray("muonHit")
self.digiMuonBranch=self.sTree.Branch("Digi_muonHits",self.digiMuon,32000,-1)
# for the digitizing step
self.v_drift = modules["Strawtubes"].StrawVdrift()
self.sigma_spatial = modules["Strawtubes"].StrawSigmaSpatial()
# optional if present, splitcalCluster
if self.sTree.GetBranch("splitcalPoint"):
self.digiSplitcal = ROOT.TClonesArray("splitcalHit")
self.digiSplitcalBranch=self.sTree.Branch("Digi_SplitcalHits",self.digiSplitcal,32000,-1)
self.recoSplitcal = ROOT.TClonesArray("splitcalCluster")
self.recoSplitcalBranch=self.sTree.Branch("Reco_SplitcalClusters",self.recoSplitcal,32000,-1)
# setup ecal reconstruction
self.caloTasks = []
if self.sTree.GetBranch("EcalPoint") and not self.sTree.GetBranch("splitcalPoint"):
# Creates. exports and fills calorimeter structure
dflag = 0
if debug: dflag = 10
ecalGeo = ecalGeoFile+'z'+str(ShipGeo.ecal.z)+".geo"
if not ecalGeo in os.listdir(os.environ["FAIRSHIP"]+"/geometry"): shipDet_conf.makeEcalGeoFile(ShipGeo.ecal.z,ShipGeo.ecal.File)
ecalFiller=ROOT.ecalStructureFiller("ecalFiller", dflag,ecalGeo)
ecalFiller.SetUseMCPoints(ROOT.kTRUE)
ecalFiller.StoreTrackInformation()
self.caloTasks.append(ecalFiller)
#GeV -> ADC conversion
ecalDigi=ROOT.ecalDigi("ecalDigi",0)
self.caloTasks.append(ecalDigi)
#ADC -> GeV conversion
ecalPrepare=ROOT.ecalPrepare("ecalPrepare",0)
self.caloTasks.append(ecalPrepare)
# Maximums locator
ecalMaximumFind=ROOT.ecalMaximumLocator("maximumFinder",dflag)
self.caloTasks.append(ecalMaximumFind)
# Cluster calibration
ecalClusterCalib=ROOT.ecalClusterCalibration("ecalClusterCalibration", 0)
#4x4 cm cells
ecalCl3PhS=ROOT.TFormula("ecalCl3PhS", "[0]+x*([1]+x*([2]+x*[3]))")
ecalCl3PhS.SetParameters(6.77797e-04, 5.75385e+00, 3.42690e-03, -1.16383e-04)
ecalClusterCalib.SetStraightCalibration(3, ecalCl3PhS)
ecalCl3Ph=ROOT.TFormula("ecalCl3Ph", "[0]+x*([1]+x*([2]+x*[3]))+[4]*x*y+[5]*x*y*y")
ecalCl3Ph.SetParameters(0.000750975, 5.7552, 0.00282783, -8.0025e-05, -0.000823651, 0.000111561)
ecalClusterCalib.SetCalibration(3, ecalCl3Ph)
#6x6 cm cells
ecalCl2PhS=ROOT.TFormula("ecalCl2PhS", "[0]+x*([1]+x*([2]+x*[3]))")
ecalCl2PhS.SetParameters(8.14724e-04, 5.67428e+00, 3.39030e-03, -1.28388e-04)
ecalClusterCalib.SetStraightCalibration(2, ecalCl2PhS)
ecalCl2Ph=ROOT.TFormula("ecalCl2Ph", "[0]+x*([1]+x*([2]+x*[3]))+[4]*x*y+[5]*x*y*y")
ecalCl2Ph.SetParameters(0.000948095, 5.67471, 0.00339177, -0.000122629, -0.000169109, 8.33448e-06)
ecalClusterCalib.SetCalibration(2, ecalCl2Ph)
self.caloTasks.append(ecalClusterCalib)
# Cluster finder
ecalClusterFind=ROOT.ecalClusterFinder("clusterFinder",dflag)
self.caloTasks.append(ecalClusterFind)
# Calorimeter reconstruction
ecalReco=ROOT.ecalReco('ecalReco',0)
self.caloTasks.append(ecalReco)
# Match reco to MC
ecalMatch=ROOT.ecalMatch('ecalMatch',0)
self.caloTasks.append(ecalMatch)
if EcalDebugDraw:
# ecal drawer: Draws calorimeter structure, incoming particles, clusters, maximums
ecalDrawer=ROOT.ecalDrawer("clusterFinder",10)
self.caloTasks.append(ecalDrawer)
# add pid reco
import shipPid
self.caloTasks.append(shipPid.Task(self))
# prepare vertexing
self.Vertexing = shipVertex.Task(h,self.sTree)
# setup random number generator
self.random = ROOT.TRandom()
ROOT.gRandom.SetSeed(13)
self.PDG = ROOT.TDatabasePDG.Instance()
# access ShipTree
self.sTree.GetEvent(0)
if len(self.caloTasks)>0:
print "** initialize Calo reconstruction **"
self.ecalStructure = ecalFiller.InitPython(self.sTree.EcalPointLite)
ecalDigi.InitPython(self.ecalStructure)
ecalPrepare.InitPython(self.ecalStructure)
self.ecalMaximums = ecalMaximumFind.InitPython(self.ecalStructure)
self.ecalCalib = ecalClusterCalib.InitPython()
self.ecalClusters = ecalClusterFind.InitPython(self.ecalStructure, self.ecalMaximums, self.ecalCalib)
self.EcalClusters = self.sTree.Branch("EcalClusters",self.ecalClusters,32000,-1)
self.ecalReconstructed = ecalReco.InitPython(self.sTree.EcalClusters, self.ecalStructure, self.ecalCalib)
self.EcalReconstructed = self.sTree.Branch("EcalReconstructed",self.ecalReconstructed,32000,-1)
ecalMatch.InitPython(self.ecalStructure, self.ecalReconstructed, self.sTree.MCTrack)
if EcalDebugDraw: ecalDrawer.InitPython(self.sTree.MCTrack, self.sTree.EcalPoint, self.ecalStructure, self.ecalClusters)
else:
ecalClusters = ROOT.TClonesArray("ecalCluster")
ecalReconstructed = ROOT.TClonesArray("ecalReconstructed")
self.EcalClusters = self.sTree.Branch("EcalClusters",ecalClusters,32000,-1)
self.EcalReconstructed = self.sTree.Branch("EcalReconstructed",ecalReconstructed,32000,-1)
#
# init geometry and mag. field
gMan = ROOT.gGeoManager
self.geoMat = ROOT.genfit.TGeoMaterialInterface()
#
self.bfield = ROOT.genfit.FairShipFields()
self.bfield.setField(fieldMaker.getGlobalField())
self.fM = ROOT.genfit.FieldManager.getInstance()
self.fM.init(self.bfield)
ROOT.genfit.MaterialEffects.getInstance().init(self.geoMat)
# init fitter, to be done before importing shipPatRec
#fitter = ROOT.genfit.KalmanFitter()
#fitter = ROOT.genfit.KalmanFitterRefTrack()
self.fitter = ROOT.genfit.DAF()
self.fitter.setMaxIterations(50)
if debug: self.fitter.setDebugLvl(1) # produces lot of printout
#set to True if "real" pattern recognition is required also
if debug == True: shipPatRec.debug = 1
# for 'real' PatRec
shipPatRec.initialize(fgeo)
