from InDetRecExample.InDetJobProperties import InDetFlags
from InDetRecExample.InDetKeys import InDetKeys
doCreation = ( InDetFlags.doNewTracking() or InDetFlags.doPseudoTracking() or InDetFlags.doLargeD0) \
and InDetFlags.doParticleCreation()
doConversion = not InDetFlags.doNewTracking() and not InDetFlags.doPseudoTracking() and not InDetFlags.doLargeD0() \
and InDetFlags.doParticleConversion()
if doCreation:
print "Creating xAOD::TrackParticles from Trk::Tracks"
if doConversion:
print "Converting Rec::TrackParticles to xAOD::TrackParticles"
_perigee_expression=InDetFlags.perigeeExpression()
# need to tread Vertex specifically because at the time of
# the track particle creation the primary vertex does not yet exist.
# The problem is solved by first creating track particles wrt. the beam line
# and correcting the parameters after the vertex finding.
if _perigee_expression == 'Vertex' :
_perigee_expression = 'BeamLine'
from TrkParticleCreator.TrkParticleCreatorConf import Trk__TrackParticleCreatorTool
InDetxAODParticleCreatorTool = Trk__TrackParticleCreatorTool(name = "InDetxAODParticleCreatorTool",
Extrapolator = InDetExtrapolator,
TrackSummaryTool = InDetTrackSummaryToolSharedHits,
BadClusterID = InDetFlags.pixelClusterBadClusterID(),
ForceTrackSummaryUpdate = False,
KeepParameters = True,
PerigeeExpression = _perigee_expression)
def __init__(self,
InputCollections=None,
NewTrackingCuts=None,
TrackCollectionKeys=[],
TrackCollectionTruthKeys=[]):
from InDetRecExample.InDetJobProperties import InDetFlags
from InDetRecExample.InDetKeys import InDetKeys
#
# get ToolSvc and topSequence
#
from AthenaCommon.AppMgr import ToolSvc
from AthenaCommon.AlgSequence import AlgSequence
topSequence = AlgSequence()
#
# --- decide if use the association tool
#
if len(InputCollections) > 0:
usePrdAssociationTool = True
else:
usePrdAssociationTool = False
# --- the PRD association tool is filled by the Segment making
# no need to run again
# ------------------------------------------------------------
#
# ---------- TRT Seeded Tracking
#
# ------------------------------------------------------------
if InDetFlags.doTRTSeededTrackFinder():
#
# --- decide which TRT seed space point finder to use
#
if InDetFlags.loadTRTSeededSPFinder():
#
# --- defaul space point finder
#
from TRT_SeededSpacePointFinderTool.TRT_SeededSpacePointFinderToolConf import InDet__TRT_SeededSpacePointFinder_ATL
InDetTRT_SeededSpacePointFinder = InDet__TRT_SeededSpacePointFinder_ATL(
name='InDetTRT_SeededSpFinder',
SpacePointsSCTName=InDetKeys.SCT_SpacePoints(),
SpacePointsOverlapName=InDetKeys.OverlapSpacePoints(),
AssociationTool=InDetPrdAssociationTool,
UseAssociationTool=usePrdAssociationTool,
NeighborSearch=True,
LoadFull=False,
DoCosmics=InDetFlags.doCosmics(),
pTmin=NewTrackingCuts.minSecondaryPt())
#InDetTRT_SeededSpacePointFinder.OutputLevel = VERBOSE
elif InDetFlags.loadSimpleTRTSeededSPFinder():
#
# --- alternative version using the region selector
#
from RegionSelector.RegSelSvcDefault import RegSelSvcDefault
InDetRegSelSvc = RegSelSvcDefault()
InDetRegSelSvc.enablePixel = DetFlags.pixel_on()
InDetRegSelSvc.enableSCT = DetFlags.SCT_on()
ServiceMgr += InDetRegSelSvc
if (InDetFlags.doPrintConfigurables()):
print InDetRegSelSvc
from TRT_SeededSpacePointFinderTool.TRT_SeededSpacePointFinderToolConf import InDet__SimpleTRT_SeededSpacePointFinder_ATL
InDetTRT_SeededSpacePointFinder = InDet__SimpleTRT_SeededSpacePointFinder_ATL(
name='InDetTRT_SeededSpFinder',
SpacePointsSCTName=InDetKeys.SCT_SpacePoints(),
SpacePointsOverlapName=InDetKeys.OverlapSpacePoints(),
PerigeeCut=1000.,
DirectionPhiCut=.3,
DirectionEtaCut=1.,
MaxHoles=2,
AssociationTool=InDetPrdAssociationTool,
RestrictROI=True)
#InDetTRT_SeededSpacePointFinder.OutputLevel = VERBOSE
if not usePrdAssociationTool:
InDetTRT_SeededSpacePointFinder.AssociationTool = None
# add either into the Tool Service
ToolSvc += InDetTRT_SeededSpacePointFinder
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededSpacePointFinder
#
# Silicon det elements road maker tool
#
from SiDetElementsRoadTool_xk.SiDetElementsRoadTool_xkConf import InDet__SiDetElementsRoadMaker_xk
InDetTRT_SeededSiRoadMaker = InDet__SiDetElementsRoadMaker_xk(
name='InDetTRT_SeededSiRoad',
PropagatorTool=InDetPatternPropagator,
usePixel=NewTrackingCuts.usePixel(),
PixManagerLocation=InDetKeys.PixelManager(),
useSCT=NewTrackingCuts.useSCT(),
SCTManagerLocation=InDetKeys.SCT_Manager(),
RoadWidth=35.,
MaxStep=20.) # NOT DEFAULT ?
#InDetTRT_SeededSiRoadMaker.OutputLevel = VERBOSE
if InDetFlags.doCosmics():
InDetTRT_SeededSiRoadMaker.RoadWidth = 50
ToolSvc += InDetTRT_SeededSiRoadMaker
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededSiRoadMaker
#
# --- TRT seeded back tracking tool
#
from TRT_SeededTrackFinderTool.TRT_SeededTrackFinderToolConf import InDet__TRT_SeededTrackFinder_ATL
InDetTRT_SeededTrackTool = InDet__TRT_SeededTrackFinder_ATL(
name='InDetTRT_SeededTrackMaker',
PropagatorTool=InDetPatternPropagator,
UpdatorTool=InDetPatternUpdator,
RoadTool=InDetTRT_SeededSiRoadMaker,
SeedTool=InDetTRT_SeededSpacePointFinder,
CombinatorialTrackFinder=InDetSiComTrackFinder,
pTmin=NewTrackingCuts.minSecondaryPt(),
nHolesMax=NewTrackingCuts.SecondarynHolesMax(),
# ME bugfix: nHolesGapMax = 2*NewTrackingCuts.SecondarynHolesGapMax(),
nHolesGapMax=NewTrackingCuts.SecondarynHolesGapMax(),
Xi2max=NewTrackingCuts.SecondaryXi2max(),
Xi2maxNoAdd=NewTrackingCuts.SecondaryXi2maxNoAdd(),
ConsistentSeeds=True,
#BremCorrection = True,
BremCorrection=False,
UseAssociationTool=usePrdAssociationTool)
if InDetFlags.doCosmics():
InDetTRT_SeededTrackTool.nWClustersMin = 0
ToolSvc += InDetTRT_SeededTrackTool
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededTrackTool
#
# --- Output key for the finder
#
self.__TRTSeededTracks = InDetKeys.TRTSeededTracks()
#
# TRT seeded back tracking algorithm
#
from TRT_SeededTrackFinder.TRT_SeededTrackFinderConf import InDet__TRT_SeededTrackFinder
InDetTRT_SeededTrackFinder = InDet__TRT_SeededTrackFinder(
name='InDetTRT_SeededTrackFinder',
RefitterTool=InDetTrackFitter,
TrackTool=InDetTRT_SeededTrackTool,
TrackExtensionTool=InDetTRTExtensionTool,
MinTRTonSegment=NewTrackingCuts.minSecondaryTRTonTrk(),
MinTRTonly=NewTrackingCuts.minTRTonly(),
TrtExtension=True,
SiExtensionCuts=NewTrackingCuts.SiExtensionCuts(),
minPt=NewTrackingCuts.minSecondaryPt(),
maxRPhiImp=NewTrackingCuts.maxSecondaryImpact(),
maxZImp=NewTrackingCuts.maxZImpact(),
maxEta=NewTrackingCuts.maxEta(),
Extrapolator=InDetExtrapolator,
RejectShortExtension=NewTrackingCuts.rejectShortExtensions(),
FinalRefit=False,
FinalStatistics=False,
OutputSegments=False,
InputSegmentsLocation=InDetKeys.TRT_Segments(),
OutputTracksLocation=self.__TRTSeededTracks)
# InDetTRT_SeededTrackFinder.OutputLevel = VERBOSE
topSequence += InDetTRT_SeededTrackFinder
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededTrackFinder
#
# ------------ Track truth.
#
if not InDetFlags.doSGDeletion():
if InDetFlags.doTruth():
#
# set up the truth info for this container
#
include("InDetRecExample/ConfiguredInDetTrackTruth.py")
InDetTracksTruth = ConfiguredInDetTrackTruth(
self.__TRTSeededTracks,
self.__TRTSeededTracks + "DetailedTruth",
self.__TRTSeededTracks + "TruthCollection")
#
# add final output for statistics
#
TrackCollectionKeys += [InDetTracksTruth.Tracks()]
TrackCollectionTruthKeys += [
InDetTracksTruth.TracksTruth()
]
else:
TrackCollectionKeys += [self.__TRTSeededTracks]
# --- output track collection
self.__BackTrackingTracks = self.__TRTSeededTracks
# ------------------------------------------------------------
#
# --- Resolve back tracking tracks ?
#
# ------------------------------------------------------------
if InDetFlags.doResolveBackTracks():
#
# --- set up special Scoring Tool for TRT seeded tracks
#
if InDetFlags.doCosmics():
from InDetTrackScoringTools.InDetTrackScoringToolsConf import InDet__InDetCosmicScoringTool
InDetTRT_SeededScoringTool = InDet__InDetCosmicScoringTool(
name='InDetCosmicScoringTool_TRT',
nWeightedClustersMin=0,
minTRTHits=NewTrackingCuts.minSecondaryTRTonTrk(),
SummaryTool=InDetTrackSummaryTool)
else:
from InDetTrackScoringTools.InDetTrackScoringToolsConf import InDet__InDetAmbiScoringTool
InDetTRT_SeededScoringTool = InDet__InDetAmbiScoringTool(
name='InDetTRT_SeededScoringTool',
Extrapolator=InDetExtrapolator,
DriftCircleCutTool=InDetTRTDriftCircleCut,
SummaryTool=InDetTrackSummaryTool,
useTRT_AmbigFcn=InDetFlags.doNewTracking(
), # full search => use NewT
useAmbigFcn=not InDetFlags.doNewTracking(
), # full search => use NewT
minPt=NewTrackingCuts.minSecondaryPt(),
maxRPhiImp=NewTrackingCuts.maxSecondaryImpact(),
maxZImp=NewTrackingCuts.maxZImpact(),
maxEta=NewTrackingCuts.maxEta(),
minSiClusters=NewTrackingCuts.minSecondaryClusters(),
maxSiHoles=NewTrackingCuts.maxSecondaryHoles(),
maxPixelHoles=NewTrackingCuts.maxSecondaryPixelHoles(),
maxSCTHoles=NewTrackingCuts.maxSecondarySCTHoles(),
maxDoubleHoles=NewTrackingCuts.maxSecondaryDoubleHoles(),
usePixel=NewTrackingCuts.usePixel(),
useSCT=NewTrackingCuts.useSCT(),
minTRTonTrk=NewTrackingCuts.minSecondaryTRTonTrk(),
minTRTPrecisionFraction=NewTrackingCuts.
minSecondaryTRTPrecFrac())
# InDetTRT_SeededScoringTool.OutputLevel = DEBUG
ToolSvc += InDetTRT_SeededScoringTool
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededScoringTool
#
# --- Load selection tool
#
from InDetAmbiTrackSelectionTool.InDetAmbiTrackSelectionToolConf import InDet__InDetAmbiTrackSelectionTool
InDetTRT_SeededAmbiTrackSelectionTool = InDet__InDetAmbiTrackSelectionTool(
name='InDetTRT_SeededAmbiTrackSelectionTool',
AssociationTool=InDetPrdAssociationTool,
DriftCircleCutTool=InDetTRTDriftCircleCut,
minScoreShareTracks=-1., # off !
minHits=NewTrackingCuts.minSecondaryClusters(),
minNotShared=NewTrackingCuts.minSecondarySiNotShared(),
maxShared=NewTrackingCuts.maxSecondaryShared(),
minTRTHits=NewTrackingCuts.minSecondaryTRTonTrk(),
UseParameterization=NewTrackingCuts.useParameterizedTRTCuts(),
Cosmics=InDetFlags.doCosmics(),
doPixelSplitting=InDetFlags.doPixelClusterSplitting())
# InDetTRT_SeededAmbiTrackSelectionTool.OutputLevel = DEBUG
ToolSvc += InDetTRT_SeededAmbiTrackSelectionTool
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededAmbiTrackSelectionTool
#
# --- load Ambiguity Processor
#
from TrkAmbiguityProcessor.TrkAmbiguityProcessorConf import Trk__SimpleAmbiguityProcessorTool
InDetTRT_SeededAmbiguityProcessor = Trk__SimpleAmbiguityProcessorTool(
name='InDetTRT_SeededAmbiguityProcessor',
Fitter=InDetTrackFitter,
SelectionTool=InDetTRT_SeededAmbiTrackSelectionTool,
RefitPrds=not InDetFlags.refitROT(),
SuppressTrackFit=False,
SuppressHoleSearch=False,
ScoringTool=InDetTRT_SeededScoringTool)
# InDetTRT_SeededAmbiguityProcessor.OutputLevel = DEBUG
if InDetFlags.materialInteractions():
InDetTRT_SeededAmbiguityProcessor.MatEffects = InDetFlags.materialInteractionsType(
)
else:
InDetTRT_SeededAmbiguityProcessor.MatEffects = 0
#
ToolSvc += InDetTRT_SeededAmbiguityProcessor
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededAmbiguityProcessor
#
# --- load the algorithm
#
self.__ResolvedTRTSeededTracks = InDetKeys.ResolvedTRTSeededTracks(
)
#
from TrkAmbiguitySolver.TrkAmbiguitySolverConf import Trk__TrkAmbiguitySolver
InDetTRT_SeededAmbiguitySolver = Trk__TrkAmbiguitySolver(
name='InDetTRT_SeededAmbiguitySolver',
TrackInput=[self.__TRTSeededTracks],
TrackOutput=self.__ResolvedTRTSeededTracks,
AmbiguityProcessor=InDetTRT_SeededAmbiguityProcessor)
topSequence += InDetTRT_SeededAmbiguitySolver
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededAmbiguitySolver
# --- Delete (non-resloved) TRT seeded tracks
from InDetRecExample.ConfiguredInDetSGDeletion import InDetSGDeletionAlg
InDetSGDeletionAlg(key=InDetKeys.TRTSeededTracks())
#
# ------------ Track truth.
#
if not InDetFlags.doSGDeletion():
if InDetFlags.doTruth():
#
# set up the truth info for this container
#
include("InDetRecExample/ConfiguredInDetTrackTruth.py")
InDetTracksTruth = ConfiguredInDetTrackTruth(
self.__ResolvedTRTSeededTracks,
self.__ResolvedTRTSeededTracks + "DetailedTruth",
self.__ResolvedTRTSeededTracks + "TruthCollection")
#
# add final output for statistics
#
TrackCollectionKeys += [InDetTracksTruth.Tracks()]
TrackCollectionTruthKeys += [
InDetTracksTruth.TracksTruth()
]
else:
TrackCollectionKeys += [self.__ResolvedTRTSeededTracks]
# --- output track collection
self.__BackTrackingTracks = self.__ResolvedTRTSeededTracks
from InDetRecExample.InDetJobProperties import InDetFlags
from InDetRecExample.InDetKeys import InDetKeys
doCreation = ( InDetFlags.doNewTracking() or InDetFlags.doPseudoTracking() or InDetFlags.doLargeD0() or InDetFlags.doLowPtLargeD0() or InDetFlags.doDisplacedSoftPion() ) \
and InDetFlags.doParticleCreation()
doConversion = not InDetFlags.doNewTracking() and not InDetFlags.doPseudoTracking() and not InDetFlags.doLargeD0() \
and not InDetFlags.doLowPtLargeD0() and not InDetFlags.doDisplacedSoftPion() and InDetFlags.doParticleConversion()
if doCreation:
print "Creating xAOD::TrackParticles from Trk::Tracks"
if doConversion:
print "Converting Rec::TrackParticles to xAOD::TrackParticles"
_perigee_expression = InDetFlags.perigeeExpression()
# need to treat Vertex specifically because at the time of
# the track particle creation the primary vertex does not yet exist.
# The problem is solved by first creating track particles wrt. the beam line
# and correcting the parameters after the vertex finding.
if _perigee_expression == 'Vertex':
_perigee_expression = 'BeamLine'
# Run the xAOD truth builder for PU if needed
if InDetFlags.doSplitReco():
from xAODTruthCnv.xAODTruthCnvConf import xAODMaker__xAODTruthCnvAlg
xAODTruthCnvPU = xAODMaker__xAODTruthCnvAlg("xAODTruthCnvPU")
xAODTruthCnvPU.WriteInTimePileUpTruth = False
xAODTruthCnvPU.WriteAllPileUpTruth = True
xAODTruthCnvPU.AODContainerName = "GEN_EVENT_PU"
xAODTruthCnvPU.xAODTruthEventContainerName = "TruthEvents_PU" #output
xAODTruthCnvPU.xAODTruthPileupEventContainerName = "TruthPileupEvents_PU" #output
xAODTruthCnvPU.xAODTruthParticleContainerName = "TruthParticles_PU" #output
def passCollectionName(coll_name, condition):
return coll_name if condition else ""
def getRecTrackParticleNameIfInFile(coll_name):
return getCollectionNameIfInFile('Rec::TrackParticle', coll_name)
def getRecVertexNameIfInFile(coll_name):
return getCollectionNameIfInFile('Rec::Vertex', coll_name)
from AthenaCommon.GlobalFlags import globalflags
is_mc = (globalflags.DataSource == 'geant4')
doCreation = ( InDetFlags.doNewTracking() or InDetFlags.doPseudoTracking() or InDetFlags.doLargeD0() or InDetFlags.doR3LargeD0() \
or InDetFlags.doLowPtLargeD0() ) and InDetFlags.doParticleCreation()
doConversion = not InDetFlags.doNewTracking() and not InDetFlags.doPseudoTracking() and not InDetFlags.doLargeD0() and not InDetFlags.doR3LargeD0()\
and not InDetFlags.doLowPtLargeD0() and InDetFlags.doParticleConversion()
if doCreation:
printfunc("Creating xAOD::TrackParticles from Trk::Tracks")
if doConversion:
printfunc("Converting Rec::TrackParticles to xAOD::TrackParticles")
# Run the xAOD truth builder for PU if needed
if InDetFlags.doSplitReco() and is_mc:
from xAODTruthCnv.xAODTruthCnvConf import xAODMaker__xAODTruthCnvAlg
xAODTruthCnvPU = xAODMaker__xAODTruthCnvAlg("xAODTruthCnvPU")
xAODTruthCnvPU.WriteInTimePileUpTruth = False
xAODTruthCnvPU.WriteAllPileUpTruth = True
def __init__(self, InputCollections = None, NewTrackingCuts = None, TrackCollectionKeys=[] , TrackCollectionTruthKeys=[]):
from InDetRecExample.InDetJobProperties import InDetFlags
from InDetRecExample.InDetKeys import InDetKeys
#
# get ToolSvc and topSequence
#
from AthenaCommon.AppMgr import ToolSvc
from AthenaCommon.AlgSequence import AlgSequence
topSequence = AlgSequence()
#
# --- decide if use the association tool
#
if len(InputCollections) > 0:
usePrdAssociationTool = True
else:
usePrdAssociationTool = False
# --- the PRD association tool is filled by the Segment making
# no need to run again
# ------------------------------------------------------------
#
# ---------- TRT Seeded Tracking
#
# ------------------------------------------------------------
if InDetFlags.doTRTSeededTrackFinder():
#
# --- decide which TRT seed space point finder to use
#
if InDetFlags.loadTRTSeededSPFinder():
#
# --- defaul space point finder
#
from TRT_SeededSpacePointFinderTool.TRT_SeededSpacePointFinderToolConf import InDet__TRT_SeededSpacePointFinder_ATL
InDetTRT_SeededSpacePointFinder = InDet__TRT_SeededSpacePointFinder_ATL(name = 'InDetTRT_SeededSpFinder' ,
SpacePointsSCTName = InDetKeys.SCT_SpacePoints(),
SpacePointsOverlapName = InDetKeys.OverlapSpacePoints(),
AssociationTool = InDetPrdAssociationTool ,
UseAssociationTool = usePrdAssociationTool ,
NeighborSearch = True,
LoadFull = False,
DoCosmics = InDetFlags.doCosmics(),
pTmin = NewTrackingCuts.minSecondaryPt())
#InDetTRT_SeededSpacePointFinder.OutputLevel = VERBOSE
elif InDetFlags.loadSimpleTRTSeededSPFinder():
#
# --- alternative version using the region selector
#
from RegionSelector.RegSelSvcDefault import RegSelSvcDefault
InDetRegSelSvc = RegSelSvcDefault()
InDetRegSelSvc.enablePixel = DetFlags.pixel_on()
InDetRegSelSvc.enableSCT = DetFlags.SCT_on()
ServiceMgr += InDetRegSelSvc
if (InDetFlags.doPrintConfigurables()):
print InDetRegSelSvc
from TRT_SeededSpacePointFinderTool.TRT_SeededSpacePointFinderToolConf import InDet__SimpleTRT_SeededSpacePointFinder_ATL
InDetTRT_SeededSpacePointFinder = InDet__SimpleTRT_SeededSpacePointFinder_ATL(name = 'InDetTRT_SeededSpFinder' ,
SpacePointsSCTName = InDetKeys.SCT_SpacePoints(),
SpacePointsOverlapName = InDetKeys.OverlapSpacePoints(),
PerigeeCut = 1000.,
DirectionPhiCut = .3,
DirectionEtaCut = 1.,
MaxHoles = 2,
AssociationTool = InDetPrdAssociationTool,
RestrictROI = True)
#InDetTRT_SeededSpacePointFinder.OutputLevel = VERBOSE
if not usePrdAssociationTool:
InDetTRT_SeededSpacePointFinder.AssociationTool = None
# add either into the Tool Service
ToolSvc += InDetTRT_SeededSpacePointFinder
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededSpacePointFinder
#
# Silicon det elements road maker tool
#
from SiDetElementsRoadTool_xk.SiDetElementsRoadTool_xkConf import InDet__SiDetElementsRoadMaker_xk
InDetTRT_SeededSiRoadMaker = InDet__SiDetElementsRoadMaker_xk(name = 'InDetTRT_SeededSiRoad' ,
PropagatorTool = InDetPatternPropagator ,
usePixel = NewTrackingCuts.usePixel(),
PixManagerLocation = InDetKeys.PixelManager() ,
useSCT = NewTrackingCuts.useSCT(),
SCTManagerLocation = InDetKeys.SCT_Manager() ,
RoadWidth = 35.,
MaxStep = 20.) # NOT DEFAULT ?
#InDetTRT_SeededSiRoadMaker.OutputLevel = VERBOSE
if InDetFlags.doCosmics():
InDetTRT_SeededSiRoadMaker.RoadWidth = 50
ToolSvc += InDetTRT_SeededSiRoadMaker
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededSiRoadMaker
#
# --- TRT seeded back tracking tool
#
from TRT_SeededTrackFinderTool.TRT_SeededTrackFinderToolConf import InDet__TRT_SeededTrackFinder_ATL
InDetTRT_SeededTrackTool = InDet__TRT_SeededTrackFinder_ATL(name = 'InDetTRT_SeededTrackMaker',
PropagatorTool = InDetPatternPropagator,
UpdatorTool = InDetPatternUpdator,
RoadTool = InDetTRT_SeededSiRoadMaker,
SeedTool = InDetTRT_SeededSpacePointFinder,
CombinatorialTrackFinder = InDetSiComTrackFinder,
pTmin = NewTrackingCuts.minSecondaryPt(),
nHolesMax = NewTrackingCuts.SecondarynHolesMax(),
# ME bugfix: nHolesGapMax = 2*NewTrackingCuts.SecondarynHolesGapMax(),
nHolesGapMax = NewTrackingCuts.SecondarynHolesGapMax(),
Xi2max = NewTrackingCuts.SecondaryXi2max(),
Xi2maxNoAdd = NewTrackingCuts.SecondaryXi2maxNoAdd(),
ConsistentSeeds = True,
#BremCorrection = True,
BremCorrection = False,
UseAssociationTool = usePrdAssociationTool)
if InDetFlags.doCosmics():
InDetTRT_SeededTrackTool.nWClustersMin = 0
ToolSvc += InDetTRT_SeededTrackTool
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededTrackTool
#
# --- Output key for the finder
#
self.__TRTSeededTracks = InDetKeys.TRTSeededTracks()
#
# TRT seeded back tracking algorithm
#
from TRT_SeededTrackFinder.TRT_SeededTrackFinderConf import InDet__TRT_SeededTrackFinder
InDetTRT_SeededTrackFinder = InDet__TRT_SeededTrackFinder(name = 'InDetTRT_SeededTrackFinder',
RefitterTool = InDetTrackFitter,
TrackTool = InDetTRT_SeededTrackTool,
TrackExtensionTool = InDetTRTExtensionTool,
MinTRTonSegment = NewTrackingCuts.minSecondaryTRTonTrk(),
MinTRTonly = NewTrackingCuts.minTRTonly(),
TrtExtension = True,
SiExtensionCuts = NewTrackingCuts.SiExtensionCuts(),
minPt = NewTrackingCuts.minSecondaryPt(),
maxRPhiImp = NewTrackingCuts.maxSecondaryImpact(),
maxZImp = NewTrackingCuts.maxZImpact(),
maxEta = NewTrackingCuts.maxEta(),
Extrapolator = InDetExtrapolator,
RejectShortExtension = NewTrackingCuts.rejectShortExtensions(),
FinalRefit = False,
FinalStatistics = False,
OutputSegments = False,
InputSegmentsLocation = InDetKeys.TRT_Segments(),
OutputTracksLocation = self.__TRTSeededTracks)
# InDetTRT_SeededTrackFinder.OutputLevel = VERBOSE
topSequence += InDetTRT_SeededTrackFinder
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededTrackFinder
#
# ------------ Track truth.
#
if not InDetFlags.doSGDeletion():
if InDetFlags.doTruth():
#
# set up the truth info for this container
#
include ("InDetRecExample/ConfiguredInDetTrackTruth.py")
InDetTracksTruth = ConfiguredInDetTrackTruth(self.__TRTSeededTracks,
self.__TRTSeededTracks+"DetailedTruth",
self.__TRTSeededTracks+"TruthCollection")
#
# add final output for statistics
#
TrackCollectionKeys += [ InDetTracksTruth.Tracks() ]
TrackCollectionTruthKeys += [ InDetTracksTruth.TracksTruth() ]
else:
TrackCollectionKeys += [ self.__TRTSeededTracks ]
# --- output track collection
self.__BackTrackingTracks = self.__TRTSeededTracks
# ------------------------------------------------------------
#
# --- Resolve back tracking tracks ?
#
# ------------------------------------------------------------
if InDetFlags.doResolveBackTracks():
#
# --- set up special Scoring Tool for TRT seeded tracks
#
if InDetFlags.doCosmics():
from InDetTrackScoringTools.InDetTrackScoringToolsConf import InDet__InDetCosmicScoringTool
InDetTRT_SeededScoringTool = InDet__InDetCosmicScoringTool(name = 'InDetCosmicScoringTool_TRT',
nWeightedClustersMin = 0,
minTRTHits = NewTrackingCuts.minSecondaryTRTonTrk(),
SummaryTool = InDetTrackSummaryTool)
else:
from InDetTrackScoringTools.InDetTrackScoringToolsConf import InDet__InDetAmbiScoringTool
InDetTRT_SeededScoringTool = InDet__InDetAmbiScoringTool(name = 'InDetTRT_SeededScoringTool',
Extrapolator = InDetExtrapolator,
DriftCircleCutTool = InDetTRTDriftCircleCut,
SummaryTool = InDetTrackSummaryTool,
useTRT_AmbigFcn = InDetFlags.doNewTracking(), # full search => use NewT
useAmbigFcn = not InDetFlags.doNewTracking(), # full search => use NewT
minPt = NewTrackingCuts.minSecondaryPt(),
maxRPhiImp = NewTrackingCuts.maxSecondaryImpact(),
maxZImp = NewTrackingCuts.maxZImpact(),
maxEta = NewTrackingCuts.maxEta(),
minSiClusters = NewTrackingCuts.minSecondaryClusters(),
maxSiHoles = NewTrackingCuts.maxSecondaryHoles(),
maxPixelHoles = NewTrackingCuts.maxSecondaryPixelHoles(),
maxSCTHoles = NewTrackingCuts.maxSecondarySCTHoles(),
maxDoubleHoles = NewTrackingCuts.maxSecondaryDoubleHoles(),
usePixel = NewTrackingCuts.usePixel(),
useSCT = NewTrackingCuts.useSCT(),
minTRTonTrk = NewTrackingCuts.minSecondaryTRTonTrk(),
minTRTPrecisionFraction = NewTrackingCuts.minSecondaryTRTPrecFrac())
# InDetTRT_SeededScoringTool.OutputLevel = DEBUG
ToolSvc += InDetTRT_SeededScoringTool
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededScoringTool
#
# --- Load selection tool
#
from InDetAmbiTrackSelectionTool.InDetAmbiTrackSelectionToolConf import InDet__InDetAmbiTrackSelectionTool
InDetTRT_SeededAmbiTrackSelectionTool = InDet__InDetAmbiTrackSelectionTool(name = 'InDetTRT_SeededAmbiTrackSelectionTool',
AssociationTool = InDetPrdAssociationTool,
DriftCircleCutTool = InDetTRTDriftCircleCut,
minScoreShareTracks = -1., # off !
minHits = NewTrackingCuts.minSecondaryClusters(),
minNotShared = NewTrackingCuts.minSecondarySiNotShared(),
maxShared = NewTrackingCuts.maxSecondaryShared(),
minTRTHits = NewTrackingCuts.minSecondaryTRTonTrk(),
UseParameterization = NewTrackingCuts.useParameterizedTRTCuts(),
Cosmics = InDetFlags.doCosmics(),
doPixelSplitting = InDetFlags.doPixelClusterSplitting())
# InDetTRT_SeededAmbiTrackSelectionTool.OutputLevel = DEBUG
ToolSvc += InDetTRT_SeededAmbiTrackSelectionTool
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededAmbiTrackSelectionTool
#
# --- load Ambiguity Processor
#
from TrkAmbiguityProcessor.TrkAmbiguityProcessorConf import Trk__SimpleAmbiguityProcessorTool
InDetTRT_SeededAmbiguityProcessor = Trk__SimpleAmbiguityProcessorTool(name = 'InDetTRT_SeededAmbiguityProcessor',
Fitter = InDetTrackFitter,
SelectionTool = InDetTRT_SeededAmbiTrackSelectionTool,
RefitPrds = not InDetFlags.refitROT(),
SuppressTrackFit = False,
SuppressHoleSearch = False,
ScoringTool = InDetTRT_SeededScoringTool)
# InDetTRT_SeededAmbiguityProcessor.OutputLevel = DEBUG
if InDetFlags.materialInteractions():
InDetTRT_SeededAmbiguityProcessor.MatEffects = InDetFlags.materialInteractionsType()
else:
InDetTRT_SeededAmbiguityProcessor.MatEffects = 0
#
ToolSvc += InDetTRT_SeededAmbiguityProcessor
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededAmbiguityProcessor
#
# --- load the algorithm
#
self.__ResolvedTRTSeededTracks = InDetKeys.ResolvedTRTSeededTracks()
#
from TrkAmbiguitySolver.TrkAmbiguitySolverConf import Trk__TrkAmbiguitySolver
InDetTRT_SeededAmbiguitySolver = Trk__TrkAmbiguitySolver(name = 'InDetTRT_SeededAmbiguitySolver',
TrackInput = [ self.__TRTSeededTracks ],
TrackOutput = self.__ResolvedTRTSeededTracks,
AmbiguityProcessor = InDetTRT_SeededAmbiguityProcessor)
topSequence += InDetTRT_SeededAmbiguitySolver
if (InDetFlags.doPrintConfigurables()):
print InDetTRT_SeededAmbiguitySolver
# --- Delete (non-resloved) TRT seeded tracks
from InDetRecExample.ConfiguredInDetSGDeletion import InDetSGDeletionAlg
InDetSGDeletionAlg(key = InDetKeys.TRTSeededTracks())
#
# ------------ Track truth.
#
if not InDetFlags.doSGDeletion():
if InDetFlags.doTruth():
#
# set up the truth info for this container
#
include ("InDetRecExample/ConfiguredInDetTrackTruth.py")
InDetTracksTruth = ConfiguredInDetTrackTruth(self.__ResolvedTRTSeededTracks,
self.__ResolvedTRTSeededTracks+"DetailedTruth",
self.__ResolvedTRTSeededTracks+"TruthCollection")
#
# add final output for statistics
#
TrackCollectionKeys += [ InDetTracksTruth.Tracks() ]
TrackCollectionTruthKeys += [ InDetTracksTruth.TracksTruth() ]
else:
TrackCollectionKeys += [ self.__ResolvedTRTSeededTracks ]
# --- output track collection
self.__BackTrackingTracks = self.__ResolvedTRTSeededTracks
