PndOutputContainer.h

Go to the documentation of this file.

#ifndef PNDOUTPUTCONTAINER_HH
#define PNDOUTPUTCONTAINER_HH

#include <vector>

#include "PndContainerI.h"
#include "PndSTDOutputContainer.h"
#include "PndTCAOutputContainer.h"

template <class T>
class PndOutputContainer : public PndOutputContainerI<T> {

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

  virtual ~PndOutputContainer(){};

  virtual Bool_t Init(const TString &t_branchname)
  {
    this->fBranchName = t_branchname;
    Bool_t success = TryTCA(t_branchname);
    if (!success) {
      success = TrySTD(t_branchname);
    }
    return success;
  }

  Bool_t TrySTD(const TString &t_branchname)
  {
    this->fImp.reset(new PndSTDOutputContainer<T>{this->fFolderName, this->fPersistency});
    Bool_t success = this->fImp->Init(t_branchname);
    if (!success) {
      LOG(debug) << "No std::vector<T> branch called " << t_branchname << " registered with the FairRootManager.";
      this->fImp.reset(nullptr);
    } else {
      LOG(debug) << "Found std::vector<T> branch called " << t_branchname << " registered with the FairRootManager. ";
    }
    return success;
  }

  Bool_t TryTCA(const TString &t_branchname)
  {
    this->fImp.reset(new PndTCAOutputContainer<T>{this->fFolderName, this->fPersistency});
    Bool_t success = this->fImp->Init(t_branchname);
    if (!success) {
      LOG(debug) << "No TClonesArray branch called " << t_branchname << " registered with the FairRootManager either. !";
      this->fImp.reset(nullptr);
      // abort();
    } else {
      LOG(debug) << "Found TClonesArray branch called " << t_branchname << " registered with the FairRootManager. ";
    }
    return success;
  }

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

  virtual T *CreateCopy(const T &t_data) final
  {
    if (this->fImp != nullptr) {
      return this->fImp->CreateCopy(t_data);
    }
    return nullptr;
  }

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

  virtual ssize_t GetSize() const final
  {
    if (this->fImp != nullptr) {
      return this->fImp->GetSize();
    }
    return 0;
  }

  virtual T *GetElement(Int_t t_index) const final
  {
    if (this->fImp != nullptr) {
      return this->fImp->GetElement(t_index);
    }
    return nullptr;
  }

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

 protected:
  std::unique_ptr<PndOutputContainerI<T>> fImp{nullptr};
};

#endif /*PNDOUTPUTCONTAINER_HH*/