PndMvdIdealTrackFinderTask.h

Go to the documentation of this file.

#ifndef PNDMVDIDEALTRACKFINDERTASK_H
#define PNDMVDIDEALTRACKFINDERTASK_H

#include "FairTask.h"
#include "PndMCTrack.h"
//#include "PndMvdGeoPar.h"
#include "PndSdsHit.h"
#include "PndSdsMCPoint.h"
//#include "PndMvdPixel.h"
#include "PndSdsCluster.h"
#include "PndTrackCand.h"
//#include "TrackCand.h"

#include <string>
#include <vector>

class TClonesArray;

class PndMvdIdealTrackFinderTask : public FairTask {
 public:
  PndMvdIdealTrackFinderTask();
  virtual ~PndMvdIdealTrackFinderTask();
  PndMvdIdealTrackFinderTask(const PndMvdIdealTrackFinderTask &) = delete;
  PndMvdIdealTrackFinderTask &operator=(const PndMvdIdealTrackFinderTask &) = delete;

  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(PndSdsCluster* clusterCand);
  void ClearTrackCandMap();
  void AddAndExpand(Int_t trackID, Int_t detnum, Int_t iHit, PndSdsHit *theHit);
  Double_t GetTrackDip(PndMCTrack *myTrack);
  Double_t GetTrackCurvature(PndMCTrack *myTrack);

  TString fHitBranchStrip;
  TString fHitBranchPixel;
  TString fClusterBranchStrip;
  TString fClusterBranchPixel;
  TString fDigiBranchStrip;
  TString fDigiBranchPixel;
  TString fMcBranch;
  TString fTrackBranch;

  TClonesArray *fStripHitArray;
  TClonesArray *fPixelHitArray;
  TClonesArray *fStripClusterArray;
  TClonesArray *fPixelClusterArray;
  TClonesArray *fStripDigiArray;
  TClonesArray *fPixelDigiArray;
  TClonesArray *fMcArray;
  TClonesArray *fTrackArray;

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

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

  ClassDef(PndMvdIdealTrackFinderTask, 2);
};

#endif /* PndMvdIdealTrackFinderTASK_H */