PndTCAInputContainer.h

Go to the documentation of this file.

#ifndef PNDTCAINPUTCONTAINER_HH
#define PNDTCAINPUTCONTAINER_HH

#include <stdexcept>
#include <vector>

#include "TClonesArray.h"
#include "TString.h"

#include "FairLogger.h"
#include "FairRootManager.h"

#include "PndInputContainerI.h"

template <class T>
class PndTCAInputContainer : public PndInputContainerI<T> {
 public:
  PndTCAInputContainer() : PndInputContainerI<T>() {}

  virtual ~PndTCAInputContainer(){};

  virtual Bool_t Init(const TString &t_branchname) final
  {
    this->fBranchName = t_branchname;
    FairRootManager *ioman = FairRootManager::Instance();
    if (ioman == nullptr) {
      throw std::runtime_error("FairRootManager::Instance() returned nullptr");
    } else {
      this->fTCA = dynamic_cast<TClonesArray *>(ioman->GetObject(t_branchname));
      if (this->fTCA == nullptr) {
        // throw std::runtime_error(
        //   ("PndTCAInputContainer<" + TString{T().ClassName()} + ">::Init(" + t_branchname + "): No " + TString{T().ClassName()} + " array with name " + t_branchname + " found!")
        //     .Data());
        LOG(error) << "PndTCAInputContainer<" << T().ClassName() << ">::Init(" << t_branchname << "): No " << T().ClassName() << " array with name " << t_branchname << " found!";
        return kFALSE;
      }
    }
    return kTRUE;
  }

  virtual T const *GetElement(Int_t t_idx) const final
  {
    if (fTCA != nullptr) {
      return static_cast<T *>(fTCA->At(t_idx));
    }
    return nullptr;
  }

  virtual ssize_t GetSize() const final
  {
    if (fTCA != nullptr) {
      return fTCA->GetEntriesFast();
    }
    return 0;
  }

  void SetTCA(TClonesArray *t_tca) { fTCA = t_tca; }

  TClonesArray *GetTCA() { return fTCA; }

 protected:
  TClonesArray *fTCA{nullptr};
};

#endif /*PNDTCAINPUTCONTAINER_HH*/