File indexing completed on 2025-01-31 02:20:02
0001 #include <string>
0002 #include <memory>
0003 #include <iostream>
0004
0005 #include "FWCore/Framework/interface/ESHandle.h"
0006 #include "FWCore/Framework/interface/ESProducer.h"
0007 #include "FWCore/Framework/interface/EventSetup.h"
0008 #include "FWCore/Framework/interface/ModuleFactory.h"
0009 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0010 #include "FWCore/Utilities/interface/ESGetToken.h"
0011 #include "TrackingTools/GeomPropagators/interface/AnalyticalPropagator.h"
0012 #include "TrackingTools/GsfTools/interface/CloseComponentsMerger.h"
0013 #include "TrackingTools/GsfTools/interface/MultiGaussianStateMerger.h"
0014 #include "TrackingTools/GsfTracking/interface/GsfChi2MeasurementEstimator.h"
0015 #include "TrackingTools/GsfTracking/interface/GsfMaterialEffectsUpdator.h"
0016 #include "TrackingTools/GsfTracking/interface/GsfMultiStateUpdator.h"
0017 #include "TrackingTools/GsfTracking/interface/GsfPropagatorWithMaterial.h"
0018 #include "TrackingTools/GsfTracking/interface/GsfTrajectoryFitter.h"
0019 #include "TrackingTools/GsfTracking/interface/MultiTrajectoryStateMerger.h"
0020 #include "TrackingTools/Records/interface/TrackingComponentsRecord.h"
0021 #include "TrackingTools/TrackFitters/interface/TrajectoryFitter.h"
0022 #include "TrackingTools/TrackFitters/interface/TrajectoryFitterRecord.h"
0023
0024
0025
0026 class GsfTrajectoryFitterESProducer : public edm::ESProducer {
0027 public:
0028 GsfTrajectoryFitterESProducer(const edm::ParameterSet& p);
0029 ~GsfTrajectoryFitterESProducer() override = default;
0030 std::unique_ptr<TrajectoryFitter> produce(const TrajectoryFitterRecord&);
0031
0032 static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
0033
0034 private:
0035 edm::ESGetToken<GsfMaterialEffectsUpdator, TrackingComponentsRecord> matUpdatorToken_;
0036 edm::ESGetToken<Propagator, TrackingComponentsRecord> propagatorToken_;
0037 edm::ESGetToken<MultiGaussianStateMerger<5>, TrackingComponentsRecord> mergerToken_;
0038 edm::ESGetToken<DetLayerGeometry, RecoGeometryRecord> geoToken_;
0039 };
0040
0041 GsfTrajectoryFitterESProducer::GsfTrajectoryFitterESProducer(const edm::ParameterSet& p) {
0042 std::string myname = p.getParameter<std::string>("ComponentName");
0043 auto cc = setWhatProduced(this, myname);
0044 matUpdatorToken_ = cc.consumes(edm::ESInputTag("", p.getParameter<std::string>("MaterialEffectsUpdator")));
0045 propagatorToken_ = cc.consumes(edm::ESInputTag("", p.getParameter<std::string>("GeometricalPropagator")));
0046 mergerToken_ = cc.consumes(edm::ESInputTag("", p.getParameter<std::string>("Merger")));
0047 geoToken_ = cc.consumes(edm::ESInputTag("", p.getParameter<std::string>("RecoGeometry")));
0048 }
0049
0050 std::unique_ptr<TrajectoryFitter> GsfTrajectoryFitterESProducer::produce(const TrajectoryFitterRecord& iRecord) {
0051
0052
0053
0054 GsfPropagatorWithMaterial propagator(iRecord.get(propagatorToken_), iRecord.get(matUpdatorToken_));
0055
0056
0057
0058 MultiTrajectoryStateMerger merger(iRecord.get(mergerToken_));
0059
0060
0061
0062
0063 double chi2Cut(100.);
0064 GsfChi2MeasurementEstimator estimator(chi2Cut);
0065
0066
0067
0068
0069 return std::make_unique<GsfTrajectoryFitter>(
0070 propagator, GsfMultiStateUpdator(), estimator, merger, &iRecord.get(geoToken_));
0071 }
0072
0073 void GsfTrajectoryFitterESProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
0074 edm::ParameterSetDescription desc;
0075 desc.add<std::string>("ComponentName", "GsfTrajectoryFitter");
0076 desc.add<std::string>("MaterialEffectsUpdator", "ElectronMaterialEffects");
0077 desc.add<std::string>("GeometricalPropagator", "fwdAnalyticalPropagator");
0078 desc.add<std::string>("Merger", "CloseComponentsMerger5D");
0079 desc.add<std::string>("RecoGeometry", "GlobalDetLayerGeometry");
0080 descriptions.addWithDefaultLabel(desc);
0081 }
0082
0083 DEFINE_FWK_EVENTSETUP_MODULE(GsfTrajectoryFitterESProducer);