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:
  Rich::RichMeasType GetBetaType() const;

  RichBetaVariable<float> GetBeta() const;

#ifdef ENABLE_PRODUCTION_CODE
  void 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;

private:
  // ugly empty-object optimization trick
  std::vector<std::pair<Rich::RichMeasType, RichBetaVariable<float>>> 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 {
    short NHitsTot = 0;             
    short NHitsUsed = 0;            
    short NPEExpUncorr = 0;         
    short NPEMeasUncorr = 0;        
    short RingPMTs = 0;             
    short NBadClusters = 0;         
    short NSecondaryHits = 0;       
    float NPETotUncorr = 0.0f;      
    float NPEHyp1UncorrTot = 0.0f;  
    double Probability = 0.0;       
    double ChargeConsistency = 0.0; 
    // clang-format off
    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.size() > 0; }

  const RichAdditionalData &GetData() const { return m_data.at(0); }
#ifdef ENABLE_PRODUCTION_CODE
  void Fill(AMSEventR *evPtr, RichRingR *ringPtr, 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