PndTrkLegendreTask.h

Go to the documentation of this file.

#ifndef PNDTRKLEGENDRETASK_H
#define PNDTRKLEGENDRETASK_H

#include "PndTrkLegendreTransform.h"
// fairroot
#include "FairTask.h"
// pandaroot
#include "PndTrack.h"
#include "PndMCTrack.h"
// stt
#include "PndGeoSttPar.h"
#include "PndSttHit.h"
#include "PndSttMapCreator.h"
// tracking
#include "PndTrkSttHitList.h"
#include "PndTrkSdsHitList.h"
#include "PndTrkCluster.h"
#include "PndTrkClusterList.h"
#include "PndTrkTrack.h"
#include "PndTrkConformalHitList.h"
#include "PndTrkFitter.h"

// ROOT
#include "TH2F.h"
#include "TCanvas.h"

#define MAXNOFHITS 1000 // CHECK

class TClonesArray;
class TObjectArray;
class TSpectrum2;
class PndTrkLegendreTransform;
class PndTrkConformalTransform;
class PndTrkLegendreTask : public FairTask {

 public:
  PndTrkLegendreTask();
  PndTrkLegendreTask(int verbose);

  ~PndTrkLegendreTask();

  virtual InitStatus Init();

  virtual void Exec(Option_t *opt);

  void SetParContainers();

  void Initialize();
  void Reset();

  Int_t FillConformalHitList();
  void FillLegendreHisto(Int_t mode);
  void FillLegendreHisto(PndTrkCluster *cluster);

  PndTrkCluster CreateSttCluster(PndTrkHit *firsthit);
  Bool_t IsSttAssociate(PndTrkHit *hit1, PndTrkHit *hit2);

  PndTrkCluster CreateClusterByConfDistance(double fitm, double fitq);
  PndTrkCluster CreateClusterByRealDistance(double xc0, double yc0, double R0);
  PndTrkCluster CreateClusterByMixedDistance(double fitm, double fitq);
  PndTrkCluster CreateClusterByDistance(Int_t mode, double fitm, double fitq);
  PndTrkCluster CreateSkewHitList(PndTrkTrack *track);

  void FromConformalToRealTrack(double fitm, double fitp, double &x0, double &y0, double &R);
  PndTrkCluster CleanUpSkewHitList(PndTrkCluster *skewhitlist);

  void RegisterTrack(PndTrkTrack *track);

  void ComputeTraAndRot(PndTrkHit *hit, Double_t &delta, Double_t trasl[2]);

  PndTrkHit *FindSttReferenceHit();
  PndTrkHit *FindMvdPixelReferenceHit();
  PndTrkHit *FindMvdStripReferenceHit();
  PndTrkHit *FindMvdReferenceHit();
  PndTrkHit *FindReferenceHit();
  PndTrkCluster Cleanup(PndTrkCluster cluster);

  void SearchSecondaryTracks() { fSecondary = kTRUE; }

  Bool_t fDisplayOn;
  TH2F *hxy, *hxz, *hzphi;
  TCanvas *display;
  inline void SwitchOnDisplay() { fDisplayOn = kTRUE; }
  void DrawGeometry();
  void DrawHits(PndTrkHitList *hitlist);
  void Refresh();
  // conformal
  TH2F *huv;
  void RefreshConf();
  void DrawGeometryConf(double x1, double y1, double x2, double y2);
  void DrawConfHit(double x, double y, double r, int marker = 2);
  void LightCluster(PndTrkCluster *cluster);
  void DrawLegendreHisto();
  // z
  void RefreshZ();
  void DrawZGeometry(int whichone = 1, double phimin = 0, double phimax = 360, double zmin = -43, double zmax = 113);
  void DontUseMvdPix() { fUseMVDPix = kFALSE; }
  void DontUseMvdStr() { fUseMVDStr = kFALSE; }
  void DontUseStt() { fUseSTT = kFALSE; }

  Bool_t DoesRealHitBelong(PndTrkHit *hit, double x0, double y0, double R);
  Bool_t DoesConfHitBelong(PndTrkConformalHit *hit, double fitm, double fitp);

  void RePrepareLegendre(PndTrkCluster *cluster);
  void PrepareLegendre();
  Int_t ApplyLegendre(double &theta_max, double &r_max);
  Int_t ApplyLegendre(PndTrkCluster *cluster, double &theta_max, double &r_max);
  Int_t ExtractLegendre(Int_t mode, double &theta_max, double &r_max);

  Bool_t ZPhiFit(int iter, PndTrkCluster *cluster, double &fitm, double &fitp);
  PndTrkCluster *CleanupZPhiFit(PndTrkCluster *cluster, double fitm, double fitp);
  double ComputeZRediduals(PndTrkCluster *cluster, double fitm, double fitp);
  double CorrectZ(PndTrkCluster *cluster, double deltaz, double fitm, double fitp);

 private:
  Int_t fNofMvdPixHits, fNofMvdStrHits, fNofSttHits, fNofTriplets, fNofHits;

  TClonesArray *fSttPointArray;
  TClonesArray *fSttHitArray;

  TClonesArray *fMvdPixelHitArray;
  TClonesArray *fMvdStripHitArray;

  TClonesArray *fTrackArray, *fTrackCandArray;

  TClonesArray *fTubeArray;

  PndGeoSttPar *fSttParameters; //  CHECK added
  char fSttBranch[200], fMvdPixelBranch[200], fMvdStripBranch[200];

  PndSttMapCreator *fMapper;

  Int_t fEventCounter; // , fVerbose;

  PndTrkSttHitList *stthitlist;
  PndTrkSdsHitList *mvdpixhitlist;
  PndTrkSdsHitList *mvdstrhitlist;
  Double_t fSttParalDistance, fSttToMvdStripDistance;

  double fDeltaThetaRad;
  //  TSpectrum2 *s;
  PndTrkLegendreTransform *legendre;
  Bool_t fPersistence, fUseMVDPix, fUseMVDStr, fUseSTT, fSecondary;

  Bool_t fInitDone;

  PndTrkConformalTransform *conform;
  PndTrkConformalHitList *conformalhitlist;
  PndTrkTools *tools;

  PndTrkHit *fRefHit;

  Double_t fMvdPix_RealDistLimit, fMvdStr_RealDistLimit, fStt_RealDistLimit, fMvdPix_ConfDistLimit, fMvdStr_ConfDistLimit, fStt_ConfDistLimit;

  std::vector<std::pair<double, double>> fFoundPeaks;
  double fTime;
  TStopwatch *fTimer;
  PndTrkFitter *fFitter;
  ClassDef(PndTrkLegendreTask, 1);
};

#endif