ArrayBranch.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTESTER_ARRAYBRANCH_H
#define MUONTESTER_ARRAYBRANCH_H
 
#include <MuonTesterTree/IMuonTesterBranch.h>
#include <MuonTesterTree/MuonTesterBranch.h>
#include <TTree.h>
 
namespace MuonVal {
class MuonTesterTree;
template <class T> class ArrayBranch : public MuonTesterBranch, virtual public IMuonTesterBranch {
public:
    ArrayBranch(TTree* tree, const std::string& name, size_t size);
    ArrayBranch(MuonTesterTree& tree, const std::string& name, size_t size);

    ArrayBranch(TTree* tree, const std::string& name, size_t size, const T& def_val);
    ArrayBranch(MuonTesterTree& tree, const std::string& name, size_t size, const T& def_val);
 
    virtual ~ArrayBranch() = default;

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

    bool initialized() const;

    size_t size() const;

    void set(size_t s, const T& val);
    const T& get(size_t s) const;

    const T& operator[](size_t s) const;
    T& operator[](size_t s);

    const T& getDefault() const;
 
    void setDefault(const T& val);
 
private:
    std::string tree_data_type() const;
    void reset();
    size_t m_size{0};
    std::vector<T> m_data{};
    std::vector<bool> m_updated{};
 
    bool m_init{false};

    T m_default{};
    bool m_failIfNotUpdated{true};
};

template <> std::string ArrayBranch<char*>::tree_data_type() const;
template <> std::string ArrayBranch<std::string>::tree_data_type() const;
template <> std::string ArrayBranch<Char_t>::tree_data_type() const;
template <> std::string ArrayBranch<UChar_t>::tree_data_type() const;
template <> std::string ArrayBranch<Short_t>::tree_data_type() const;
template <> std::string ArrayBranch<Int_t>::tree_data_type() const;
template <> std::string ArrayBranch<Long64_t>::tree_data_type() const;
template <> std::string ArrayBranch<UShort_t>::tree_data_type() const;
template <> std::string ArrayBranch<UInt_t>::tree_data_type() const;
template <> std::string ArrayBranch<ULong64_t>::tree_data_type() const;
template <> std::string ArrayBranch<Float_t>::tree_data_type() const;
template <> std::string ArrayBranch<Double_t>::tree_data_type() const;
template <> std::string ArrayBranch<Bool_t>::tree_data_type() const;
}
#include <MuonTesterTree/ArrayBranch.icc>
#endif