PndSttCellTrackFinderTask.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

#ifndef PndSttCellTrackFinderTask_H_
#define PndSttCellTrackFinderTask_H_

#include "FairTask.h"
#include "PndGeoSttPar.h"
#include "PndSttGeometryMap.h"
#include "PndSttStrawMap.h"
#include "PndSttCellTrackFinder.h"
#include "FairTSBufferFunctional.h"

#include <vector>

#ifdef RUNCUDA
extern "C" int *AllocateStaticData(int *, int);
extern "C" void FreeStaticData(int *);
#endif

class PndSttCellTrackFinderTask : public FairTask {
 public:
  PndSttCellTrackFinderTask()
    : FairTask("Stt Cell Track Finder"), fClusterTime(250.0), fUseGPU(kFALSE), fDev_tubeNeighborings(nullptr), fRunTimeBased(kFALSE), fRunWithSortedHits(kFALSE),
      fCalcWithCorrectedIsochrones(kFALSE), fVerbose(0), fPersistence(kTRUE), fAnalyseSteps(kFALSE), fTrackFinder(nullptr)
  {
    fOutBranchNamePrefix = "";
    fInBranchNamePrefix = "";
  };

  virtual ~PndSttCellTrackFinderTask(){};

  virtual void SetParContainers();
  virtual InitStatus Init();
  // virtual InitStatus ReInit();

  virtual void Exec(Option_t *opt);

  virtual void FinishEvent();

  virtual void FinishTask();

  // void AddHitBranch(TString branchName);//Has to be called before Init() is used otherwise the default names are taken!

  void SetVerbose(Int_t verbose) { fVerbose = verbose; };
  void SetPersistence(Bool_t val) { fPersistence = val; };
  void SetAnalyseSteps(Bool_t val) { fAnalyseSteps = val; };
  void SetUseGPU(Bool_t val) { fUseGPU = val; };
  void SetRunTimeBased(Bool_t val) { fRunTimeBased = val; };
  void SetRunWithSortedHits(Bool_t val) { fRunWithSortedHits = val; };
  void SetClusterTime(double val) { fClusterTime = val; }; // J.R. 20/04-2018
  void SetCalcWithCorrectedIsochrones(Bool_t val) { fCalcWithCorrectedIsochrones = val; }
  // define a Prefix for the Output in case you run more Instances of the CellTrackFinderTask
  void SetOutBranchNamePrefix(TString prefix) { fOutBranchNamePrefix = prefix + "_"; };
  // set a prefix for the defoult inbut Branch names
  void SetInBranchNamePrefix(TString prefix) { fInBranchNamePrefix = prefix + "_"; };
  // if nothing is defined, the standard STTHit branches are used.
  // If InputBranches are defined, so far only Branches holding data of type STTHit are further processed (See "initHitArray" method)
  void AddHitBranch(TString branchName) { fHitBranch.push_back(branchName); };
  // initialise all STTHit type data,  If someone wants to change this also the "Addhit" method of the Trackfinder must be adapted!
  void InitHitArray(TString branchName);

 private:
  BinaryFunctor *fFunctorCellTrack;

  double fStopTimeValueCellTrack;

  double fClusterTime;
  // for GPU computing
  Bool_t fUseGPU;
  int *fDev_tubeNeighborings;

  Bool_t fRunTimeBased;
  Bool_t fRunWithSortedHits;

  Bool_t fCalcWithCorrectedIsochrones;

  TString fOutBranchNamePrefix;
  TString fInBranchNamePrefix;

  std::vector<TString> fHitBranch;
  std::vector<TString> fSTTHitBranch;
  // TString fTrackBranch;

  std::vector<TClonesArray *> fSTTHitArray;

  // first step of trackfinding
  TClonesArray *fFirstTrackCandArray;
  TClonesArray *fFirstTrackArray;
  TClonesArray *fFirstRiemannTrackArray;

  // second step of trackfinding
  TClonesArray *fCombiTrackCandArray;
  TClonesArray *fCombiRiemannTrackArray;
  TClonesArray *fCombiTrackArray;
  TClonesArray *fCorrectedIsochronesArray;

  Int_t fVerbose;
  Bool_t fPersistence;  // safe data?
  Bool_t fAnalyseSteps; // if true, results of first step (generating tracklets) will be stored for AnalysisTask

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

  PndSttGeometryMap *fSttGeometryMap;
  PndSttStrawMap *fSttStrawMap;
  PndSttCellTrackFinder *fTrackFinder;

  TClonesArray *fEventHeader;

  // for runtime analysis: numHits, numHitsWithoutDouble, numUnambiguousHits
  std::vector<std::vector<int>> fNumHitsPerEvent;

  int eventNumber; // J.R. For testing purpouses

  ClassDef(PndSttCellTrackFinderTask, 1);
};

#endif /*PndSttCellTrackFinderTask_H_*/