/data/refman/pasoursint/CMSSW_4_1_8_patch13/src/TrackingTools/MaterialEffects/interface/PropagatorWithMaterial.h

Go to the documentation of this file.

#ifndef _COMMONRECO_PROPAGATORWITHMATERIAL_H_
#define _COMMONRECO_PROPAGATORWITHMATERIAL_H_

#include "DataFormats/GeometryCommonDetAlgo/interface/DeepCopyPointerByClone.h"
#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/MaterialEffects/interface/MaterialEffectsUpdator.h"
#include "MagneticField/Engine/interface/MagneticField.h"

class PropagatorWithMaterial : public Propagator {
  
public:
  PropagatorWithMaterial (PropagationDirection dir, const float mass,  
                          const MagneticField * mf=0,const float maxDPhi=1.6,
                          bool useRungeKutta=false, float ptMin=-1.);
  PropagatorWithMaterial(const Propagator& Propagator,
                         const MaterialEffectsUpdator& MEUpdator,
                         const MagneticField * mf=0,
                         bool useRungeKutta=false);

  ~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;

private:
  // Geometrical propagator
  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