CbmRichRingRecoQa.h

Go to the documentation of this file.

/* Copyright (C) 2025 Justus-Liebig-Universitaet Giessen, Giessen
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Martin Beyer [committer] */

 
#pragma once
 
#include "CbmQaTask.h"
 
class CbmDigiManager;
class CbmMCDataArray;
class CbmMCTrack;
class CbmRichRing;
class TClonesArray;

class CbmRichRingRecoQa : public CbmQaTask {
 public:
  CbmRichRingRecoQa(int verbose, bool isMcUsed) : CbmQaTask("CbmRichRingRecoQa", verbose, isMcUsed){};
 
  InitStatus InitQa() override;
 
  void ExecQa() override;
 
  void Check() override;
 
  void CreateSummary() override{};
 
 private:
  enum ERecoRingType
  {
    Reco     = -1,  // Reconstruced ring
    TrueReco = 0,   // [MC] True reconstructed ring, all particle types
    PrimEl   = 1,   // [MC] True reconstructed primary electron ring
    SecEl    = 2,   // [MC] True reconstructed secondary electron ring
    Pion     = 3,   // [MC] True reconstructed pion ring
    Other    = 4,   // [MC] True reconstructed other ring, e.g. muon
    Fake     = 5,   // [MC] Fake/Ghost ring (e.g. sub quota)
    Clone    = 6,   // [MC] Clone ring, i.e. multiple reconstructed rings for same accepted ring
  };
 
  enum ERingFeature
  {
    Radius         = 0,   // Ring radius
    NHits          = 1,   // Number of ring hits
    CenterX        = 2,   // Center x of ellipse fit
    CenterY        = 3,   // Center y of ellipse fit
    Aaxis          = 4,   // A axis of ellipse fit
    Baxis          = 5,   // B axis of ellipse fit
    BoA            = 6,   // B axis / A axis
    Phi            = 7,   // Ring tilt/rotation angle
    Chi2oNDF       = 8,   // Chi2 / (nHits - 5) [5 for ellipse]
    Angle          = 9,   // Largest sum of 3 consecutive angles between hits
    RadialPosition = 10,  // Radial position
    RadialAngle    = 11,  // Radial angle
  };
 
  void InitHistograms();
 
  void FillHistsTrueRecoRing(const CbmRichRing* ring, const CbmMCTrack* mcTrack);
 
  void FillHistsFakeRing(const CbmRichRing* ring);
 
  void FillHistsRingFeatures(const CbmRichRing* ring, const CbmMCTrack* mcTrack, const ERecoRingType ringType);
 
  bool IsMcPrimaryElectron(const CbmMCTrack* mcTrack);
 
  bool IsMcSecondaryElectron(const CbmMCTrack* mcTrack);
 
  bool IsMcPion(const CbmMCTrack* mcTrack);
 
  static constexpr int fAccRingMinNumHits{7};  // [MC] Minimum amount of hits for a accepted ring
  static constexpr double fQuotaRich{0.7};     // [MC] Fraction of hits from a mother particle needed to be true reco
 
  // [MC] For now very generous checks for efficiencies and purity
  static constexpr std::array<double, 4> fRingRecoEffsMin{
    0.5, 0.8, 0.5, 0.2};                        // Minimum valid efficiency for TrueReco, PrimEl, SecEl and pion rings
  static constexpr double fRingPurityMin{0.7};  // Minimum valid ring reco purity
 
  // Input data
  CbmDigiManager* fDigiManager{nullptr};
  TClonesArray* fCbmEvents{nullptr};
  TClonesArray* fRichHits{nullptr};
  TClonesArray* fRichRings{nullptr};
  TClonesArray* fRichProjections{nullptr};
 
  // [MC] Input data
  CbmMCDataArray* fMcTracks{nullptr};
  CbmMCDataArray* fRichPoints{nullptr};
  TClonesArray* fRichRingMatches{nullptr};
 
  // Histograms
  TH1D* fhRichRingsPerEvent{nullptr};      // Number of rings per CbmEvent
  TH1D* fhRingTimeInEvent{nullptr};        // Ring time in CbmEvent
  TH1D* fhRingHitsTime{nullptr};           // Hit times in ring
  TH2D* fhRingCenterXY{nullptr};           // Position of rings (center of ellipse fit) in camera
  std::array<TH1D*, 12> fhRingFeatures{};  // Ring features (i.e. radius, number of hits, center x, etc.)
 
  // [MC] Histograms
  TH1D* fhRingTypeCounts{nullptr};           // Count rings for various particles types & fake, clone rings
  TProfile2D* fhRingRecoPurityXY{nullptr};   // Ring reconstruction purity in XY camera plane
  std::array<TProfile*, 5> fhRingRecoEff{};  // Ring reconstruction efficiency vs MC momentum for various particle types
  std::array<std::array<TH1D*, 12>, 6> fhRingTypeFeatures{};  // Ring features for all reco ring types, except Clones
  std::array<std::array<TH2D*, 12>, 5> fhRingTypeFeaturesMomentum{};  // Ring features vs MC momentum all particle types
 
  ClassDefOverride(CbmRichRingRecoQa, 1);
};