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"

#include "GblTrajectory.h"

class TrajectoryStateOnSurface;
class MagneticField;
class MaterialEffectsUpdator;
class Plane;
class BeamSpotTransientTrackingRecHit;

namespace reco { class BeamSpot; }

class ReferenceTrajectory : public ReferenceTrajectoryBase
{

public:

  typedef SurfaceSideDefinition::SurfaceSide SurfaceSide;

  ReferenceTrajectory(const TrajectoryStateOnSurface& referenceTsos,
                      const TransientTrackingRecHit::ConstRecHitContainer& recHits,
                      const MagneticField* magField,
                      const reco::BeamSpot& beamSpot,
                      const ReferenceTrajectoryBase::Config& config);

  ~ReferenceTrajectory() override {}

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

protected:

  // ReferenceTrajectory(unsigned int nPar, unsigned int nHits, MaterialEffects materialEffects);
  ReferenceTrajectory(unsigned int nPar, unsigned int nHits,
              const ReferenceTrajectoryBase::Config& config);

  virtual bool construct(const TrajectoryStateOnSurface &referenceTsos, 
             const TransientTrackingRecHit::ConstRecHitContainer &recHits,
             const MagneticField *magField,
             const reco::BeamSpot &beamSpot);

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

  virtual bool propagate(const Plane &previousSurface, const TrajectoryStateOnSurface &previousTsos,
             const Plane &newSurface, TrajectoryStateOnSurface &newTsos, AlgebraicMatrix &newJacobian, 
             AlgebraicMatrix &newCurvlinJacobian, double &nextStep,
             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);
  
  virtual bool addMaterialEffectsLocalGbl(const std::vector<AlgebraicMatrix> &allJacobians,
                                     const std::vector<AlgebraicMatrix> &allProjections,
                                     const std::vector<AlgebraicSymMatrix> &allCurvatureChanges,
                                     const std::vector<AlgebraicSymMatrix> &allDeltaParameterCovs);

  virtual bool addMaterialEffectsCurvlinGbl(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);
          
  inline SurfaceSide surfaceSide(const PropagationDirection dir) const
  {
    return ( dir == alongMomentum ) ?
      SurfaceSideDefinition::beforeSurface :
      SurfaceSideDefinition::afterSurface;
  }

  AlgebraicMatrix
    getHitProjectionMatrix(const TransientTrackingRecHit::ConstRecHitPointer &recHit) const;

  template <unsigned int N>
    AlgebraicMatrix
    getHitProjectionMatrixT(const TransientTrackingRecHit::ConstRecHitPointer &recHit) const;
private:
  template <typename Derived>
  void clhep2eigen(const AlgebraicVector& in, Eigen::MatrixBase<Derived>& out);
  template <typename Derived>
  void clhep2eigen(const AlgebraicMatrix& in, Eigen::MatrixBase<Derived>& out);
  template <typename Derived>
  void clhep2eigen(const AlgebraicSymMatrix& in, Eigen::MatrixBase<Derived>& out);

  const double mass_;
  const MaterialEffects materialEffects_;
  const PropagationDirection propDir_;
  const bool useBeamSpot_;
  const bool includeAPEs_;
  const bool allowZeroMaterial_;
};

#endif