CkfDebugTrackCandidateMaker.h

Go to the documentation of this file.

#ifndef CkfDebugTrackCandidateMaker_h
#define CkfDebugTrackCandidateMaker_h

#include "RecoTracker/CkfPattern/interface/CkfTrackCandidateMakerBase.h"
#include "CkfDebugTrajectoryBuilder.h"
#include "FWCore/Framework/interface/EDProducer.h"
#include "DataFormats/TrackReco/interface/SeedStopInfo.h"

namespace cms {
  class CkfDebugTrackCandidateMaker : public edm::EDProducer, public CkfTrackCandidateMakerBase {
  public:
    CkfDebugTrackCandidateMaker(const edm::ParameterSet& conf) : CkfTrackCandidateMakerBase(conf, consumesCollector()) {
      produces<TrackCandidateCollection>();
      produces<SeedStopInfo>();
    }

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

    void produce(edm::Event& e, const edm::EventSetup& es) override { produceBase(e, es); }
    void endJob() override { delete dbg; }

  private:
    TrajectorySeedCollection::const_iterator lastSeed(TrajectorySeedCollection const& theSeedColl) override {
      return theSeedColl.begin() + 1;
    }

    void initDebugger(edm::EventSetup const& es) {
      dbg = new CkfDebugger(es, consumesCollector());
      myTrajectoryBuilder = dynamic_cast<const CkfDebugTrajectoryBuilder*>(theTrajectoryBuilder.get());
      if (myTrajectoryBuilder)
        myTrajectoryBuilder->setDebugger(dbg);
      else
        throw cms::Exception("CkfDebugger") << "please use CkfDebugTrajectoryBuilder";
      //theTrajectoryBuilder->setDebugger( dbg);
    };

    void printHitsDebugger(edm::Event& e) override { dbg->printSimHits(e); };
    void countSeedsDebugger() override { dbg->countSeed(); };
    void deleteAssocDebugger() override { dbg->deleteHitAssociator(); };
    void deleteDebugger() { delete dbg; };
    CkfDebugger* dbg;
    const CkfDebugTrajectoryBuilder* myTrajectoryBuilder;
  };
}  // namespace cms

#endif