PndCurlingTrackFinderTask.h

Go to the documentation of this file.

//  PndCurlingTrackFinderTask
//  Finds Track

#pragma once

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

#include "PndCurlingTrackFinder.h"
#include "PndSttStrawMap.h"
#include "PndGeoSttPar.h"
#include "PndSttGeometryMap.h"

#include "PndPreselectSttHits.h"
#include "PndSttCA.h"

struct TrackSolution;

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

  virtual ~PndCurlingTrackFinderTask();

  virtual InitStatus Init();

  virtual void AddBranchName(TString name) { fBranchMap[name] = nullptr; } 
  virtual void AddBranchNameUnassigned(TString name)
  {
    fBranchMapUnassigned[name] = nullptr;
  } 
  virtual void AddFoundTrackName(TString name) { fBranchNameFoundTrack = name; }
  virtual void AddFoundTrackCandName(TString name) { fBranchNameFoundTrackCand = name; }
  void SetParContainers();

  virtual void Exec(Option_t *opt);

  virtual void Finish();

 protected:
  void FillPndTrack(std::vector<TripletSolution> &solutions);
  void FillPndTrackDebug(std::vector<TripletSolution> &solutions, TClonesArray *tracks, TClonesArray *trackCands);
  PndTrack GetPndTrack(double B, std::vector<FairHit *> &hits, TVector3 &circle, PndTrackCand &cand);

 private:
  FairRootManager *ioman = nullptr;

  PndGeoSttPar *fSttParameters = nullptr; // needed for SttStrawMap
  TClonesArray *fTubeArray = nullptr;     // needed for SttStrawMap

  //  PndSttStrawMap fStrawMap; // for getting more information about the tubes
  PndSttGeometryMap *fGeometryMap = nullptr;

  PndCurlingTrackFinder *fFinder = nullptr;

  std::vector<TString> fBranchNames;
  std::map<TString, TClonesArray *> fBranchMap;
  std::vector<TString> fBranchNamesUnassigned;
  std::map<TString, TClonesArray *> fBranchMapUnassigned;
  TString fBranchNameFoundTrack;
  TString fBranchNameFoundTrackCand;
  TClonesArray *fBranchFoundTrack;
  TClonesArray *fBranchFoundTrackCand;
  TClonesArray *fTrackCandArray = nullptr;
  TClonesArray *fTrackArray = nullptr;
  TClonesArray *fTripletTracksCands = nullptr;
  TClonesArray *fTripletTracks = nullptr;
  TClonesArray *fContinuousTripletTracksCands = nullptr;
  TClonesArray *fContinuousTripletTracks = nullptr;
  TClonesArray *fTripletTracksCandsAfterAdding = nullptr;
  TClonesArray *fTripletTracksAfterAdding = nullptr;
  TClonesArray *fFinalTrackCandArray = nullptr;
  TClonesArray *fFinalTrackArray = nullptr;
  PndSttCA *fCATrackFinder = nullptr;

  bool fWithTubeReduction;
  bool fWithCombiReduction;
  int fVerbose;

  ClassDef(PndCurlingTrackFinderTask, 1);
};