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

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;

  TrTrack::Span GetBestSpan(TrTrack::Fit fit = TrTrack::Fit::Choutko) const;

  bool FitIDExists(TrTrack::Fit fit, TrTrack::Span span) const;

  std::bitset<9> GetTrackPattern(TrTrack::Side side);

private:
  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