@@ -38,7 +38,6 @@ namespace trklet {
3838 ~ProducerKF () override {}
3939
4040 private:
41- void beginRun (const Run&, const EventSetup&) override ;
4241 void produce (Event&, const EventSetup&) override ;
4342 void endStream () override {
4443 if (printDebug_)
@@ -72,8 +71,6 @@ namespace trklet {
7271 KFParamsComb* tmtt_;
7372 // print end job internal unused MSB
7473 bool printDebug_;
75- //
76- bool use5ParameterFit_;
7774 };
7875
7976 ProducerKF::ProducerKF (const ParameterSet& iConfig)
@@ -123,9 +120,6 @@ namespace trklet {
123120 iConfig_.baseShiftC23_ = iConfig.getParameter <int >(" BaseShiftC23" );
124121 iConfig_.baseShiftC33_ = iConfig.getParameter <int >(" BaseShiftC33" );
125122 printDebug_ = iConfig.getParameter <bool >(" PrintKFDebug" );
126- use5ParameterFit_ = iConfig.getParameter <bool >(" Use5ParameterFit" );
127- if (use5ParameterFit_)
128- tmtt_ = &tmtt5_;
129123 const string& label = iConfig.getParameter <string>(" InputLabelKF" );
130124 const string& branchStubs = iConfig.getParameter <string>(" BranchStubs" );
131125 const string& branchTracks = iConfig.getParameter <string>(" BranchTracks" );
@@ -143,21 +137,13 @@ namespace trklet {
143137 esGetTokenDataFormats_ = esConsumes ();
144138 }
145139
146- void ProducerKF::beginRun ( const Run& iRun , const EventSetup& iSetup) {
140+ void ProducerKF::produce (Event& iEvent , const EventSetup& iSetup) {
147141 // helper class to store configurations
148142 const Setup* setup = &iSetup.getData (esGetTokenSetup_);
149143 settings_.setMagneticField (setup->bField ());
150144 // helper class to extract structured data from tt::Frames
151145 const DataFormats* dataFormats = &iSetup.getData (esGetTokenDataFormats_);
152- // provides dataformats of Kalman filter internals
153146 kalmanFilterFormats_.consume (dataFormats, iConfig_);
154- }
155-
156- void ProducerKF::produce (Event& iEvent, const EventSetup& iSetup) {
157- // helper class to store configurations
158- const Setup* setup = &iSetup.getData (esGetTokenSetup_);
159- // helper class to extract structured data from tt::Frames
160- const DataFormats* dataFormats = &iSetup.getData (esGetTokenDataFormats_);
161147 auto valid = [](int sum, const FrameTrack& f) { return sum + (f.first .isNull () ? 0 : 1 ); };
162148 // empty KF products
163149 StreamsStub streamsStub (setup->numRegions () * setup->numLayers ());
@@ -170,7 +156,8 @@ namespace trklet {
170156 // prep TTTracks
171157 TTTracks ttTracks;
172158 vector<TTTrackRef> ttTrackRefs;
173- if (use5ParameterFit_) {
159+ if (setup->kfUse5ParameterFit ()) {
160+ tmtt_ = &tmtt5_;
174161 int nTracks (0 );
175162 for (const StreamTrack& stream : tracks)
176163 nTracks += accumulate (stream.begin (), stream.end (), 0 , valid);
@@ -189,7 +176,7 @@ namespace trklet {
189176 // fill output products
190177 kf.produce (streamsStub, streamsTrack, numStatesAccepted, numStatesTruncated);
191178 }
192- if (use5ParameterFit_ ) {
179+ if (setup-> kfUse5ParameterFit () ) {
193180 // store ttTracks
194181 const OrphanHandle<TTTracks> oh = iEvent.emplace (edPutTokenTTTracks_, std::move (ttTracks));
195182 // replace ttTrackRefs in track streams
0 commit comments