CollectionBase.h

#ifndef PandaTree_Framework_CollectionBase_h
#define PandaTree_Framework_CollectionBase_h

#include "ContainerBase.h"

#include <vector>
#include <utility>

namespace panda {


  class CollectionBase : public ContainerBase {
  public:
    CollectionBase() = delete;
    CollectionBase(CollectionBase const& src) = delete;
    ~CollectionBase();

    void setStatus(TTree&, utils::BranchList const& blist) final;
    utils::BranchList getStatus(TTree&) const final;
    utils::BranchList getBranchNames(Bool_t fullName = kTRUE, Bool_t = kFALSE) const final;
    UInt_t setAddress(TTree&, utils::BranchList const& blist = {"*"}, Bool_t setStatus = kTRUE) final;
    void book(TTree&, utils::BranchList const& blist = {"*"}) final;
    using ReaderObject::getEntry;
    Int_t getEntry(UInt_t, Long64_t entry, Bool_t localEntry = kFALSE) final;
    Int_t fill(TTree&) final;
    void init() final { clear(); }
    void print(std::ostream& = std::cout, UInt_t level = 1) const override;
    void dump(std::ostream& = std::cout) const override;
    void unlink(TTree&) final;
    UInt_t size() const final { return size_; }

    Bool_t empty() const { return size_ == 0; }


    void resize(UInt_t size);


    void reserve(UInt_t size);

    void clear() { size_ = 0; getData().resizeVectors_(0); }

    void prepareFill(TTree&);

  protected:
    CollectionBase(char const* name, UInt_t unitSize, Bool_t dummy) : ContainerBase(name, unitSize) {}

    virtual void reallocate_(UInt_t) = 0;

  private:
    void doSetAddress_(TTree&, utils::BranchList const&, Bool_t setStatus, Bool_t asInput);

    UInt_t size_{0};


    UInt_t sizeIn_{0};

    std::vector<Bool_t> inputInSynch_{};


    std::vector<std::pair<TTree*, Bool_t>> outputs_{};
  };

}

#endif