PixelNtupletsFitterProducer.cc

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/global/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"

#include "RecoPixelVertexing/PixelTrackFitting/interface/PixelFitter.h"
#include "RecoPixelVertexing/PixelTrackFitting/interface/PixelNtupletsFitter.h"

#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "RecoTracker/TkMSParametrization/interface/PixelRecoUtilities.h"

class PixelNtupletsFitterProducer: public edm::global::EDProducer<> {
public:
  explicit PixelNtupletsFitterProducer(const edm::ParameterSet& iConfig)
    : useRiemannFit_(iConfig.getParameter<bool>("useRiemannFit"))
  {
    produces<PixelFitter>();
  }
  ~PixelNtupletsFitterProducer() override {}

  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
    edm::ParameterSetDescription desc;
    desc.add<bool>("useRiemannFit", false);
    descriptions.add("pixelNtupletsFitterDefault", desc);
  }

private:
  bool useRiemannFit_;
  void produce(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const override;
};


void PixelNtupletsFitterProducer::produce(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const {
  edm::ESHandle<MagneticField> fieldESH;
  iSetup.get<IdealMagneticFieldRecord>().get(fieldESH);
  float bField = 1 / PixelRecoUtilities::fieldInInvGev(iSetup);
  auto impl = std::make_unique<PixelNtupletsFitter>(bField,
      fieldESH.product(), useRiemannFit_);
  auto prod = std::make_unique<PixelFitter>(std::move(impl));
  iEvent.put(std::move(prod));
}

DEFINE_FWK_MODULE(PixelNtupletsFitterProducer);