SmartPropagatorESProducer.cc

Go to the documentation of this file.

 
#include "FWCore/Framework/interface/ESProducer.h"
 
#include "TrackingTools/GeomPropagators/interface/SmartPropagator.h"
#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"
 
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ModuleFactory.h"
#include "FWCore/Utilities/interface/Exception.h"
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "TrackingTools/Records/interface/TrackingComponentsRecord.h"
 
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
 
#include <memory>
 
class SmartPropagatorESProducer : public edm::ESProducer {
public:
  SmartPropagatorESProducer(const edm::ParameterSet&);
 
  ~SmartPropagatorESProducer() override;
 
  // Operations
  std::unique_ptr<Propagator> produce(const TrackingComponentsRecord&);
 
private:
  PropagationDirection thePropagationDirection;
  double theEpsilon;
  edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> magToken_;
  edm::ESGetToken<Propagator, TrackingComponentsRecord> trackerToken_;
  edm::ESGetToken<Propagator, TrackingComponentsRecord> muonToken_;
};
 
using namespace edm;
using namespace std;
 
SmartPropagatorESProducer::SmartPropagatorESProducer(const ParameterSet& parameterSet) {
  string myname = parameterSet.getParameter<string>("ComponentName");
 
  string propDir = parameterSet.getParameter<string>("PropagationDirection");
 
  if (propDir == "oppositeToMomentum")
    thePropagationDirection = oppositeToMomentum;
  else if (propDir == "alongMomentum")
    thePropagationDirection = alongMomentum;
  else if (propDir == "anyDirection")
    thePropagationDirection = anyDirection;
  else
    throw cms::Exception("SmartPropagatorESProducer") << "Wrong fit direction chosen in SmartPropagatorESProducer";
 
  theEpsilon = parameterSet.getParameter<double>("Epsilon");
 
  setWhatProduced(this, myname)
      .setConsumes(magToken_)
      .setConsumes(trackerToken_, edm::ESInputTag("", parameterSet.getParameter<string>("TrackerPropagator")))
      .setConsumes(muonToken_, edm::ESInputTag("", parameterSet.getParameter<string>("MuonPropagator")));
}
 
SmartPropagatorESProducer::~SmartPropagatorESProducer() {}
 
std::unique_ptr<Propagator> SmartPropagatorESProducer::produce(const TrackingComponentsRecord& iRecord) {
  return std::make_unique<SmartPropagator>(
      iRecord.get(trackerToken_), iRecord.get(muonToken_), &iRecord.get(magToken_), thePropagationDirection, theEpsilon);
}
 
DEFINE_FWK_EVENTSETUP_MODULE(SmartPropagatorESProducer);