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>
 
 
class CbmL1Track;
class CbmL1HitDebugInfo;
 
class TFolder;
class TH1F;
class TProfile;
class CbmQaCanvas;
 
namespace cbm::algo::ca
{
  class McPoint;
}
 
namespace cbm::ca
{
  namespace phys      = cbm::algo::ca::constants::phys;
  namespace constants = cbm::algo::ca::constants;
  using cbm::algo::ca::McPoint;

  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::TrackParamD& trPar, const 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);

    void SetDefaultProperties();
 
    // ************************
    // ** 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;  
 
    // ** Chi^2 **
    TH1F* fph_chiSqNdf = nullptr;  
    TH1F* fph_prob     = 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:
    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