Exemple #1
0
############################################################
# (15) additional validation - if needed
if FatrasFlags.ReconstructionID():
    # check whether the ValidationNtupleWriter has been set up
    if 'TrkValNtupleWriter' in dir():
        # get the keys from outside
        from FatrasExample.FatrasKeys import FatrasKeyFlags
        from FatrasExample.FatrasValidation import FatrasValidationFlags
        # add the simulation track collection
        if FatrasFlags.SimulationID() and FatrasValidationFlags.Simulation():
            # add the simulation track collection
            TrkValNtupleWriter.DoTruth = True
            TrkValNtupleWriter.TrackSelectorTool = ToolSvc.FatrasDummyTrackSelectorTool
            TrkValNtupleWriter.TrackCollection += [
                FatrasKeyFlags.TrackCollection()
            ]
            TrkValNtupleWriter.TrackTruthCollection += [
                FatrasKeyFlags.TrackCollection() + 'Truth'
            ]
        # add the simulation track collection
        if FatrasFlags.RefittingID() and FatrasValidationFlags.Refitting():
            # add the simulation track collection
            TrkValNtupleWriter.TrackCollection += [
                FatrasKeyFlags.RefittedTrackCollection()
            ]
            TrkValNtupleWriter.TrackTruthCollection += [
                FatrasKeyFlags.RefittedTrackCollection() + 'Truth'
            ]

if not FatrasFlags.ReconstructionID():
Exemple #2
0
# include the Fast Calo Sim reconstruction setup
include('FastSimulationJobTransforms/FastCaloSimRecConfig.py')

jobproperties.FastSimulation.doFatrasID = True
jobproperties.FastSimulation.doFatrasMuon = True
jobproperties.FastSimulation.doParametrisedMuon = False

from FatrasExample.FatrasJobProperties import FatrasFlags
#FatrasFlags.FastCaloSim is NOT needed when running in Reco_trf: FastCaloSim is naturally integrated in the calorec part
#FatrasFlags.FastCaloSim = True

# set Muon Reco and ID reco on
FatrasFlags.RefittingID = False
FatrasFlags.ReconstructionID = True
FatrasFlags.EventPreProcessing = True
# (MS)
FatrasFlags.RefittingMS = False
FatrasFlags.DigitizationMS = True
FatrasFlags.ReconstructionMS = True

from FatrasExample.FatrasKeys import FatrasKeyFlags
FatrasKeyFlags.InputMcEventCollection = 'GEN_EVENT'

from FatrasExample.FatrasKeys import FatrasKeyFlags
from FastCaloSim.FastCaloSimConf import FastShowerCellBuilderTool
theFastShowerCellBuilderTool = FastShowerCellBuilderTool()
theFastShowerCellBuilderTool.McLocation = FatrasKeyFlags.McEventCollection()
#theFastShowerCellBuilderTool.OutputLevel = ERROR

rec.abortOnUncheckedStatusCode = False
        #elif InDetFlags.doxKalman():
        #InputTrackCollection = InDetKeys.XKalConvertedTracks()
        #else:
        # --- in case of reading from ESD, so we just set the Collection and truth
        #InputTrackCollection      = InDetKeys.Tracks()
        #InputTrackCollectionTruth = InDetKeys.TracksTruth()

        InputTrackCollection = FatrasKeyFlags.RefittedTrackCollection()
        InputTrackCollectionTruth = FatrasKeyFlags.RefittedTrackCollection(
        ) + 'Truth'
        TrackCollectionKeys += [InputTrackCollection]

        from FatrasExample.FatrasKeys import FatrasKeyFlags
        include("InDetRecExample/ConfiguredInDetTrackTruth.py")
        InDetTracksTruth = ConfiguredInDetTrackTruth(
            FatrasKeyFlags.RefittedTrackCollection(),
            FatrasKeyFlags.RefittedTrackCollection() + 'DetailedTruth',
            FatrasKeyFlags.RefittedTrackCollection() + 'Truth')

        TrackCollectionTruthKeys += [InputTrackCollectionTruth]

        # ------------------------------------------------------------
        #
        # ----------- now we do post-processing
        #
        # ------------------------------------------------------------
        #
        #
        include("InDetRecExample/InDetRecPostProcessing.py")

        # ------------------------------------------------------------