CbmCaTrackFitQa.h

Go to the documentation of this file.

/* Copyright (C) 2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergei Zharko [committer] */
 
 
 
#pragma once
 
#include "CaDefs.h"
#include "CaEnumArray.h"
#include "CbmL1DetectorID.h"
#include "CbmQaIO.h"
#include "KfFieldRegion.h"
#include "KfTrackParam.h"
 
#include <array>
 
// Forward declarations
namespace cbm::ca::tools
{
  class MCData;
  class MCPoint;
}  // namespace cbm::ca::tools
class CbmL1Track;
class CbmL1HitDebugInfo;
 
class TFolder;
class TH1F;
class TProfile;
class CbmQaCanvas;
 
namespace cbm::ca
{
  namespace phys      = cbm::algo::ca::constants::phys;
  namespace constants = cbm::algo::ca::constants;
 
  enum class ETrackParType
  {
    kX,        
    kY,        
    kTX,       
    kTY,       
    kQP,       
    kTIME,     
    kVI,       
    END,       
    BEGIN = 0  
  };
 
  ETrackParType operator++(ETrackParType& type);
 
  ETrackParType operator++(ETrackParType& type, int);
 
 
  template<typename T>
  using TrackParArray_t = ca::EnumArray<ETrackParType, T>;
 
  class TrackFitQa : public CbmQaIO {
   public:
    TrackFitQa(const char* pointTag, const char* prefix, std::shared_ptr<ObjList_t> pObjList);
 
    ~TrackFitQa() = default;
 
    TrackFitQa(const TrackFitQa&) = delete;
 
    TrackFitQa(TrackFitQa&&) = delete;
 
    TrackFitQa& operator=(const TrackFitQa&) = delete;
 
    TrackFitQa& operator=(TrackFitQa&&) = delete;
 
    const char* GetTitle() const { return fsTitle; }
 
    void Init();
 
    void Fill(const cbm::algo::kf::TrackParamV& trPar, const tools::MCPoint& mcPoint, bool bTimeMeasured,
              double weight = 1);
 
    void SetParticleMass(double mass) { fMass = mass; }
 
    void SetTitle(const char* title) { fsTitle = title; }
 
    void FitHistograms() {}
 
    CbmQaCanvas* CreateResidualPlot();
 
    CbmQaCanvas* CreatePullPlot();
 
    CbmQaCanvas* CreateResolutionPlot() { return nullptr; }
 
    void SetResidualHistoProperties(ETrackParType type, int nBins, double lo, double up);
 
    void SetPullHistoProperties(ETrackParType type, int nBins, double lo, double up);
 
    // ************************
    // ** List of histograms **
    // ************************
 
 
    TH1F* fph_res_x  = nullptr;  
    TH1F* fph_res_y  = nullptr;  
    TH1F* fph_res_tx = nullptr;  
    TH1F* fph_res_ty = nullptr;  
    TH1F* fph_res_qp = nullptr;  
    TH1F* fph_res_t  = nullptr;  
    TH1F* fph_res_vi = nullptr;  
 
    // ** Pulls **
    TH1F* fph_pull_x  = nullptr;  
    TH1F* fph_pull_y  = nullptr;  
    TH1F* fph_pull_tx = nullptr;  
    TH1F* fph_pull_ty = nullptr;  
    TH1F* fph_pull_qp = nullptr;  
    TH1F* fph_pull_t  = nullptr;  
    TH1F* fph_pull_vi = nullptr;  
 
    // ** Resolution profiles **
    TProfile* fph_res_p_pMC         = nullptr;  
    TProfile* fph_res_phi_phiMC     = nullptr;  
    TProfile* fph_res_theta_thetaMC = nullptr;  
 
    // **************************
    // ** Histogram properties **
    // **************************
 
    // ** Binning **
    static constexpr int kCXSIZEPX = 600;  
    static constexpr int kCYSIZEPX = 600;  
 
   private:
    void SetDefaultProperties();
 
    using FnVal_t = std::function<double()>;
    void FillResAndPull(ETrackParType type, double recoVal, double recoErr, double trueVal);
 
    TrackParArray_t<TH1F*> fvphResiduals = {{0}};  
    TrackParArray_t<TH1F*> fvphPulls     = {{0}};  
 
    TrackParArray_t<bool> fvbParIgnored = {{0}};  
 
    // ** Distribution properties **
    TrackParArray_t<int> fvRBins  = {{0}};  
    TrackParArray_t<double> fvRLo = {{0}};  
    TrackParArray_t<double> fvRUp = {{0}};  
 
    TrackParArray_t<int> fvPBins  = {{0}};  
    TrackParArray_t<double> fvPLo = {{0}};  
    TrackParArray_t<double> fvPUp = {{0}};  
 
    TString fsTitle = "";  
 
    double fMass = constants::phys::MuonMass;  
 
    ClassDefNV(TrackFitQa, 0);
  };
 
}  // namespace cbm::ca