PndSTDOutputContainer.h

Go to the documentation of this file.

#ifndef PNDSTDOUTPUTCONTAINER_HH
#define PNDSTDOUTPUTCONTAINER_HH

#include <stdexcept>
#include <vector>

#include "TString.h"

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

#include "PndOutputContainerI.h"

template <class T>
class PndSTDOutputContainer : public PndOutputContainerI<T> {
 public:
  PndSTDOutputContainer(const TString &t_foldername = "", Bool_t t_persistency = kTRUE) : PndOutputContainerI<T>(t_foldername, t_persistency) {}

  virtual ~PndSTDOutputContainer() {}

  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 {
      if (ioman->CheckBranch(t_branchname) == 0) {
        this->fSTD = new std::vector<T>{};
        ioman->RegisterAny(t_branchname, this->fSTD, this->fPersistency);
      } else {
        // throw std::runtime_error(("PndSTDOutputContainer<" + TString{T().ClassName()} + ">::Init(" + t_branchname + "): A branch with that name already exists.").Data());
        LOG(error) << "PndSTDOutputContainer<" << T().ClassName() << ">::Init(" << t_branchname << "): No " << T().ClassName() << " array with name " << t_branchname
                   << " created!";
        return kFALSE;
      }
    }
    return kTRUE;
  }

  virtual T *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;
  }

  virtual T *CreateCopy(const T &t_element)
  {
    if (fSTD != nullptr) {
      fSTD->push_back(T{t_element});
    } else {
      LOG(ERROR) << "PndSTDInputContainer::CreateCopy() no underlying std::vector";
      return nullptr;
    }
    return GetElement(GetSize() - 1);
  }

  virtual void RemoveAt(Int_t t_index) final
  {
    if (this->fSTD != nullptr) {
      this->fSTD->erase(this->fSTD->begin() + t_index);
    }
  }

  virtual void Compress() {}

  virtual void Reset() final
  {
    if (this->fSTD != nullptr) {
      this->fSTD->clear();
    }
  }

  void SetData(std::vector<T> *t_container) { fSTD = t_container; }

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

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

#endif /*PNDSTDOUTPUTCONTAINER_HH*/