TwoBodyDecayTrajectoryState.h

Go to the documentation of this file.

#ifndef Alignment_ReferenceTrajectories_TwoBodyDecayTrajectoryState_h
#define Alignment_ReferenceTrajectories_TwoBodyDecayTrajectoryState_h

#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "Alignment/TwoBodyDecay/interface/TwoBodyDecay.h"

class GlobalTrajectoryParameters;


class TwoBodyDecayTrajectoryState
{

public:

  typedef std::pair< TrajectoryStateOnSurface, TrajectoryStateOnSurface > TsosContainer;
  typedef std::pair< AlgebraicMatrix, AlgebraicMatrix > Derivatives;

  TwoBodyDecayTrajectoryState( const TsosContainer & tsos,
                   const TwoBodyDecay & tbd,
                   double particleMass,
                   const MagneticField* magField,
                   bool propagateErrors = false );

  ~TwoBodyDecayTrajectoryState( void ) {}

  inline bool isValid( void ) const { return theValidityFlag; }

  inline double particleMass( void ) const { return theParticleMass; }
  inline const TwoBodyDecayParameters & decayParameters( void ) const { return theParameters; }
  inline const TsosContainer& trajectoryStates( bool useRefittedState = true ) const { return useRefittedState ? theRefittedTsos : theOriginalTsos; }
  inline const Derivatives& derivatives( void ) const { return theDerivatives; }

  void rescaleError( double scale );

  inline double primaryMass( void ) const { return thePrimaryMass; }
  inline double primaryWidth( void ) const { return thePrimaryWidth; }

private:

  void construct( const MagneticField* magField,
          bool propagateErrors );

  bool propagateSingleState( const FreeTrajectoryState & fts,
                 const GlobalTrajectoryParameters & gtp,
                 const AlgebraicMatrix & startDeriv,
                 const Surface & surface,
                 const MagneticField* magField,
                 TrajectoryStateOnSurface & tsos,
                 AlgebraicMatrix & endDeriv ) const;


  void setError( FreeTrajectoryState& fts,
         AlgebraicMatrix& derivative ) const;

  bool theValidityFlag;

  double theParticleMass;

  TwoBodyDecayParameters theParameters;
  Derivatives theDerivatives;
  TsosContainer theOriginalTsos;
  TsosContainer theRefittedTsos;

  double thePrimaryMass;
  double thePrimaryWidth;

  static const unsigned int nLocalParam = 5;
  static const unsigned int nDecayParam = TwoBodyDecayParameters::dimension;
};


#endif