Functions
	EventLoop (SmearedHits)

Variables
int	nEvents = 999999

	parser = OptionParser()

	dest

	help

	default

	options

	args

	cheated

	debug

	inputFile = options.input

	geoFile

	monitor

	printhelp

	reconstructiblerequired

	threeprong

	saveDisk = options.saveDisk

	outFile = inputFile

	tmp = inputFile.replace('ship.','geofile_full.')

	run = ROOT.FairRunSim()

dict	modules = {}

	fgeo = ROOT.TFile(shipPatRec_prev.geoFile)

	sGeo = fgeo.FAIRGeom

	bfield = ROOT.genfit.BellField(shipPatRec_prev.ship_geo.Bfield.max ,shipPatRec_prev.ship_geo.Bfield.z,2,shipPatRec_prev.ship_geo.Yheight/2.*u.m)

	fM = ROOT.genfit.FieldManager.getInstance()

	geoMat = ROOT.genfit.TGeoMaterialInterface()

	fn = ROOT.TFile(outFile,'update')

	sTree = fn.cbmsim

	fout = outFile

	f = ROOT.TFile(fout,'update')

	rawFile = fout.replace("_patrec.root","_raw.root")

	recf = ROOT.TFile(rawFile,"recreate")

	newTree = sTree.CloneTree(0)

	rc = newTree.Fill()

	fPartArray_PR = ROOT.TClonesArray("TParticle")

	fGenFitArray_PR = ROOT.TClonesArray("genfit::Track")

	fitTrack2MC_PR = ROOT.std.vector('int')()

	SmearedHits = ROOT.TClonesArray("TVectorD")

	Particles_PR = sTree.Branch("Particles_PR", fPartArray_PR,32000,-1)

	SHbranch = sTree.Branch("SmearedHits",SmearedHits,32000,-1)

	fitTracks_PR = sTree.Branch("FitTracks_PR", fGenFitArray_PR,32000,-1)

	mcLink_PR = sTree.Branch("fitTrack2MC_PR",fitTrack2MC_PR,32000,-1)

	debugrootfile = ROOT.TFile(str(shipPatRec_prev.reconstructiblerequired)+"track-debug-mumunu-"+str(nEvents)+".root","RECREATE")

int	scale = 1.

	totalafterpatrec

	totalaftermatching

Function Documentation

◆ EventLoop()

shipStrawTracking_prev.EventLoop ( SmearedHits )

Definition at line 145 of file shipStrawTracking_prev.py.

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 : 
        if global_variables.debug:
          print(
              "Number of reconstructible tracks =", len(shipPatRec_prev.ReconstructibleMCTracks),
              "but number of reconstructible required=", shipPatRec_prev.reconstructiblerequired,
              ". Rejecting event."
              )
        continue
 
     if global_variables.debug:
       print("Reconstructible track ids", shipPatRec_prev.ReconstructibleMCTracks)
     
     
  #n = current event number, False=wire endpoints, True=MC truth
    
  SmearedHits  = shipPatRec_prev.SmearHits(n,sTree,modules,SmearedHits,shipPatRec_prev.ReconstructibleMCTracks)  
 
  fittedtrackids=shipPatRec_prev.execute(n,SmearedHits,sTree,shipPatRec_prev.ReconstructibleMCTracks)
  if fittedtrackids:
     tracknbr=0
     for ids in fittedtrackids:
        nTrack   = fGenFitArray_PR.GetEntries()
        fGenFitArray_PR[nTrack] = shipPatRec_prev.theTracks[tracknbr]
        fitTrack2MC_PR.push_back(ids) 
        tracknbr+=1  
  
  Particles_PR.Fill()
  fitTracks_PR.Fill()
  mcLink_PR.Fill()    
  SHbranch.Fill()  
    
    
 if global_variables.debug: 
   print(shipPatRec_prev.falsenegative,"matched tracks with wrong negative charge from deflection.")
   print(shipPatRec_prev.falsepositive,"matched tracks with wrong positive charge from deflection.")
   print(shipPatRec_prev.morethan500,"events with more than 500 hits.")  
   print(shipPatRec_prev.morethan100tracks,"events with more than 100 tracks.") 
  
 return   
            

Variable Documentation

◆ args

shipStrawTracking_prev.args

Definition at line 27 of file shipStrawTracking_prev.py.

◆ bfield

shipStrawTracking_prev.bfield = ROOT.genfit.BellField(shipPatRec_prev.ship_geo.Bfield.max ,shipPatRec_prev.ship_geo.Bfield.z,2,shipPatRec_prev.ship_geo.Yheight/2.*u.m)

Definition at line 77 of file shipStrawTracking_prev.py.

◆ cheated

shipStrawTracking_prev.cheated

Definition at line 29 of file shipStrawTracking_prev.py.

◆ debug

shipStrawTracking_prev.debug

Definition at line 30 of file shipStrawTracking_prev.py.

◆ debugrootfile

shipStrawTracking_prev.debugrootfile = ROOT.TFile(str(shipPatRec_prev.reconstructiblerequired)+"track-debug-mumunu-"+str(nEvents)+".root","RECREATE")

Definition at line 200 of file shipStrawTracking_prev.py.

◆ default

shipStrawTracking_prev.default

Definition at line 17 of file shipStrawTracking_prev.py.

◆ dest

shipStrawTracking_prev.dest

Definition at line 17 of file shipStrawTracking_prev.py.

◆ f

shipStrawTracking_prev.f = ROOT.TFile(fout,'update')

Definition at line 92 of file shipStrawTracking_prev.py.

◆ fGenFitArray_PR

shipStrawTracking_prev.fGenFitArray_PR = ROOT.TClonesArray("genfit::Track")

Definition at line 118 of file shipStrawTracking_prev.py.

◆ fgeo

shipStrawTracking_prev.fgeo = ROOT.TFile(shipPatRec_prev.geoFile)

Definition at line 73 of file shipStrawTracking_prev.py.

◆ fitTrack2MC_PR

shipStrawTracking_prev.fitTrack2MC_PR = ROOT.std.vector('int')()

Definition at line 120 of file shipStrawTracking_prev.py.

◆ fitTracks_PR

shipStrawTracking_prev.fitTracks_PR = sTree.Branch("FitTracks_PR", fGenFitArray_PR,32000,-1)

Definition at line 125 of file shipStrawTracking_prev.py.

◆ fM

shipStrawTracking_prev.fM = ROOT.genfit.FieldManager.getInstance()

Definition at line 78 of file shipStrawTracking_prev.py.

◆ fn

shipStrawTracking_prev.fn = ROOT.TFile(outFile,'update')

Definition at line 85 of file shipStrawTracking_prev.py.

◆ fout

shipStrawTracking_prev.fout = outFile

Definition at line 87 of file shipStrawTracking_prev.py.

◆ fPartArray_PR

shipStrawTracking_prev.fPartArray_PR = ROOT.TClonesArray("TParticle")

Definition at line 117 of file shipStrawTracking_prev.py.

◆ geoFile

shipStrawTracking_prev.geoFile

Definition at line 32 of file shipStrawTracking_prev.py.

◆ geoMat

shipStrawTracking_prev.geoMat = ROOT.genfit.TGeoMaterialInterface()

Definition at line 81 of file shipStrawTracking_prev.py.

◆ help

shipStrawTracking_prev.help

Definition at line 17 of file shipStrawTracking_prev.py.

◆ inputFile

shipStrawTracking_prev.inputFile = options.input

Definition at line 31 of file shipStrawTracking_prev.py.

◆ mcLink_PR

shipStrawTracking_prev.mcLink_PR = sTree.Branch("fitTrack2MC_PR",fitTrack2MC_PR,32000,-1)

Definition at line 126 of file shipStrawTracking_prev.py.

◆ modules

dict shipStrawTracking_prev.modules = {}

Definition at line 70 of file shipStrawTracking_prev.py.

◆ monitor

shipStrawTracking_prev.monitor

Definition at line 33 of file shipStrawTracking_prev.py.

◆ nEvents

shipStrawTracking_prev.nEvents = 999999

Definition at line 13 of file shipStrawTracking_prev.py.

◆ newTree

shipStrawTracking_prev.newTree = sTree.CloneTree(0)

Definition at line 101 of file shipStrawTracking_prev.py.

◆ options

shipStrawTracking_prev.options

Definition at line 27 of file shipStrawTracking_prev.py.

◆ outFile

shipStrawTracking_prev.outFile = inputFile

Definition at line 57 of file shipStrawTracking_prev.py.

◆ parser

shipStrawTracking_prev.parser = OptionParser()

Definition at line 16 of file shipStrawTracking_prev.py.

◆ Particles_PR

shipStrawTracking_prev.Particles_PR = sTree.Branch("Particles_PR", fPartArray_PR,32000,-1)

Definition at line 123 of file shipStrawTracking_prev.py.

◆ printhelp

shipStrawTracking_prev.printhelp

Definition at line 34 of file shipStrawTracking_prev.py.

◆ rawFile

shipStrawTracking_prev.rawFile = fout.replace("_patrec.root","_raw.root")

Definition at line 99 of file shipStrawTracking_prev.py.

◆ rc

shipStrawTracking_prev.rc = newTree.Fill()

Definition at line 104 of file shipStrawTracking_prev.py.

◆ recf

shipStrawTracking_prev.recf = ROOT.TFile(rawFile,"recreate")

Definition at line 100 of file shipStrawTracking_prev.py.

◆ reconstructiblerequired

shipStrawTracking_prev.reconstructiblerequired

Definition at line 35 of file shipStrawTracking_prev.py.

◆ run

shipStrawTracking_prev.run = ROOT.FairRunSim()

Definition at line 68 of file shipStrawTracking_prev.py.

◆ saveDisk

shipStrawTracking_prev.saveDisk = options.saveDisk

Definition at line 37 of file shipStrawTracking_prev.py.

◆ scale

int shipStrawTracking_prev.scale = 1.

Definition at line 208 of file shipStrawTracking_prev.py.

◆ sGeo

shipStrawTracking_prev.sGeo = fgeo.FAIRGeom

Definition at line 74 of file shipStrawTracking_prev.py.

◆ SHbranch

shipStrawTracking_prev.SHbranch = sTree.Branch("SmearedHits",SmearedHits,32000,-1)

Definition at line 124 of file shipStrawTracking_prev.py.

◆ SmearedHits

shipStrawTracking_prev.SmearedHits = ROOT.TClonesArray("TVectorD")

Definition at line 121 of file shipStrawTracking_prev.py.

◆ sTree

shipStrawTracking_prev.sTree = fn.cbmsim

Definition at line 86 of file shipStrawTracking_prev.py.

◆ threeprong

shipStrawTracking_prev.threeprong

Definition at line 36 of file shipStrawTracking_prev.py.

◆ tmp

shipStrawTracking_prev.tmp = inputFile.replace('ship.','geofile_full.')

Definition at line 65 of file shipStrawTracking_prev.py.

◆ totalaftermatching

shipStrawTracking_prev.totalaftermatching

Definition at line 231 of file shipStrawTracking_prev.py.

◆ totalafterpatrec

shipStrawTracking_prev.totalafterpatrec

Definition at line 230 of file shipStrawTracking_prev.py.

Functions

Variables

Function Documentation

◆ EventLoop()

Variable Documentation

◆ args

◆ bfield

◆ cheated

◆ debug

◆ debugrootfile

◆ default

◆ dest

◆ f

◆ fGenFitArray_PR

◆ fgeo

◆ fitTrack2MC_PR

◆ fitTracks_PR

◆ fM

◆ fn

◆ fout

◆ fPartArray_PR

◆ geoFile

◆ geoMat

◆ help

◆ inputFile

◆ mcLink_PR

◆ modules

◆ monitor

◆ nEvents

◆ newTree

◆ options

◆ outFile

◆ parser

◆ Particles_PR

◆ printhelp

◆ rawFile

◆ rc

◆ recf

◆ reconstructiblerequired

◆ run

◆ saveDisk

◆ scale

◆ sGeo

◆ SHbranch

◆ SmearedHits

◆ sTree

◆ threeprong

◆ tmp

◆ totalaftermatching

◆ totalafterpatrec