PndTCAMutableContainer.h

Go to the documentation of this file.

#ifndef PNDTCAMUTABLECONTAINER_HH
#define PNDTCAMUTABLECONTAINER_HH

#include <stdexcept>
#include <vector>

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

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

#include "PndMutableContainerI.h"

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

  virtual ~PndTCAMutableContainer() {}

  virtual void Reset() final
  {
    if (this->fTCA != nullptr) {
      this->fTCA->Delete();
    }
  }
  virtual T *CreateCopy(const T &t_element)
  {
    if (fTCA != nullptr) {
      new ((*fTCA)[fTCA->GetEntriesFast()]) T{t_element};
    } else {
      LOG(error) << "PndTCAMutableContainer::CreateCopy() no underlying TClonesArray";
      return nullptr;
    }
    return GetElement(GetSize() - 1);
  }

  virtual void RemoveAt(Int_t t_index)
  {
    if (this->fTCA != nullptr) {
      this->fTCA->RemoveAt(t_index);
    }
  }

  virtual void Compress()
  {
    if (this->fTCA != nullptr) {
      this->fTCA->Compress();
    }
  }

  virtual T *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 /*PNDTCAMUTABLECONTAINER_HH*/