PndSTDInputContainer.h

Go to the documentation of this file.

#ifndef PNDSTDINPUTCONTAINER_HH
#define PNDSTDINPUTCONTAINER_HH

#include <stdexcept>
#include <vector>

#include "TString.h"

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

#include "PndInputContainerI.h"

template <class T>
class PndSTDInputContainer : public PndInputContainerI<T> {
 public:
  PndSTDInputContainer() : PndInputContainerI<T>() {}
  virtual ~PndSTDInputContainer(){};

  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->fSTD = ioman->InitObjectAs<std::vector<T> const *>(t_branchname);
      if (this->fSTD == nullptr) {
        // throw std::runtime_error(
        //   ("PndSTDInputContainer<" + TString{T().ClassName()} + ">::Init(" + t_branchname + "): No " + TString{T().ClassName()} + " array with name " + t_branchname + " found!")
        //     .Data());
        LOG(error) << "PndSTDInputContainer<" << T().ClassName() << ">::Init(" << t_branchname << "): No " << T().ClassName() << " array with name " << t_branchname << " found!";
        return kFALSE;
      }
    }
    LOG(INFO) << "virtual Bool_t PndSTDInputContainer::Init(const TString &t_branchname) final successful";
    return kTRUE;
  }

  virtual T const *GetElement(Int_t t_idx) const final
  {
    if (fSTD != nullptr) {
      return &(fSTD->at(t_idx));
    }

    return nullptr;
  }

  virtual ssize_t GetSize() const final
  {
    if (fSTD != nullptr) {
      return fSTD->size();
    }
    return 0;
  }
  void SetData(std::vector<T> const *t_container) { fSTD = t_container; }

  std::vector<T> const *GetData() { return fSTD; }

 private:
  std::vector<T> const *fSTD{nullptr};
};

#endif /*PNDSTDINPUTCONTAINER_HH*/