Rich.h

Go to the documentation of this file.

 
#ifndef NAIA_RICH_H
#define NAIA_RICH_H
 
// gbatch headers
#ifdef ENABLE_PRODUCTION_CODE
#include "root.h"
#endif
 
#include "Containers/OnDemandContainer.h"
#include "Containers/Tof.h"
#include "Containers/Utils.h"
 
#include <array>
#include <map>
#include <vector>
 
namespace NAIA {
 
template <class T> using RichBetaVariable = std::map<Rich::BetaType, T>;
 
class RichBaseData : public TObject {
public:
  enum RingGeom { Borders, BadAglTiles, AglTilesEdge, BadAglRegions };
  [[nodiscard]] bool IsNaF() const;
 
  [[nodiscard]] bool HasBetaLIP() const;
 
  [[nodiscard]] RichBetaVariable<float> GetBeta() const;
 
  [[nodiscard]] RichBetaVariable<float> GetBetaError() const;
 
  [[nodiscard]] std::array<float, 2> GetTrackEmissionPoint() const;
 
#ifdef ENABLE_PRODUCTION_CODE
  bool Fill(RichRingR *ringPtr, RichRingBR *ringBPtr);
 
  // void SetMC(bool isMC) { _isMC = isMC; }
#endif
 
  void Clear();
 
  void Dump() const;
 
  [[nodiscard]] double BetaConsistency() const;
 
  [[nodiscard]] double BetaConsistencyTof(const TofBaseData &tofData) const;
 
  [[nodiscard]] int GetTileIndex() const; //<  Radiator tile generating the ring
 
  [[nodiscard]] bool RingGeomTest(RingGeom geom = RingGeom::Borders) const;
 
  // clang-format off
  struct RichBaseDataR {
    RichBetaVariable<float> beta;         //< Measured beta of type defined by Rich::BetaType, CIEMAT: Refit and corrected, SIMPLE: Reconstructed velocity using only hits (no charge), REFIT:  Beta estimate taking into account the number of photoelectrons for each hit, LIP: Beta with LIP standalone algo
    RichBetaVariable<float> ebeta;        //< Error on Measured beta of type defined by Rich::BetaType
    bool isNaF = false;                   //<  Ring on NaF radiator (true) or AGL (false)
    bool isGood = false;                  //< IsClean (i.e. (Status&2)!=0)  && getExpectedPhotoelectrons(false)>1e-6
    std::array<float, 2> Pos{0.0f, 0.0f}; //< (X,Y) position on the radiator where the ring is generated
    float Theta{0.0f};                    //< Track theta angle on the radiator
    float Phi{0.0f};                      //< Track phi angle on the radiator
    bool hasBetaLIP = false;              //< RICH LIP standalone Reconstruction is available
  };
  // clang-format on
 
private:
  // ugly empty-object optimization trick
  std::vector<RichBaseDataR> m_beta;
 
  ClassDef(RichBaseData, 1)
};
 
class RichBase : public OnDemandContainer<RichBase>, public RichBaseData {
public:
  static const std::string BranchName;
 
  RichBase() : RichBaseData(), OnDemandContainer(){};
};
 
class RichPlusData : public TObject {
public:
  enum HitType { Direct, Reflected };
  struct RichAdditionalData {
    // clang-format off
    short NHitsTot = 0; 
    float NPETotUncorr = 0.0f; 
 
    short NHitsUsed = 0;      
    float NPEExpUncorr = 0;   
    float NPEExpCorr = 0;     
    float NPEMeasUncorr = 0;  
    float NPEMeasCorr = 0;    
    double Probability = 0.0; 
    short RingPMTs = 0;       
    short RingPMTs2 = 0;      
    short RichPMTs = 0;       
    double ChargeConsistency = 0.0f; 
    float RichCharge = 0.0f;         
    float RichCharge2 = 0.0f;        
 
    float LIP_ringProb = 0.0f; 
 
    short NBadClusters = 0;        
    short NSecondaryHits = 0;      
    float NPEHyp1UncorrTot = 0.0f; 
 
    std::array<short, 2> NHitsHyp1Tot{0, 0}; 
    std::array<short, 2> NHitsHyp1OutOfRing{0, 0}; 
    float NPEHyp1UncorrOutOfRing =0.0f; 
    // clang-format on
  };
 
  [[nodiscard]] bool IsAvailable() const { return !m_data.empty(); }
 
  [[nodiscard]] const RichAdditionalData &GetData() const { return m_data.at(0); }
#ifdef ENABLE_PRODUCTION_CODE
  bool Fill(AMSEventR *evPtr, RichRingR *ringPtr, RichRingBR *ringPtrB, TrTrackR *trackPtr);
  void FillHyp1Data(RichAdditionalData &data, AMSEventR *evPtr, RichRingR *ringPtr, TrTrackR *trackPtr);
 
  void SetMC(bool isMC) { m_isMC = isMC; }
#endif
 
  void Clear();
 
  void Dump() const;
 
private:
  bool m_isMC = false; 
  std::vector<RichAdditionalData> m_data;
 
  ClassDef(RichPlusData, 1)
};
 
class RichPlus : public OnDemandContainer<RichPlus>, public RichPlusData {
public:
  static const std::string BranchName;
 
  RichPlus() : RichPlusData(), OnDemandContainer() {}
};
 
} // namespace NAIA
 
#endif