PerigeeLinearizedTrackState.h

Go to the documentation of this file.

#ifndef PerigeeLinearizedTrackState_H
#define PerigeeLinearizedTrackState_H

#include "RecoVertex/VertexPrimitives/interface/LinearizedTrackState.h"
#include "TrackingTools/PatternTools/interface/TSCPBuilderNoMaterial.h"
#include "DataFormats/GeometrySurface/interface/ReferenceCounted.h"
#include "RecoVertex/VertexPrimitives/interface/LinearizedTrackState.h"
#include "Math/SMatrix.h"
#include "DataFormats/CLHEP/interface/Migration.h"

#include "FWCore/Utilities/interface/GCC11Compatibility.h"

class PerigeeLinearizedTrackState final : public LinearizedTrackState<5> {
public:
  friend class LinearizedTrackStateFactory;
  typedef ReferenceCountingPointer<LinearizedTrackState<5> > RefCountedLinearizedTrackState;

  RefCountedLinearizedTrackState stateWithNewLinearizationPoint(const GlobalPoint& newLP) const override;

  const GlobalPoint& linearizationPoint() const override { return theLinPoint; }

  reco::TransientTrack track() const override { return theTrack; }

  const TrajectoryStateOnSurface state() const { return theTSOS; }

  const AlgebraicVector5& constantTerm() const override;

  const AlgebraicMatrix53& positionJacobian() const override;

  const AlgebraicMatrix53& momentumJacobian() const override;

  const AlgebraicVector5& parametersFromExpansion() const override;

  const TrajectoryStateClosestToPoint& predictedState() const;

  AlgebraicVector5 predictedStateParameters() const override;

  AlgebraicVector3 predictedStateMomentumParameters() const override;

  AlgebraicSymMatrix55 predictedStateWeight(int& error) const override;

  AlgebraicSymMatrix55 predictedStateError() const override;

  AlgebraicSymMatrix33 predictedStateMomentumError() const override;

  //   /** Method returning the impact point measurement
  //    */
  //   ImpactPointMeasurement impactPointMeasurement() const;

  TrackCharge charge() const override { return theCharge; }

  bool hasError() const override;

  bool operator==(LinearizedTrackState<5>& other) const override;

  bool operator==(ReferenceCountingPointer<LinearizedTrackState<5> >& other) const;

  RefCountedRefittedTrackState createRefittedTrackState(const GlobalPoint& vertexPosition,
                                                        const AlgebraicVector3& vectorParameters,
                                                        const AlgebraicSymMatrix66& covarianceMatrix) const override;

  AlgebraicVector5 refittedParamFromEquation(const RefCountedRefittedTrackState& theRefittedState) const override;

  double weightInMixture() const override { return theTSOS.weight(); }

  void inline checkParameters(AlgebraicVector5& parameters) const override;

  std::vector<ReferenceCountingPointer<LinearizedTrackState<5> > > components() const override;

  bool isValid() const override;

private:
  PerigeeLinearizedTrackState(const GlobalPoint& linP,
                              const reco::TransientTrack& track,
                              const TrajectoryStateOnSurface& tsos)
      : theLinPoint(linP), theTrack(track), theTSOS(tsos), theCharge(theTrack.charge()), jacobiansAvailable(false) {}

  void computeJacobians() const;
  void computeChargedJacobians() const;
  void computeNeutralJacobians() const;
  void compute3DImpactPoint() const;

  GlobalPoint theLinPoint;
  reco::TransientTrack theTrack;
  const TrajectoryStateOnSurface theTSOS;

  mutable AlgebraicMatrix53 thePositionJacobian, theMomentumJacobian;
  mutable TrajectoryStateClosestToPoint thePredState;
  mutable AlgebraicVector5 theConstantTerm;
  mutable AlgebraicVector5 theExpandedParams;

  TSCPBuilderNoMaterial builder;

  TrackCharge theCharge;
  mutable bool jacobiansAvailable;
};

inline const AlgebraicVector5& PerigeeLinearizedTrackState::constantTerm() const {
  if
    UNLIKELY(!jacobiansAvailable) computeJacobians();
  return theConstantTerm;
}

inline const AlgebraicMatrix53& PerigeeLinearizedTrackState::positionJacobian() const {
  if
    UNLIKELY(!jacobiansAvailable) computeJacobians();
  return thePositionJacobian;
}

inline const AlgebraicMatrix53& PerigeeLinearizedTrackState::momentumJacobian() const {
  if
    UNLIKELY(!jacobiansAvailable) computeJacobians();
  return theMomentumJacobian;
}

inline const AlgebraicVector5& PerigeeLinearizedTrackState::parametersFromExpansion() const {
  if
    UNLIKELY(!jacobiansAvailable) computeJacobians();
  return theExpandedParams;
}

inline const TrajectoryStateClosestToPoint& PerigeeLinearizedTrackState::predictedState() const {
  if
    UNLIKELY(!jacobiansAvailable) computeJacobians();
  return thePredState;
}

inline bool PerigeeLinearizedTrackState::hasError() const {
  if
    UNLIKELY(!jacobiansAvailable) computeJacobians();
  return thePredState.hasError();
}

inline AlgebraicVector5 PerigeeLinearizedTrackState::predictedStateParameters() const {
  if (!jacobiansAvailable)
    computeJacobians();
  return thePredState.perigeeParameters().vector();
}

inline AlgebraicVector3 PerigeeLinearizedTrackState::predictedStateMomentumParameters() const {
  if
    UNLIKELY(!jacobiansAvailable) computeJacobians();
  auto v = thePredState.perigeeParameters().vector();
  return AlgebraicVector3(v[0], v[1], v[2]);
}

inline AlgebraicSymMatrix55 PerigeeLinearizedTrackState::predictedStateWeight(int& error) const {
  if
    UNLIKELY(!jacobiansAvailable) computeJacobians();
  if (!thePredState.isValid()) {
    error = 1;
    return AlgebraicSymMatrix55();
  }
  return thePredState.perigeeError().weightMatrix(error);
}

inline AlgebraicSymMatrix55 PerigeeLinearizedTrackState::predictedStateError() const {
  if
    UNLIKELY(!jacobiansAvailable) computeJacobians();
  return thePredState.perigeeError().covarianceMatrix();
}

inline AlgebraicSymMatrix33 PerigeeLinearizedTrackState::predictedStateMomentumError() const {
  if
    UNLIKELY(!jacobiansAvailable) computeJacobians();
  return thePredState.perigeeError().covarianceMatrix().Sub<AlgebraicSymMatrix33>(0, 2);
}

inline bool PerigeeLinearizedTrackState::isValid() const {
  if
    UNLIKELY(!theTSOS.isValid()) return false;
  if
    UNLIKELY(!jacobiansAvailable) computeJacobians();
  return jacobiansAvailable;
}

#endif