def MuonClusterSegmentFinderCfg(flags, **kwargs):
#declareProperty("MuonClusterizationTool", m_clusterTool);
#declareProperty("MuonEDMPrinterTool",m_printer );
#declareProperty("MuonPRDSelectionTool", m_muonPRDSelectionTool );
#declareProperty("MdtSegmentMaker",m_segmentMaker);
#declareProperty("SLFitter", m_slTrackFitter);
#declareProperty("TrackToSegmentTool", m_trackToSegmentTool);
#declareProperty("AmbiguityProcessor",m_ambiguityProcessor);
#declareProperty("TrackCleaner", m_trackCleaner);
result = ComponentAccumulator()
# FIXME - rest of the tools.
acc = DCMathSegmentMakerCfg(flags, name="DCMathSegmentMaker")
segment_maker = acc.getPrimary()
acc.addPublicTool(segment_maker)
kwargs.setdefault('MdtSegmentMaker', segment_maker)
result.merge(acc)
acc = MCTBFitterCfg(flags, name="SLFitter", StraightLine=True)
slfitter = acc.getPrimary()
acc.addPublicTool(slfitter)
result.merge(acc)
kwargs.setdefault("SLFitter", slfitter)
acc = MuonAmbiProcessorCfg(flags, name='NewMuonAmbiProcessor')
ambi = acc.getPrimary()
acc.addPublicTool(ambi)
result.merge(acc)
kwargs.setdefault("AmbiguityProcessor", ambi)
acc = MuonTrackCleanerCfg(flags)
cleaner = acc.getPrimary()
acc.addPublicTool(cleaner)
result.merge(acc)
kwargs.setdefault("TrackCleaner", cleaner)
result.addPublicTool(Muon__MuonClusterSegmentFinder(**kwargs),
primary=True)
return result
def MooTrackBuilderCfg(flags, name="MooTrackBuilderTemplate", **kwargs):
Muon__MooTrackBuilder = CompFactory.Muon.MooTrackBuilder
Trk__STEP_Propagator = CompFactory.Trk.STEP_Propagator
from MuonConfig.MuonRIO_OnTrackCreatorConfig import MdtDriftCircleOnTrackCreatorCfg, TriggerChamberClusterOnTrackCreatorCfg
from MuonConfig.MuonRecToolsConfig import MuonTrackToSegmentToolCfg
from MagFieldServices.MagFieldServicesConfig import MagneticFieldSvcCfg
# Based on this: https://gitlab.cern.ch/atlas/athena/blob/release/22.0.3/MuonSpectrometer/MuonReconstruction/MuonRecExample/python/MooreTools.py#L221
# ignoring all the name_prefix stuff for the moment, since I'm not sure it's necessary any more.
result = MooTrackFitterCfg(flags)
moo_track_fitter = result.getPrimary()
result.addPublicTool(moo_track_fitter)
kwargs.setdefault("Fitter", moo_track_fitter)
# Now setup SL fitter (MooSLTrackFitter as defined here:
# https://gitlab.cern.ch/atlas/athena/blob/release/22.0.3/MuonSpectrometer/MuonReconstruction/MuonRecExample/python/MuonRecExampleConfigDb.py#L203)
acc = MCTBFitterCfg(flags, name='MCTBSLFitter', StraightLine=True)
mctbslfitter = acc.getPrimary()
result.addPublicTool(mctbslfitter)
result.merge(acc)
# Just take the default configuration, as per https://gitlab.cern.ch/atlas/athena/blob/release/22.0.3/MuonSpectrometer/MuonReconstruction/MuonRecExample/python/MuonRecExampleConfigDb.py#L56
# FIXME - this should be updated once there is a proper CA implementation for the STEP_Propagator
prop = Trk__STEP_Propagator(name='MuonStraightLinePropagator')
result.addPublicTool(prop)
# TODO - check why Fitter and FitterPreFit are identical
acc = MooTrackFitterCfg(flags,
name="MooSLTrackFitter",
Fitter=mctbslfitter,
FitterPreFit=mctbslfitter,
Propagator=prop,
ReducedChi2Cut=10.0,
SLFit=True)
moo_sl_track_fitter = acc.getPrimary()
result.addPublicTool(moo_sl_track_fitter)
result.merge(acc)
kwargs.setdefault("SLFitter", moo_sl_track_fitter)
kwargs.setdefault(
"RecalibrateMDTHitsOnTrack",
((not flags.Muon.doSegmentT0Fit) and flags.Beam.Type == 'collisions'))
acc = MuonSeededSegmentFinderCfg(flags)
muon_seeded_segment_finder = acc.getPrimary()
result.addPublicTool(muon_seeded_segment_finder)
result.merge(acc)
kwargs.setdefault("SeededSegmentFinder", muon_seeded_segment_finder)
acc = MdtDriftCircleOnTrackCreatorCfg(flags)
mdt_dcot_creator = acc.getPrimary()
kwargs.setdefault("MdtRotCreator", mdt_dcot_creator)
result.merge(acc)
acc = TriggerChamberClusterOnTrackCreatorCfg(flags)
muon_comp_cluster_creator = acc.getPrimary()
kwargs.setdefault("CompetingClustersCreator", muon_comp_cluster_creator)
result.merge(acc)
acc = MuonSTEP_PropagatorCfg(flags)
muon_prop = acc.getPrimary()
kwargs.setdefault("Propagator", muon_prop)
result.merge(acc)
acc = MuonChamberHoleRecoveryToolCfg(flags)
hole_recovery_tool = acc.getPrimary()
result.addPublicTool(hole_recovery_tool)
result.merge(acc)
kwargs.setdefault("HitRecoveryTool", hole_recovery_tool)
kwargs.setdefault("ChamberHoleRecoveryTool",
hole_recovery_tool) # FIXME? Remove duplicate from cxx?
acc = MagneticFieldSvcCfg(flags)
result.merge(acc)
acc = MuonTrackToSegmentToolCfg(flags)
track_to_segment_tool = acc.getPrimary()
kwargs.setdefault("TrackToSegmentTool", track_to_segment_tool)
result.merge(acc)
kwargs.setdefault("Printer", MuonEDMPrinterTool(flags))
# FIXME - remove ErrorOptimisationTool from cxx?
# declareProperty("ErrorOptimisationTool","" );
acc = MuPatCandidateToolCfg(flags)
cand_tool = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("CandidateTool", cand_tool)
acc = MooCandidateMatchingToolCfg(flags)
track_segment_matching_tool = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("CandidateMatchingTool", track_segment_matching_tool)
from MuonConfig.MuonRecToolsConfig import MuonTrackSummaryToolCfg
acc = MuonTrackSummaryToolCfg(flags)
track_summary = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("TrackSummaryTool", track_summary)
builder = Muon__MooTrackBuilder(name, **kwargs)
result.setPrivateTools(builder)
return result
def MooTrackFitterCfg(flags, name='MooTrackFitter', **kwargs):
# Based on https://gitlab.cern.ch/atlas/athena/blob/release/22.0.3/MuonSpectrometer/MuonReconstruction/MuonRecExample/python/MooreTools.py#L179
# (Tons of explicit configuration missing from there though)
Muon__MooTrackFitter = CompFactory.Muon.MooTrackFitter
MuonSegmentMomentum = CompFactory.MuonSegmentMomentum
from MuonConfig.MuonRecToolsConfig import MuonPhiHitSelector, MuonTrackToSegmentToolCfg, MuonTrackSummaryHelperToolCfg, MuPatHitToolCfg
from MuonConfig.MuonRIO_OnTrackCreatorConfig import MdtDriftCircleOnTrackCreatorCfg
result = MCTBFitterCfg(flags)
mctb_fitter = result.getPrimary()
result.addPublicTool(mctb_fitter)
kwargs.setdefault("Fitter", mctb_fitter)
kwargs.setdefault("FitterPreFit", mctb_fitter)
# FIXME MuPatHitTool currently isn't configured, but should be.
MuPatHitToolCfg
acc = MuPatHitToolCfg(flags)
mu_pat_hit_tool = acc.getPrimary()
result.addPublicTool(mu_pat_hit_tool)
result.merge(acc)
kwargs.setdefault("HitTool", mu_pat_hit_tool)
acc = MuonSTEP_PropagatorCfg(flags)
muon_prop = acc.getPrimary()
result.addPublicTool(muon_prop)
result.merge(acc)
kwargs.setdefault("Propagator", muon_prop)
# kwargs.setdefault("SLFit" , ) # Was "not jobproperties.BField.allToroidOn()" but do not have access to Field here.
kwargs.setdefault("ReducedChi2Cut", flags.Muon.Chi2NDofCut)
momentum_estimator = ""
if flags.Beam.Type == 'cosmics':
momentum_estimator = MuonSegmentMomentum(DoCosmics=True)
else:
acc = MuonSegmentMomentumFromFieldCfg(flags)
momentum_estimator = acc.getPrimary()
result.merge(acc)
result.addPublicTool(momentum_estimator)
kwargs.setdefault("SegmentMomentum", momentum_estimator)
kwargs.setdefault("MuonPrinterTool", MuonEDMPrinterTool(flags))
acc = MuonTrackToSegmentToolCfg(flags)
track_to_segment_tool = acc.getPrimary()
kwargs.setdefault("TrackToSegmentTool", track_to_segment_tool)
result.merge(acc)
acc = MdtDriftCircleOnTrackCreatorCfg(flags)
mdt_dcot_creator = acc.getPrimary()
kwargs.setdefault("MdtRotCreator", mdt_dcot_creator)
result.merge(acc)
kwargs.setdefault("PhiHitSelector", MuonPhiHitSelector(flags))
acc = MuonTrackCleanerCfg(flags)
track_cleaner = acc.getPrimary()
result.merge(acc)
result.addPublicTool(track_cleaner)
kwargs.setdefault("TrackCleaner", track_cleaner)
# Leaving "SegmentInOverlapTool" as default, which is what happens in the current configuration.
acc = MuonTrackSummaryHelperToolCfg(flags)
track_summary_helper = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("TrackSummaryTool", track_summary_helper)
kwargs.setdefault("CleanPhiHits", True)
kwargs.setdefault("UsePreciseHits", True)
kwargs.setdefault("UsePrefit", False)
kwargs.setdefault("SeedAtStartOfTrack", False)
fitter = Muon__MooTrackFitter(name, **kwargs)
result.setPrivateTools(fitter)
return result
def MuonTrackSteeringCfg(flags, name="MuonTrackSteering", **kwargs):
Muon__MuonTrackSteering = CompFactory.Muon.MuonTrackSteering
# This is based on https://gitlab.cern.ch/atlas/athena/blob/release/22.0.3/MuonSpectrometer/MuonReconstruction/MuonRecExample/python/MuonStandalone.py#L35
result = ComponentAccumulator()
# This was originally defined here: https://gitlab.cern.ch/atlas/athena/blob/release/22.0.3/MuonSpectrometer/MuonReconstruction/MuonRecExample/python/MuonStandaloneFlags.py#L137
MoorelikeStrategy = [
"BarrelCombined[CombineSegInStation,DoRefinement,DoAmbiSolving,CutSeedsOnTracks]:BML,BMS;BOL,BOS;BIL,BIS",
"EndcapCombined[CombineSegInStation,DoRefinement,DoAmbiSolving,CutSeedsOnTracks]:EML,EMS;EOL,EOS;EIL,EIS;CSL,CSS;EEL,EES;BEE",
"BarrelEndcap[CombineSegInStation,DoRefinement,DoAmbiSolving,CutSeedsOnTracks,BarrelEndcapFilter]:EML,EMS;EEL,EES;BML,BMS;EIL,EIS;BIL,BIS;BOL,BOS"
]
kwargs.setdefault("StrategyList", MoorelikeStrategy)
# Set properties which were set in MuonStandalone.py, and which aren't default (fix this later, because we don't want to break existing configuration)
# extraFlags are ignored (don't believe they're necessary)
result = MCTBFitterCfg(flags, name='MCTBSLFitter', StraightLine=True)
mctbslfitter = result.getPrimary()
acc = MuonSegmentRegionRecoveryToolCfg(flags,
name='MuonEORecoveryTool',
OnlyEO=True,
Fitter=mctbslfitter,
UseFitterOutlierLogic=False)
muon_eo_recovery_tool = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("HoleRecoveryTool", muon_eo_recovery_tool)
kwargs.setdefault("SegSeedQCut", 2)
kwargs.setdefault("Seg2ndQCut", 1)
kwargs.setdefault("OutputSingleStationTracks", True) # default is False
if "TrackBuilderTool" not in kwargs:
#Old config had: kwargs["TrackBuilderTool"] = getPublicToolClone("MooMuonTrackBuilder", "MooTrackBuilderTemplate",
# extraFlags=extraFlags)
# For some strange reason, MooTrackBuilderTemplate was just a MooTrackBuilder i.e. they had:
# addTool( "MuonRecExample.MooreTools.MooTrackBuilder", "MooTrackBuilderTemplate")
# I have simplified this. Also, in the old configuration there was the possiblity to set optimiseMomentumResolutionUsingChi2 to true in extraflags (which were passed to TrackBuilderTool but not TrackRefinementTool) - however I don't think this was used, and have removed it.
# However some other things were set in extraFlags here: https://gitlab.cern.ch/atlas/athena/blob/release/22.0.3/MuonSpectrometer/MuonReconstruction/MuonRecExample/python/MuonStandalone.py#L39
# So I might well have made a mistake. TODO check
acc = MooTrackBuilderCfg(flags)
builder = acc.getPrimary()
kwargs["TrackBuilderTool"] = builder
result.merge(acc)
if "TrackRefinementTool" not in kwargs:
kwargs["TrackRefinementTool"] = builder
# FIXME. Need to see if TrackRefinementTool is actually used & possibly remove.
# --- Now let's setup tools which were ignored in the old configuration. Just want to get this working for the moment. Will clean it up later.
acc = MuPatCandidateToolCfg(flags)
cand_tool = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("MuPatCandidateTool", cand_tool)
from MuonConfig.MuonRecToolsConfig import MuonAmbiProcessorCfg
acc = MuonAmbiProcessorCfg(flags, name='MuonAmbiProcessor')
ambi = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("AmbiguityTool", ambi)
kwargs.setdefault("MooBuilderTool", builder)
kwargs.setdefault("TrackRefinementTool", builder)
acc = MooCandidateMatchingToolCfg(flags)
cand_matching_tool = acc.getPrimary()
kwargs["CandidateMatchingTool"] = cand_matching_tool
result.merge(acc)
acc = MuonSegmentFittingToolCfg(flags)
seg_fitting_tool = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("MuonSegmentFittingTool", seg_fitting_tool)
kwargs.setdefault("MuonTrackSelector", MuonTrackSelector(flags))
acc = MuonChamberHoleRecoveryToolCfg(flags)
hole_recovery_tool = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("HoleRecoveryTool", hole_recovery_tool)
from MuonConfig.MuonRecToolsConfig import MuonTrackSummaryToolCfg
acc = MuonTrackSummaryToolCfg(flags)
kwargs.setdefault("TrackSummaryTool", acc.getPrimary())
result.merge(acc)
track_maker_steering = Muon__MuonTrackSteering(name, **kwargs)
result.setPrivateTools(track_maker_steering)
return result
def MuonSegmentRegionRecoveryToolCfg(flags,
name="MuonSegmentRegionRecoveryTool",
**kwargs):
Muon__MuonSegmentRegionRecoveryTool = CompFactory.Muon.MuonSegmentRegionRecoveryTool
from MuonConfig.MuonRecToolsConfig import MuonExtrapolatorCfg, MuonStationIntersectSvcCfg, MuonTrackSummaryToolCfg
# Based on https://gitlab.cern.ch/atlas/athena/blob/release/22.0.3/MuonSpectrometer/MuonReconstruction/MuonRecExample/python/MooreTools.py#L426
result = MuonSeededSegmentFinderCfg(flags)
muon_seeded_segment_finder = result.getPrimary()
kwargs.setdefault("SeededSegmentFinder", muon_seeded_segment_finder)
acc = MooCandidateMatchingToolCfg(flags)
track_segment_matching_tool = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("TrackSegmentMatchingTool", track_segment_matching_tool)
acc = MuonChamberHoleRecoveryToolCfg(flags)
hole_recovery_tool = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("ChamberHoleRecoveryTool", hole_recovery_tool)
acc = MuonExtrapolatorCfg(flags)
extrap = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("Extrapolator", extrap)
acc = MCTBFitterCfg(flags, name='MCTBSLFitter', StraightLine=True)
mctbslfitter = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("Fitter", mctbslfitter)
acc = MuonStationIntersectSvcCfg(flags)
msis = acc.getPrimary()
result.merge(acc)
kwargs.setdefault("MuonStationIntersectSvc", msis)
# Not bothering to handle IDHelper or EDMHelper or HitSummaryTool. Default is okay.
from RegionSelector.RegSelToolConfig import regSelTool_MDT_Cfg, regSelTool_RPC_Cfg, regSelTool_TGC_Cfg
acc = regSelTool_MDT_Cfg(flags)
kwargs.setdefault("MDTRegionSelector", acc.popPrivateTools())
result.merge(acc)
acc = regSelTool_TGC_Cfg(flags)
kwargs.setdefault("TGCRegionSelector", acc.popPrivateTools())
result.merge(acc)
acc = regSelTool_RPC_Cfg(flags)
kwargs.setdefault("RPCRegionSelector", acc.popPrivateTools())
result.merge(acc)
if flags.Detector.GeometryCSC:
from RegionSelector.RegSelToolConfig import regSelTool_CSC_Cfg
acc = regSelTool_CSC_Cfg(flags)
kwargs.setdefault("CSCRegionSelector", acc.popPrivateTools())
result.merge(acc)
else:
kwargs.setdefault("CSCRegionSelector", "")
if flags.Detector.GeometrysTGC:
from RegionSelector.RegSelToolConfig import regSelTool_STGC_Cfg
acc = regSelTool_STGC_Cfg(flags)
kwargs.setdefault("STGCRegionSelector", acc.popPrivateTools())
result.merge(acc)
else:
kwargs.setdefault("STGCRegionSelector", "")
if flags.Detector.GeometryMM:
from RegionSelector.RegSelToolConfig import regSelTool_MM_Cfg
acc = regSelTool_MM_Cfg(flags)
kwargs.setdefault("MMRegionSelector", acc.popPrivateTools())
result.merge(acc)
else:
kwargs.setdefault("MMRegionSelector", "")
acc = MuonTrackSummaryToolCfg(flags)
kwargs.setdefault("TrackSummaryTool", acc.getPrimary())
result.merge(acc)
segment_region_recovery_tool = Muon__MuonSegmentRegionRecoveryTool(
name, **kwargs)
result.setPrivateTools(segment_region_recovery_tool)
return result
def MuonClusterSegmentFinderToolCfg(flags, **kwargs):
#m_slTrackFitter("Trk::GlobalChi2Fitter/MCTBSLFitter"),
#m_ambiTool("Trk::SimpleAmbiguityProcessorTool/MuonAmbiProcessor"),
#m_trackToSegmentTool("Muon::MuonTrackToSegmentTool/MuonTrackToSegmentTool"),
#m_printer("Muon::MuonEDMPrinterTool/MuonEDMPrinterTool"),
#m_edmHelperSvc("Muon::MuonEDMHelperSvc/MuonEDMHelperSvc"),
#m_trackCleaner("Muon::MuonTrackCleaner/MuonTrackCleaner") {
#declareProperty("SLFitter", m_slTrackFitter);
#declareProperty("SegmentAmbiguityTool",m_ambiTool);
#declareProperty("TrackToSegmentTool", m_trackToSegmentTool);
#declareProperty("TrackCleaner", m_trackCleaner);
result = ComponentAccumulator()
acc = MCTBFitterCfg(flags, name="SLFitter", StraightLine=True)
slfitter = acc.getPrimary()
result.addPublicTool(slfitter)
result.merge(acc)
kwargs.setdefault("SLFitter", slfitter)
acc = MuonTrackCleanerCfg(flags)
cleaner = acc.getPrimary()
result.addPublicTool(cleaner)
kwargs.setdefault("TrackCleaner", cleaner)
result.merge(acc)
acc = MuonTrackSummaryToolCfg(flags)
track_summary = acc.getPrimary()
result.addPublicTool(track_summary)
kwargs.setdefault('TrackSummaryTool', track_summary)
result.merge(acc)
acc = MuonAmbiProcessorCfg(flags, name='MuonAmbiProcessor')
ambi = acc.popPrivateTools()
acc.addPublicTool(ambi)
result.merge(acc)
kwargs.setdefault("SegmentAmbiguityTool", ambi)
# FIXME - remaining tools
result.setPrivateTools(Muon__MuonClusterSegmentFinderTool(**kwargs))
return result
def MuonSegmentFittingToolCfg(flags, **kwargs):
# declareProperty("SLPropagator", m_slPropagator);
# declareProperty("SLFitter", m_slTrackFitter);
# declareProperty("CurvedFitter", m_curvedTrackFitter);
# declareProperty("TrackCleaner", m_trackCleaner);
# declareProperty("UpdatePrecisionCoordinate", m_updatePrecisionCoordinate = false );
result = ComponentAccumulator()
# FIXME! Add this.
# acc, propagator = AtlasRungeKuttaPropagatorCfg(flags)
# result.merge(acc)
# kwargs.setdefault("SLPropagator", propagator)
acc = MCTBFitterCfg(flags, name="SLFitter", StraightLine=True)
slfitter = acc.getPrimary()
acc.addPublicTool(slfitter)
result.merge(acc)
kwargs.setdefault("SLFitter", slfitter)
acc = MCTBFitterCfg(flags, name="CurvedFitter")
fitter = acc.getPrimary()
acc.addPublicTool(fitter)
result.merge(acc)
kwargs.setdefault("CurvedFitter", fitter)
acc = MuonTrackCleanerCfg(flags)
cleaner = acc.getPrimary()
acc.addPublicTool(cleaner)
result.merge(acc)
kwargs.setdefault("TrackCleaner", cleaner)
result.setPrivateTools(Muon__MuonSegmentFittingTool(**kwargs))
return result