TMTrackProducer.h

Go to the documentation of this file.

#ifndef L1Trigger_TrackFindingTMTT_TMTrackProducer_h
#define L1Trigger_TrackFindingTMTT_TMTrackProducer_h
 
#include "L1Trigger/TrackFindingTMTT/interface/Settings.h"
#include "L1Trigger/TrackFindingTMTT/interface/Histos.h"
#include "L1Trigger/TrackFindingTMTT/interface/Stub.h"
#include "L1Trigger/TrackFindingTMTT/interface/L1track3D.h"
#include "L1Trigger/TrackFindingTMTT/interface/TrackerModule.h"
#include "L1Trigger/TrackFindingTMTT/interface/StubFEWindows.h"
#include "L1Trigger/TrackFindingTMTT/interface/StubWindowSuggest.h"
#include "L1Trigger/TrackFindingTMTT/interface/GlobalCacheTMTT.h"
 
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/L1TrackTrigger/interface/TTTypes.h"
#include "DataFormats/Phase2TrackerDigi/interface/Phase2TrackerDigi.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
#include "SimTracker/TrackTriggerAssociation/interface/TTClusterAssociationMap.h"
#include "SimTracker/TrackTriggerAssociation/interface/TTStubAssociationMap.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "L1Trigger/TrackTrigger/interface/TTStubAlgorithmRecord.h"
#include "L1Trigger/TrackTrigger/interface/TTStubAlgorithm_official.h"
//#include "L1Trigger/TrackTrigger/interface/TTStubAlgorithm.h"
 
#include <vector>
#include <list>
#include <string>
#include <memory>
 
namespace tmtt {
 
  class TrackFitGeneric;
 
  typedef TTStubAlgorithm<Ref_Phase2TrackerDigi_> StubAlgorithm;
  typedef TTStubAlgorithm_official<Ref_Phase2TrackerDigi_> StubAlgorithmOfficial;
 
  class TMTrackProducer : public edm::stream::EDProducer<edm::GlobalCache<GlobalCacheTMTT>> {
  public:
    explicit TMTrackProducer(const edm::ParameterSet &, GlobalCacheTMTT const *globalCacheTMTT);
    ~TMTrackProducer() override {}
 
    static std::unique_ptr<GlobalCacheTMTT> initializeGlobalCache(edm::ParameterSet const &iConfig);
 
    static void globalEndJob(GlobalCacheTMTT *globalCacheTMTT);
 
  private:
    typedef std::vector<TTTrack<Ref_Phase2TrackerDigi_>> TTTrackCollection;
 
    void beginRun(const edm::Run &, const edm::EventSetup &) override;
 
    void produce(edm::Event &, const edm::EventSetup &) override;
 
  private:
    // ES tokens
    edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> magneticFieldToken_;
    edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> trackerGeometryToken_;
    edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> trackerTopologyToken_;
    edm::ESGetToken<StubAlgorithm, TTStubAlgorithmRecord> ttStubAlgoToken_;
    // ED tokens
    edm::EDGetTokenT<TTStubDetSetVec> stubToken_;
    edm::EDGetTokenT<TrackingParticleCollection> tpToken_;
    edm::EDGetTokenT<TTStubAssMap> stubTruthToken_;
    edm::EDGetTokenT<TTClusterAssMap> clusterTruthToken_;
    edm::EDGetTokenT<reco::GenJetCollection> genJetToken_;
 
    // Info about tracker geometry
    const TrackerGeometry *trackerGeometry_;
    const TrackerTopology *trackerTopology_;
    std::list<TrackerModule> listTrackerModule_;
 
    // Configuration parameters
    Settings settings_;
    std::vector<std::string> trackFitters_;
    std::vector<std::string> useRZfilter_;
    bool runRZfilter_;
 
    // Stub window sizes used by FE electronics.
    const StubAlgorithmOfficial *stubAlgo_;
    std::unique_ptr<StubFEWindows> stubFEWindows_;
    StubWindowSuggest &stubWindowSuggest_;
    std::unique_ptr<DegradeBend> degradeBend_;
 
    Histos &hists_;
    HTrphi::ErrorMonitor &htRphiErrMon_;
 
    std::map<std::string, std::unique_ptr<TrackFitGeneric>> fitterWorkerMap_;
 
    bool debug_;
  };
 
}  // namespace tmtt
 
#endif