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 final : public Propagator {

  public:

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

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


    BeamHaloPropagator( const BeamHaloPropagator& );

    ~BeamHaloPropagator() override ;

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


    void setPropagationDirection (PropagationDirection dir) override
    {
      Propagator::setPropagationDirection(dir);
      theEndCapTkProp->setPropagationDirection(dir);
      theCrossTkProp->setPropagationDirection(dir);
    }

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

 private:

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


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


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

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

  private:
    void directionCheck(PropagationDirection dir);

    Propagator* theEndCapTkProp;
    Propagator* theCrossTkProp;
    const MagneticField* theField;

  protected:

};

#endif