PndLmdTrackFinderCATask.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----            PndLmdTrackFinderCATask                    -----
// -----       Created 18/05/11  by A. Karavdina              -----
// -------------------------------------------------------------------------

#ifndef PNDLMDTRACKFINDERCATASK_H
#define PNDLMDTRACKFINDERCATASK_H
#include "FairTask.h"
#include "PndMCTrack.h"
#include "PndSdsClusterStrip.h"
#include "PndSdsGeoPar.h"
#include "PndSdsHit.h"
#include "PndSdsMCPoint.h"
#include "PndTrackCand.h"

#include <string>
#include <vector>
#include "PndSdsCell.h"
#include "TH2.h"
#include "TNtuple.h"

class TClonesArray;

class PndLmdTrackFinderCATask : public FairTask {
 public:
  PndLmdTrackFinderCATask();
  PndLmdTrackFinderCATask(const bool missPl, const double setdmax, Int_t innSensPP = 8, Int_t innP = 4, TString hitBranch = "LMDHitsStrip");
  virtual ~PndLmdTrackFinderCATask();

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

  virtual void Exec(Option_t *opt);

  void SetVerbose(Int_t verbose) { fVerbose = verbose; };
  void SetInaccuracy(Double_t accu) { dXY = accu; };
  void SetSensStripFlag(bool fS) { flagStipSens = fS; };
  void SetSensPixelFlag(bool fS) { flagPixelSens = fS; };
  void SetTrkCandCutsFlag(bool fS) { flagTrkCandCuts = fS; };
  void FinishTask();

 private:
  Double_t dXY;
  double rule_max;
  int nSensPP; // number of sensors on one plane
  int nP;      // number of planes
  bool SortHitsByDet(std::vector<std::vector<std::pair<Int_t, bool>>> &hitsd, Int_t nStripHits);
  bool SortHitsByDetSimple(std::vector<std::vector<Int_t>> &hitsd, Int_t nStripHits);
  bool SortHitsByDetSimple2(std::vector<std::vector<Int_t>> &hitsd,
                            Int_t nStripHits); // Uses PmdLmdDim class
  bool SortHitsByZ(std::vector<std::vector<std::pair<Int_t, bool>>> &hitsd, Int_t nStripHits);
  Double_t GetTrackDip(PndMCTrack *myTrack);
  Double_t GetTrackCurvature(PndMCTrack *myTrack);
  bool Neighbor(int &icell0, int &icell1);
  bool Neighbor(PndSdsCell *cell0, PndSdsCell *cell1);
  void Evolution(int &pv0, int &pv1, int &pv0_n, int &pv1_n, bool isch);
  TClonesArray *ForwardEvolution(TClonesArray *fCellArray, int niter = 100);
  TClonesArray *CookAllCells(std::vector<std::vector<Int_t>> hitsd);
  TClonesArray *CookCells(std::vector<std::vector<Int_t>> hitsd, int &pl0, int &pl1, TClonesArray *tCellArray);
  TString fHitBranchStrip;

  bool missPlAlgo;
  bool flagStipSens;
  bool flagPixelSens;
  bool flagTrkCandCuts;
  TClonesArray *fStripHitArray;

  TClonesArray *fTrackCandArray;
  TClonesArray *fCellArray;

  void Register();
  void Reset();
  void ProduceHits();

  ClassDef(PndLmdTrackFinderCATask, 4);
};

#endif /* PndLmdTrackFinderCATASK_H */