/data/refman/pasoursint/CMSSW_4_4_5_patch3/src/Alignment/ReferenceTrajectories/interface/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( const TsosContainer & tsos,
                               const TwoBodyDecayParameters & param,
                               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 const double primaryMass( void ) const { return thePrimaryMass; }
  inline const 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