Skip to content
Snippets Groups Projects
Commit 11550b61 authored by Wouter Deconinck's avatar Wouter Deconinck
Browse files

Use algorithms list for all steps

parent 27060ed4
No related branches found
No related tags found
No related merge requests found
This commit is part of merge request !168. Comments created here will be created in the context of that merge request.
Hide whitespace changes
Inline Side-by-side
benchmarks/full/options/full_reconstruction.py
+ 59
− 59
View file @ 11550b61
...@@ -103,25 +103,25 @@ sim_coll = [ ...@@ -103,25 +103,25 @@ sim_coll = [
# list of algorithms # list of algorithms
algorithms = [] algorithms = []
# input and output # input
podin = PodioInput("PodioReader", collections=sim_coll) podin = PodioInput("PodioReader", collections=sim_coll)
podout = PodioOutput("out", filename=output_rec) algorithms.append(podin)
## copiers to get around input --> output copy bug. Note the "2" appended to the output collection. ## copiers to get around input --> output copy bug. Note the "2" appended to the output collection.
copier = MCCopier("MCCopier", copier = MCCopier("MCCopier",
inputCollection="mcparticles", inputCollection="mcparticles",
outputCollection="mcparticles2") outputCollection="mcparticles2")
algorithms.append( copier ) algorithms.append(copier)
trkcopier = TrkCopier("TrkCopier", trkcopier = TrkCopier("TrkCopier",
inputCollection="TrackerBarrelHits", inputCollection="TrackerBarrelHits",
outputCollection="TrackerBarrelHits2") outputCollection="TrackerBarrelHits2")
algorithms.append( trkcopier ) algorithms.append(trkcopier)
pmtcopier = PMTCopier("PMTCopier", pmtcopier = PMTCopier("PMTCopier",
inputCollection="DRICHHits", inputCollection="DRICHHits",
outputCollection="DRICHHits2") outputCollection="DRICHHits2")
algorithms.append( pmtcopier ) algorithms.append(pmtcopier)
# Crystal Endcap Ecal # Crystal Endcap Ecal
ce_ecal_daq = dict( ce_ecal_daq = dict(
...@@ -135,7 +135,7 @@ ce_ecal_digi = CalHitDigi("ce_ecal_digi", ...@@ -135,7 +135,7 @@ ce_ecal_digi = CalHitDigi("ce_ecal_digi",
outputHitCollection="EcalEndcapNHitsDigi", outputHitCollection="EcalEndcapNHitsDigi",
energyResolutions=[0., 0.02, 0.], energyResolutions=[0., 0.02, 0.],
**ce_ecal_daq) **ce_ecal_daq)
algorithms.append( ce_ecal_digi ) algorithms.append(ce_ecal_digi)
ce_ecal_reco = CalHitReco("ce_ecal_reco", ce_ecal_reco = CalHitReco("ce_ecal_reco",
inputHitCollection=ce_ecal_digi.outputHitCollection, inputHitCollection=ce_ecal_digi.outputHitCollection,
...@@ -144,7 +144,7 @@ ce_ecal_reco = CalHitReco("ce_ecal_reco", ...@@ -144,7 +144,7 @@ ce_ecal_reco = CalHitReco("ce_ecal_reco",
readoutClass="EcalEndcapNHits", readoutClass="EcalEndcapNHits",
sectorField="sector", sectorField="sector",
**ce_ecal_daq) **ce_ecal_daq)
algorithms.append( ce_ecal_reco ) algorithms.append(ce_ecal_reco)
ce_ecal_cl = IslandCluster("ce_ecal_cl", ce_ecal_cl = IslandCluster("ce_ecal_cl",
inputHitCollection=ce_ecal_reco.outputHitCollection, inputHitCollection=ce_ecal_reco.outputHitCollection,
...@@ -154,7 +154,7 @@ ce_ecal_cl = IslandCluster("ce_ecal_cl", ...@@ -154,7 +154,7 @@ ce_ecal_cl = IslandCluster("ce_ecal_cl",
minClusterCenterEdep=30*units.MeV, minClusterCenterEdep=30*units.MeV,
sectorDist=5.0*units.cm, sectorDist=5.0*units.cm,
dimScaledLocalDistXY=[1.8, 1.8]) # dimension scaled dist is good for hybrid sectors with different module size dimScaledLocalDistXY=[1.8, 1.8]) # dimension scaled dist is good for hybrid sectors with different module size
algorithms.append( ce_ecal_cl ) algorithms.append(ce_ecal_cl)
ce_ecal_clreco = RecoCoG("ce_ecal_clreco", ce_ecal_clreco = RecoCoG("ce_ecal_clreco",
inputHitCollection=ce_ecal_cl.inputHitCollection, inputHitCollection=ce_ecal_cl.inputHitCollection,
...@@ -162,7 +162,7 @@ ce_ecal_clreco = RecoCoG("ce_ecal_clreco", ...@@ -162,7 +162,7 @@ ce_ecal_clreco = RecoCoG("ce_ecal_clreco",
outputClusterCollection="EcalEndcapNClusters", outputClusterCollection="EcalEndcapNClusters",
samplingFraction=0.998, # this accounts for a small fraction of leakage samplingFraction=0.998, # this accounts for a small fraction of leakage
logWeightBase=4.6) logWeightBase=4.6)
algorithms.append( ce_ecal_clreco ) algorithms.append(ce_ecal_clreco)
# Endcap Sampling Ecal # Endcap Sampling Ecal
ci_ecal_daq = dict( ci_ecal_daq = dict(
...@@ -175,14 +175,14 @@ ci_ecal_digi = CalHitDigi("ci_ecal_digi", ...@@ -175,14 +175,14 @@ ci_ecal_digi = CalHitDigi("ci_ecal_digi",
inputHitCollection="EcalEndcapPHits", inputHitCollection="EcalEndcapPHits",
outputHitCollection="EcalEndcapPHitsDigi", outputHitCollection="EcalEndcapPHitsDigi",
**ci_ecal_daq) **ci_ecal_daq)
algorithms.append( ci_ecal_digi ) algorithms.append(ci_ecal_digi)
ci_ecal_reco = CalHitReco("ci_ecal_reco", ci_ecal_reco = CalHitReco("ci_ecal_reco",
inputHitCollection=ci_ecal_digi.outputHitCollection, inputHitCollection=ci_ecal_digi.outputHitCollection,
outputHitCollection="EcalEndcapPHitsReco", outputHitCollection="EcalEndcapPHitsReco",
thresholdFactor=5.0, thresholdFactor=5.0,
**ci_ecal_daq) **ci_ecal_daq)
algorithms.append( ci_ecal_reco ) algorithms.append(ci_ecal_reco)
# merge hits in different layer (projection to local x-y plane) # merge hits in different layer (projection to local x-y plane)
ci_ecal_merger = CalHitsMerger("ci_ecal_merger", ci_ecal_merger = CalHitsMerger("ci_ecal_merger",
...@@ -191,7 +191,7 @@ ci_ecal_merger = CalHitsMerger("ci_ecal_merger", ...@@ -191,7 +191,7 @@ ci_ecal_merger = CalHitsMerger("ci_ecal_merger",
fields=["layer", "slice"], fields=["layer", "slice"],
fieldRefNumbers=[1, 0], fieldRefNumbers=[1, 0],
readoutClass="EcalEndcapPHits") readoutClass="EcalEndcapPHits")
algorithms.append( ci_ecal_merger ) algorithms.append(ci_ecal_merger)
ci_ecal_cl = IslandCluster("ci_ecal_cl", ci_ecal_cl = IslandCluster("ci_ecal_cl",
inputHitCollection=ci_ecal_merger.outputHitCollection, inputHitCollection=ci_ecal_merger.outputHitCollection,
...@@ -199,7 +199,7 @@ ci_ecal_cl = IslandCluster("ci_ecal_cl", ...@@ -199,7 +199,7 @@ ci_ecal_cl = IslandCluster("ci_ecal_cl",
splitCluster=False, splitCluster=False,
minClusterCenterEdep=10.*units.MeV, minClusterCenterEdep=10.*units.MeV,
localDistXY=[10*units.mm, 10*units.mm]) localDistXY=[10*units.mm, 10*units.mm])
algorithms.append( ci_ecal_cl ) algorithms.append(ci_ecal_cl)
ci_ecal_clreco = RecoCoG("ci_ecal_clreco", ci_ecal_clreco = RecoCoG("ci_ecal_clreco",
inputHitCollection=ci_ecal_cl.inputHitCollection, inputHitCollection=ci_ecal_cl.inputHitCollection,
...@@ -208,7 +208,7 @@ ci_ecal_clreco = RecoCoG("ci_ecal_clreco", ...@@ -208,7 +208,7 @@ ci_ecal_clreco = RecoCoG("ci_ecal_clreco",
outputInfoCollection="EcalEndcapPClustersInfo", outputInfoCollection="EcalEndcapPClustersInfo",
logWeightBase=6.2, logWeightBase=6.2,
samplingFraction=ci_ecal_sf) samplingFraction=ci_ecal_sf)
algorithms.append( ci_ecal_clreco ) algorithms.append(ci_ecal_clreco)
# Central Barrel Ecal (Imaging Cal.) # Central Barrel Ecal (Imaging Cal.)
cb_ecal_daq = dict( cb_ecal_daq = dict(
...@@ -222,7 +222,7 @@ cb_ecal_digi = CalHitDigi("cb_ecal_digi", ...@@ -222,7 +222,7 @@ cb_ecal_digi = CalHitDigi("cb_ecal_digi",
outputHitCollection="EcalBarrelHitsDigi", outputHitCollection="EcalBarrelHitsDigi",
energyResolutions=[0., 0.02, 0.], # 2% flat resolution energyResolutions=[0., 0.02, 0.], # 2% flat resolution
**cb_ecal_daq) **cb_ecal_daq)
algorithms.append( cb_ecal_digi ) algorithms.append(cb_ecal_digi)
cb_ecal_reco = ImCalPixelReco("cb_ecal_reco", cb_ecal_reco = ImCalPixelReco("cb_ecal_reco",
inputHitCollection=cb_ecal_digi.outputHitCollection, inputHitCollection=cb_ecal_digi.outputHitCollection,
...@@ -232,7 +232,7 @@ cb_ecal_reco = ImCalPixelReco("cb_ecal_reco", ...@@ -232,7 +232,7 @@ cb_ecal_reco = ImCalPixelReco("cb_ecal_reco",
layerField="layer", # field to get layer id layerField="layer", # field to get layer id
sectorField="module", # field to get sector id sectorField="module", # field to get sector id
**cb_ecal_daq) **cb_ecal_daq)
algorithms.append( cb_ecal_reco ) algorithms.append(cb_ecal_reco)
cb_ecal_cl = ImagingCluster("cb_ecal_cl", cb_ecal_cl = ImagingCluster("cb_ecal_cl",
inputHitCollection=cb_ecal_reco.outputHitCollection, inputHitCollection=cb_ecal_reco.outputHitCollection,
...@@ -241,7 +241,7 @@ cb_ecal_cl = ImagingCluster("cb_ecal_cl", ...@@ -241,7 +241,7 @@ cb_ecal_cl = ImagingCluster("cb_ecal_cl",
layerDistEtaPhi=[10*units.mrad, 10*units.mrad], # adjacent layer layerDistEtaPhi=[10*units.mrad, 10*units.mrad], # adjacent layer
neighbourLayersRange=2, # id diff for adjacent layer neighbourLayersRange=2, # id diff for adjacent layer
sectorDist=3.*units.cm) # different sector sectorDist=3.*units.cm) # different sector
algorithms.append( cb_ecal_cl ) algorithms.append(cb_ecal_cl)
cb_ecal_clreco = ImagingClusterReco("cb_ecal_clreco", cb_ecal_clreco = ImagingClusterReco("cb_ecal_clreco",
samplingFraction=cb_ecal_sf, samplingFraction=cb_ecal_sf,
...@@ -250,7 +250,7 @@ cb_ecal_clreco = ImagingClusterReco("cb_ecal_clreco", ...@@ -250,7 +250,7 @@ cb_ecal_clreco = ImagingClusterReco("cb_ecal_clreco",
outputClusterCollection="EcalBarrelClusters", outputClusterCollection="EcalBarrelClusters",
outputInfoCollection="EcalBarrelClustersInfo", outputInfoCollection="EcalBarrelClustersInfo",
outputLayerCollection="EcalBarrelLayers") outputLayerCollection="EcalBarrelLayers")
algorithms.append( cb_ecal_clreco ) algorithms.append(cb_ecal_clreco)
# Central ECAL SciFi # Central ECAL SciFi
scfi_barrel_daq = dict( scfi_barrel_daq = dict(
...@@ -263,6 +263,7 @@ scfi_barrel_digi = CalHitDigi("scfi_barrel_digi", ...@@ -263,6 +263,7 @@ scfi_barrel_digi = CalHitDigi("scfi_barrel_digi",
inputHitCollection="EcalBarrelScFiHits", inputHitCollection="EcalBarrelScFiHits",
outputHitCollection="EcalBarrelScFiHitsDigi", outputHitCollection="EcalBarrelScFiHitsDigi",
**scfi_barrel_daq) **scfi_barrel_daq)
algorithms.append(scfi_barrel_digi)
scfi_barrel_reco = CalHitReco("scfi_barrel_reco", scfi_barrel_reco = CalHitReco("scfi_barrel_reco",
inputHitCollection=scfi_barrel_digi.outputHitCollection, inputHitCollection=scfi_barrel_digi.outputHitCollection,
...@@ -273,6 +274,7 @@ scfi_barrel_reco = CalHitReco("scfi_barrel_reco", ...@@ -273,6 +274,7 @@ scfi_barrel_reco = CalHitReco("scfi_barrel_reco",
sectorField="module", sectorField="module",
localDetFields=["system", "module"], # use local coordinates in each module (stave) localDetFields=["system", "module"], # use local coordinates in each module (stave)
**scfi_barrel_daq) **scfi_barrel_daq)
algorithms.append(scfi_barrel_reco)
# merge hits in different layer (projection to local x-y plane) # merge hits in different layer (projection to local x-y plane)
scfi_barrel_merger = CalHitsMerger("scfi_barrel_merger", scfi_barrel_merger = CalHitsMerger("scfi_barrel_merger",
...@@ -281,6 +283,7 @@ scfi_barrel_merger = CalHitsMerger("scfi_barrel_merger", ...@@ -281,6 +283,7 @@ scfi_barrel_merger = CalHitsMerger("scfi_barrel_merger",
fields=["fiber"], fields=["fiber"],
fieldRefNumbers=[1], fieldRefNumbers=[1],
readoutClass="EcalBarrelScFiHits") readoutClass="EcalBarrelScFiHits")
algorithms.append(scfi_barrel_merger)
scfi_barrel_cl = IslandCluster("scfi_barrel_cl", scfi_barrel_cl = IslandCluster("scfi_barrel_cl",
inputHitCollection=scfi_barrel_merger.outputHitCollection, inputHitCollection=scfi_barrel_merger.outputHitCollection,
...@@ -288,6 +291,7 @@ scfi_barrel_cl = IslandCluster("scfi_barrel_cl", ...@@ -288,6 +291,7 @@ scfi_barrel_cl = IslandCluster("scfi_barrel_cl",
splitCluster=False, splitCluster=False,
minClusterCenterEdep=10.*MeV, minClusterCenterEdep=10.*MeV,
localDistXZ=[30*mm, 30*mm]) localDistXZ=[30*mm, 30*mm])
algorithms.append(scfi_barrel_cl)
scfi_barrel_clreco = RecoCoG("scfi_barrel_clreco", scfi_barrel_clreco = RecoCoG("scfi_barrel_clreco",
inputHitCollection=scfi_barrel_cl.inputHitCollection, inputHitCollection=scfi_barrel_cl.inputHitCollection,
...@@ -296,6 +300,7 @@ scfi_barrel_clreco = RecoCoG("scfi_barrel_clreco", ...@@ -296,6 +300,7 @@ scfi_barrel_clreco = RecoCoG("scfi_barrel_clreco",
outputInfoCollection="EcalBarrelScFiClustersInfo", outputInfoCollection="EcalBarrelScFiClustersInfo",
logWeightBase=6.2, logWeightBase=6.2,
samplingFraction= scifi_barrel_sf) samplingFraction= scifi_barrel_sf)
algorithms.append(scfi_barrel_clreco)
# Central Barrel Hcal # Central Barrel Hcal
cb_hcal_daq = dict( cb_hcal_daq = dict(
...@@ -308,6 +313,7 @@ cb_hcal_digi = CalHitDigi("cb_hcal_digi", ...@@ -308,6 +313,7 @@ cb_hcal_digi = CalHitDigi("cb_hcal_digi",
inputHitCollection="HcalBarrelHits", inputHitCollection="HcalBarrelHits",
outputHitCollection="HcalBarrelHitsDigi", outputHitCollection="HcalBarrelHitsDigi",
**cb_hcal_daq) **cb_hcal_daq)
algorithms.append(cb_hcal_digi)
cb_hcal_reco = CalHitReco("cb_hcal_reco", cb_hcal_reco = CalHitReco("cb_hcal_reco",
inputHitCollection=cb_hcal_digi.outputHitCollection, inputHitCollection=cb_hcal_digi.outputHitCollection,
...@@ -317,6 +323,7 @@ cb_hcal_reco = CalHitReco("cb_hcal_reco", ...@@ -317,6 +323,7 @@ cb_hcal_reco = CalHitReco("cb_hcal_reco",
layerField="layer", layerField="layer",
sectorField="module", sectorField="module",
**cb_hcal_daq) **cb_hcal_daq)
algorithms.append(cb_hcal_reco)
cb_hcal_merger = CalHitsMerger("cb_hcal_merger", cb_hcal_merger = CalHitsMerger("cb_hcal_merger",
inputHitCollection=cb_hcal_reco.outputHitCollection, inputHitCollection=cb_hcal_reco.outputHitCollection,
...@@ -324,6 +331,7 @@ cb_hcal_merger = CalHitsMerger("cb_hcal_merger", ...@@ -324,6 +331,7 @@ cb_hcal_merger = CalHitsMerger("cb_hcal_merger",
readoutClass="HcalBarrelHits", readoutClass="HcalBarrelHits",
fields=["layer", "slice"], fields=["layer", "slice"],
fieldRefNumbers=[1, 0]) fieldRefNumbers=[1, 0])
algorithms.append(cb_hcal_merger)
cb_hcal_cl = IslandCluster("cb_hcal_cl", cb_hcal_cl = IslandCluster("cb_hcal_cl",
inputHitCollection=cb_hcal_merger.outputHitCollection, inputHitCollection=cb_hcal_merger.outputHitCollection,
...@@ -331,6 +339,7 @@ cb_hcal_cl = IslandCluster("cb_hcal_cl", ...@@ -331,6 +339,7 @@ cb_hcal_cl = IslandCluster("cb_hcal_cl",
splitCluster=False, splitCluster=False,
minClusterCenterEdep=30.*units.MeV, minClusterCenterEdep=30.*units.MeV,
localDistXY=[15.*units.cm, 15.*units.cm]) localDistXY=[15.*units.cm, 15.*units.cm])
algorithms.append(cb_hcal_cl)
cb_hcal_clreco = RecoCoG("cb_hcal_clreco", cb_hcal_clreco = RecoCoG("cb_hcal_clreco",
inputHitCollection=cb_hcal_cl.inputHitCollection, inputHitCollection=cb_hcal_cl.inputHitCollection,
...@@ -339,7 +348,7 @@ cb_hcal_clreco = RecoCoG("cb_hcal_clreco", ...@@ -339,7 +348,7 @@ cb_hcal_clreco = RecoCoG("cb_hcal_clreco",
outputInfoCollection="HcalBarrelClustersInfo", outputInfoCollection="HcalBarrelClustersInfo",
logWeightBase=6.2, logWeightBase=6.2,
samplingFraction=cb_hcal_sf) samplingFraction=cb_hcal_sf)
algorithms.append(cb_hcal_clreco)
# Hcal Hadron Endcap # Hcal Hadron Endcap
ci_hcal_daq = dict( ci_hcal_daq = dict(
...@@ -347,16 +356,19 @@ ci_hcal_daq = dict( ...@@ -347,16 +356,19 @@ ci_hcal_daq = dict(
capacityADC=32768, capacityADC=32768,
pedestalMean=400, pedestalMean=400,
pedestalSigma=10) pedestalSigma=10)
ci_hcal_digi = CalHitDigi("ci_hcal_digi", ci_hcal_digi = CalHitDigi("ci_hcal_digi",
inputHitCollection="HcalEndcapPHits", inputHitCollection="HcalEndcapPHits",
outputHitCollection="HcalEndcapPHitsDigi", outputHitCollection="HcalEndcapPHitsDigi",
**ci_hcal_daq) **ci_hcal_daq)
algorithms.append(ci_hcal_digi)
ci_hcal_reco = CalHitReco("ci_hcal_reco", ci_hcal_reco = CalHitReco("ci_hcal_reco",
inputHitCollection=ci_hcal_digi.outputHitCollection, inputHitCollection=ci_hcal_digi.outputHitCollection,
outputHitCollection="HcalEndcapPHitsReco", outputHitCollection="HcalEndcapPHitsReco",
thresholdFactor=5.0, thresholdFactor=5.0,
**ci_hcal_daq) **ci_hcal_daq)
algorithms.append(ci_hcal_reco)
ci_hcal_merger = CalHitsMerger("ci_hcal_merger", ci_hcal_merger = CalHitsMerger("ci_hcal_merger",
inputHitCollection=ci_hcal_reco.outputHitCollection, inputHitCollection=ci_hcal_reco.outputHitCollection,
...@@ -364,6 +376,7 @@ ci_hcal_merger = CalHitsMerger("ci_hcal_merger", ...@@ -364,6 +376,7 @@ ci_hcal_merger = CalHitsMerger("ci_hcal_merger",
readoutClass="HcalEndcapPHits", readoutClass="HcalEndcapPHits",
fields=["layer", "slice"], fields=["layer", "slice"],
fieldRefNumbers=[1, 0]) fieldRefNumbers=[1, 0])
algorithms.append(ci_hcal_merger)
ci_hcal_cl = IslandCluster("ci_hcal_cl", ci_hcal_cl = IslandCluster("ci_hcal_cl",
inputHitCollection=ci_hcal_merger.outputHitCollection, inputHitCollection=ci_hcal_merger.outputHitCollection,
...@@ -371,6 +384,7 @@ ci_hcal_cl = IslandCluster("ci_hcal_cl", ...@@ -371,6 +384,7 @@ ci_hcal_cl = IslandCluster("ci_hcal_cl",
splitCluster=False, splitCluster=False,
minClusterCenterEdep=30.*units.MeV, minClusterCenterEdep=30.*units.MeV,
localDistXY=[15.*units.cm, 15.*units.cm]) localDistXY=[15.*units.cm, 15.*units.cm])
algorithms.append(ci_hcal_cl)
ci_hcal_clreco = RecoCoG("ci_hcal_clreco", ci_hcal_clreco = RecoCoG("ci_hcal_clreco",
inputHitCollection=ci_hcal_cl.inputHitCollection, inputHitCollection=ci_hcal_cl.inputHitCollection,
...@@ -379,6 +393,7 @@ ci_hcal_clreco = RecoCoG("ci_hcal_clreco", ...@@ -379,6 +393,7 @@ ci_hcal_clreco = RecoCoG("ci_hcal_clreco",
outputInfoCollection="HcalEndcapPClustersInfo", outputInfoCollection="HcalEndcapPClustersInfo",
logWeightBase=6.2, logWeightBase=6.2,
samplingFraction=ci_hcal_sf) samplingFraction=ci_hcal_sf)
algorithms.append(ci_hcal_clreco)
# Hcal Electron Endcap # Hcal Electron Endcap
ce_hcal_daq = dict( ce_hcal_daq = dict(
...@@ -391,12 +406,14 @@ ce_hcal_digi = CalHitDigi("ce_hcal_digi", ...@@ -391,12 +406,14 @@ ce_hcal_digi = CalHitDigi("ce_hcal_digi",
inputHitCollection="HcalEndcapNHits", inputHitCollection="HcalEndcapNHits",
outputHitCollection="HcalEndcapNHitsDigi", outputHitCollection="HcalEndcapNHitsDigi",
**ce_hcal_daq) **ce_hcal_daq)
algorithms.append(ce_hcal_digi)
ce_hcal_reco = CalHitReco("ce_hcal_reco", ce_hcal_reco = CalHitReco("ce_hcal_reco",
inputHitCollection=ce_hcal_digi.outputHitCollection, inputHitCollection=ce_hcal_digi.outputHitCollection,
outputHitCollection="HcalEndcapNHitsReco", outputHitCollection="HcalEndcapNHitsReco",
thresholdFactor=5.0, thresholdFactor=5.0,
**ce_hcal_daq) **ce_hcal_daq)
algorithms.append(ce_hcal_reco)
ce_hcal_merger = CalHitsMerger("ce_hcal_merger", ce_hcal_merger = CalHitsMerger("ce_hcal_merger",
inputHitCollection=ce_hcal_reco.outputHitCollection, inputHitCollection=ce_hcal_reco.outputHitCollection,
...@@ -404,6 +421,7 @@ ce_hcal_merger = CalHitsMerger("ce_hcal_merger", ...@@ -404,6 +421,7 @@ ce_hcal_merger = CalHitsMerger("ce_hcal_merger",
readoutClass="HcalEndcapNHits", readoutClass="HcalEndcapNHits",
fields=["layer", "slice"], fields=["layer", "slice"],
fieldRefNumbers=[1, 0]) fieldRefNumbers=[1, 0])
algorithms.append(ce_hcal_merger)
ce_hcal_cl = IslandCluster("ce_hcal_cl", ce_hcal_cl = IslandCluster("ce_hcal_cl",
inputHitCollection=ce_hcal_merger.outputHitCollection, inputHitCollection=ce_hcal_merger.outputHitCollection,
...@@ -411,6 +429,7 @@ ce_hcal_cl = IslandCluster("ce_hcal_cl", ...@@ -411,6 +429,7 @@ ce_hcal_cl = IslandCluster("ce_hcal_cl",
splitCluster=False, splitCluster=False,
minClusterCenterEdep=30.*units.MeV, minClusterCenterEdep=30.*units.MeV,
localDistXY=[15.*units.cm, 15.*units.cm]) localDistXY=[15.*units.cm, 15.*units.cm])
algorithms.append(ce_hcal_cl)
ce_hcal_clreco = RecoCoG("ce_hcal_clreco", ce_hcal_clreco = RecoCoG("ce_hcal_clreco",
inputHitCollection=ce_hcal_cl.inputHitCollection, inputHitCollection=ce_hcal_cl.inputHitCollection,
...@@ -419,53 +438,64 @@ ce_hcal_clreco = RecoCoG("ce_hcal_clreco", ...@@ -419,53 +438,64 @@ ce_hcal_clreco = RecoCoG("ce_hcal_clreco",
outputInfoCollection="HcalEndcapNClustersInfo", outputInfoCollection="HcalEndcapNClustersInfo",
logWeightBase=6.2, logWeightBase=6.2,
samplingFraction=ce_hcal_sf) samplingFraction=ce_hcal_sf)
algorithms.append(ce_hcal_clreco)
# Tracking # Tracking
trk_b_digi = TrackerDigi("trk_b_digi", trk_b_digi = TrackerDigi("trk_b_digi",
inputHitCollection="TrackerBarrelHits", inputHitCollection="TrackerBarrelHits",
outputHitCollection="TrackerBarrelRawHits", outputHitCollection="TrackerBarrelRawHits",
timeResolution=8) timeResolution=8)
algorithms.append(trk_b_digi)
trk_ec_digi = TrackerDigi("trk_ec_digi", trk_ec_digi = TrackerDigi("trk_ec_digi",
inputHitCollection="TrackerEndcapHits", inputHitCollection="TrackerEndcapHits",
outputHitCollection="TrackerEndcapRawHits", outputHitCollection="TrackerEndcapRawHits",
timeResolution=8) timeResolution=8)
algorithms.append(trk_ec_digi)
vtx_b_digi = TrackerDigi("vtx_b_digi", vtx_b_digi = TrackerDigi("vtx_b_digi",
inputHitCollection="VertexBarrelHits", inputHitCollection="VertexBarrelHits",
outputHitCollection="VertetimeResolution=10xBarrelRawHits", outputHitCollection="VertetimeResolution=10xBarrelRawHits",
timeResolution=8) timeResolution=8)
algorithms.append(vtx_b_digi)
vtx_ec_digi = TrackerDigi("vtx_ec_digi", vtx_ec_digi = TrackerDigi("vtx_ec_digi",
inputHitCollection="VertexEndcapHits", inputHitCollection="VertexEndcapHits",
outputHitCollection="VertexEndcapRawHits", outputHitCollection="VertexEndcapRawHits",
timeResolution=8) timeResolution=8)
algorithms.append(vtx_ec_digi)
gem_ec_digi = TrackerDigi("gem_ec_digi", gem_ec_digi = TrackerDigi("gem_ec_digi",
inputHitCollection="GEMTrackerEndcapHits", inputHitCollection="GEMTrackerEndcapHits",
outputHitCollection="GEMTrackerEndcapRawHits", outputHitCollection="GEMTrackerEndcapRawHits",
timeResolution=10) timeResolution=10)
algorithms.append(gem_ec_digi)
# Tracker and vertex reconstruction # Tracker and vertex reconstruction
trk_b_reco = TrackerHitReconstruction("trk_b_reco", trk_b_reco = TrackerHitReconstruction("trk_b_reco",
inputHitCollection = trk_b_digi.outputHitCollection, inputHitCollection = trk_b_digi.outputHitCollection,
outputHitCollection="TrackerBarrelRecHits") outputHitCollection="TrackerBarrelRecHits")
algorithms.append(trk_b_reco)
trk_ec_reco = TrackerHitReconstruction("trk_ec_reco", trk_ec_reco = TrackerHitReconstruction("trk_ec_reco",
inputHitCollection = trk_ec_digi.outputHitCollection, inputHitCollection = trk_ec_digi.outputHitCollection,
outputHitCollection="TrackerEndcapRecHits") outputHitCollection="TrackerEndcapRecHits")
algorithms.append(trk_ec_reco)
vtx_b_reco = TrackerHitReconstruction("vtx_b_reco", vtx_b_reco = TrackerHitReconstruction("vtx_b_reco",
inputHitCollection = vtx_b_digi.outputHitCollection, inputHitCollection = vtx_b_digi.outputHitCollection,
outputHitCollection="VertexBarrelRecHits") outputHitCollection="VertexBarrelRecHits")
algorithms.append(vtx_b_reco)
vtx_ec_reco = TrackerHitReconstruction("vtx_ec_reco", vtx_ec_reco = TrackerHitReconstruction("vtx_ec_reco",
inputHitCollection = vtx_ec_digi.outputHitCollection, inputHitCollection = vtx_ec_digi.outputHitCollection,
outputHitCollection="VertexEndcapRecHits") outputHitCollection="VertexEndcapRecHits")
algorithms.append(vtx_ec_reco)
gem_ec_reco = TrackerHitReconstruction("gem_ec_reco", gem_ec_reco = TrackerHitReconstruction("gem_ec_reco",
inputHitCollection=gem_ec_digi.outputHitCollection, inputHitCollection=gem_ec_digi.outputHitCollection,
outputHitCollection="GEMTrackerEndcapRecHits") outputHitCollection="GEMTrackerEndcapRecHits")
algorithms.append(gem_ec_reco)
# Tracking hit collector # Tracking hit collector
trk_hit_col = TrackingHitsCollector("trk_hit_col", trk_hit_col = TrackingHitsCollector("trk_hit_col",
...@@ -476,34 +506,20 @@ trk_hit_col = TrackingHitsCollector("trk_hit_col", ...@@ -476,34 +506,20 @@ trk_hit_col = TrackingHitsCollector("trk_hit_col",
str(vtx_ec_reco.outputHitCollection), str(vtx_ec_reco.outputHitCollection),
str(gem_ec_reco.outputHitCollection) ], str(gem_ec_reco.outputHitCollection) ],
trackingHits="trackingHits") trackingHits="trackingHits")
algorithms.append(trk_hit_col)
# Hit Source linker # Hit Source linker
sourcelinker = TrackerSourcesLinker("trk_srcslnkr", sourcelinker = TrackerSourcesLinker("trk_srcslnkr",
inputHitCollections = trk_hit_col.trackingHits, inputHitCollections = trk_hit_col.trackingHits,
outputSourceLinks = "TrackerSourceLinks", outputSourceLinks = "TrackerSourceLinks",
outputMeasurements = "TrackerMeasurements") outputMeasurements = "TrackerMeasurements")
algorithms.append(sourcelinker)
#trk_hits_srclnkr = TrackerSourcesLinker("trk_srcslnkr",
# ITrackerBarrelHits = vtx_b_reco.outputHitCollection,
# ITrackerEndcapHits = vtx_ec_reco.outputHitCollection,
# OTrackerBarrelHits = trk_b_reco.outputHitCollection,
# OTrackerEndcapHits = trk_ec_reco.outputHitCollection,
# outputSourceLinks="TrackerMeasurements")
## Track param init ## Track param init
truth_trk_init = TrackParamTruthInit("truth_trk_init", truth_trk_init = TrackParamTruthInit("truth_trk_init",
inputMCParticles="mcparticles", inputMCParticles="mcparticles",
outputInitialTrackParameters="InitTrackParams") outputInitialTrackParameters="InitTrackParams")
algorithms.append(truth_trk_init)
#clust_trk_init = TrackParamClusterInit("clust_trk_init",
# inputClusters="SimpleClusters",
# outputInitialTrackParameters="InitTrackParamsFromClusters")
#vtxcluster_trk_init = TrackParamVertexClusterInit("vtxcluster_trk_init",
# inputVertexHits="VertexBarrelRecHits",
# inputClusters="EcalBarrelClusters",
# outputInitialTrackParameters="InitTrackParamsFromVtxClusters",
# maxHitRadius=40.0*units.mm)
# Tracking algorithms # Tracking algorithms
trk_find_alg = TrackFindingAlgorithm("trk_find_alg", trk_find_alg = TrackFindingAlgorithm("trk_find_alg",
...@@ -511,42 +527,24 @@ trk_find_alg = TrackFindingAlgorithm("trk_find_alg", ...@@ -511,42 +527,24 @@ trk_find_alg = TrackFindingAlgorithm("trk_find_alg",
inputMeasurements = sourcelinker.outputMeasurements, inputMeasurements = sourcelinker.outputMeasurements,
inputInitialTrackParameters = "InitTrackParams", inputInitialTrackParameters = "InitTrackParams",
outputTrajectories = "trajectories") outputTrajectories = "trajectories")
algorithms.append(trk_find_alg)
parts_from_fit = ParticlesFromTrackFit("parts_from_fit", parts_from_fit = ParticlesFromTrackFit("parts_from_fit",
inputTrajectories = "trajectories", inputTrajectories = "trajectories",
outputParticles = "ReconstructedParticles", outputParticles = "ReconstructedParticles",
outputTrackParameters = "outputTrackParameters") outputTrackParameters = "outputTrackParameters")
algorithms.append(parts_from_fit)
#trk_find_alg1 = TrackFindingAlgorithm("trk_find_alg1",
# inputSourceLinks = sourcelinker.outputSourceLinks,
# inputMeasurements = sourcelinker.outputMeasurements,
# inputInitialTrackParameters= "InitTrackParamsFromClusters",
# outputTrajectories="trajectories1")
#parts_from_fit1 = ParticlesFromTrackFit("parts_from_fit1",
# inputTrajectories="trajectories1",
# outputParticles="ReconstructedParticlesInitFromClusters",
# outputTrackParameters="outputTrackParameters1")
#trk_find_alg2 = TrackFindingAlgorithm("trk_find_alg2",
# inputSourceLinks = trk_hits_srclnkr.outputSourceLinks,
# inputMeasurements = trk_hits_srclnkr.outputMeasurements,
# inputInitialTrackParameters= "InitTrackParamsFromVtxClusters",
# outputTrajectories="trajectories2")
#parts_from_fit2 = ParticlesFromTrackFit("parts_from_fit2",
# inputTrajectories="trajectories2",
# outputParticles="ReconstructedParticles2",
# outputTrackParameters="outputTrackParameters2")
pmtdigi = PhotoMultiplierDigi("pmtdigi", pmtdigi = PhotoMultiplierDigi("pmtdigi",
inputHitCollection="DRICHHits", inputHitCollection="DRICHHits",
outputHitCollection="DRICHHitsDigi", outputHitCollection="DRICHHitsDigi",
quantumEfficiency=[(a*units.eV, b) for a, b in qe_data]) quantumEfficiency=[(a*units.eV, b) for a, b in qe_data])
algorithms.append(pmtdigi)
pmtreco = PhotoMultiplierReco("pmtreco", pmtreco = PhotoMultiplierReco("pmtreco",
inputHitCollection="DRICHHitsDigi", inputHitCollection="DRICHHitsDigi",
outputHitCollection="DRICHHitsReco") outputHitCollection="DRICHHitsReco")
algorithms.append(pmtreco)
# FIXME # FIXME
#richcluster = PhotoRingClusters("richcluster", #richcluster = PhotoRingClusters("richcluster",
...@@ -555,6 +553,7 @@ pmtreco = PhotoMultiplierReco("pmtreco", ...@@ -555,6 +553,7 @@ pmtreco = PhotoMultiplierReco("pmtreco",
# outputClusterCollection="ForwardRICHClusters") # outputClusterCollection="ForwardRICHClusters")
# Output # Output
podout = PodioOutput("out", filename=output_rec)
podout.outputCommands = [ podout.outputCommands = [
"keep *", "keep *",
"drop *Digi", "drop *Digi",
...@@ -565,6 +564,7 @@ podout.outputCommands = [ ...@@ -565,6 +564,7 @@ podout.outputCommands = [
"drop InitTrackParams", "drop InitTrackParams",
"drop mcparticles" "drop mcparticles"
] ]
algorithms.append(podout)
ApplicationMgr( ApplicationMgr(
TopAlg = algorithms, TopAlg = algorithms,
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment