RecoTracker/CkfPattern/plugins/CkfTrajectoryMaker.h - master - githubmirror/cmssw

RecoTracker/CkfPattern/plugins/CkfTrajectoryMaker.h

54 строки2 KB

Christopher Jones

Added edm::stream::Watch* to RecoTracker modules

25 мар 2026, 23:55

25 мар 2026, 23:55214f0aa

#ifndef CkfTrajectoryMaker_h
#define CkfTrajectoryMaker_h

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"

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

#include "TrackingTools/TrajectoryCleaning/interface/TrajectoryCleaner.h"

#include "MagneticField/Engine/interface/MagneticField.h"
#include "TrackingTools/DetLayers/interface/NavigationSchool.h"
#include "RecoTracker/TkDetLayers/interface/GeometricSearchTracker.h"

#include "RecoTracker/CkfPattern/interface/RedundantSeedCleaner.h"
#include "RecoTracker/CkfPattern/interface/CkfTrackCandidateMakerBase.h"
#include "DataFormats/TrackCandidate/interface/TrackCandidateCollection.h"
#include "DataFormats/TrackReco/interface/SeedStopInfo.h"

class TransientInitialStateEstimator;

namespace cms {
  class dso_internal CkfTrajectoryMaker : public edm::stream::EDProducer<edm::stream::WatchRuns>,
                                          public CkfTrackCandidateMakerBase {
  public:
    typedef std::vector<Trajectory> TrajectoryCollection;

    explicit CkfTrajectoryMaker(const edm::ParameterSet& conf) : CkfTrackCandidateMakerBase(conf, consumesCollector()) {
      theTrackCandidateOutput = conf.getParameter<bool>("trackCandidateAlso");
      theTrajectoryOutput = true;
      if (theTrackCandidateOutput)
        produces<TrackCandidateCollection>();
      produces<TrajectoryCollection>();
      produces<std::vector<SeedStopInfo> >();
    }

    ~CkfTrajectoryMaker() override { ; }

    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
      edm::ParameterSetDescription desc;
      desc.setComment("Make CKF trajectories");
      desc.add<bool>("trackCandidateAlso", false);
      CkfTrackCandidateMakerBase::fillPSetDescription(desc);
      descriptions.addWithDefaultLabel(desc);
    }

    void beginRun(edm::Run const& run, edm::EventSetup const& es) override { beginRunBase(run, es); }

    void produce(edm::Event& e, const edm::EventSetup& es) override { produceBase(e, es); }
  };
}  // namespace cms

#endif