CbmL1Track.h

Go to the documentation of this file.

/* Copyright (C) 2006-2022 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Ivan Kisel,  Sergey Gorbunov, Denis Bertini [committer], Igor Kulakov, Maksym Zyzak, Sergei Zharko */
 
/*
 *====================================================================
 *
 *  CBM Level 1 Reconstruction 
 *  
 *  Authors: I.Kisel,  S.Gorbunov
 *
 *  e-mail : ikisel@kip.uni-heidelberg.de 
 *
 *====================================================================
 *
 *  L1 track class
 *
 *====================================================================
 */
 
// TODO: SZh 06.06.2024: Rework this class (inherit the new CbmTrack class with additional functionality)
 
#ifndef CbmL1Track_H
#define CbmL1Track_H
 
#include "CaVector.h"
#include "CbmL1Constants.h"
#include "KfTrackParam.h"
#include "TMath.h"
 
#include <iterator>
#include <map>
#include <string>
#include <vector>
 
namespace
{
  namespace ca = cbm::algo::ca;
}
 
 
class CbmL1Track : public cbm::algo::kf::TrackParamD {
 public:
  CbmL1Track() = default;

  void AddMCTrackIndex(int iMT) { fvMcTrackIndexes.push_back_no_warning(iMT); }

  void ClearMatchedMCTracks() { fvMcTrackIndexes.clear(); }
 
  bool IsGhost() const { return (maxPurity < CbmL1Constants::MinPurity); }

  int GetFirstHitIndex() const { return Hits.front(); }

  int GetLastHitIndex() const { return Hits.back(); }

  const auto& GetHitIndexes() const { return Hits; }
 

  const auto& GetMCTrackIndexes() const { return fvMcTrackIndexes; }

  double GetMaxPurity() const { return maxPurity; }

  int GetMatchedMCTrackIndex() const { return fvMcTrackIndexes.size() ? fvMcTrackIndexes[0] : -1; }

  int GetNofHits() const { return Hits.size(); }

  int GetNofMCTracks() const { return fvMcTrackIndexes.size(); }

  int GetNofStations() const { return nStations; }
 

  void SetMaxPurity(double maxPurity_) { maxPurity = maxPurity_; }

  double GetProb() const { return TMath::Prob(GetChiSq(), GetNdf()); }

  std::string ToString(int verbose = 10, bool header = false) const;
 
  cbm::algo::kf::TrackParamD Tpv;    
  cbm::algo::kf::TrackParamD TLast;  
 
  std::vector<int> Hits;  
  int nStations{0};       
  int index{0};           
 
  double fTrackTime{0.};  
 
  std::map<int, int> hitMap;  // N hits (second) from each mcTrack (first is a MC track ID) belong to current recoTrack
                              // FIXME: SZh 14.12.2022:  map => unordered_map
 
 private:
  ca::Vector<int> fvMcTrackIndexes = {"CbmL1Track::fvMcTrackIndexes"};  // global indexes of MC tracks
  // NOTE: mcTracks should be replaced with fvMcTrackIndexes
 
  double maxPurity{-1.};  
};
 
#endif