IMuonTesterBranch.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTESTER_IMUONTESTERBRANCH_H
#define MUONTESTER_IMUONTESTERBRANCH_H
 
#include <StoreGate/ReadHandleKey.h>
#include <xAODBase/IParticle.h>
 
#include <algorithm>
#include <iostream>
#include <memory>
#include <string>
#include <vector>
 
class TTree;
 
namespace MuonVal {
class IMuonTesterBranch {
public:
    virtual ~IMuonTesterBranch() = default;
    virtual bool fill(const EventContext& ctx) = 0;
    virtual bool init() = 0;
    virtual std::string name() const = 0;
 
    using DataDependency = SG::VarHandleKey*;
    virtual std::vector<DataDependency> data_dependencies() = 0;

    virtual TTree* tree() = 0;
    virtual const TTree* tree() const = 0;
 
};
 
// Extension of the interface in case someone wants to store information directly associated
// with the particles
// Particles are usually saved in vectors
class IAuxElementDecorationBranch : virtual public IMuonTesterBranch {
public:
    virtual void push_back(const SG::AuxElement* ele) = 0;
    virtual void push_back(const SG::AuxElement& ele) = 0;
 
    virtual void operator+=(const SG::AuxElement& ele) = 0;
    virtual void operator+=(const SG::AuxElement* ele) = 0;
    virtual ~IAuxElementDecorationBranch() = default;
};
 
class IParticleDecorationBranch : virtual public IMuonTesterBranch {
public:
    virtual void push_back(const xAOD::IParticle* p) = 0;
    virtual void push_back(const xAOD::IParticle& p) = 0;
    virtual void operator+=(const xAOD::IParticle* p) = 0;
    virtual void operator+=(const xAOD::IParticle& p) = 0;
    virtual ~IParticleDecorationBranch() = default;
};
}
#endif