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