TruthCollectionMaker.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORK_TRUTHCOLLECTIONMAKER_H
#define DERIVATIONFRAMEWORK_TRUTHCOLLECTIONMAKER_H
 
#include <string>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "Gaudi/Property.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/ServiceHandle.h"
#include "xAODTruth/TruthParticleContainer.h"
// Handles and keys
#include "StoreGate/WriteDecorHandleKey.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/WriteHandle.h"
// Exprssion evaluator
#include "ExpressionEvaluation/ExpressionParserUser.h"
 
// Forward declarations
class StoreGateSvc;
 
namespace DerivationFramework {
 
  class TruthCollectionMaker : public extends<ExpressionParserUser<AthAlgTool>, IAugmentationTool> {
    public:
      TruthCollectionMaker(const std::string& t, const std::string& n, const IInterface* p);
      ~TruthCollectionMaker();
      StatusCode initialize();
      StatusCode finalize();
      virtual StatusCode addBranches(const EventContext& ctx) const;
 
    private:
      mutable std::atomic<unsigned int> m_ntotpart, m_npasspart;
      // R/W handles
      SG::ReadHandleKey<xAOD::TruthParticleContainer> m_particlesKey
         {this, "ParticlesKey", "TruthParticles", "ReadHandleKey for input TruthParticleContainer"};
      SG::WriteHandleKey<xAOD::TruthParticleContainer> m_outputParticlesKey
         {this, "OutputCollectionName", "", "WriteHandleKey for new TruthParticleContainer"};
      // Non-handle properties
      Gaudi::Property<std::string> m_partString
         {this, "ParticleSelectionString", "", "ExpressionEvaluation string for particle selection"};
      Gaudi::Property<bool> m_do_compress
         {this, "Do_Compress", false, "Removes particles with the same pdgId in a decay chain (but keeps first and last)"};
      Gaudi::Property<bool> m_do_sherpa
         {this, "Do_Sherpa", false, "Checks if there are truth W bosons in the current record.  If not, tries to combine W daughters to create one"};
      Gaudi::Property<bool> m_keep_navigation_info
         {this, "KeepNavigationInfo", true, "m_do_sherpa currently only works for W+jets"};
      // Decor handles
      SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> m_linkDecoratorKey
         {this, "originalTruthParticle", m_outputParticlesKey, "originalTruthParticle", "Name of the decoration linking to the original truth particle"};
      SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> m_originDecoratorKey
         {this, "classifierParticleOrigin", m_outputParticlesKey, "classifierParticleOrigin", "Name of the decoration which records the particle origin as determined by the MCTruthClassifier"};
      SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> m_typeDecoratorKey
         {this, "classifierParticleType", m_outputParticlesKey, "classifierParticleType", "Name of the decoration which records the particle type as determined by the MCTruthClassifier"};
      SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> m_outcomeDecoratorKey
         {this, "classifierParticleOutCome", m_outputParticlesKey, "classifierParticleOutCome", "Name of the decoration which records the particle outcome as determined by the MCTruthClassifier"};
      SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> m_classificationDecoratorKey
         {this, "Classification", m_outputParticlesKey, "Classification", "Name of the decoration which records the particle classification as determined by the MCTruthClassifier"};
      SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> m_motherIDDecoratorKey
         {this, "motherID", m_outputParticlesKey, "motherID", "Name of the decoration which records the ID of the particle's mother"};
      SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> m_daughterIDDecoratorKey
         {this, "daughterID", m_outputParticlesKey, "daughterID", "Name of the decoration which records the ID of the particle's daughter"};
      SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> m_hadronOriginDecoratorKey
         {this, "TopHadronOriginFlag", m_outputParticlesKey, "TopHadronOriginFlag", "Name of the decoration which records the origin of hadrons from top decays"};
 
      // Decorations to be read from elements of the "TruthParticles" container
      // to be copied onto elements of the output container.
      SG::ReadDecorHandleKey<xAOD::TruthParticleContainer> m_originReadDecorKey
         {this, "inputClassifierParticleOrigin", m_particlesKey, "classifierParticleOrigin", "Particle origin"};
      SG::ReadDecorHandleKey<xAOD::TruthParticleContainer> m_typeReadDecorKey
         {this, "inputClassifierParticleType", m_particlesKey, "classifierParticleType", "Particle type"};
      SG::ReadDecorHandleKey<xAOD::TruthParticleContainer> m_outcomeReadDecorKey
         {this, "inputClassifierParticleOutCome", m_particlesKey, "classifierParticleOutCome", "Particle outcome"};
      SG::ReadDecorHandleKey<xAOD::TruthParticleContainer> m_classificationReadDecorKey
         {this, "inputClassification", m_particlesKey, "Classification", "Classification code"};
 
      ServiceHandle<StoreGateSvc> m_metaStore; 
  };
}
 
#endif // DERIVATIONFRAMEWORK_TRUTHCOLLECTIONMAKER_H