SparseChain.h

Go to the documentation of this file.

/*
 * SparseChain.h
 *
 *  Created on: 7 Aug 2019
 *      Author: Nicola Mori
 */

#ifndef SPARSECHAIN_H_
#define SPARSECHAIN_H_

// Root headers
#include "TChain.h"

// C/C++ headers
#include <algorithm>
#include <string>
#include <vector>

namespace EA {

class SparseChain : public TChain {
public:
  using TChain::TChain;

  template <typename T> Int_t SetBranchAddress(const char *bname, T **add, TBranch **ptr = 0);

  Int_t SetBranchAddress(const char *bname, void *add, TClass *realClass, EDataType datatype, Bool_t isptr);

  Int_t GetEntry(Long64_t entry = 0, Int_t getall = 0);

private:
  std::vector<std::string> _branchesWithAddress;

  ClassDef(SparseChain, 1)
};

template <typename T> Int_t SparseChain::SetBranchAddress(const char *bname, T **add, TBranch **ptr) {

  auto *branch = GetBranch(bname);
  if (branch) {
    std::string branchName(bname);
    if (std::find(_branchesWithAddress.begin(), _branchesWithAddress.end(), branchName) == _branchesWithAddress.end()) {
      _branchesWithAddress.push_back(std::move(branchName));
    }
    return TChain::SetBranchAddress<T>(bname, add, ptr);
  }
  return 0;
}

} // namespace EA

#endif /* SPARSECHAIN_H_ */