/data/refman/pasoursint/CMSSW_4_4_5_patch3/src/Alignment/ReferenceTrajectories/interface/ReferenceTrajectory.h

Go to the documentation of this file.

#ifndef Alignment_ReferenceTrajectories_ReferenceTrajectory_H
#define Alignment_ReferenceTrajectories_ReferenceTrajectory_H

#include "Alignment/ReferenceTrajectories/interface/ReferenceTrajectoryBase.h"
#include "TrackingTools/TransientTrackingRecHit/interface/TransientTrackingRecHit.h"
#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"
#include "TrackingTools/TrajectoryState/interface/SurfaceSideDefinition.h"

class TrajectoryStateOnSurface;
class MagneticField;
class MaterialEffectsUpdator;
class BoundPlane;
class BeamSpotTransientTrackingRecHit;

namespace reco { class BeamSpot; }

class ReferenceTrajectory : public ReferenceTrajectoryBase
{

public:

  typedef SurfaceSideDefinition::SurfaceSide SurfaceSide;

  ReferenceTrajectory(const TrajectoryStateOnSurface &referenceTsos,
                      const TransientTrackingRecHit::ConstRecHitContainer &recHits,
                      bool hitsAreReverse,
                      const MagneticField *magField,
                      MaterialEffects materialEffects,
                      PropagationDirection propDir,
                      double mass,
                      bool useBeamSpot,
                      const reco::BeamSpot &beamSpot);
  virtual ~ReferenceTrajectory() {}

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

protected:

  ReferenceTrajectory(unsigned int nPar, unsigned int nHits, MaterialEffects materialEffects);

  virtual bool construct(const TrajectoryStateOnSurface &referenceTsos, 
                         const TransientTrackingRecHit::ConstRecHitContainer &recHits,
                         double mass, MaterialEffects materialEffects,
                         const PropagationDirection propDir, const MagneticField *magField,
                         bool useBeamSpot,
                         const reco::BeamSpot &beamSpot);

  MaterialEffectsUpdator* createUpdator(MaterialEffects materialEffects, double mass) const;

  virtual bool propagate(const BoundPlane &previousSurface, const TrajectoryStateOnSurface &previousTsos,
                         const BoundPlane &newSurface, TrajectoryStateOnSurface &newTsos, AlgebraicMatrix &newJacobian, 
                         AlgebraicMatrix &newCurvlinJacobian, double &nextStep,
                         const PropagationDirection propDir, const MagneticField *magField) const;
  
  virtual void fillMeasurementAndError(const TransientTrackingRecHit::ConstRecHitPointer &hitPtr,
                                       unsigned int iRow,
                                       const TrajectoryStateOnSurface &updatedTsos);

  virtual void fillDerivatives(const AlgebraicMatrix &projection,
                               const AlgebraicMatrix &fullJacobian, unsigned int iRow);

  virtual void fillTrajectoryPositions(const AlgebraicMatrix &projection, 
                                       const AlgebraicVector &mixedLocalParams, 
                                       unsigned int iRow);

  virtual bool addMaterialEffectsCov(const std::vector<AlgebraicMatrix> &allJacobians, 
                                     const std::vector<AlgebraicMatrix> &allProjections,
                                     const std::vector<AlgebraicSymMatrix> &allCurvChanges,
                                     const std::vector<AlgebraicSymMatrix> &allDeltaParaCovs);
                                     
  virtual bool addMaterialEffectsBp (const std::vector<AlgebraicMatrix> &allJacobians, 
                                     const std::vector<AlgebraicMatrix> &allProjections,
                                     const std::vector<AlgebraicSymMatrix> &allCurvChanges,
                                     const std::vector<AlgebraicSymMatrix> &allDeltaParaCovs,
                                     const std::vector<AlgebraicMatrix> &allLocalToCurv);
                                     
  virtual bool addMaterialEffectsBrl(const std::vector<AlgebraicMatrix> &allJacobians, 
                                     const std::vector<AlgebraicMatrix> &allProjections,
                                     const std::vector<AlgebraicSymMatrix> &allCurvChanges,
                                     const std::vector<AlgebraicSymMatrix> &allDeltaParaCovs,
                                     const std::vector<AlgebraicMatrix> &allLocalToCurv,
                                     const GlobalTrajectoryParameters &gtp);
  virtual bool addMaterialEffectsBrl(const std::vector<AlgebraicMatrix> &allProjections,
                                     const std::vector<AlgebraicSymMatrix> &allDeltaParaCovs,
                                     const std::vector<AlgebraicMatrix> &allLocalToCurv,
                                     const std::vector<double> &allSteps,
                                     const GlobalTrajectoryParameters &gtp,   
                                     const double minStep = 1.0);
          
  // Don't care for propagation direction 'anyDirection' - in that case the material effects
  // are anyway not updated ...
  inline const SurfaceSide surfaceSide(const PropagationDirection dir) const
  {
    return ( dir == alongMomentum ) ?
      SurfaceSideDefinition::beforeSurface :
      SurfaceSideDefinition::afterSurface;
  }
};

#endif