CbmStsTracksConverter.h

Go to the documentation of this file.

#ifndef ANALYSIS_TREE_STSTRACKSCONVERTER_H_
#define ANALYSIS_TREE_STSTRACKSCONVERTER_H_
 
#include "CbmConverterTask.h"
 
#include "AnalysisTree/Detector.hpp"
 
class TClonesArray;
class CbmVertex;
class CbmStsTrack;
class CbmTrackMatchNew;
 
namespace AnalysisTree {
  class Matching;
}
 
class CbmStsTracksConverter final : public CbmConverterTask {
 
public:
  enum kInBranches {
    eStsTracks,
    ePrimiryVertex,
    eSimTracks,
    eNumberOfInputBranches
  };
 
  explicit CbmStsTracksConverter(std::string out_branch_name,
                                 std::string match_to = "")
    : CbmConverterTask(std::move(out_branch_name), std::move(match_to)) {
    in_branches_.resize(eNumberOfInputBranches);
    in_branches_.at(eStsTracks)     = "StsTrack";
    in_branches_.at(ePrimiryVertex) = "PrimaryVertex.";
    in_branches_.at(eSimTracks)     = "MCTrack";
  }
 
  ~CbmStsTracksConverter() final;
 
  void Init(std::map<std::string, void*>&) final;
  void Exec() final;
  void Finish() final {}
 
  void SetIsWriteKFInfo(bool is = true) { is_write_kfinfo_ = is; }
  void SetIsReproduceCbmKFPF(bool is = true) { is_reproduce_cbmkfpf_ = is; }
 
private:
  void ReadVertexTracks();
  void MapTracks();
  void InitInput();
  float ExtrapolateToVertex(CbmStsTrack* sts_track,
                            AnalysisTree::Track* track,
                            int pdg);
  void WriteKFInfo(AnalysisTree::Track* track,
                   const CbmStsTrack* sts_track,
                   bool is_good_track) const;
  bool IsGoodCovMatrix(const CbmStsTrack* sts_track) const;
  int GetMcPid(const CbmTrackMatchNew* match, AnalysisTree::Track* track) const;
 
  AnalysisTree::TrackDetector* vtx_tracks_ {
    nullptr};  
  AnalysisTree::Matching* vtx_tracks_2_sim_ {
    nullptr};  
 
  CbmVertex* cbm_prim_vertex_ {nullptr};    
  TClonesArray* cbm_mc_tracks_ {nullptr};   
  TClonesArray* cbm_sts_tracks_ {nullptr};  
  TClonesArray* cbm_sts_match_ {nullptr};   
 
  bool is_write_kfinfo_ {true};
  bool is_reproduce_cbmkfpf_ {true};
 
  int ipar_ {-1};
  int imf_ {-1};
  int icov_ {-1};
  int imc_pdg_ {-1};
  int imother_pdg_ {-1};
  int ipasscuts_ {-1};
 
  ClassDef(CbmStsTracksConverter, 1)
};
 
#endif  // ANALYSIS_TREE_STSTRACKSCONVERTER_H_