CbmKFParticleFinder.h

Go to the documentation of this file.

/* Copyright (C) 2014-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Maksym Zyzak, Volker Friese [committer] */
 
//-----------------------------------------------------------
//-----------------------------------------------------------
 
#ifndef CbmKFParticleFinder_HH
#define CbmKFParticleFinder_HH
 
#include "CbmStsTrack.h"
#include "FairTask.h"
#include "TString.h"
 
#include <vector>
 
class CbmKFParticleFinderPID;
class KFParticleTopoReconstructor;
class TClonesArray;
class KFParticleFinder;
class KFPTrackVector;
class CbmMCEventList;
class CbmMCDataArray;
class CbmVertex;
 
struct KFFieldVector {
  float fField[10];
};
 
class CbmKFParticleFinder : public FairTask {
 public:
  // Constructors/Destructors ---------
  CbmKFParticleFinder(const char* name = "CbmKFParticleFinder", Int_t iVerbose = 0);
  ~CbmKFParticleFinder();
 
  void UseMCPV() { fPVFindMode = 0; }
  void ReconstructSinglePV() { fPVFindMode = 1; }
  void RconstructMultiplePV() { fPVFindMode = 2; }
  void UseReconstructedPV() { fPVFindMode = 3; }
 
  void SetStsTrackBranchName(const TString& name) { fStsTrackBranchName = name; }
 
  virtual InitStatus Init();
  virtual void Exec(Option_t* opt);
  virtual void Finish();
 
  const KFParticleTopoReconstructor* GetTopoReconstructor() const { return fTopoReconstructor; }
 
  void SetPIDInformation(CbmKFParticleFinderPID* pid) { fPID = pid; }
 
 
  // set cuts
  void SetPrimaryProbCut(float prob);
 
  // Set SE analysis
  void SetSuperEventAnalysis();
 
  void SetTarget(const std::array<float, 3>& target);
 
  //KF Particle Finder cuts
  void SetMaxDistanceBetweenParticlesCut(float cut);
  void SetLCut(float cut);
  void SetChiPrimaryCut2D(float cut);
  void SetChi2Cut2D(float cut);
  void SetLdLCut2D(float cut);
  void SetLdLCutXiOmega(float cut);
  void SetChi2TopoCutXiOmega(float cut);
  void SetChi2CutXiOmega(float cut);
  void SetChi2TopoCutResonances(float cut);
  void SetChi2CutResonances(float cut);
  void SetPtCutLMVM(float cut);
  void SetPCutLMVM(float cut);
  void SetPtCutJPsi(float cut);
  void SetPtCutCharm(float cut);
  void SetChiPrimaryCutCharm(float cut);
  void SetLdLCutCharmManybodyDecays(float cut);
  void SetChi2TopoCutCharmManybodyDecays(float cut);
  void SetChi2CutCharmManybodyDecays(float cut);
  void SetLdLCutCharm2D(float cut);
  void SetChi2TopoCutCharm2D(float cut);
  void SetChi2CutCharm2D(float cut);
 
  //Add decay to the reconstruction list. By default, if list is empty - all decays are reconstructed.
  void AddDecayToReconstructionList(int pdg);
 
  static double InversedChi2Prob(double p, int ndf);
 
 private:
  void FillKFPTrackVector(KFPTrackVector* tracks, const std::vector<CbmStsTrack>& vRTracks,
                          const std::vector<KFFieldVector>& vField, const std::vector<int>& pdg,
                          const std::vector<int>& trackId, const std::vector<float>& vChiToPrimVtx,
                          bool atFirstPoint = 1) const;
 
  const CbmKFParticleFinder& operator=(const CbmKFParticleFinder&);
  CbmKFParticleFinder(const CbmKFParticleFinder&);
 
  //direct access to the KF Particle Finder object
  KFParticleFinder* GetKFParticleFinder();
 
  //names of input branches
  TString fStsTrackBranchName;  
 
  //input branches
  TClonesArray* fTrackArray;
  TClonesArray* fEvents;
  CbmMCDataArray* fMcTrackArray;            //mc tracks in timeslices
  CbmMCEventList* fMcEventList;             //mc event list in timeslice
  TClonesArray* fTrackMatchArray{nullptr};  //track match
 
  //topology reconstructor
  KFParticleTopoReconstructor* fTopoReconstructor;
  //   KFParticleTopoReconstructor* eventTopoReconstructor;
  int fPVFindMode;
 
  //PID information
  CbmKFParticleFinderPID* fPID{nullptr};
 
  //for super event analysis
  bool fSuperEventAnalysis;
  std::vector<CbmStsTrack> fSETracks;
  std::vector<KFFieldVector> fSEField;
  std::vector<int> fSEpdg;
  std::vector<int> fSETrackId;
  std::vector<float> fSEChiPrim;
 
  ClassDef(CbmKFParticleFinder, 0);
};
 
#endif