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