ParticleVariables.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONPRDTEST_PARTICLEVARIABLES_H
#define MUONPRDTEST_PARTICLEVARIABLES_H
 
#include "MuonPRDTest/PrdTesterModule.h"
 
#include "xAODBase/IParticleContainer.h"
#include "StoreGate/ReadDecorHandleKeyArray.h"
 
namespace MuonPRDTest{
    class ParticleVariables : public PrdTesterModule {
        public:
            ParticleVariables(MuonTesterTree& tree, 
                              const std::string& containerKey,
                              const std::string& outName,
                              MSG::Level msglvl);
 
            bool fill(const EventContext& ctx) override final;
            bool declare_keys() override final;
            template <typename T>
                bool addVariable(const std::string& variable, const bool declareDep = false) {
                    if (declareDep) {
                        declare_decorator(variable);
                    } 
                    return m_branch->addVariable<T>(variable);
                }

            template <typename T>
            bool addVariable(const std::string& variable, const std::string& accName, 
                             const bool declareDep = false) {
                if (declareDep){
                    declare_decorator(variable);
                }
                return m_branch->addVariable<T>(variable, accName);
            }

            template <typename T>
                bool addVariableGeV(const std::string& variable, const bool declareDep = false) {
                    if (declareDep) {
                        declare_decorator(variable);
                    } 
                    return m_branch->addVariableGeV<T>(variable);
                }

            template <typename T>
            bool addVariableGeV(const std::string& variable, const std::string& accName, 
                                const bool declareDep = false) {
                if (declareDep){
                    declare_decorator(variable);
                }
                return m_branch->addVariableGeV<T>(variable, accName);
            }

            bool addVariable(std::shared_ptr<IParticleDecorationBranch> branch);
            size_t push_back(const xAOD::IParticle* p);

            void declare_decorator(const std::string& decorName);
 
        private:
            SG::ReadHandleKey<xAOD::IParticleContainer> m_key{};
            using DecorKey_t = SG::ReadDecorHandleKey<xAOD::IParticleContainer>;
            SG::ReadDecorHandleKeyArray<xAOD::IParticleContainer> m_decorKeys{};
            std::shared_ptr<IParticleFourMomBranch> m_branch{};
    };
}
#endif