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(0),
      fOutputBranch("MVDHitsPixelPosCorrected"),
      fFolder(),
      fOutputArray(0),
      fEntryNr(0)
          {
      SetVerbose(2);
    }

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

    PndPosCorrectorTask(TString inputBranch, TString outputBranch, TString folderName):
      FairTask("Corrector"),
      fPersistance(kTRUE),
      fInputBranch(inputBranch),
      fInputArray(0),
      fOutputBranch(outputBranch),
      fFolder(folderName),
      fOutputArray(0),
      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