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:
  bool IsNaF() const;

  bool HasBetaLIP() const;

  RichBetaVariable<float> GetBeta() const;

  RichBetaVariable<float> GetBetaError() const;

#ifdef ENABLE_PRODUCTION_CODE
  bool Fill(RichRingR *ringPtr, RichRingBR *ringBPtr);

  // void SetMC(bool isMC) { _isMC = isMC; }
#endif

  void Clear();

  void Dump() const;

  double BetaConsistency() const;

  double BetaConsistencyTof(const TofBaseData &tofData) const;

  int GetTileIndex() const; //<  Radiator tile generating the ring

  bool RingGeomTest() const;

  struct RichBaseDataR {
    RichBetaVariable<float> beta;
    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) postion on the radiator where the ring is generated
    bool hasBetaLIP = false;              //< RICH LIP standalone Reconstruction is available
  };

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
  };

  bool IsAvailable() const { return !m_data.empty(); }

  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