KFTrajectorySmootherESProducer.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ESProducer.h"

#include "TrackingTools/PatternTools/interface/TrajectoryStateUpdator.h"
#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/PatternTools/interface/TrajectoryStateUpdator.h"
#include "TrackingTools/KalmanUpdators/interface/Chi2MeasurementEstimatorBase.h"


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


#include "TrackingTools/TrackFitters/interface/TrajectoryFitterRecord.h"
#include "TrackingTools/Records/interface/TrackingComponentsRecord.h"
#include "TrackingTools/TrackFitters/interface/KFTrajectorySmoother.h"
#include <memory>


namespace {

  class  KFTrajectorySmootherESProducer final : public edm::ESProducer{
  public:
    KFTrajectorySmootherESProducer(const edm::ParameterSet & p);
    ~KFTrajectorySmootherESProducer() override; 
    std::unique_ptr<TrajectorySmoother> produce(const TrajectoryFitterRecord &);

    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
      edm::ParameterSetDescription desc;
       desc.add<std::string>("ComponentName","KFSmoother");
       desc.add<std::string>("Propagator","PropagatorWithMaterial");
       desc.add<std::string>("Updator","KFUpdator");
       desc.add<std::string>("Estimator","Chi2");
       desc.add<std::string>("RecoGeometry","GlobalDetLayerGeometry");
       desc.add<double>("errorRescaling",100);
       desc.add<int>("minHits",3);
       descriptions.add("KFTrajectorySmoother", desc);
    }
 
    
  private:
    edm::ParameterSet pset_;
  };


  KFTrajectorySmootherESProducer::KFTrajectorySmootherESProducer(const edm::ParameterSet & p) 
  {
    std::string myname = p.getParameter<std::string>("ComponentName");
    pset_ = p;
    setWhatProduced(this,myname);
  }
  
  KFTrajectorySmootherESProducer::~KFTrajectorySmootherESProducer() {}
  
  std::unique_ptr<TrajectorySmoother> 
    KFTrajectorySmootherESProducer::produce(const TrajectoryFitterRecord & iRecord){ 
    
    std::string pname = pset_.getParameter<std::string>("Propagator");
    std::string uname = pset_.getParameter<std::string>("Updator");
    std::string ename = pset_.getParameter<std::string>("Estimator");
    std::string gname = pset_.getParameter<std::string>("RecoGeometry");
    double rescaleFactor =  pset_.getParameter<double>("errorRescaling");
    int minHits = pset_.getParameter<int>("minHits");
    
    edm::ESHandle<Propagator> prop;
    edm::ESHandle<TrajectoryStateUpdator> upd;
    edm::ESHandle<Chi2MeasurementEstimatorBase> est;
    edm::ESHandle<DetLayerGeometry> geo;
    
    
    iRecord.getRecord<TrackingComponentsRecord>().get(pname, prop);
    iRecord.getRecord<TrackingComponentsRecord>().get(uname, upd);
    iRecord.getRecord<TrackingComponentsRecord>().get(ename, est);
    iRecord.getRecord<RecoGeometryRecord>().get(gname,geo);
    
    return std::make_unique<KFTrajectorySmoother>(prop.product(),
                          upd.product(),
                          est.product(),
                          rescaleFactor,
                          minHits,
                          geo.product());
    
  }
}


#include "FWCore/Framework/interface/ModuleFactory.h"
DEFINE_FWK_EVENTSETUP_MODULE(KFTrajectorySmootherESProducer);