PndSTDMutableContainer.h

Go to the documentation of this file.

#ifndef PNDSTDMUTABLECONTAINER_HH
#define PNDSTDMUTABLECONTAINER_HH

#include <stdexcept>
#include <vector>

#include "TString.h"

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

#include "PndMutableContainerI.h"

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

  virtual ~PndSTDMutableContainer() {}

  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) << "PndSTDMutableContainer::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 /*PNDSTDMUTABLECONTAINER_HH*/