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

  virtual TrajectoryStateOnSurface propagate (const TrajectoryStateOnSurface& tsos, 
                          const Surface& surface) const
  {
    return Propagator::propagate(tsos,surface);
  }
  virtual TrajectoryStateOnSurface propagate (const TrajectoryStateOnSurface& tsos, 
                          const Plane& plane) const
  {
    return propagateWithPath(tsos,plane).first;
  }
  virtual TrajectoryStateOnSurface propagate (const TrajectoryStateOnSurface& tsos, 
                          const Cylinder& cylinder) const
  {
    return propagateWithPath(tsos,cylinder).first;
  }

  virtual std::pair<TrajectoryStateOnSurface,double> 
  propagateWithPath (const TrajectoryStateOnSurface& tsos, 
             const Surface& surface) const
  {
    return Propagator::propagateWithPath(tsos,surface);
  }
  virtual std::pair<TrajectoryStateOnSurface,double> 
  propagateWithPath (const TrajectoryStateOnSurface&, 
             const Plane&) const;
  virtual std::pair<TrajectoryStateOnSurface,double> 
  propagateWithPath (const TrajectoryStateOnSurface&, 
             const Cylinder&) const;

  virtual TrajectoryStateOnSurface propagate (const FreeTrajectoryState& fts, 
                          const Surface& surface) const
  {
    return Propagator::propagate(fts,surface);
  }
  virtual TrajectoryStateOnSurface propagate (const FreeTrajectoryState& fts, 
                          const Plane& plane) const
  {
    return propagateWithPath(fts,plane).first;
  }
  virtual TrajectoryStateOnSurface propagate (const FreeTrajectoryState& fts, 
                          const Cylinder& cylinder) const
  {
    return propagateWithPath(fts,cylinder).first;
  }

  virtual std::pair<TrajectoryStateOnSurface,double> propagateWithPath (const FreeTrajectoryState& fts, 
                                   const Surface& surface) const
  {
    return Propagator::propagateWithPath(fts,surface);
  }
  virtual std::pair<TrajectoryStateOnSurface,double> propagateWithPath (const FreeTrajectoryState&, 
                                   const Plane&) const;
  virtual std::pair<TrajectoryStateOnSurface,double> propagateWithPath (const FreeTrajectoryState&, 
                                   const 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 FullConvolutionWithMaterial& convolutionWithMaterial() const {
    return *theConvolutor;
  }

  virtual GsfPropagatorWithMaterial* clone() const
  {
    return new GsfPropagatorWithMaterial(*theGeometricalPropagator,*theConvolutor);
  }

  const MagneticField* magneticField() const {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