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>;

class TrTrackBaseData : public TObject {
  friend class TrdKBaseData;
  friend class TrTrackPlusData;

public:
  // clang-format off
  // fit-related
  TrackFitVariable<float> RigidityCorr; 
  TrackFitVariable<TrackSideVariable<float>> TrChiSq; 

  // charge-related
  TrackChargeVariable<float> Charge; 
  TrackChargeVariable<float> InnerCharge; 
  LayerVariable<TrackChargeVariable<float>> LayerChargeXY; 
  LayerVariable<int> LayerChargeStatus; 
  // clang-format on

#ifdef ENABLE_PRODUCTION_CODE
  void Fill(TrTrackR *trackPtr);
  void FillElectronVars(TrTrackR *trackPtr, 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;

  float GetGeomCorrectedRigidityCorr(TrTrack::Fit fit, TrTrack::Span span,
                                     TrTrack::TrGeometry trGeom = TrTrack::TrGeometry::Default) const;

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

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

private:
  bool _isMC = false;            
  float _beta = 1.0f;            
  TrackFitVariable<int> _fit_ID; 

  ClassDef(TrTrackBaseData, 1)
};

class TrTrackPlusData : public TObject {
public:
  // hit-related
  LayerVariable<TrackSideVariable<float>> TrTrackHitPos; 
  LayerVariable<float> TrackFeetDistance; 

  // fit-related
  float Theta = 0; 
  float Phi = 0;   
  // clang-format off
  TrackFitVariable<float> Rigidity; 
  TrackFitVariable<float> RigidityTOI; 
  TrackFitVariable<float> InvRigErr; 
  LayerVariable<TrackFitVariable<TrackSideVariable<float>>> TrTrackResidual; 
  LayerVariable<TrackFitOnlyVariable<float>> PartialRigidity; 
  LayerVariable<TrackFitOnlyVariable<float>> PartialInvRigErr; 
  LayerVariable<TrackFitOnlyVariable<TrackSideVariable<float>>> PartialTrChiSq; 
  TrackFitPosVariable<TrackFitVariable<TrackSideVariable<float>>> TrTrackFitPos; 

  // charge-related
  TrackChargeVariable<float> UnbiasedCharge; 
  TrackChargeVariable<float> ChargeRMS; 
  TrackChargeVariable<float> InnerChargeRMS; 
  LayerVariable<TrackChargeVariable<TrackSideVariable<float>>> LayerCharge; 
  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; 
  // clang-format on

#ifdef ENABLE_PRODUCTION_CODE
  void Fill(TrTrackR *trackPtr);
  void FillElectronVars(TrTrackR *trackPtr, float zEcalCOG, bool refitKalman = false);
#endif

  void Clear();

  void Dump() const;

  float GetGeomCorrectedRigidity(TrTrack::Fit fit, TrTrack::Span span,
                                 TrTrack::TrGeometry trGeom = TrTrack::TrGeometry::Default) const;

  float GetGeomCorrectedRigidityTOI(TrTrack::Fit fit, TrTrack::Span span,
                                    TrTrack::TrGeometry trGeom = TrTrack::TrGeometry::Default) const;

  const TrTrackBaseData *trkBase = nullptr; 

private:
  TrackFitVariable<int> _fit_ID; 

  ClassDef(TrTrackPlusData, 1)
};

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() {}
};

} // namespace NAIA
#endif