GsfPropagatorWithMaterial.h

Go to the documentation of this file.

#ifndef GsfPropagatorWithMaterial_h_
#define GsfPropagatorWithMaterial_h_
 
#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/GsfTools/interface/GsfPropagatorAdapter.h"
#include "TrackingTools/GsfTracking/interface/GsfMaterialEffectsUpdator.h"
#include "TrackingTools/GsfTracking/interface/FullConvolutionWithMaterial.h"
#include "DataFormats/GeometryCommonDetAlgo/interface/DeepCopyPointerByClone.h"
 
class MagneticField;
 
// #include "Utilities/Timing/interface/TimingReport.h"
 
class GsfPropagatorWithMaterial : public Propagator {
public:
  // Constructors
  GsfPropagatorWithMaterial(const Propagator& Propagator, const GsfMaterialEffectsUpdator& MEUpdator);
  GsfPropagatorWithMaterial(const GsfPropagatorAdapter& Propagator, const FullConvolutionWithMaterial& Convolutor);
 
  ~GsfPropagatorWithMaterial() override {}
 
  using Propagator::propagate;
  using Propagator::propagateWithPath;
 
  std::pair<TrajectoryStateOnSurface, double> propagateWithPath(const TrajectoryStateOnSurface&,
                                                                const Plane&) const override;
 
  std::pair<TrajectoryStateOnSurface, double> propagateWithPath(const TrajectoryStateOnSurface&,
                                                                const Cylinder&) const override;
 
  std::pair<TrajectoryStateOnSurface, double> propagateWithPath(const FreeTrajectoryState&,
                                                                const Plane&) const override;
  std::pair<TrajectoryStateOnSurface, double> propagateWithPath(const FreeTrajectoryState&,
                                                                const Cylinder&) const override;
 
  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 FullConvolutionWithMaterial& convolutionWithMaterial() const { return *theConvolutor; }
 
  GsfPropagatorWithMaterial* clone() const override {
    return new GsfPropagatorWithMaterial(*theGeometricalPropagator, *theConvolutor);
  }
 
  const MagneticField* magneticField() const override { return theGeometricalPropagator->magneticField(); }
 
private:
  //   /// Definition of timers (temporary)
  //   void defineTimer();
  std::pair<TrajectoryStateOnSurface, double> convoluteWithMaterial(
      const std::pair<TrajectoryStateOnSurface, double>&) const;
  TrajectoryStateOnSurface convoluteStateWithMaterial(const TrajectoryStateOnSurface, const PropagationDirection) const;
  bool materialAtSource() const;
 
private:
  // Geometrical propagator
  DeepCopyPointerByClone<GsfPropagatorAdapter> theGeometricalPropagator;
  // Material effects & convolution
  DeepCopyPointerByClone<FullConvolutionWithMaterial> theConvolutor;
  // Use material at source?
  MaterialLocation theMaterialLocation;
 
  typedef std::pair<TrajectoryStateOnSurface, double> TsosWP;
  typedef std::vector<TrajectoryStateOnSurface> MultiTSOS;
 
  //   static TimingReport::Item* propWithPathTimer1;
  //   static TimingReport::Item* propWithPathTimer2;
};
 
#endif