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;

  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