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