PndInputContainer.h

Go to the documentation of this file.

#ifndef PNDINPUTCONTAINER_HH
#define PNDINPUTCONTAINER_HH

#include <vector>

#include "PndContainerI.h"
#include "PndSTDInputContainer.h"
#include "PndTCAInputContainer.h"

template <class T>
class PndInputContainer : public PndInputContainerI<T> {

 public:
  PndInputContainer() : PndInputContainerI<T>() {}

  virtual ~PndInputContainer(){};

  virtual Bool_t Init(const TString &t_branchname)
  {
    this->fBranchName = t_branchname;
    this->fImp.reset(nullptr);

    Bool_t success = TryTCA(t_branchname);
    if (!success) {
      success = TrySTD(t_branchname);
    }
    if (!success) {
      this->fImp.reset(nullptr);
      // throw std::runtime_error(
      //   ("PndInputContainer<" + TString{T().ClassName()} + ">::Init(" + t_branchname + "): No " + TString{T().ClassName()} + " array with name " + t_branchname +
      //   "found!").Data());
      LOG(error) << "PndInputContainer<" << T().ClassName() << ">::Init(" << t_branchname << "): No " << T().ClassName() << " array with name " << t_branchname << " found!";
    }
    return success;
  }

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

  Bool_t TrySTD(const TString &t_branchname)
  {
    if (FairRootManager::Instance()->CheckBranch(t_branchname) ==
        0) { // This test is required here, as you get a segmentation violation if you try to access a not existing branch t_branchname via std
      return kFALSE;
    }
    this->fImp.reset(new PndSTDInputContainer<T>{});
    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> TClonesArray branch called " << t_branchname << " registered with the FairRootManager.";
    }
    return success;
  }

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

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

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

#endif /*PNDINPUTCONTAINER_HH*/