PndProcessTask.h

Go to the documentation of this file.

#ifndef PNDPROCESSTASK_H
#define PNDPROCESSTASK_H

#include <algorithm>
#include <iterator>
#include <stdexcept>
#include <string>
#include <vector>

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

#include "FairLogger.h"
#include "FairRun.h"
#include "FairRuntimeDb.h"

#include "PndPersistencyTask.h"
#include "PndProcess.h"
#include "PndProcessHandler.h"

template <class DataStruct>
class PndProcessTask : public PndPersistencyTask, public PndProcessHandler<DataStruct> {
 public:
  PndProcessTask(const std::string &t_detectorName);
  virtual ~PndProcessTask();

  virtual InitStatus Init() /*override*/ final;

  virtual void Exec(Option_t *t_opt = nullptr) /*override*/;

  virtual void Finish() /*override*/;

 protected:
  void SetParContainers() /*override*/ final;

  virtual void PreProcessing(Option_t * /*unused*/);

  virtual void PostProcessing(Option_t * /*unused*/);

  virtual void SetupDataArrays() = 0;

 private:
  ClassDef(PndProcessTask, 1);
};

template <class DataStruct>
PndProcessTask<DataStruct>::PndProcessTask(const std::string &t_detectorName) : PndPersistencyTask(), PndProcessHandler<DataStruct>(t_detectorName)
{
}

template <class DataStruct>
PndProcessTask<DataStruct>::~PndProcessTask()
{
}

template <class DataStruct>
InitStatus PndProcessTask<DataStruct>::Init()
{
  InitStatus result = kSUCCESS;
  try {
    SetupDataArrays();
    this->SetupProcesses();
  } catch (const std::exception &e) {
    result = kERROR;
  }
  return result;
}

template <class DataStruct>
void PndProcessTask<DataStruct>::Exec(Option_t *t_option)
{
  PreProcessing(t_option);
  this->CallProcesses();
  PostProcessing(t_option);
}

template <class DataStruct>
void PndProcessTask<DataStruct>::Finish()
{
  LOG(INFO) << "PndProcessTask: Finish - " << this->fDetectorName;
  for (PndProcess<DataStruct> *process : this->fProcesses) {
    process->TearDown();
  }
}

template <class DataStruct>
void PndProcessTask<DataStruct>::PreProcessing(Option_t * /*unused*/)
{
  LOG(trace) << "PndProcessTask: PreProcessing - " << this->fDetectorName;
}

template <class DataStruct>
void PndProcessTask<DataStruct>::PostProcessing(Option_t * /*unused*/)
{
  LOG(trace) << "PndProcessTask: PostProcessing - " << this->fDetectorName;
}

template <class DataStruct>
void PndProcessTask<DataStruct>::SetParContainers()
{
  // Get run and runtime database
  FairRun *run = FairRun::Instance();
  if (run == nullptr) {
    Fatal("SetParContainers", "No analysis run");
  }
  FairRuntimeDb *db = run->GetRuntimeDb();
  if (db == nullptr) {
    Fatal("SetParContainers", "No runtime database");
  }

  LOG(DEBUG) << "PndProcessTask - " << this->fDetectorName << " SetParContainers()";
  for (const std::string &parameterName : this->fParameterRegister->GetRequestList()) {
    LOG(DEBUG) << parameterName;
    this->fParameterRegister->SetParameter(parameterName, db->getContainer(parameterName.c_str()));
  }
}

#endif /*PNDPROCESSTASK_H*/