Skip to content
Open
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
93 changes: 93 additions & 0 deletions StandardConfig/production/ILDReconstruction.py
Original file line number Diff line number Diff line change
Expand Up @@ -163,6 +163,34 @@
action="store_true",
)

parser.add_argument(
Comment thread
armin-ilg marked this conversation as resolved.
"--VXDBarrelDigitiserResolutionU",
action="store",
default=["0.003"],
help="Resolution in U direction for VXD Barrel digitisation", nargs="+"
)

parser.add_argument(
"--VXDBarrelDigitiserResolutionV",
action="store",
default=["0.003"],
help="Resolution in V direction for VXD Barrel digitisation", nargs="+"
)

parser.add_argument(
"--VXDEndcapDigitiserResolutionU",
action="store",
default=["0.003"],
help="Resolution in U direction for VXD Endcap digitisation", nargs="+"
)

parser.add_argument(
"--VXDEndcapDigitiserResolutionV",
action="store",
default=["0.003"],
help="Resolution in V direction for VXD Endcap digitisation", nargs="+"
)


def get_compact_file_path(detector_model: str):
"""returns the compact file path to a specified detector model starting with the path stored in the 'K4GEO' environment variable"""
Expand Down Expand Up @@ -241,6 +269,10 @@ def get_cms_energy_config(
"CONSTANTS": CONSTANTS,
"cms_energy_config": cms_energy_config,
"using_particle_gun": reco_args.usingParticleGun,
"VXDBarrelDigitiserResolutionU": reco_args.VXDBarrelDigitiserResolutionU,
"VXDBarrelDigitiserResolutionV": reco_args.VXDBarrelDigitiserResolutionV,
"VXDEndcapDigitiserResolutionU": reco_args.VXDEndcapDigitiserResolutionU,
"VXDEndcapDigitiserResolutionV": reco_args.VXDEndcapDigitiserResolutionV,
},
)

Expand Down Expand Up @@ -295,6 +327,67 @@ def get_cms_energy_config(

if not is_FCCee_model:
sequenceLoader.load("HighLevelReco/HighLevelReco")
else:
MyRecoMCTruthLinker = MarlinProcessorWrapper("MyRecoMCTruthLinker")
MyRecoMCTruthLinker.ProcessorType = "RecoMCTruthLinker"
MyRecoMCTruthLinker.Parameters = {
"CalohitMCTruthLinkName": ["CalohitMCTruthLink"],
"ClusterCollection": ["PandoraClusters"],
"ClusterMCTruthLinkName": ["ClusterMCTruthLink"],
"FullRecoRelation": ["true"],
"KeepDaughtersPDG": ["22", "111", "310", "13", "211", "321"],
"MCParticleCollection": ["MCParticle"],
"MCParticlesSkimmedName": ["MCParticlesSkimmed"],
"MCTruthClusterLinkName": ["MCTruthClusterLink"],
"MCTruthRecoLinkName": ["MCTruthRecoLink"],
"MCTruthTrackLinkName": ["MCTruthMarlinTrkTracksLink"],
"RecoMCTruthLinkName": ["RecoMCTruthLink"],
"RecoParticleCollection": ["PandoraPFOs"],
"SimCaloHitCollections": [
"BeamCalCollection",
"LHCalCollection",
"LumiCalCollection",
CONSTANTS["ECalSimHitCollections"],
CONSTANTS["HCalSimHitCollections"],
"YokeBarrelCollection",
"YokeEndcapsCollection",
],
"SimCalorimeterHitRelationNames": [
"EcalBarrelRelationsSimRec",
"EcalEndcapRingRelationsSimRec",
"EcalEndcapsRelationsSimRec",
"HcalBarrelRelationsSimRec",
"HcalEndcapRingRelationsSimRec",
"HcalEndcapsRelationsSimRec",
"RelationLHcalHit",
"RelationMuonHit",
"RelationLcalHit",
"RelationBCalHit",
],
"SimTrackerHitCollections": [
"VXDCollection",
"SITCollection",
"FTD_PIXELCollection",
"FTD_STRIPCollection",
Comment thread
armin-ilg marked this conversation as resolved.
Outdated
"TPCCollection",
"SETCollection",
],
"TrackCollection": ["MarlinTrkTracks"],
"TrackMCTruthLinkName": ["MarlinTrkTracksMCTruthLink"],
"TrackerHitsRelInputCollections": [
"VXDTrackerHitRelations",
"SITTrackerHitRelations",
"FTDPixelTrackerHitRelations",
"FTDSpacePointRelations",
Comment thread
armin-ilg marked this conversation as resolved.
Outdated
"TPCTrackerHitRelations",
"SETSpacePointRelations",
],
"UseTrackerHitRelations": ["true"],
"UsingParticleGun": ["true"],
}
algList.append(MyRecoMCTruthLinker)
Comment thread
armin-ilg marked this conversation as resolved.
Outdated



if not reco_args.noPFO:
MyPfoAnalysis = MarlinProcessorWrapper("MyPfoAnalysis")
Expand Down
12 changes: 6 additions & 6 deletions StandardConfig/production/Tracking/TrackingDigi_FCCeeMDI.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,23 +10,23 @@
VertexBarrelDigitiser.ProcessorType = "DDPlanarDigiProcessor"
VertexBarrelDigitiser.Parameters = {
"IsStrip": ["false"],
"ResolutionU": ["0.003", "0.003", "0.003", "0.003", "0.003", "0.003"],
"ResolutionV": ["0.003", "0.003", "0.003", "0.003", "0.003", "0.003"],
"ResolutionU": [str(x) for x in VXDBarrelDigitiserResolutionU],
"ResolutionV": [str(x) for x in VXDBarrelDigitiserResolutionV],
Comment thread
armin-ilg marked this conversation as resolved.
"SimTrackHitCollectionName": ["VertexBarrelCollection"],
"SimTrkHitRelCollection": ["VertexBarrelTrackerHitRelations"],
"SubDetectorName": ["VertexBarrel"],
"SubDetectorName": ["Vertex"],
"TrackerHitCollectionName": ["VertexBarrelTrackerHits"],
}

VertexEndcapDigitiser = MarlinProcessorWrapper("VertexEndcapDigitiser")
VertexEndcapDigitiser.ProcessorType = "DDPlanarDigiProcessor"
VertexEndcapDigitiser.Parameters = {
"IsStrip": ["false"],
"ResolutionU": ["0.003", "0.003", "0.003", "0.003", "0.003", "0.003"],
"ResolutionV": ["0.003", "0.003", "0.003", "0.003", "0.003", "0.003"],
"ResolutionU": [str(x) for x in VXDEndcapDigitiserResolutionU],
"ResolutionV": [str(x) for x in VXDEndcapDigitiserResolutionV],
"SimTrackHitCollectionName": ["VertexEndcapCollection"],
"SimTrkHitRelCollection": ["VertexEndcapTrackerHitRelations"],
"SubDetectorName": ["VertexEndcap"],
"SubDetectorName": ["Vertex"],
"TrackerHitCollectionName": ["VertexEndcapTrackerHits"],
}

Expand Down
36 changes: 18 additions & 18 deletions StandardConfig/production/Tracking/TrackingReco_FCCeeMDI.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

from py_utils import encode_CT_steps_dict_to_legacy_list

CT_MAX_DIST = "0.03" # RANDOM VALUE COPYIED FROM CLDRECO
CT_MAX_DIST = "0.05" # RANDOM VALUE COPYIED FROM CLDRECO
Comment thread
armin-ilg marked this conversation as resolved.
Outdated
MCPartColName = ["MCParticle"] # MCParticleCollectionName
VertexBarrelHitCollectionNames = ["VertexBarrelTrackerHits"]
VertexEndcapHitCollectionNames = ["VertexEndcapTrackerHits"]
Expand Down Expand Up @@ -38,15 +38,15 @@
"TrackStartsInnerDist": ["25"],
"TrackSystemName": ["DDKalTest"],
"VXDHitCollection": ["VertexBarrelTrackerHits"],
"VXDDetectorName": ["VertexBarrel"],
"VXDDetectorName": ["Vertex"],
"pickUpSiHits": ["true"],
}

MyConformalTracking = MarlinProcessorWrapper("MyConformalTracking")
MyConformalTracking.ProcessorType = "ConformalTrackingV2"
conformal_tracking_steps_config = {
# Based on CLD's Reconstruction in CLDConfig
"VertexBarrel": {
"Vertex": {
"collections": ["VertexBarrelTrackerHits"],
"params": {
"MaxCellAngle": 0.01,
Expand All @@ -60,7 +60,7 @@
"flags": ["HighPTFit", "VertexToTracker"],
"functions": ["CombineCollections", "BuildNewTracks"],
},
"VertexEncap": {
"Vertex": {
"collections": ["VertexEndcapTrackerHits"],
"params": {
"MaxCellAngle": 0.01,
Expand Down Expand Up @@ -92,20 +92,20 @@
"flags": ["HighPTFit", "VertexToTracker", "RadialSearch"],
"functions": ["CombineCollections", "BuildNewTracks"],
},
# "LowerCellAngle2": {
# "collections": "",
# "params": {
# "MaxCellAngle": 0.1,
# "MaxCellAngleRZ": 0.1,
# "Chi2Cut": 2000,
# "MinClustersOnTrack": 4,
# "MaxDistance": CT_MAX_DIST,
# "SlopeZRange": 10.0,
# "HighPTCut": 10.0,
# },
# "flags": ["HighPTFit", "VertexToTracker", "RadialSearch"],
# "functions": ["BuildNewTracks","SortTracks"],
# },
"LowerCellAngle2": {
"collections": [""],
"params": {
"MaxCellAngle": 0.1,
"MaxCellAngleRZ": 0.1,
"Chi2Cut": 2000,
"MinClustersOnTrack": 4,
"MaxDistance": CT_MAX_DIST,
"SlopeZRange": 10.0,
"HighPTCut": 10.0,
},
"flags": ["HighPTFit", "VertexToTracker", "RadialSearch"],
"functions": ["BuildNewTracks","SortTracks"],
},
Comment thread
armin-ilg marked this conversation as resolved.
"Tracker": {
"collections": ["InnerTrackerBarrelHits", "InnerTrackerEndcapHits"],
"params": {
Expand Down