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=0,const float maxDPhi=1.6,
              bool useRungeKutta=false, float ptMin=-1.,bool useOldGeoPropLogic=true);

  virtual ~PropagatorWithMaterial();



  using Propagator::propagate;
  using Propagator::propagateWithPath;


private:
  virtual std::pair<TrajectoryStateOnSurface,double> propagateWithPath (const TrajectoryStateOnSurface& tsos,
                                    const Plane& plane) const override;

  virtual std::pair<TrajectoryStateOnSurface,double> propagateWithPath (const FreeTrajectoryState& fts,
                                    const Plane& plane) const override;

  virtual std::pair<TrajectoryStateOnSurface,double> propagateWithPath (const TrajectoryStateOnSurface& tsos,
                                    const Cylinder& cylinder) const override;

  virtual std::pair<TrajectoryStateOnSurface,double> propagateWithPath (const FreeTrajectoryState& fts,
                                    const Cylinder& cylinder) const override;

public:
  virtual bool setMaxDirectionChange( float phiMax) {
    return theGeometricalPropagator->setMaxDirectionChange(phiMax);
  }
  virtual 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;
  }

  virtual const MagneticField* magneticField() const {return field;}


  virtual PropagatorWithMaterial* clone() const
    {
      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