CkfTrackCandidateMakerBase.h

Go to the documentation of this file.

#ifndef CkfTrackCandidateMakerBase_h
#define CkfTrackCandidateMakerBase_h

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

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

#include "TrackingTools/PatternTools/interface/TrajectoryBuilder.h"
#include "RecoTracker/CkfPattern/interface/BaseCkfTrajectoryBuilder.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 "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "DataFormats/Common/interface/DetSetVectorNew.h"
#include "DataFormats/Common/interface/ContainerMask.h"
#include "DataFormats/SiStripCluster/interface/SiStripCluster.h"
#include "DataFormats/SiPixelCluster/interface/SiPixelCluster.h"
#include "DataFormats/Phase2TrackerCluster/interface/Phase2TrackerCluster1D.h"
#include "RecoTracker/MeasurementDet/interface/MeasurementTrackerEvent.h"

#include <memory>

class TransientInitialStateEstimator;

namespace cms {
  class CkfTrackCandidateMakerBase {
  public:
    explicit CkfTrackCandidateMakerBase(const edm::ParameterSet& conf, edm::ConsumesCollector&& iC);

    virtual ~CkfTrackCandidateMakerBase() noexcept(false);

    virtual void beginRunBase(edm::Run const&, edm::EventSetup const& es);

    virtual void produceBase(edm::Event& e, const edm::EventSetup& es);

  protected:
    bool theTrackCandidateOutput;
    bool theTrajectoryOutput;
    bool useSplitting;
    bool doSeedingRegionRebuilding;
    bool cleanTrajectoryAfterInOut;
    bool reverseTrajectories;
    bool produceSeedStopReasons_;

    unsigned int theMaxNSeeds;

    std::unique_ptr<BaseCkfTrajectoryBuilder> theTrajectoryBuilder;

    std::string theTrajectoryCleanerName;
    const TrajectoryCleaner* theTrajectoryCleaner;

    std::unique_ptr<TransientInitialStateEstimator> theInitialState;

    const std::string theMagFieldName;
    edm::ESHandle<MagneticField> theMagField;
    edm::ESHandle<GeometricSearchTracker> theGeomSearchTracker;

    std::string theNavigationSchoolName;
    const NavigationSchool* theNavigationSchool;

    std::unique_ptr<RedundantSeedCleaner> theSeedCleaner;

    unsigned int maxSeedsBeforeCleaning_;

    edm::EDGetTokenT<edm::View<TrajectorySeed> > theSeedLabel;
    edm::EDGetTokenT<MeasurementTrackerEvent> theMTELabel;

    bool skipClusters_;
    bool phase2skipClusters_;
    typedef edm::ContainerMask<edmNew::DetSetVector<SiPixelCluster> > PixelClusterMask;
    typedef edm::ContainerMask<edmNew::DetSetVector<SiStripCluster> > StripClusterMask;
    typedef edm::ContainerMask<edmNew::DetSetVector<Phase2TrackerCluster1D> > Phase2OTClusterMask;
    edm::EDGetTokenT<PixelClusterMask> maskPixels_;
    edm::EDGetTokenT<StripClusterMask> maskStrips_;
    edm::EDGetTokenT<Phase2OTClusterMask> maskPhase2OTs_;

    // methods for debugging
    virtual TrajectorySeedCollection::const_iterator lastSeed(TrajectorySeedCollection const& theSeedColl) {
      return theSeedColl.end();
    }
    virtual void printHitsDebugger(edm::Event& e) { ; }
    virtual void countSeedsDebugger() { ; }
    virtual void deleteAssocDebugger() { ; }

  private:
    void setEventSetup(const edm::EventSetup& es);
  };
}  // namespace cms

#endif