SmartPropagator.h

Go to the documentation of this file.

#ifndef GeomPropagators_SmartPropagator_H
#define GeomPropagators_SmartPropagator_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 Cylinder;
class Plane;


/* Class SmartPropagator Interface */

class SmartPropagator final : public Propagator {

  public:

    /* Constructor */
    SmartPropagator(const Propagator* aTkProp, const Propagator* aGenProp, const MagneticField* field,
        PropagationDirection dir = alongMomentum, float epsilon = 5) ;

    SmartPropagator(const Propagator& aTkProp, const Propagator& aGenProp,const MagneticField* field,
        PropagationDirection dir = alongMomentum, float epsilon = 5) ;

    SmartPropagator( const SmartPropagator& );

    virtual ~SmartPropagator() ;

    virtual SmartPropagator* clone() const override {
      return new SmartPropagator(getTkPropagator(),getGenPropagator(),magneticField());
    }

    void setPropagationDirection (PropagationDirection dir) override
    {
      Propagator::setPropagationDirection (dir);
      theTkProp->setPropagationDirection(dir);
      theGenProp->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;
    
    virtual std::pair< TrajectoryStateOnSurface, double>
      propagateWithPath (const TrajectoryStateOnSurface& tsos, const Plane& sur) const override;
    
    virtual std::pair< TrajectoryStateOnSurface, double>
      propagateWithPath (const TrajectoryStateOnSurface& tsos, const Cylinder& sur) const override;

 public:

    bool insideTkVol(const FreeTrajectoryState& fts) const ;
    bool insideTkVol(const Surface& surface) const ;
    bool insideTkVol(const Cylinder& cylin)  const ;
    bool insideTkVol(const Plane& plane)  const ;

    const Propagator* getTkPropagator() const ;
    const Propagator* getGenPropagator() const ;
    virtual const MagneticField* magneticField() const override {return theField;}

  private:
    void initTkVolume(float epsilon);

    Propagator* theTkProp;
    Propagator* theGenProp;
    const MagneticField* theField;
    ReferenceCountingPointer<Cylinder> theTkVolume;

  protected:

};

#endif // SMARTPROPAGATOR_H