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"
#include "MagneticField/Engine/interface/MagneticField.h"

class PropagatorWithMaterial GCC11_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();

  virtual TrajectoryStateOnSurface propagate (const TrajectoryStateOnSurface& tsos, 
                          const Plane& plane) const
  {
    // should be implemented (in case underlying propagator has an independent
    // implementation)
    return propagateWithPath(tsos,plane).first;
  }

  virtual TrajectoryStateOnSurface propagate (const FreeTrajectoryState& fts, 
                          const Plane& plane) const
  {
    // should be implemented (in case underlying propagator has an independent
    // implementation)
    return propagateWithPath(fts,plane).first;
  }

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

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

  virtual TrajectoryStateOnSurface propagate (const TrajectoryStateOnSurface& tsos, 
                          const Cylinder& cylinder) const
  {
    // should be implemented (in case underlying propagator has an independent
    // implementation)
    return propagateWithPath(tsos,cylinder).first;
  }

  virtual TrajectoryStateOnSurface propagate (const FreeTrajectoryState& fts, 
                          const Cylinder& cylinder) const
  {
    // should be implemented (in case underlying propagator has an independent
    // implementation)
    return propagateWithPath(fts,cylinder).first;
  }

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

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

  virtual bool setMaxDirectionChange( float phiMax) { 
    return theGeometricalPropagator->setMaxDirectionChange(phiMax);
  }
  virtual void setPropagationDirection (PropagationDirection dir) const;

  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