Element.h

#ifndef PandaTree_Framework_Element_h
#define PandaTree_Framework_Element_h

#include "Object.h"
#include "IOUtils.h"

#include <map>

namespace panda {

  class ContainerBase;
  class ArrayBase;
  class CollectionBase;


  class Element : public Object {
  public:
    struct datastore {
      datastore() {}
      virtual ~datastore() { deallocate(); }

      virtual void allocate(UInt_t n) { nmax_ = n; }
      virtual void deallocate() {}
      virtual void setStatus(TTree&, TString const&, utils::BranchList const&) {}
      virtual utils::BranchList getStatus(TTree&, TString const&) const { return utils::BranchList(); }
      virtual utils::BranchList getBranchNames(TString const& = "") const { return utils::BranchList(); }
      virtual void setAddress(TTree&, TString const&, utils::BranchList const& = {"*"}, Bool_t setStatus = kTRUE) {}
      virtual void book(TTree&, TString const&, utils::BranchList const& = {"*"}, Bool_t dynamic = kTRUE) {}
      virtual void releaseTree(TTree&, TString const&) {}

      UInt_t nmax() const { return nmax_; }


      virtual void resizeVectors_(UInt_t) {}

    protected:
      UInt_t nmax_;
    };

    typedef ArrayBase array_type;
    typedef CollectionBase collection_type;

    Element() = delete;
    Element(datastore&, UInt_t) {}

    Element(Element const& src) : Object(src) {}
    ~Element();
    Element& operator=(Element const&) { return *this; }

    static char const* typeName() { return "Element"; }

    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;
    Int_t getEntry(TTree& tree, Long64_t entry, Bool_t localEntry = kFALSE) final;
    Int_t getEntry(UInt_t treeId, Long64_t entry, Bool_t localEntry = kFALSE) final;
    void init() final { doInit_(); }
    char const* getName() const final;
    void setName(char const*) final;

  protected:

    Element(ArrayBase*);

    virtual void doBook_(TTree&, TString const&, utils::BranchList const&) = 0;
    virtual void doInit_() = 0;


    class StoreManager : public std::map<Element const*, ArrayBase*> {
    public:
      ArrayBase& getArray(Element const*) const;
      template<class O> typename O::datastore& getData(O const*) const;
      char const* getName(Element const*) const;
    };

    static StoreManager gStore;
  };

  template<class O>
  typename O::datastore&
  Element::StoreManager::getData(O const* _obj) const
  {
    return static_cast<typename O::datastore&>(getArray(_obj).getData());
  }
}

#endif