PndTCAOutputContainer.h

Go to the documentation of this file.

#ifndef PNDTCAOUTPUTCONTAINER_HH
#define PNDTCAOUTPUTCONTAINER_HH

#include <stdexcept>
#include <vector>

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

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

#include "PndOutputContainerI.h"

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

  virtual ~PndTCAOutputContainer() {}

  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) {
        LOG(debug) << "Attempting to register " << t_branchname << " as a container for " << T().ClassName() << " in folder " << this->fFolderName;
        this->fTCA = ioman->Register(t_branchname, T().ClassName(), this->fFolderName, this->fPersistency);
      } 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) << "PndTCAOutputContainer<" << T().ClassName() << ">::Init(" << t_branchname << "): No " << T().ClassName() << " array with name " << t_branchname
                     << " already exists but could not be fetched!";

          return kFALSE;
        }
        LOG(debug) << "Found a branch already called " << t_branchname << ". Using this for container for " << T().ClassName() << " in folder " << this->fFolderName;
      }
    }
    return kTRUE;
  }

  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) << "PndTCAOutputContainer::CreateCopy() no underlying std::vector";
      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 /*PNDTCAOUTPUTCONTAINER_HH*/