BeamHaloPropagator.h

Go to the documentation of this file.

#ifndef GeomPropagators_BeamHaloPropagator_H
#define GeomPropagators_BeamHaloPropagator_H

/* Collaborating Class Declarations */
#include "TrackingTools/TrajectoryState/interface/FreeTrajectoryState.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "DataFormats/GeometrySurface/interface/ReferenceCounted.h"
#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"

/* Class BeamHaloPropagator Interface */

class BeamHaloPropagator : public Propagator {

  public:

    /* Constructor */ 
    BeamHaloPropagator(Propagator* aEndCapTkProp, Propagator* aCrossTkProp, const MagneticField* field,
               PropagationDirection dir = alongMomentum);

    BeamHaloPropagator(const Propagator& aEndCapTkProp,const Propagator& aCrossTkProp, const MagneticField* field,
               PropagationDirection dir = alongMomentum);


    BeamHaloPropagator( const BeamHaloPropagator& );

    virtual ~BeamHaloPropagator() ;

    virtual BeamHaloPropagator* clone() const {
      return new BeamHaloPropagator(getEndCapTkPropagator(),getCrossTkPropagator(),magneticField(),propagationDirection());
    }


    void setPropagationDirection (PropagationDirection dir) const
    {
      Propagator::setPropagationDirection(dir);
      getEndCapTkPropagator()->setPropagationDirection(dir);
      getCrossTkPropagator()->setPropagationDirection(dir);
    }



    /* Operations as propagator*/ 
    TrajectoryStateOnSurface propagate(const FreeTrajectoryState& fts, 
                                       const Surface& surface) const;

    TrajectoryStateOnSurface propagate(const TrajectoryStateOnSurface& tsos, 
                                       const Surface& surface) const {
      return Propagator::propagate(tsos,surface);
    }

    TrajectoryStateOnSurface propagate(const FreeTrajectoryState& fts,
                                       const Plane& plane) const;

    TrajectoryStateOnSurface propagate(const TrajectoryStateOnSurface& tsos,
                                       const Plane& plane) const {
      return Propagator::propagate(tsos, plane);
    }

    TrajectoryStateOnSurface propagate(const FreeTrajectoryState& fts, 
                                       const Cylinder& cylinder) const;

    TrajectoryStateOnSurface propagate(const TrajectoryStateOnSurface& tsos, 
                                       const Cylinder& cylinder) const {
      return Propagator::propagate(tsos, cylinder);
    }

    std::pair<TrajectoryStateOnSurface,double> 
      propagateWithPath(const FreeTrajectoryState& fts, 
                        const Surface& surface) const {
        return Propagator::propagateWithPath(fts,surface);
      }

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

    std::pair<TrajectoryStateOnSurface,double> 
      propagateWithPath(const FreeTrajectoryState& fts, 
                        const Plane& plane) const;

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

    std::pair<TrajectoryStateOnSurface,double> 
      propagateWithPath(const FreeTrajectoryState& fts, 
                        const Cylinder& cylinder) const;

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

      bool crossingTk(const FreeTrajectoryState& fts, const Plane& plane)  const ;

    Propagator* getEndCapTkPropagator() const ;
    Propagator* getCrossTkPropagator() const ;
    virtual const MagneticField* magneticField() const {return theField;}

  private:
    void directionCheck(PropagationDirection dir)const;

    mutable Propagator* theEndCapTkProp;
    mutable Propagator* theCrossTkProp;
    const MagneticField* theField;
    
  protected:

};

#endif