DualKalmanTrajectory.h

Go to the documentation of this file.

#ifndef Alignment_ReferenceTrajectories_DualKalmanTrajectory_H
#define Alignment_ReferenceTrajectories_DualKalmanTrajectory_H



#include "Alignment/ReferenceTrajectories/interface/ReferenceTrajectoryBase.h"
#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"
#include "DataFormats/GeometrySurface/interface/LocalError.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"

#include <vector>

class ReferenceTrajectory;
namespace reco { class BeamSpot;}

class DualKalmanTrajectory : public ReferenceTrajectoryBase
{

public:

  DualKalmanTrajectory(const Trajectory::DataContainer &trajMeasurements,
               const TrajectoryStateOnSurface &referenceTsos,
               const std::vector<unsigned int> &forwardRecHitNums,
               const std::vector<unsigned int> &backwardRecHitNums,
               const MagneticField *magField,
               MaterialEffects materialEffects,
               PropagationDirection propDir,
               double mass,
               bool useBeamSpot,
               const reco::BeamSpot &beamSpot,
               int residualMethod);

  virtual ~DualKalmanTrajectory() {}

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

protected:

  DualKalmanTrajectory( unsigned int nPar = 0, unsigned int nHits = 0 );

  virtual bool construct(const Trajectory::DataContainer &trajMeasurements,
             const TrajectoryStateOnSurface &referenceTsos,
             const std::vector<unsigned int> &forwardRecHitNums,
             const std::vector<unsigned int> &backwardRecHitNums,
             double mass, MaterialEffects materialEffects,
             const PropagationDirection propDir, const MagneticField *magField,
             bool useBeamSpot, const reco::BeamSpot &beamSpot,
             int residualMethod);

  virtual ReferenceTrajectory* construct(const Trajectory::DataContainer &trajMeasurements,
                     const TrajectoryStateOnSurface &referenceTsos, 
                     const std::vector<unsigned int> &recHits,
                     double mass, MaterialEffects materialEffects,
                     const PropagationDirection propDir,
                     const MagneticField *magField,
                     bool useBeamSpot,
                     const reco::BeamSpot &beamSpot) const;
  bool fillKalmanPart(const Trajectory::DataContainer &trajMeasurements,
              const std::vector<unsigned int> &recHitNums, bool startFirst,
              unsigned int iNextHit, int residualMethod);
  TrajectoryStateOnSurface 
  fillMeasurementAndError1(const TransientTrackingRecHit::ConstRecHitPointer &hitPtr,
               unsigned int iHit, const TrajectoryMeasurement &trajMeasurement);
  TrajectoryStateOnSurface 
  fillMeasurementAndError2(const TransientTrackingRecHit::ConstRecHitPointer &hitPtr,
               unsigned int iHit, const TrajectoryMeasurement &trajMeasurement,
               bool doPull);

  void fillTrajectoryPositions(const AlgebraicMatrix &projection,
                   const TrajectoryStateOnSurface &tsos, unsigned int iHit);

  LocalError hitErrorWithAPE(const TransientTrackingRecHit::ConstRecHitPointer &hitPtr) const;

  virtual AlgebraicVector extractParameters(const TrajectoryStateOnSurface &referenceTsos) const;

  inline const PropagationDirection oppositeDirection( const PropagationDirection propDir ) const
  { return ((propDir == anyDirection) ? anyDirection 
        : ((propDir == alongMomentum) ? oppositeToMomentum : alongMomentum));
  }

};

#endif