LinkerBranch.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTESTER_LINKERBRANCH_H
#define MUONTESTER_LINKERBRANCH_H
 
#include <MuonTesterTree/VectorBranch.h>
#include <MuonTesterTree/IParticleFourMomBranch.h>
 
namespace MuonVal{
    class LinkerBranch : public VectorBranch<unsigned short>,
                         virtual public IParticleDecorationBranch {
    public:
        using ParticleBranch_ptr  = std::shared_ptr<IParticleFourMomBranch>;

        using Linker_t = std::function<const xAOD::IParticle*(const xAOD::IParticle*)>;

        LinkerBranch(IParticleFourMomBranch& parent,
                     ParticleBranch_ptr linkColl,
                     Linker_t  linker,
                     const std::string& altName = "");
        using VectorBranch<unsigned short>::push_back;
        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;
 
    private:
        IParticleFourMomBranch* m_linkColl{nullptr};
        Linker_t m_linkerFunc;
    };
 
    class BilateralLinkerBranch: public VectorBranch<unsigned short>,
                                 virtual public IParticleDecorationBranch {
        public:
            using Linker_t = LinkerBranch::Linker_t;
            using ParticleBranch_ptr = LinkerBranch::ParticleBranch_ptr;
            using VectorBranch<unsigned short>::push_back;
            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;
        
            bool fill(const EventContext& ctx) override;
 
            static bool connectCollections(ParticleBranch_ptr primColl,
                                           ParticleBranch_ptr secondColl,
                                           Linker_t fromPrimToSec,
                                           const std::string& altPrimName ="",
                                           const std::string& altSecName = "");
        private:
            BilateralLinkerBranch(IParticleFourMomBranch& bilatColl,
                                  ParticleBranch_ptr primColl,
                                  Linker_t linker,
                                  const std::string& altName);
            const IParticleFourMomBranch& m_parent;
            IParticleFourMomBranch* m_linkColl{nullptr};
            Linker_t m_linkerFunc;
    
 
    };
}
#endif