defaultRKPropagator.h

Go to the documentation of this file.

#ifndef RKPropagatorDefault_H
#define RKPropagatorDefault_H
 
#include "TrackPropagation/RungeKutta/interface/RKPropagatorInS.h"
#include "MagneticField/VolumeGeometry/interface/MagneticFieldProvider.h"
 
namespace defaultRKPropagator {
 
  using RKPropagator = RKPropagatorInS;
 
  // not clear why we need all this
  class TrivialFieldProvider final : public MagneticFieldProvider<float> {
  public:
    TrivialFieldProvider(const MagneticField* field) : theField(field) {}
 
    LocalVectorType valueInTesla(const LocalPointType& lp) const override {
      // NOTE: the following transformation only works for the central volume
      // where global and local coordinates are numerically equal !
      GlobalPoint gp(lp.basicVector());
      return LocalVectorType(theField->inTesla(gp).basicVector());
    }
 
  private:
    const MagneticField* theField;
  };
 
  class RKMagVolume final : public MagVolume {
  public:
    RKMagVolume(const PositionType& pos, const RotationType& rot, const MagneticFieldProvider<float>* mfp)
        : MagVolume(pos, rot, mfp) {}
 
    bool inside(const GlobalPoint& gp, double tolerance = 0.) const override { return true; }
 
    const std::vector<VolumeSide>& faces() const override { return theFaces; }
 
  private:
    std::vector<VolumeSide> theFaces;
  };
 
  struct Product {
    explicit Product(const MagneticField* field, PropagationDirection dir = alongMomentum, double tolerance = 5.e-5)
        : mpf(field),
          volume(MagVolume::PositionType(0, 0, 0), MagVolume::RotationType(), &mpf),
          propagator(volume, dir, tolerance) {}
    TrivialFieldProvider mpf;
    RKMagVolume volume;
    RKPropagator propagator;
  };
 
}  // namespace defaultRKPropagator
 
#endif  // RKPropagatorDefault