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;

  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

  const TVector3 &GetLinePoint() const { return m_pointAndDir.at(0); }
  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