VectorBranch.h

Go to the documentation of this file.

/*
Copyright (C) 2002-2023 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 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;
 
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