TwoBodyDecayEstimator.h

Go to the documentation of this file.

#ifndef Alignment_TwoBodyDecay_TwoBodyDecayEstimator_h
#define Alignment_TwoBodyDecay_TwoBodyDecayEstimator_h

#include "Alignment/TwoBodyDecay/interface/TwoBodyDecay.h"
#include "Alignment/TwoBodyDecay/interface/TwoBodyDecayVirtualMeasurement.h"

#include "RecoVertex/VertexTools/interface/PerigeeLinearizedTrackState.h" 
#include "DataFormats/CLHEP/interface/AlgebraicObjects.h" 

#include "FWCore/ParameterSet/interface/ParameterSet.h"

class TwoBodyDecayEstimator
{

public:

  typedef PerigeeLinearizedTrackState::RefCountedLinearizedTrackState RefCountedLinearizedTrackState;

  TwoBodyDecayEstimator( const edm::ParameterSet & config );
  virtual ~TwoBodyDecayEstimator( void ) {}

  virtual TwoBodyDecay estimate( const std::vector< RefCountedLinearizedTrackState > & linTracks,
                 const TwoBodyDecayParameters & linearizationPoint,
                 const TwoBodyDecayVirtualMeasurement & vm ) const;

  inline int ndf( void ) const { return theNdf; }
  inline const AlgebraicVector& pulls( void ) const { return thePulls; }

  virtual TwoBodyDecayEstimator* clone( void ) const { return new TwoBodyDecayEstimator( *this ); }

protected:

  virtual bool constructMatrices( const std::vector< RefCountedLinearizedTrackState > & linTracks,
                  const TwoBodyDecayParameters & linearizationPoint,
                  const TwoBodyDecayVirtualMeasurement & vm,
                  AlgebraicVector & vecM, AlgebraicSymMatrix & matG, AlgebraicMatrix & matA ) const;

private:

  bool checkValues( const AlgebraicVector & vec ) const;

  double theRobustificationConstant;
  double theMaxIterDiff;
  int theMaxIterations;
  bool theUseInvariantMass;

  mutable int theNdf;
  mutable AlgebraicVector thePulls;

};


#endif