PndApolloniusTripletTrackFinderTask.h

Go to the documentation of this file.

//  PndApolloniusTripletTrackFinderTask
//  Finds Track

#pragma once

#include "FairTask.h"
#include <vector>

#include "PndApolloniusTripletTrackFinder.h"
#include "PndSttStrawMap.h"
#include "PndGeoSttPar.h"

#include "PndPreselectSttHits.h"
#include "PndSttCA.h"
using namespace PndApollonius;

struct TrackSolution;

class PndApolloniusTripletTrackFinderTask : public FairTask {
 public:
  PndApolloniusTripletTrackFinderTask();

  virtual ~PndApolloniusTripletTrackFinderTask();

  virtual InitStatus Init();

  virtual void AddBranchName(TString name) { fBranchMap[name] = nullptr; } 

  void SetParContainers();

  virtual void Exec(Option_t *opt);

  virtual void Finish();
  virtual void SetWithTubeReduction(bool reduction) { fWithTubeReduction = reduction; };
  virtual void SetWithCombiReduction(bool reduction) { fWithCombiReduction = reduction; };
  virtual void SetSTTName(TString name) { fSTT = name; };
  virtual void AddFoundTrackName(TString name) { fBranchNameFoundTrack = name; }
  void SetPreselector(PndPreselectSttHits *val) { fPreselector = val; }

 protected:
  void FillPndTrack(std::vector<TripletSolution> &solutions);
  void FillPndTrackDebug(std::vector<TripletSolution> &solutions, TClonesArray *tracks, TClonesArray *trackCands);
  bool CheckZInfo(PndTrackCand &cand);

 private:
  FairRootManager *ioman = nullptr; 

  PndGeoSttPar *fSttParameters = nullptr; 
  TClonesArray *fTubeArray = nullptr;     

  PndApolloniusTripletTrackFinder *fFinder = nullptr; 

  PndSttStrawMap *fStrawMap;
  std::vector<TString> fBranchNames;            
  std::map<TString, TClonesArray *> fBranchMap; 

  TClonesArray *fTrackCandArray = nullptr; 
  TClonesArray *fTrackArray = nullptr;     

  PndPreselectSttHits *fPreselector = nullptr; 
  PndSttCA *fCATrackFinder = nullptr;          

  double fBz = 2.; 

  bool fWithTubeReduction;  
  bool fWithCombiReduction; 
  TString fSTT;
  TString fBranchNameFoundTrack;

  TClonesArray *fPreselectedTracksCands = nullptr;        
  TClonesArray *fPreselectedTracks = nullptr;             
  TClonesArray *fTripletsFirstCands = nullptr;            
  TClonesArray *fTripletsFirst = nullptr;                 
  TClonesArray *fTripletsMidCands = nullptr;              
  TClonesArray *fTripletsMid = nullptr;                   
  TClonesArray *fTripletsLastCands = nullptr;             
  TClonesArray *fTripletsLast = nullptr;                  
  TClonesArray *fTripletsCombiCands = nullptr;            
  TClonesArray *fTripletsCombi = nullptr;                 
  TClonesArray *fTripletTracksCands = nullptr;            
  TClonesArray *fTripletTracks = nullptr;                 
  TClonesArray *fContinuousTripletTracksCands = nullptr;  
  TClonesArray *fContinuousTripletTracks = nullptr;       
  TClonesArray *fCombinedTripletTracksCands = nullptr;    
  TClonesArray *fCombinedTripletTracks = nullptr;         
  TClonesArray *fTripletTracksCandsAfterAdding = nullptr; 
  TClonesArray *fTripletTracksAfterAdding = nullptr;      
  TClonesArray *fBranchFoundTrack = nullptr;              
  TClonesArray *fSttBranchOriginal = nullptr;             

  ClassDef(PndApolloniusTripletTrackFinderTask, 1);
};