Added tracking scripts and mod options file .

• modified: tracking/options/tracker_reconstruction.py
• modified: tracking/scripts/rec_central_electrons.cxx

tracking/options/tracker_reconstruction.py

@@ -29,6 +29,8 @@ from Configurables import Jug__Reco__TrackerHitReconstruction as TrackerHitRecon
 from Configurables import Jug__Reco__TrackerSourceLinker as TrackerSourceLinker
 from Configurables import Jug__Reco__TrackParamTruthInit as TrackParamTruthInit
 from Configurables import Jug__Reco__TrackParamClusterInit as TrackParamClusterInit
+from Configurables import Jug__Reco__TrackParamVertexClusterInit as TrackParamVertexClusterInit
+
 from Configurables import Jug__Reco__TrackFindingAlgorithm as TrackFindingAlgorithm
 from Configurables import Jug__Reco__ParticlesFromTrackFit as ParticlesFromTrackFit
 from Configurables import Jug__Reco__EMCalReconstruction as EMCalReconstruction
@@ -38,77 +40,106 @@ from Configurables import Jug__Reco__SimpleClustering as SimpleClustering
 
 
 podioinput = PodioInput("PodioReader", 
-                        collections=["mcparticles","SiTrackerBarrelHits","EcalBarrelHits"])#, OutputLevel=DEBUG)
+                        collections=["mcparticles","SiTrackerBarrelHits","SiVertexBarrelHits","EcalBarrelHits"])#, OutputLevel=DEBUG)
 
 ## copiers to get around input --> output copy bug. Note the "2" appended to the output collection.
-copier = MCCopier("MCCopier", inputCollection="mcparticles", outputCollection="mcparticles2",OutputLevel=DEBUG) 
-trkcopier = TrkCopier("TrkCopier", inputCollection="SiTrackerBarrelHits", outputCollection="SiTrackerBarrelHits2",OutputLevel=DEBUG) 
-#caldigi = ExampleCaloDigi(inputHitCollection="FAEC_ShHits",outputHitCollection="RawFAECShowerHits")
-#ufsd_digi = UFSDTrackerDigi("ufsd_digi", inputHitCollection="SiVertexBarrelHits",outputHitCollection="VertexRawHits",timeResolution=8)
+copier = MCCopier("MCCopier", 
+        inputCollection="mcparticles", 
+        outputCollection="mcparticles2") 
+trkcopier = TrkCopier("TrkCopier", 
+        inputCollection="SiTrackerBarrelHits", 
+        outputCollection="SiTrackerBarrelHits2") 
 
 ecal_digi = EMCalorimeterDigi("ecal_digi", 
         inputHitCollection="EcalBarrelHits", 
         outputHitCollection="RawEcalBarrelHits")
 
+ufsd_digi = UFSDTrackerDigi("ufsd_digi", 
+        inputHitCollection="SiTrackerBarrelHits",
+        outputHitCollection="SiTrackerBarrelRawHits",
+        timeResolution=8)
+
+vtx_digi = UFSDTrackerDigi("vtx_digi", 
+        inputHitCollection="SiVertexBarrelHits",
+        outputHitCollection="SiVertexBarrelRawHits",
+        timeResolution=8)
+
+
 ecal_reco = EMCalReconstruction("ecal_reco", 
         inputHitCollection="RawEcalBarrelHits", 
         outputHitCollection="RecEcalBarrelHits",
-        minModuleEdep=1.0*units.MeV,OutputLevel=DEBUG)
+        minModuleEdep=1.0*units.MeV)
 
 simple_cluster = SimpleClustering("simple_cluster", 
         inputHitCollection="RecEcalBarrelHits", 
         outputClusters="SimpleClusters",
         minModuleEdep=1.0*units.MeV,OutputLevel=DEBUG)
 
-cluster_init = TrackParamClusterInit("trk_clust_init",
-        inputClusters="SimpleClusters",
-        outputInitialTrackParameters="InitTrackParamsFromClusters",
-        OutputLevel=DEBUG)
+trk_barrel_reco = TrackerHitReconstruction("trk_barrel_reco",
+        inputHitCollection="SiTrackerBarrelRawHits",
+        outputHitCollection="TrackerBarrelRecHits")
 
-ufsd_digi = UFSDTrackerDigi("ufsd_digi", 
-        inputHitCollection="SiTrackerBarrelHits",
-        outputHitCollection="SiTrackerBarrelRawHits",
-        timeResolution=8)
+vtx_barrel_reco = TrackerHitReconstruction("vtx_barrel_reco",
+        inputHitCollection = vtx_digi.outputHitCollection,
+        outputHitCollection="VertexBarrelRecHits")
 
-trackpartruth = TrackParamTruthInit("trk_par_init",
+# Source linker 
+sourcelinker = TrackerSourceLinker("trk_srclinker",
+        inputHitCollection="TrackerBarrelRecHits",
+        outputSourceLinks="BarrelTrackSourceLinks")
+
+## Track param init
+truth_trk_init = TrackParamTruthInit("truth_trk_init",
         inputMCParticles="mcparticles",
         outputInitialTrackParameters="InitTrackParams",
         OutputLevel=DEBUG)
 
-trackerhit = TrackerHitReconstruction("trk_hit_reco",
-        inputHitCollection="SiTrackerBarrelRawHits",
-        outputHitCollection="TrackerBarrelRecHits",
+clust_trk_init = TrackParamClusterInit("clust_trk_init",
+        inputClusters="SimpleClusters",
+        outputInitialTrackParameters="InitTrackParamsFromClusters",
         OutputLevel=DEBUG)
 
-sourcelinker = TrackerSourceLinker("trk_srclinker",
-        inputHitCollection="TrackerBarrelRecHits",
-        outputSourceLinks="BarrelTrackSourceLinks",
+vtxcluster_trk_init = TrackParamVertexClusterInit("vtxcluster_trk_init",
+        inputVertexHits="VertexBarrelRecHits",
+        inputClusters="SimpleClusters",
+        outputInitialTrackParameters="InitTrackParamsFromVtxClusters",
         OutputLevel=DEBUG)
 
+# Tracking algorithms
 trk_find_alg = TrackFindingAlgorithm("trk_find_alg",
         inputSourceLinks="BarrelTrackSourceLinks",
         inputInitialTrackParameters= "InitTrackParams",#"InitTrackParamsFromClusters", 
         outputTrajectories="trajectories",
         OutputLevel=DEBUG)
-
-trk_find_alg1 = TrackFindingAlgorithm("trk_find_alg1",
-        inputSourceLinks="BarrelTrackSourceLinks",
-        inputInitialTrackParameters= "InitTrackParamsFromClusters", 
-        outputTrajectories="trajectories1",
-        OutputLevel=DEBUG)
-
 parts_from_fit = ParticlesFromTrackFit("parts_from_fit",
         inputTrajectories="trajectories",
         outputParticles="ReconstructedParticles",
         outputTrackParameters="outputTrackParameters",
         OutputLevel=DEBUG)
 
+trk_find_alg1 = TrackFindingAlgorithm("trk_find_alg1",
+        inputSourceLinks="BarrelTrackSourceLinks",
+        inputInitialTrackParameters= "InitTrackParamsFromClusters", 
+        outputTrajectories="trajectories1",
+        OutputLevel=DEBUG)
 parts_from_fit1 = ParticlesFromTrackFit("parts_from_fit1",
         inputTrajectories="trajectories1",
         outputParticles="ReconstructedParticles1",
         outputTrackParameters="outputTrackParameters1",
         OutputLevel=DEBUG)
 
+trk_find_alg2 = TrackFindingAlgorithm("trk_find_alg2",
+        inputSourceLinks="BarrelTrackSourceLinks",
+        inputInitialTrackParameters= "InitTrackParamsFromVtxClusters", 
+        outputTrajectories="trajectories2",
+        OutputLevel=DEBUG)
+parts_from_fit2 = ParticlesFromTrackFit("parts_from_fit2",
+        inputTrajectories="trajectories2",
+        outputParticles="ReconstructedParticles2",
+        outputTrackParameters="outputTrackParameters2",
+        OutputLevel=DEBUG)
+
+
 #types = []
 ## this printout is useful to check that the type information is passed to python correctly
 #print("---------------------------------------\n")
@@ -140,15 +171,18 @@ out.outputCommands = ["keep *",
 ApplicationMgr(
     TopAlg = [podioinput, 
               copier, trkcopier,
-              ecal_digi,ecal_reco, 
+              ecal_digi, ufsd_digi, vtx_digi, 
+              ecal_reco, 
               simple_cluster,
-              ufsd_digi, trackerhit, 
+              trk_barrel_reco, 
+              vtx_barrel_reco, 
               sourcelinker, 
-              cluster_init, trackpartruth, 
-              trk_find_alg, 
-              trk_find_alg1,
-              parts_from_fit,
-              parts_from_fit1,
+              clust_trk_init, 
+              truth_trk_init, 
+              vtxcluster_trk_init, 
+              trk_find_alg, parts_from_fit,
+              trk_find_alg1, parts_from_fit1,
+              trk_find_alg2, parts_from_fit2,
               out
               ],
     EvtSel = 'NONE',