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;

  virtual  RefCountedLinearizedTrackState stateWithNewLinearizationPoint
    (const GlobalPoint & newLP) const;


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

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

  const TrajectoryStateOnSurface state() const { return theTSOS; }

  const AlgebraicVector5 & constantTerm() const;

  const AlgebraicMatrix53 & positionJacobian() const;

  const AlgebraicMatrix53 & momentumJacobian() const;

  const AlgebraicVector5 & parametersFromExpansion() const;

  const TrajectoryStateClosestToPoint & predictedState() const;

  AlgebraicVector5 predictedStateParameters() const;

  virtual AlgebraicVector3 predictedStateMomentumParameters() const;

  AlgebraicSymMatrix55 predictedStateWeight(int & error) const;

  AlgebraicSymMatrix55 predictedStateError() const;

  AlgebraicSymMatrix33 predictedStateMomentumError() const;

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

  TrackCharge charge() const {return theCharge;}

  bool hasError() const;

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

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

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


  virtual AlgebraicVector5 refittedParamFromEquation(
    const RefCountedRefittedTrackState & theRefittedState) const;

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

  virtual void inline checkParameters(AlgebraicVector5 & parameters) const;

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

  virtual bool isValid() const;

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