PndSorterTask.h

Go to the documentation of this file.

#ifndef PndSorterTask_H
#define PndSorterTask_H

#include "FairTask.h"

#include "TClonesArray.h"

#include "PndRingSorter.h"

class TClonesArray;

class PndSorterTask : public FairTask {
 public:
  PndSorterTask() : FairTask("SorterTask"), fNumberOfCells(1000), fWidthOfCells(10), fEntryNr(0)
  {
    SetVerbose(3);
    SetPersistance();
  }

  PndSorterTask(const char *name) : FairTask(name), fNumberOfCells(1000), fWidthOfCells(10), fEntryNr(0)
  {
    SetVerbose(3);
    SetPersistance();
  };

  PndSorterTask(Int_t numberOfCells, Double_t widthOfCells, TString inputBranch, TString outputBranch, TString folderName)
    : FairTask("Sorter"), fNumberOfCells(numberOfCells), fWidthOfCells(widthOfCells), fInputBranch(inputBranch), fOutputBranch(outputBranch), fFolder(folderName)
  {
    SetVerbose(3);
    SetPersistance();
  }

  virtual ~PndSorterTask()
  {
    if (fSorter != nullptr)
      delete fSorter;
  }

  virtual InitStatus Init();
  virtual InitStatus ReInit();

  virtual void Exec(Option_t *opt);
  virtual void FinishEvent();
  virtual void FinishTask();

  virtual void SetParContainers(){};

  void SetPersistance(Bool_t p = kTRUE) { fPersistance = p; };
  Bool_t GetPersistance() { return fPersistance; };

  virtual void AddNewDataToTClonesArray(FairTimeStamp *data) = 0;
  virtual PndRingSorter *InitSorter(Int_t numberOfCells, Double_t widthOfCells) = 0;

 protected:
  PndRingSorter *fSorter;

  Bool_t fPersistance;     // switch to turn on/off storing the arrays to a file
  Bool_t fDigiPixelMCInfo; // switch to turn on/off storing additional MC Info of Digis

  Int_t fNumberOfCells;
  Double_t fWidthOfCells; // in ns

  TString fInputBranch;
  TClonesArray *fInputArray;

  TString fOutputBranch;
  TString fFolder;
  TClonesArray *fOutputArray;

  Int_t fEntryNr;

  ClassDef(PndSorterTask, 2);
};

#endif