CbmKFV0FinderQa.h

Go to the documentation of this file.

/* Copyright (C) 2025 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergei Zharko [committer] */

 
// FIXME: Later move this class into a separate QA-task
 
#pragma once
 
#include "CbmKFV0FinderTask.h"
#include "CbmQaTask.h"
 
#include <TString.h>
 
#include <array>
#include <string>
#include <vector>
 
#include <Math/Vector3D.h>  // Radius-vector
#include <Math/Vector4D.h>  // Lorentz vector
 
class TH1D;
class TH2D;
 
// NOTE: works only for Lambda
// FIXME: generalize for V0
 
namespace cbm::kfp
{
  class V0FinderQa : public CbmQaTask {
   public:
    using Cutter_t   = typename V0FinderTask::Cutter_t;
    using XVector4_t = ROOT::Math::XYZTVector;
    using PVector4_t = ROOT::Math::PxPyPzEVector;
    using Vector3_t  = ROOT::Math::XYZVector;

    V0FinderQa(int verbose, bool bUseMc, ECbmRecoMode recoMode = ECbmRecoMode::Timeslice)
      : CbmQaTask("v0Finder", verbose, bUseMc, recoMode)
    {
    }

    ~V0FinderQa() = default;
 
    V0FinderQa(const V0FinderQa&) = delete;
    V0FinderQa(V0FinderQa&&)      = delete;
    V0FinderQa& operator=(const V0FinderQa&) = delete;
    V0FinderQa& operator=(V0FinderQa&&) = delete;

    void Check() {}

    void ExecQa();

    InitStatus InitQa();

    void SetFrom(const V0FinderTask* pFinder);

    void InitHistograms();

    void WriteHistograms(const TString&);
 
 
   private:
    struct RefHistogramsPrt {
      TH1D* beta{nullptr};           
      TH1D* dca_rt_origin{nullptr};  
      TH1D* p{nullptr};              
      TH1D* pt{nullptr};             
    };

    struct RefHistograms {
      RefHistogramsPrt pion;
      RefHistogramsPrt proton;
      RefHistogramsPrt all;
    };

    struct RefHistogramsV0 {
      TH1D* opening_angle{nullptr};     
      TH1D* opening_angle_kf{nullptr};  
      TH1D* dca_rt_pv{nullptr};         
      TH1D* dca_daughters{nullptr};     
      TH1D* decay_length{nullptr};      
      TH1D* decay_vtx_z{nullptr};       
      TH2D* decay_vtx_xy{nullptr};      
      TH1D* chi2ndf_topo{nullptr};      
      TH1D* chi2ndf_geo{nullptr};       
      TH1D* mass{nullptr};              
    };

    void FillParticles();

    void FillTracks();

    template<bool IsMixedEvent>
    void FillV0Candidates(const KFParticle& v0cand);

    void FillSecondary(RefHistogramsPrt& h, const KFParticle& prt, const V0FinderTask::ParticleInfo& prtInfo);

    void Refit(const XVector4_t& aX, const XVector4_t& bX, XVector4_t& mX, PVector4_t& aP, PVector4_t& bP,
               PVector4_t& mP);
 
    //* Control flags
    bool fbDataBranchesInitialized{false};
    bool fbUseMc{false};
 
    //* Data branches
    std::shared_ptr<const KFParticleTopoReconstructor> fpTopoReconstructor{nullptr};
    std::shared_ptr<const Cutter_t> fpCutter{nullptr};
    const std::vector<V0FinderTask::TrackInfo>* fpTrackInfo{nullptr};
    const std::vector<V0FinderTask::ParticleInfo>* fpParticleInfo{nullptr};
    TClonesArray* fpBrRecoEvents{nullptr};
    TClonesArray* fpBrGlobalTracks{nullptr};
 
    //* Histogram properties (binning)
    static constexpr int kBbeta{200};               
    static constexpr double kLbeta{-0.5};           
    static constexpr double kUbeta{+1.5};           
    static constexpr int kBdca_rt_origin{100};      
    static constexpr double kLdca_rt_origin{+0.0};  
    static constexpr double kUdca_rt_origin{+10.};  
    static constexpr int kBp{80};                   
    static constexpr double kLp{+0.0};              
    static constexpr double kUp{+4.0};              
    static constexpr int kBpt{80};                  
    static constexpr double kLpt{+0.0};             
    static constexpr double kUpt{+4.0};             
 
    static constexpr int kBopening_angle{150};      
    static constexpr double kLopening_angle{+0.0};  
    static constexpr double kUopening_angle{+1.5};  
    static constexpr int kBdca_rt_pv{100};          
    static constexpr double kLdca_rt_pv{+0.0};      
    static constexpr double kUdca_rt_pv{+1.0};      
    static constexpr int kBdca_daughters{100};      
    static constexpr double kLdca_daughters{+0.0};  
    static constexpr double kUdca_daughters{+1.0};  
    static constexpr int kBdecay_length{100};       
    static constexpr double kLdecay_length{0.0};    
    static constexpr double kUdecay_length{30.};    
    static constexpr int kBdecay_vtx_x{200};        
    static constexpr double kLdecay_vtx_x{-25.};    
    static constexpr double kUdecay_vtx_x{+25.};    
    static constexpr int kBdecay_vtx_y{200};        
    static constexpr double kLdecay_vtx_y{-25.};    
    static constexpr double kUdecay_vtx_y{+25.};    
    static constexpr int kBdecay_vtx_z{200};        
    static constexpr double kLdecay_vtx_z{0.0};     
    static constexpr double kUdecay_vtx_z{50.};     
    static constexpr int kBchi2ndf_topo{50};        
    static constexpr double kLchi2ndf_topo{0.0};    
    static constexpr double kUchi2ndf_topo{500.};   
    static constexpr int kBchi2ndf_geo{50};         
    static constexpr double kLchi2ndf_geo{0.0};     
    static constexpr double kUchi2ndf_geo{50.};     
    static constexpr int kBmass{100};               
    static constexpr double kLmass{1.08};           
    static constexpr double kUmass{1.28};           
 
    //* Histograms
    RefHistograms fhRefTracksBefore;     
    RefHistograms fhRefTracksAfter;      
    RefHistograms fhRefSecondaryStored;  
    RefHistograms fhRefSecondaryUnused;  
    RefHistograms fhRefDaughtersBefore;  
    RefHistograms fhRefDaughtersAfter;   
    RefHistogramsV0 fhRefV0Before;       
    RefHistogramsV0 fhRefV0After;        
    RefHistogramsV0 fhRefMeBefore;       
    RefHistogramsV0 fhRefMeAfter;        
 
    ClassDef(V0FinderQa, 0);
  };
}  // namespace cbm::kfp