PndStraightLineTrackFinderTask.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                      PndStraightLineTrackFinderTask                    -----
// -----                  Created 22/10/09  by M. Michel               -----
// -------------------------------------------------------------------------

#ifndef PndStraightLineTrackFinderTask_H
#define PndStraightLineTrackFinderTask_H

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

#include <string>
#include <vector>
#include <utility>

class TClonesArray;

class PndStraightLineTrackFinderTask : public FairTask {
 public:
  PndStraightLineTrackFinderTask(Int_t inFinderMode = 0, TString hitBranch = "LMDHitsStrip", TString clusterBranch = "LMDStripClusterCand", TString digiBranch = "LMDStripDigis",
                                 Int_t innSensPP = 8);
  virtual ~PndStraightLineTrackFinderTask();

  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; };

 private:
  bool flagStipSens;
  bool flagPixelSens;
  Double_t dXY;
  Int_t fFinderMode;
  Int_t nSensPP;
  //    std::vector<Int_t> GetHitPerCluster(PndSdsClusterStrip* clusterCand);
  Double_t GetTrackDip(PndMCTrack *myTrack);
  Double_t GetTrackCurvature(PndMCTrack *myTrack);

  TString fHitBranchStrip;
  TString fClusterBranchStrip;
  TString fDigiBranchStrip;

  TClonesArray *fStripHitArray;
  TClonesArray *fStripClusterArray;
  TClonesArray *fStripDigiArray;

  TClonesArray *fTrackCandArray;

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

  bool SortHitsByZ(std::vector<std::vector<std::pair<Int_t, bool>>> &hitsd, Int_t nStripHits);
  bool SortHitsByDet(std::vector<std::vector<std::pair<Int_t, bool>>> &hitsd, Int_t nStripHits);
  bool SortHitsByDet2(std::vector<std::vector<std::pair<Int_t, bool>>> &hitsd, Int_t nStripHits); // uses PndLmdDim
  void FindHitsI(std::vector<PndTrackCand> &tofill, std::vector<std::vector<std::pair<Int_t, bool>>> &hitsd, Int_t nStripHits);
  void FindHitsII(std::vector<PndTrackCand> &tofill, std::vector<std::vector<std::pair<Int_t, bool>>> &hitsd, Int_t nStripHits);
  void FindHitsIII(std::vector<PndTrackCand> &tofill, std::vector<std::vector<std::pair<Int_t, bool>>> &hitsd, Int_t nStripHits);

  ClassDef(PndStraightLineTrackFinderTask, 2);
};

#endif /* PndStraightLineTrackFinderTask_H */