PndSdsStripClusterTask.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                 PndSdsStripClusterTask header file             -----
// -------------------------------------------------------------------------

#ifndef PNDSDSSTRIPCLUSTERTASK_H
#define PNDSDSSTRIPCLUSTERTASK_H

#include "PndSdsTask.h"
#include "PndSdsStripDigiPar.h"
#include "PndSdsHit.h"
#include "PndSdsMCPoint.h"
// #include "PndSdsCluster.h"
//#include "PndSdsRecoCharge.h"
#include "FairGeoVector.h"
#include "FairGeoTransform.h"
#include "TVector3.h"
#include "TRandom.h"
#include "TGeoMatrix.h"
#include "TGeoBBox.h"
#include "TH2F.h"
#include "PndSdsStripHitProducer.h"
#include "PndSdsStripClusterer.h"
#include "PndDetectorList.h"
#include "PndSdsChargeConversion.h"

#include "FairTSBufferFunctional.h"

#include <string>
#include <vector>

class TClonesArray;
class PndSdsChargeWeightingAlgorithms;

class PndSdsStripClusterTask : public PndSdsTask {
 public:
  PndSdsStripClusterTask();
  PndSdsStripClusterTask(const char *name);
  PndSdsStripClusterTask(PndSdsStripClusterTask &other)
    : PndSdsTask(), fPath(other.fPath), fDigiArray(other.fDigiArray), fClusterArray(other.fClusterArray), fHitArray(other.fHitArray), fClustBranchName(other.fClustBranchName),
      fClusterType(other.fClusterType), fFEcolumns(other.fFEcolumns), fFErows(other.fFErows), fChargeCut(other.fChargeCut), fRadChannel(other.fRadChannel),
      fRadTime(other.fRadTime), fSingleStripChargeThreshold(other.fSingleStripChargeThreshold), fEventHeader(other.fEventHeader), fDigiParameterList(other.fDigiParameterList),
      fCurrentDigiPar(other.fCurrentDigiPar), fSensorNamePar(other.fSensorNamePar), fChargeDigiParameterList(other.fChargeDigiParameterList), fStripCalcTop(other.fStripCalcTop),
      fStripCalcBot(other.fStripCalcBot), fChargeConverter(other.fChargeConverter), fCurrentStripCalcTop(other.fCurrentStripCalcTop),
      fCurrentStripCalcBot(other.fCurrentStripCalcBot), fChargeAlgos(other.fChargeAlgos), fCurrentChargeConverter(other.fCurrentChargeConverter), fDigiPar(other.fDigiPar),
      fGeoH(other.fGeoH), fCurrentClusterfinder(other.fCurrentClusterfinder), fClusterFinderList(other.fClusterFinderList), fFunctor(other.fFunctor), eta_rect(other.eta_rect),
      eta_trap(other.eta_trap), etahistofile(other.etahistofile){};
  virtual ~PndSdsStripClusterTask();
  PndSdsStripClusterTask &operator=(PndSdsStripClusterTask &other)
  {
    fPath = other.fPath;
    fDigiArray = other.fDigiArray;
    fClusterArray = other.fClusterArray;
    fHitArray = other.fHitArray;
    fClustBranchName = other.fClustBranchName;
    fClusterType = other.fClusterType;
    fFEcolumns = other.fFEcolumns;
    fFErows = other.fFErows;
    fChargeCut = other.fChargeCut;
    fRadChannel = other.fRadChannel;
    fRadTime = other.fRadTime;
    fSingleStripChargeThreshold = other.fSingleStripChargeThreshold;
    fEventHeader = other.fEventHeader;
    fDigiParameterList = other.fDigiParameterList;
    fCurrentDigiPar = other.fCurrentDigiPar;
    fSensorNamePar = other.fSensorNamePar;
    fChargeDigiParameterList = other.fChargeDigiParameterList;
    fStripCalcTop = other.fStripCalcTop;
    fStripCalcBot = other.fStripCalcBot;
    fChargeConverter = other.fChargeConverter;
    fCurrentStripCalcTop = other.fCurrentStripCalcTop;
    fCurrentStripCalcBot = other.fCurrentStripCalcBot;
    fChargeAlgos = other.fChargeAlgos;
    fCurrentChargeConverter = other.fCurrentChargeConverter;
    fDigiPar = other.fDigiPar;
    fGeoH = other.fGeoH;
    fCurrentClusterfinder = other.fCurrentClusterfinder;
    fClusterFinderList = other.fClusterFinderList;
    fFunctor = other.fFunctor;
    eta_rect = other.eta_rect;
    eta_trap = other.eta_trap;
    etahistofile = other.etahistofile;
    return *this;
  };

  virtual void SetParContainers();
  virtual InitStatus Init();
  virtual InitStatus ReInit();

  virtual void InitMQ(TList *tempList);
  virtual void GetParList(TList *){}; // tempList //[R.K.03/2017] unused variable(s)
  virtual void ExecMQ(TList *inputList, TList *outputList);
  virtual void SetParContainersMQ(TList *){}; // tempList //[R.K.03/2017] unused variable(s)

  virtual void SetInBranchId()
  {
    FairRootManager *ioman = FairRootManager::Instance();
    fInBranchId = ioman->GetBranchId(fInBranchName);
    std::cout << "InBranchId: " << fInBranchId << " for Branch: " << fInBranchName.Data() << std::endl;
    fClusterType = ioman->GetBranchId(fClustBranchName);
    std::cout << "fClusterType: " << fClusterType << " for Branch: " << fClustBranchName.Data() << std::endl;
  }

  virtual void Exec(Option_t *opt);
  virtual void Finish();

 protected:
  TString fPath;

  virtual void SetCalculators();
  TVector2 CalcLineCross(TVector2 point1, TVector2 dir1, TVector2 point2, TVector2 dir2) const;
  void SetCurrentCalculators(PndSdsStripDigiPar *digipar);
  Bool_t SelectSensorParams(Int_t sensorID);
  void CalcMeanCharge(PndSdsClusterStrip *onecluster, Double_t &meanstrip, Double_t &meanerr, Double_t &charge, Double_t &timestamp, Double_t &timestampError);
  Bool_t Backmap(TVector2 meantopPoint, Double_t toperr, TVector2 meanbotPoint, Double_t boterr, TVector3 &hitpos, TMatrixD &hitCov, Int_t &sensorID);
  TClonesArray *fDigiArray;    // Input array of PndSdsDigis
  TClonesArray *fClusterArray; // Output array of PndSdsClusters
  TClonesArray *fHitArray;     // Output array of PndSdsHits

  TString fClustBranchName;
  Int_t fClusterType;

  void Register();
  void Reset();
  void ProduceHits();
  void ResetClusterFinders();
  void FillClusterFinders();
  void ClearCalculators();

  Int_t fFEcolumns;
  Int_t fFErows;
  Double_t fChargeCut;
  Int_t fRadChannel;
  Int_t fRadTime;
  Double_t fSingleStripChargeThreshold;
  FairEventHeader *fEventHeader;

  TList *fDigiParameterList;
  PndSdsStripDigiPar *fCurrentDigiPar;
  PndSensorNamePar *fSensorNamePar;
  TList *fChargeDigiParameterList;
  std::map<const char *, PndSdsCalcStrip *> fStripCalcTop;           
  std::map<const char *, PndSdsCalcStrip *> fStripCalcBot;           
  std::map<const char *, PndSdsChargeConversion *> fChargeConverter; 

  PndSdsCalcStrip *fCurrentStripCalcTop;
  PndSdsCalcStrip *fCurrentStripCalcBot;
  PndSdsChargeWeightingAlgorithms *fChargeAlgos;
  PndSdsChargeConversion *fCurrentChargeConverter;
  PndSdsStripDigiPar *fDigiPar;

  PndGeoHandling *fGeoH; 
  PndSdsStripClusterer *fCurrentClusterfinder;
  std::map<const char *, PndSdsStripClusterer *> fClusterFinderList; 

  BinaryFunctor *fFunctor; 

  ClassDef(PndSdsStripClusterTask, 2);

 private:
  TH2F *eta_rect;      
  TH2F *eta_trap;      
  TFile *etahistofile; 
};

#endif /* SDSCLUSTERTASK_H */