TrTrack.h

Go to the documentation of this file.

 
#ifndef NAIA_TRTRACK_H
#define NAIA_TRTRACK_H
 
// gbatch headers
#ifdef ENABLE_PRODUCTION_CODE
#include "root.h"
#endif
 
#include "Containers/OnDemandContainer.h"
#include "Containers/Utils.h"
 
#include <array>
#include <map>
#include <vector>
 
namespace NAIA {
 
template <class T> using TrackChargeVariable = std::map<TrTrack::ChargeRecoType, T>;
template <class T> using TrackFitVariable = std::map<TrTrack::Fit, std::map<TrTrack::Span, T>>;
template <class T> using TrackFitOnlyVariable = std::map<TrTrack::Fit, T>;
template <class T> using TrackSideVariable = std::map<TrTrack::Side, T>;
template <class T> using TrackFitPosVariable = std::map<TrTrack::FitPositionHeight, T>;
template <class T> using TrackDistanceVariable = std::map<TrTrack::DistanceFromTrack, T>;
 
#ifdef ENABLE_PRODUCTION_CODE
enum class TrTrackFillType { Full, SecondTrack, Standalone };
#endif
 
enum class HitClusterAssociation { None = 0, Y, XY };
 
class TrTrackBaseData : public TObject {
  friend class TrdKBaseData;
  friend class TrTrackPlusData;
 
public:
  // clang-format off
  // hit-related
  LayerVariable<std::array<float, 3>> TrTrackHitPos; 
 
  // fit-related
  TrackFitVariable<float> Rigidity; 
  TrackFitVariable<TrackSideVariable<float>> TrChiSq; 
  TrackFitPosVariable<TrackFitVariable<TrackSideVariable<float>>> TrTrackFitPos; 
 
  // charge-related
  TrackChargeVariable<float> Charge; 
  TrackChargeVariable<float> InnerCharge; 
  TrackChargeVariable<float> ChargeRMS; 
  TrackChargeVariable<float> InnerChargeRMS; 
  LayerVariable<int> LayerChargeStatus; 
  LayerVariable<TrackChargeVariable<float>> LayerChargeXY; 
 
  float FitCharge; 
  // clang-format on
 
#ifdef ENABLE_PRODUCTION_CODE
  bool Fill(TrTrackR *trackPtr, TrTrackFillType type = TrTrackFillType::Full);
  bool FillElectronVars(TrTrackR *trackPtr, float zEcalCOG, bool refitKalman = false);
 
  void SetBeta(double beta) { _beta = beta; }
  void SetMC(bool isMC) { _isMC = isMC; }
 
  TrTrack::Span _GetBestSpan(TrTrack::Fit fit = TrTrack::Fit::Choutko) const;
  bool _FitIDExists(TrTrack::Fit fit, TrTrack::Span span) const;
#endif
 
  void Clear();
 
  void Dump() const;
 
  [[nodiscard]] TrTrack::Span GetBestSpan(TrTrack::Fit fit = TrTrack::Fit::Choutko) const;
 
  [[nodiscard]] bool FitIDExists(TrTrack::Fit fit, TrTrack::Span span) const;
 
  [[nodiscard]] LayerVariable<HitClusterAssociation> GetTrackPattern() const;
 
private:
  uint32_t m_trackPattern{0};
 
  bool _isMC = false;            
  float _beta = 1.0f;            
  TrackFitVariable<int> _fit_ID; 
  int _tmpCharge = 1;            
 
  ClassDef(TrTrackBaseData, 3)
};
 
class TrTrackPlusData : public TObject {
public:
  // clang-format off
  // hit-related
  LayerVariable<float> TrackFeetDistance; 
 
  // fit-related
  TrackFitVariable<float> Theta; 
  TrackFitVariable<float> Phi; 
 
  TrackFitVariable<float> RigidityTOI; 
  TrackFitVariable<float> RigidityRICH; 
  TrackFitVariable<float> InvRigErr; 
  LayerVariable<TrackFitVariable<TrackSideVariable<float>>> TrTrackResidual; 
  LayerVariable<TrackFitOnlyVariable<TrackSideVariable<float>>> PartialTrTrackResidual; 
  LayerVariable<TrackFitOnlyVariable<float>> PartialRigidity; 
  LayerVariable<TrackFitOnlyVariable<float>> PartialInvRigErr; 
  LayerVariable<TrackFitOnlyVariable<TrackSideVariable<float>>> PartialTrChiSq; 
  LayerVariable<TrackFitOnlyVariable<TrackSideVariable<float>>> PartialTrTrackFitPos;
 
  // charge-related
  TrackChargeVariable<float> UnbiasedCharge; 
  LayerVariable<TrackSideVariable<float>> LayerEdep; 
  LayerVariable<TrackDistanceVariable<TrackSideVariable<unsigned int>>> NClusters; 
  LayerVariable<TrackDistanceVariable<TrackSideVariable<float>>> ClustersEdep; 
  LayerVariable<TrackSideVariable<float>> MaxClusterEdep; 
  LayerVariable<TrackSideVariable<float>> MaxClusterDistance; 
  LayerVariable<float> ClusterSignalRatio; 
  LayerVariable<TrackChargeVariable<TrackSideVariable<float>>> LayerCharge; 
 
  std::array<float, 2> DirectionalStoermerCutoff{0.0f, 0.0f}; 
  // clang-format on
 
#ifdef ENABLE_PRODUCTION_CODE
  bool Fill(TrTrackR *trackPtr);
  bool FillElectronVars(TrTrackR *trackPtr, bool refitKalman = false);
#endif
 
  void Clear();
 
  void Dump() const;
 
  const TrTrackBaseData *trkBase = nullptr; 
 
private:
  TrackFitVariable<int> _fit_ID; 
 
  ClassDef(TrTrackPlusData, 2)
};
 
class TrTrackBase : public OnDemandContainer<TrTrackBase>, public TrTrackBaseData {
public:
  static const std::string BranchName;
 
  TrTrackBase() : TrTrackBaseData(), OnDemandContainer() {}
};
 
class TrTrackPlus : public OnDemandContainer<TrTrackPlus>, public TrTrackPlusData {
public:
  static const std::string BranchName;
 
  TrTrackPlus() : TrTrackPlusData(), OnDemandContainer() {}
};
 
class SecondTrTrackBase : public OnDemandContainer<SecondTrTrackBase>, public TrTrackBaseData {
public:
  static const std::string BranchName;
 
  SecondTrTrackBase() : TrTrackBaseData(), OnDemandContainer() {}
};
 
class TrTrackBaseStandalone : public OnDemandContainer<TrTrackBaseStandalone>, public TrTrackBaseData {
public:
  static const std::string BranchName;
 
  TrTrackBaseStandalone() : TrTrackBaseData(), OnDemandContainer() {}
};
 
} // namespace NAIA
#endif