VectorBranch.h

Go to the documentation of this file.

/*
Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTESTER_MUONVECTORBRANCH_H
#define MUONTESTER_MUONVECTORBRANCH_H
 
#include <MuonTesterTree/MuonTesterBranch.h>
#include <vector>
#include <string>
class TTree;
// Implementation to store vector like variables
namespace MuonVal {
class MuonTesterTree;
template <class T> class VectorBranch : public MuonTesterBranch, virtual public IMuonTesterBranch {
public:
    VectorBranch(TTree* tree, const std::string& name);
    VectorBranch(MuonTesterTree& tree, const std::string& name);

    VectorBranch(TTree* tree, const std::string& name, const T& def);
    VectorBranch(MuonTesterTree& tree, const std::string& name, const T& def);
 
    virtual ~VectorBranch() = default;

    bool fill(const EventContext& ctx) override;
    bool init() override;

    inline size_t size() const;

    inline void push_back(const T& value);
    inline void operator+=(const T& value);
 
    inline const std::vector<T>& operator=(const std::vector<T>& values);

    inline T& operator[](size_t idx);
    inline T& get(size_t idx);
 
    inline const T& getDefault() const;
    void setDefault(const T& def);
 
    inline bool isUpdated() const;
    inline bool hasDefault() const;
 
    inline friend std::ostream& operator<<(std::ostream& ostr, const VectorBranch& br) {
        ostr<<br.name()<<": ";
        if (br.isUpdated()){
            ostr<<br.m_variable;
        } else ostr<<"[]";
        return ostr;
    };
 
private:
    std::vector<T> m_variable{};
    T m_default{};
    bool m_hasDefault{false};
    bool m_updated{false};
};
 
template <> bool& VectorBranch<bool>::get(size_t idx);
}
#include <MuonTesterTree/VectorBranch.icc>
#endif