UnbExtHit.h

Go to the documentation of this file.

 
#ifndef NAIA_UNBEXTHIT_H
#define NAIA_UNBEXTHIT_H
 
// gbatch headers
#ifdef ENABLE_PRODUCTION_CODE
#include "root.h"
#endif
 
#include "Containers/OnDemandContainer.h"
#include "Containers/Tof.h"
#include "Containers/TrdK.h"
#include "Containers/Utils.h"
 
#include <array>
#include <map>
#include <vector>
 
namespace NAIA {
 
template <class T> using HitChargeVariable = std::map<TrTrack::ChargeRecoType, T>;
 
class UnbExtHitBaseData : public TObject {
public:
  enum class ExtHit : char { L1 = 0, L9 = 1 };
 
#ifdef ENABLE_PRODUCTION_CODE
  bool Fill(AMSEventR *evPtr, bool use_trd = false);
 
  void SetBeta(double beta) { _beta = beta; }
  void SetMC(bool isMC) { _isMC = isMC; }
#endif
 
  void Clear();
 
  void Dump() const;
 
  [[nodiscard]] bool IsHitPresent(ExtHit hit) const { return m_hits.find(hit) != end(m_hits); }
 
  int ChargeStatus(ExtHit hit) { return m_hits[hit].ChargeStatus; }
 
  float HitPosX(ExtHit hit) { return m_hits[hit].HitPosX; }
 
  float HitPosY(ExtHit hit) { return m_hits[hit].HitPosY; }
 
  float Charge(ExtHit hit, TrTrack::ChargeRecoType chType) { return m_hits[hit].Charge[chType]; }
 
  const TofBaseData *tofBase = nullptr;  
  const TrdKBaseData *trdBase = nullptr; 
 
  struct HitData {
    int ChargeStatus = 0xFFFFFFFF;
 
    float HitPosX = std::numeric_limits<float>::max();
    float HitPosY = std::numeric_limits<float>::max();
 
    HitChargeVariable<float> Charge;
  };
 
private:
  bool _isMC = false; 
  float _beta = 1.0f; 
 
  std::map<ExtHit, HitData> m_hits;
 
  ClassDef(UnbExtHitBaseData, 1)
};
 
class UnbExtHitBase : public OnDemandContainer<UnbExtHitBase>, public UnbExtHitBaseData {
public:
  static const std::string BranchName;
 
  UnbExtHitBase() : UnbExtHitBaseData(), OnDemandContainer() {}
};
 
} // namespace NAIA
#endif