CbmKfFitTracksTask.h

Go to the documentation of this file.

/* Copyright (C) 2023-2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: S.Gorbunov[committer] */

 
 
#pragma once
 
#include "CbmKfTrackFitter.h"
#include "FairTask.h"
 
class TClonesArray;

class CbmKfFitTracksTask : public FairTask {
 public:
  enum FitMode
  {
    kSts,
    kMcbm,
    kGlobal
  };
 
  // Constructors/Destructors ---------
  CbmKfFitTracksTask(FitMode mode = FitMode::kSts, Int_t iVerbose = 0);
 
  const CbmKfFitTracksTask& operator=(const CbmKfFitTracksTask&) = delete;
  CbmKfFitTracksTask(const CbmKfFitTracksTask&)                  = delete;
 
  virtual ~CbmKfFitTracksTask();
 
  InitStatus Init() override;
  void Exec(Option_t* opt) override;
  void Finish() override;
 
  void SetFitGlobalTracks() { fFitMode = FitMode::kGlobal; }
  void SetFitStsTracks() { fFitMode = FitMode::kSts; }
  void SetFitMcbmTracks() { fFitMode = FitMode::kMcbm; }
 
  void SetMaxExtrapolationStep(double step) { fFitter.SetMaxExtrapolationStep(step); }
 
  void SetNthreads(int nthreads) { fNthreads = nthreads; }
  int GetNthreads() const { return fNthreads; }
 
 private:
  FitMode fFitMode{FitMode::kGlobal};  

 
  TClonesArray* fGlobalTracks{nullptr};  
  TClonesArray* fStsTracks{nullptr};     
  TClonesArray* fMuchTracks{nullptr};    
  TClonesArray* fTrdTracks{nullptr};     
  TClonesArray* fTofTracks{nullptr};     
 
  int fNthreads{-1};  
 
  CbmKfTrackFitter<> fFitter;  
 
  Int_t fNeventsProcessed{0};   
  Int_t fStatNtracks{0};        
  Int_t fStatNfittedTracks{0};  
 
  ClassDefOverride(CbmKfFitTracksTask, 0);
};