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/TkNavigation/interface/SimpleNavigationSchool.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 "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();

    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;

    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;
    
    RedundantSeedCleaner*  theSeedCleaner;

    unsigned int maxSeedsBeforeCleaning_;
    
    edm::EDGetTokenT<edm::View<TrajectorySeed> >  theSeedLabel;
    edm::EDGetTokenT<MeasurementTrackerEvent>     theMTELabel;

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

    // 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 ) ; 
  };
}

#endif