TrdK.h

Go to the documentation of this file.

 
#ifndef NAIA_TRDK_H
#define NAIA_TRDK_H
 
// gbatch headers
#ifdef ENABLE_PRODUCTION_CODE
#include "TrdKCluster.h"
#include "root.h"
#endif
 
#include "Containers/OnDemandContainer.h"
#include "Containers/TrTrack.h"
#include "Containers/Utils.h"
 
#include "TVector3.h"
 
#include <array>
#include <map>
#include <vector>
 
namespace NAIA {
 
template <class T> using TrdChargeVariable = std::vector<T>;
template <class T> using TrdLikelihoodVariable = std::vector<T>;
template <class T> using TrdLikelihoodRVariable = std::vector<T>;
template <class T> using TrdOnTrackVariable = std::vector<T>;
 
namespace TrdK {
enum class BuildType { NONE = 0, TrkTrack, BetaH, TrdTrack };
}
 
class TrdKBaseData : public TObject {
public:
  TrdKBaseData() = default;
 
  explicit TrdKBaseData(bool standalone) : _isStandalone{standalone} {};
 
  TrdK::BuildType Type{TrdK::BuildType::NONE};
 
  // quality-related
  bool AlignStatus{false};  
  bool CalibStatus{false};  
  bool ChargeStatus{false}; 
 
  // clang-format off
  TrdOnTrackVariable<int> NHits; 
  TrdOnTrackVariable<float> Amps; 
 
  // charge-related
  TrdChargeVariable<float> Charge; 
 
  // likelihood-related
  TrdLikelihoodVariable<double> Likelihood; 
  TrdLikelihoodRVariable<double> LikelihoodRatio; 
  // clang-format off
 
  LayerVariable<float> Edep; 
  LayerVariable<float> Pathlength; 
 
  std::array<float, 2> DirectionalStoermerCutoff; 
 
  float Cleanliness{0}; 
  float IPChi2{std::numeric_limits<float>::max()}; 
 
#ifdef ENABLE_PRODUCTION_CODE
  bool Fill(TrdKCluster *trdkPtr, TrTrackR *pTrTrack);
 
  void SetBeta(double beta) { _beta = beta; }
  // void SetTrTrackPtr(TrTrackR *trackPtr) { pTrTrack = trackPtr; }
#endif
 
  void Clear();
 
  void Dump() const;
 
  [[nodiscard]] TVector3 InterpolateAtZ(float z) const;
 
  // ugly but better than saving 6 doubles for every event
  // NB: it's gonna throw if the TrdK object is not valid
  // NB2: thanks ROOT for not supporting std::vector<std::array>. Hopefully this is the only case where we have
  // to use TVector3
 
  [[nodiscard]] const TVector3 &GetLinePoint() const { return m_pointAndDir.at(0); }
  [[nodiscard]] const TVector3 &GetLineDirection() const { return m_pointAndDir.at(1); }
 
  const TrTrackBaseData *trkBase{nullptr}; 
 
protected:
  bool _isStandalone = false; 
 
private:
  void ResizeMembers();
 
  // this is used to store point and direction of the TRD reconstruction
  std::vector<TVector3> m_pointAndDir;
 
  float _beta = 1.0; 
 
  ClassDef(TrdKBaseData, 1)
};
 
class TrdKBase : public OnDemandContainer<TrdKBase>, public TrdKBaseData {
public:
  static const std::string BranchName;
 
  TrdKBase() : TrdKBaseData(), OnDemandContainer() {}
};
 
class TrdKBaseStandalone : public OnDemandContainer<TrdKBaseStandalone>, public TrdKBaseData {
public:
  static const std::string BranchName;
 
  TrdKBaseStandalone() : TrdKBaseData(true), OnDemandContainer() {}
};
 
} // namespace NAIA
 
#endif