PndPosCorrectorTask.h

Go to the documentation of this file.

/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/

#ifndef PndPosCorrectorTask_H
#define PndPosCorrectorTask_H

#include "FairTask.h" // for FairTask, InitStatus

#include "PndMapSorter.h"

#include "Rtypes.h"  // for Bool_t, Int_t, kTRUE, etc
#include "TString.h" // for TString

class FairTimeStamp;
class TClonesArray;

class PndPosCorrectorTask : public FairTask {
 public:
  PndPosCorrectorTask()
    : FairTask("SorterTask"), fPersistance(kTRUE), fInputBranch("MVDHitsPixelCorrectedSorted_event"), fInputArray(nullptr), fOutputBranch("MVDHitsPixelPosCorrected"), fFolder(),
      fOutputArray(nullptr), fEntryNr(0)
  {
    SetVerbose(2);
  }

  PndPosCorrectorTask(const char *name) : FairTask(name), fPersistance(kTRUE), fInputBranch(), fInputArray(nullptr), fOutputBranch(), fFolder(), fOutputArray(nullptr), fEntryNr(0)
  {
    SetVerbose(2);
  }

  PndPosCorrectorTask(TString inputBranch, TString outputBranch, TString folderName)
    : FairTask("Corrector"), fPersistance(kTRUE), fInputBranch(inputBranch), fInputArray(nullptr), fOutputBranch(outputBranch), fFolder(folderName), fOutputArray(nullptr),
      fEntryNr(0)
  {
    SetVerbose(2);
  }

  virtual ~PndPosCorrectorTask() {}

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

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

  void SetCorrectionValue(Int_t sensorId, Double_t xvalue, Double_t yvalue) { fPosCorrectionMap[sensorId] = std::make_pair(xvalue, yvalue); }

  virtual void SetParContainers(){};

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

 protected:
  Bool_t fPersistance;
  TString fInputBranch;
  TClonesArray *fInputArray;
  TString fOutputBranch;
  TString fFolder;
  TClonesArray *fOutputArray;
  Int_t fEntryNr;
  std::map<Int_t, std::pair<Double_t, Double_t>> fPosCorrectionMap; //< first value is sensor ID, second parameter is correction values for x and y
  PndPosCorrectorTask(const PndPosCorrectorTask &);
  PndPosCorrectorTask &operator=(const PndPosCorrectorTask &);

  ClassDef(PndPosCorrectorTask, 1);
};

#endif