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() {}


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


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

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


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



  virtual bool setMaxDirectionChange( float phiMax) override {
    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 FullConvolutionWithMaterial& convolutionWithMaterial() const {
    return *theConvolutor;
  }

  virtual 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