PndFtsTrackFinderTask.h

Go to the documentation of this file.

#pragma once

#include "TClonesArray.h"

#include "FairTask.h"

#include "PndTrackCand.h"
#include "PndFtsContext.h"
#include "PndFtsReconRunner.h"
#include "FairTask.h"
#include "FairRootManager.h"
#include "PndPersistencyTask.h"

#include "PndFtsTrackFinderPndTrackProducer.h"

#include "tool/PndFtsSimpleHit.h"

class PndFtsTrackFinderTask : public PndPersistencyTask {
 public:
  explicit PndFtsTrackFinderTask(const TString &name = "PndFtsTrackFinder");

  PndFtsTrackFinderTask(const PndFtsTrackFinderTask &) = delete;

  PndFtsTrackFinderTask(const PndFtsTrackFinderTask &&) = delete;

  PndFtsTrackFinderTask operator=(const PndFtsTrackFinderTask &) = delete;

  PndFtsTrackFinderTask operator=(const PndFtsTrackFinderTask &&) = delete;

  ~PndFtsTrackFinderTask() override;

  void Exec(Option_t *option) override;

  void SetOutputBranchName(const TString &name);

  void SetMomentumEstimationMethod(PndFtsTrackFinder::PndFtsMomEstMethod momEstMethod);

 protected:
  InitStatus Init() override;

  void Finish() override;

 private:
  void PrintTmpTracks() const;

  [[nodiscard]] static PndFtsTrackFinder::PndFtsGeom ReadRootGeom(std::string_view filename);

  FairRootManager *ioman; 

  int32_t fInBranchId;                                 
  TString fInBranchName;                               
  TString fOutBranchName;                              
  TString fOutCandBranchName;                          
  TString fOutAnalyticBranchName;                      
  PndFtsTrackFinder::PndFtsMomEstMethod fMomEstMethod; 

  TClonesArray *fInHits; 

  TClonesArray fOutTrackCands;     
  TClonesArray fOutTracks;         
  TClonesArray fOutAnalyticTracks; 

  std::vector<PndFtsTrackFinder::PndFtsSimpleHit> fTmpHits;   
  std::vector<PndFtsTrackFinder::PndFtsFullTrack> fTmpTracks; 

  // Those are unique_ptr for the sake of lazy initialization of the task ( Init() method ).
  std::unique_ptr<PndFtsTrackFinder::PndFtsContext> fReconContext;      
  std::unique_ptr<PndFtsTrackFinder::PndFtsReconRunner> fReconRunner;   
  std::unique_ptr<PndFtsTrackFinderPndTrackProducer> fPndTrackProducer; 

  ClassDef(PndFtsTrackFinderTask, 1);
};