Tof.h

Go to the documentation of this file.

#ifndef NAIA_TOF_H
#define NAIA_TOF_H

// gbatch headers
#ifdef ENABLE_PRODUCTION_CODE
#include "root.h"
#endif

#include "Containers/OnDemandContainer.h"
#include "Containers/Utils.h"

#include "TVector3.h"

#include <array>
#include <map>
#include <vector>

namespace NAIA {

template <class T> using TofChargeVariable = std::map<Tof::ChargeType, T>;
template <class T> using TofBetaVariable = std::map<Tof::BetaType, T>;
template <class T> using TofClusterTypeVariable = std::map<Tof::BetaClusterType, T>;

class TofBaseData : public TObject {
public:
  TofBetaVariable<float> Beta;     
  TofChargeVariable<float> Charge; 

#ifdef ENABLE_PRODUCTION_CODE
  void Fill(BetaR *betaPtr, BetaHR *betahPtr, bool standalone = false);

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

  void Clear();

  void Dump() const;

  TVector3 InterpolateAtZ(float z) const;

  // ugly but better than saving 6 doubles for every event
  // NB: it's gonna throw if the Tof object is not valid
  // NB2: thanks ROOT for not supporting std::vector<std::array>.

  const TVector3 &GetLinePoint() const { return m_pointAndDir.at(0); }
  const TVector3 &GetLineDirection() const { return m_pointAndDir.at(1); }

private:
  // this is used to store point and direction of the TOF reconstruction
  std::vector<TVector3> m_pointAndDir;

  bool _isMC = false; 

  ClassDef(TofBaseData, 1)
};

class TofPlusData : public TObject {
public:
  // clang-format off
  float Chi2Coo;  
  float Chi2Time; 

  short NTrkClusters;    
  short NBetaClusters;   
  short NChargeClusters; 

  float ChargeLikelihood; 

  TofBetaVariable<short> BetaPattern; 
  std::vector<bool> LayerGoodPathl;  
  std::vector<float> LayerCharge;    
  LayerVariable<float> LayerEdep;    
  LayerVariable<TofClusterTypeVariable<unsigned int>> NClusters; 
  LayerVariable<TofClusterTypeVariable<float>> AverageDTime;     
  LayerVariable<TofClusterTypeVariable<float>> AverageEdep;      
  // clang-format on

#ifdef ENABLE_PRODUCTION_CODE
  void Fill(BetaR *betaPtr, BetaHR *betahPtr, bool standalone = false);
#endif

  void Clear();

  void Dump() const;

private:
  void ResizeMembers();

  ClassDef(TofPlusData, 1)
};

class TofBase : public OnDemandContainer<TofBase>, public TofBaseData {
public:
  static const std::string BranchName;

  TofBase() : TofBaseData(), OnDemandContainer() {}
};

class TofPlus : public OnDemandContainer<TofPlus>, public TofPlusData {
public:
  static const std::string BranchName;

  TofPlus() : TofPlusData(), OnDemandContainer() {}
};

class TofBaseStandalone : public OnDemandContainer<TofBaseStandalone>, public TofBaseData {
public:
  static const std::string BranchName;

  TofBaseStandalone() : TofBaseData(), OnDemandContainer() {}
};

class TofPlusStandalone : public OnDemandContainer<TofPlusStandalone>, public TofPlusData {
public:
  static const std::string BranchName;

  TofPlusStandalone() : TofPlusData(), OnDemandContainer() {}
};

} // namespace NAIA

#endif