CbmLitFindGlobalTracks.h

Go to the documentation of this file.

 
#ifndef CBMLITFINDGLOBALTRACKS_H_
#define CBMLITFINDGLOBALTRACKS_H_
 
#include "FairTask.h"
 
#include "CbmVertex.h"
#include "base/CbmLitDetectorSetup.h"
#include "base/CbmLitPtrTypes.h"
#include "base/CbmLitTypes.h"
#include "data/CbmLitHit.h"
 
#include "TStopwatch.h"
 
#include <string>
#include <vector>
 
class TClonesArray;
class CbmEvent;
 
using std::string;
using std::vector;
 
class CbmLitFindGlobalTracks : public FairTask {
public:
  CbmLitFindGlobalTracks();
 
  virtual ~CbmLitFindGlobalTracks();
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* opt);
 
  virtual void Finish();
 
  virtual void SetParContainers();
 
  /* Setters */
  void SetTrackingType(const string& trackingType) {
    fTrackingType = trackingType;
  }
  void SetMergerType(const string& mergerType) { fMergerType = mergerType; }
  void SetFitterType(const string& fitterType) { fFitterType = fitterType; }
 
private:
  void InitTrackReconstruction();
 
  void ConvertInputData(CbmEvent* event);
 
  void ConvertOutputData(CbmEvent* event);
 
  /*
    * \brief Calculate length of the global track
    */
  void CalculateLength(CbmEvent* event);
 
  /*
    * \brief Calculate and set the parameter set in the primary vertex of the global track
    */
  void CalculatePrimaryVertexParameters(CbmEvent* event);
 
  void ClearArrays();
 
  void ReadInputBranches();
 
  void RegisterOutputBranches();
 
  void RunTrackReconstruction();
 
  void PrintStopwatchStatistics();
 
  void SelectTracksForTofMerging();
 
  CbmLitDetectorSetup fDet;
 
  // Pointers to data arrays
  // Input
  TClonesArray* fMvdHits;        // CbmMvdHit array
  TClonesArray* fStsTracks;      // CbmStsTrack array
  TClonesArray* fStsHits;        // CbmStsHit
  TClonesArray* fMuchPixelHits;  // CbmMuchPixelHit array
  TClonesArray* fTrdHits;        // CbmTrdHit array
  TClonesArray* fTofHits;        // CbmTofHit array
  TClonesArray* fEvents;
  // Output
  TClonesArray* fMuchTracks;    // output CbmMuchTrack array
  TClonesArray* fTrdTracks;     // output CbmTrdTrack array
  TClonesArray* fTofTracks;     // output CbmTofTrack array
  TClonesArray* fGlobalTracks;  //output CbmGlobalTrack array
 
  CbmVertex* fPrimVertex;  // Pointer to the primary vertex
 
  // LIT data arrays
  TrackPtrVector fLitStsTracks;           // STS tracks
  HitPtrVector fLitHits;                  // MUCH+TRD hits
  HitPtrVector fLitTofHits;               // TOF hits
  TrackPtrVector fLitOutputTracks;        // output Lit tracks
  TofTrackPtrVector fLitOutputTofTracks;  // output Lit TOF tracks
 
  // Tools
  TrackFinderPtr fFinder;       // track finder
  HitToTrackMergerPtr fMerger;  // hit-to-track merger
  TrackFitterPtr fFitter;       // track fitter
  // track propagator
  // Used to propagate STS track to the last STS station!!!
  // Since this cannot be done in parallel mode!!!
  TrackPropagatorPtr fPropagator;
 
  // Settings
  // Tracking method to be used
  // "branch" - branching method
  // "nn" - nearest neighbor method
  string fTrackingType;
 
  // Merger method to be used
  // "nearest_hit" - assigns nearest hit to the track
  string fMergerType;
 
  // Track fitter to be used for the final track fit
  // "lit_kalman" - forward Kalman track fit with LIT propagation and TGeo navigation
  string fFitterType;
 
  // stopwatches
  TStopwatch fTrackingWatch;  // stopwatch for tracking
  TStopwatch fMergerWatch;    // stopwatch for merger
 
  Int_t fEventNo;  // event counter
 
  CbmLitFindGlobalTracks(const CbmLitFindGlobalTracks&);
  CbmLitFindGlobalTracks& operator=(const CbmLitFindGlobalTracks&);
 
  ClassDef(CbmLitFindGlobalTracks, 1);
};
 
#endif /* CBMLITFINDGLOBALTRACKS_H_ */