python.TruthCollectionsFixerConfig.TruthCollectionsFixerBlock Class Reference

Inheritance diagram for python.TruthCollectionsFixerConfig.TruthCollectionsFixerBlock:

Collaboration diagram for python.TruthCollectionsFixerConfig.TruthCollectionsFixerBlock:

Public Member Functions
	__init__ (self)
	makeAlgs (self, config)

Public Attributes
return	fixDAODTruthRecord

Detailed Description

This Config Block is meant to help fix older DAOD samples that have the old HepMC barcode instead of uid.
   See also https://gitlab.cern.ch/atlas/athena/-/merge_requests/82613
   and https://gitlab.cern.ch/atlas/athena/-/merge_requests/82746

Definition at line 8 of file TruthCollectionsFixerConfig.py.

Constructor & Destructor Documentation

__init__()

python.TruthCollectionsFixerConfig.TruthCollectionsFixerBlock.__init__

(

self

)

Definition at line 14 of file TruthCollectionsFixerConfig.py.

    def __init__(self):
        super(TruthCollectionsFixerBlock, self).__init__()
        self.addOption(
            "truthParticleContainersToFix",
            None,
            type=list,
            info="list of input DAOD truthParticle containers to fix",
        )
        self.addOption(
            "truthVertexContainersToFix",
            None,
            type=list,
            info="list of input DAOD truthVertex containers to fix",
        )
        self.addOption("fixDAODTruthRecord", True, type=bool,
                       info="older derivations have the old HepMC barcodes and need to be fixed, otherwise we get "
                       "a crash on 'missing ::uid'. Schedules an instance of TruthCollectionsFixerBlock. "
                       "Not needed for recent derivations.")
 

Member Function Documentation

makeAlgs()

python.TruthCollectionsFixerConfig.TruthCollectionsFixerBlock.makeAlgs	(		self,
			config )

Definition at line 33 of file TruthCollectionsFixerConfig.py.

    def makeAlgs(self, config):
 
        # we need this boolean because the responsibility of scheduling the whole block is passed onto CommonServicesConfig
        if not self.fixDAODTruthRecord: return
 
        if config.dataType() is DataType.Data: return
 
        partContainers = None
        if self.truthParticleContainersToFix is not None:
            partContainers = self.truthParticleContainersToFix
        elif config.isPhyslite():
            partContainers = [
                "TruthBoson", "TruthBosonsWithDecayParticles",
                "TruthElectrons", "TruthMuons", "TruthPhotons", "TruthNeutrinos",
                "TruthTaus",
                "TruthTop", "TruthBottom",
                "TruthForwardProtons",
                "TruthBSM", "TruthBSMWithDecayParticles",
                "BornLeptons"
            ]
        else:
            partContainers = [
                "TruthBoson", "TruthBosonsWithDecayParticles",
                "TruthElectrons", "TruthMuons", "TruthPhotons", "TruthNeutrinos",
                "TruthTaus", "TruthTausWithDecayParticles",
                "TruthTop", "TruthBottom", "TruthCharm", "TruthHFWithDecayParticles",
                "TruthForwardProtons", "TruthPileupParticles",
                "TruthBSM", "TruthBSMWithDecayParticles",
                "BornLeptons"
            ]
            
        vertContainers = None
        if self.truthVertexContainersToFix is not None:
            vertContainers = self.truthVertexContainersToFix
        elif config.isPhyslite():
            vertContainers = [
                "TruthBosonsWithDecayVertices",
                "TruthBSMWithDecayVertices"
            ]
        else:
            vertContainers = [
                "TruthBosonsWithDecayVertices",
                "TruthHFWithDecayVertices",
                "TruthTausWithDecayVertices",
                "TruthBSMWithDecayVertices"
            ]
 
        # in Athena, we have to rename the containers. In AnalysisBase, we can just overwrite in place
        if DualUseConfig.isAthena:
            # replicate the behaviour of AddressRemappingCfg
            ars = config.createService("AddressRemappingSvc", "AddressRemappingSvc")
            pps = config.createService("ProxyProviderSvc", "ProxyProviderSvc")
            if "AddressRemappingSvc" not in pps.ProviderNames:
                pps.ProviderNames += ["AddressRemappingSvc"]
            for container in partContainers:
                ars.TypeKeyRenameMaps += [
                    f"xAOD::TruthParticleContainer#{container}->InFile{container}",
                    f"xAOD::AuxContainerBase#{container}Aux.->InFile{container}Aux.",
                ]
            for container in vertContainers:
                ars.TypeKeyRenameMaps += [
                    f"xAOD::TruthVertexContainer#{container}->InFile{container}",
                    f"xAOD::AuxContainerBase#{container}Aux.->InFile{container}Aux.",
                ]
 
        # the actual fix for old DAOD truth schema
        for container in partContainers:
            alg = config.createAlgorithm(
                "xAODMaker::TruthParticleFixerAlg",
                "TruthParticleFixerAlg_" + container,
                reentrant=True,
            )
            alg.InputContainer = (
                container if not DualUseConfig.isAthena else f"InFile{container}"
            )
            alg.OutputContainer = container
 
            # in Athena, we further need to remap relevant ElementLinks for containers that have them
            containers_without_parent_child_links = [
                "TruthBosonsWithDecayParticles",
                "TruthTausWithDecayParticles",
                "BornLeptons", "TruthPileupParticles",
                "TruthForwardProtons",
                "TruthBSMWithDecayParticles"]
            if DualUseConfig.isAthena and container not in containers_without_parent_child_links:
                alg.LinkPrefixToRemove = "InFile"
                alg.ParticleLinks = ["parentLinks", "childLinks"]
 
        for container in vertContainers:
            alg = config.createAlgorithm(
                "xAODMaker::TruthVertexFixerAlg",
                "TruthVertexFixerAlg_" + container,
                reentrant=True,
            )
            alg.InputContainer = (
                container if not DualUseConfig.isAthena else f"InFile{container}"
            )
            alg.OutputContainer = container
 
            if DualUseConfig.isAthena :
                alg.LinkPrefixToRemove = "InFile"

Member Data Documentation

fixDAODTruthRecord

return python.TruthCollectionsFixerConfig.TruthCollectionsFixerBlock.fixDAODTruthRecord

Definition at line 36 of file TruthCollectionsFixerConfig.py.

---

The documentation for this class was generated from the following file:

• TruthCollectionsFixerConfig.py