PndHypIdealTrackFinderTask.h

Go to the documentation of this file.

#ifndef PndHypIdealTrackFinderTASK_H
#define PndHypIdealTrackFinderTASK_H

#include "FairTask.h"
#include "PndMCTrack.h"
#include "PndGeoHypPar.h"
#include "PndHypHit.h"
#include "PndHypPoint.h"
//#include "PndHypPixel.h"
//#include "PndHypClusterCand.h"
#include "GFTrackCand.h"
#include "PndHypCluster.h"
#include <string>
#include <vector>

class TClonesArray;

class PndHypIdealTrackFinderTask : public FairTask {
 public:
  PndHypIdealTrackFinderTask();
  virtual ~PndHypIdealTrackFinderTask();

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

  virtual void Exec(Option_t *opt);

  void PrintResult();
  void SetVerbose(Int_t verbose) { fVerbose = verbose; };

 private:
  // std::vector<Int_t> GetHitPerCluster(PndHypCluster* clusterCand);
  void ClearTrackCandMap();
  void AddAndExpand(Int_t trackID, Int_t detnum, Int_t iHit);
  Double_t GetTrackDip(PndMCTrack *myTrack);
  Double_t GetTrackCurvature(PndMCTrack *myTrack);

  TString fHitBranchStrip;
  // TString fClusterBranchStrip;
  // TString fDigiBranchStrip;
  TString fMcBranch;
  TString fTrackBranch;

  TClonesArray *fStripHitArray;
  // TClonesArray* fStripClusterArray;
  // TClonesArray* fStripDigiArray;
  TClonesArray *fMcArray;
  TClonesArray *fTrackArray;

  TClonesArray *fTrackCandArray;
  std::map<Int_t, GFTrackCand *> fTrackCandMap;

  Int_t fVerbose;

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

  ClassDef(PndHypIdealTrackFinderTask, 2);
};

#endif /* PndHypIdealTrackFinderTASK_H */