PropagatorWithMaterial.h

Go to the documentation of this file.

#ifndef _COMMONRECO_PROPAGATORWITHMATERIAL_H_
#define _COMMONRECO_PROPAGATORWITHMATERIAL_H_
 
#include "DataFormats/GeometryCommonDetAlgo/interface/DeepCopyPointerByClone.h"
#include "TrackPropagation/RungeKutta/interface/defaultRKPropagator.h"
 
#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/MaterialEffects/interface/MaterialEffectsUpdator.h"
 
class MagneticField;
class PropagatorWithMaterial final : public Propagator {
public:
  PropagatorWithMaterial(PropagationDirection dir,
                         const float mass,
                         const MagneticField* mf = nullptr,
                         const float maxDPhi = 1.6,
                         bool useRungeKutta = false,
                         float ptMin = -1.,
                         bool useOldGeoPropLogic = true);
 
  ~PropagatorWithMaterial() override;
 
  using Propagator::propagate;
  using Propagator::propagateWithPath;
 
private:
  std::pair<TrajectoryStateOnSurface, double> propagateWithPath(const TrajectoryStateOnSurface& tsos,
                                                                const Plane& plane) const override;
 
  std::pair<TrajectoryStateOnSurface, double> propagateWithPath(const FreeTrajectoryState& fts,
                                                                const Plane& plane) const override;
 
  std::pair<TrajectoryStateOnSurface, double> propagateWithPath(const TrajectoryStateOnSurface& tsos,
                                                                const Cylinder& cylinder) const override;
 
  std::pair<TrajectoryStateOnSurface, double> propagateWithPath(const FreeTrajectoryState& fts,
                                                                const Cylinder& cylinder) const override;
 
public:
  bool setMaxDirectionChange(float phiMax) override { return theGeometricalPropagator->setMaxDirectionChange(phiMax); }
  void setPropagationDirection(PropagationDirection dir) override;
 
  enum MaterialLocation { atSource, atDestination, fromDirection };
  void setMaterialLocation(const MaterialLocation location) { theMaterialLocation = location; }
  const Propagator& geometricalPropagator() const { return *theGeometricalPropagator; }
  const MaterialEffectsUpdator& materialEffectsUpdator() const { return *theMEUpdator; }
 
  const MagneticField* magneticField() const override { return field; }
 
  PropagatorWithMaterial* clone() const override { return new PropagatorWithMaterial(*this); }
 
private:
  bool materialAtSource() const dso_internal;
 
private:
  // Geometrical propagator
 
  defaultRKPropagator::Product rkProduct;
  DeepCopyPointerByClone<Propagator> theGeometricalPropagator;
 
  // Material effects
  DeepCopyPointerByClone<MaterialEffectsUpdator> theMEUpdator;
  typedef std::pair<TrajectoryStateOnSurface, double> TsosWP;
  // Use material at source?
  MaterialLocation theMaterialLocation;
  const MagneticField* field;
  bool useRungeKutta_;
};
 
#endif