IOUtils.h

#ifndef PandaTree_Interface_IOUtils_h
#define PandaTree_Interface_IOUtils_h

#include "TString.h"
#include "TTree.h"

#include <vector>
#include <iostream>

namespace panda {
  class ReaderObject;

  namespace utils {

    class BranchList;


    class BranchName : public std::vector<TString> {
    public:
      BranchName() {}
      BranchName(BranchName const&);
      BranchName(char const*);
      BranchName(std::string const& s) : BranchName(s.c_str()) {}
      BranchName(TString const& s) : BranchName(s.Data()) {}
      template<class InputIterator> BranchName(InputIterator begin, InputIterator end, bool isVeto) : std::vector<TString>(begin, end), isVeto_(isVeto) {}
      operator TString() const;
      TString fullName(TString const& objName = "") const;
      bool isVeto() const { return isVeto_; }

      bool match(BranchName const&) const;

      bool in(BranchList const&) const;

      bool vetoed(BranchList const&) const;

    private:
      bool isVeto_{false};
    };


    class BranchList : public std::vector<BranchName> {
    public:
      BranchList() {}
      BranchList(std::initializer_list<value_type> il, const allocator_type& alloc = allocator_type()) : std::vector<BranchName>(il, alloc) {}
      BranchList subList(TString const& objName) const;
      bool matchesAny(BranchList const&) const;
      BranchList& operator+=(BranchList const&);
      void collapse();
      BranchList fullNames(TString const& objName = "") const;
      static BranchList makeList(TTree&);

      void setVerbosity(int i) { verbosity = i; }
      int getVerbosity() const { return verbosity; }
    private:
      int verbosity{0};
    };


    Int_t checkStatus(TTree&, TString const& fullName, Bool_t status);

    Int_t setStatus(TTree&, TString const& objName, BranchName const& bName, BranchList const&);

    BranchName getStatus(TTree&, TString const& objName, BranchName const& bName);

    Int_t setAddress(TTree&, TString const& objName, BranchName const& bName, void* bPtr, BranchList const&, Bool_t setStatus);

    Int_t book(TTree&, TString const& objName, BranchName const& bName, TString const& size, char lType, void* bPtr, BranchList const&);
    Int_t resetAddress(TTree&, TString const& objName, BranchName const& bName);

    template<class O>
    Int_t
    book(TTree& _tree, TString const& _objName, BranchName const& _bName, TString const& _objType, O** _bPtr, BranchList const& _bList)
    {
      // objName: electrons
      // bName: tags
      // objType: std::vector<int>

      if (!_bName.in(_bList))
        return -1;

      _tree.Branch(_bName.fullName(_objName), _objType, _bPtr);

      return 0;
    }


    TTree*
    makeDocTree(TString const& treeName, TString names[], UInt_t size);


    class TNotify : public TObjArray {
    public:
      TNotify();
      ~TNotify() {}
      Bool_t Notify() override;
    };


    class BranchArrayUpdator : public TObject {
    public:
      BranchArrayUpdator(ReaderObject&, TTree&);
      ~BranchArrayUpdator();

      char const* GetName() const override;
      Bool_t Notify() override;

      ReaderObject const& getObject() const { return obj_; }

    private:
      ReaderObject& obj_;
      TTree& tree_;
    };


    Bool_t removeBranchArrayUpdator(ReaderObject& obj, TTree& tree);

  }
}

std::ostream& operator<<(std::ostream&, panda::utils::BranchName const&);
std::ostream& operator<<(std::ostream&, panda::utils::BranchList const&);

#endif