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 */