IParticleFourMomBranch.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTESTERTREE_IPARTICLEFOURMOMBRANCH_H
#define MUONTESTERTREE_IPARTICLEFOURMOMBRANCH_H
 
#include <MuonTesterTree/IMuonTesterBranch.h>
#include <MuonTesterTree/MuonTesterTree.h>
#include <MuonTesterTree/AuxElementBranch.h>
#include <AthenaBaseComps/AthMessaging.h>
 
namespace MuonVal {
    class IParticleFourMomBranch : public AthMessaging, public IParticleDecorationBranch {
        public:
            IParticleFourMomBranch(MuonTesterTree& tree, const std::string& particle);
            bool fill(const EventContext& ctx) override;
            bool init() override;
            std::string name() const override;
            const TTree* tree() const override;
            TTree* tree() override;

            MuonTesterTree& getTree();
            bool initialized() const;
            std::vector<DataDependency> data_dependencies() override;
            size_t size() const;
   
            void push_back(const xAOD::IParticle* p) override;
            void push_back(const xAOD::IParticle& p) override;
            void operator+=(const xAOD::IParticle* p) override;
            void operator+=(const xAOD::IParticle& p) override;
 
    
            size_t find(const xAOD::IParticle& p) const;
            size_t find(const xAOD::IParticle* p) const;
            size_t find(std::function<bool(const xAOD::IParticle*)> func) const;
            const std::vector<const xAOD::IParticle*>& getCached() const;
            template <typename T>
                bool addVariable(const std::string& variable, 
                                 const std::string& accName = "");
            template <typename T>
                bool addVariable(T defaultValue, 
                                 const std::string& variable,
                                 const std::string& accName = "");
            
            template <typename T>
                bool addVariableGeV(const std::string& variable, 
                                    const std::string& accName = "");
            template <typename T>
                bool addVariableGeV(T defaultValue,
                                    const std::string& variable,
                                    const std::string& accName = "");

            template <typename T>
                std::shared_ptr<ParticleVariableBranch<T>> getBranch(const std::string& var) const;
            
            bool addVariable(std::shared_ptr<IParticleDecorationBranch> branch);
 
    
        private:
            MuonTesterTree& m_parent;
            std::string m_name{};
            bool m_init{false};
            bool m_updated{false};
            VectorBranch<float> m_pt{m_parent.tree(), m_name +"_pt"};
            VectorBranch<float> m_eta{m_parent.tree(), m_name + "_eta"};
            VectorBranch<float> m_phi{m_parent.tree(), m_name + "_phi"};
            VectorBranch<float> m_e{m_parent.tree(), m_name + "_e"};
            VectorBranch<int> m_q{m_parent.tree(), m_name+"_q"};
 
            std::vector<std::shared_ptr<IParticleDecorationBranch>> m_variables{};
            std::vector<const xAOD::IParticle*> m_cached_particles{};
};
 
}
#include <MuonTesterTree/IParticleFourMomBranch.icc>
#endif