TrackMergeremb.h

Go to the documentation of this file.

#ifndef TauAnalysis_MCEmbeddingTools_TrackMergeremb_H
#define TauAnalysis_MCEmbeddingTools_TrackMergeremb_H

#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/EgammaReco/interface/SuperCluster.h"
#include "DataFormats/EgammaReco/interface/SuperClusterFwd.h"

#include "DataFormats/EgammaReco/interface/ElectronSeed.h"
#include "DataFormats/EgammaReco/interface/ElectronSeedFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PreId.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"

#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/MuonReco/interface/MuonQuality.h"

#include "DataFormats/MuonReco/interface/MuonToMuonMap.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"

#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrackExtra.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrackExtraFwd.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrackFwd.h"
#include "DataFormats/TrackReco/interface/TrackToTrackMap.h"

#include "Geometry/CommonDetUnit/interface/GlobalTrackingGeometry.h"
#include "Geometry/Records/interface/GlobalTrackingGeometryRecord.h"

#include <iostream>
#include <map>
#include <string>

template <typename T1>
class TrackMergeremb : public edm::stream::EDProducer<> {
public:
  explicit TrackMergeremb(const edm::ParameterSet &);
  ~TrackMergeremb() override;

private:
  void produce(edm::Event &, const edm::EventSetup &) override;

  typedef T1 TrackCollectionemb;

  void willproduce(std::string instance, std::string alias);
  void willconsume(const edm::ParameterSet &iConfig);
  void merg_and_put(edm::Event &, std::string, std::vector<edm::EDGetTokenT<TrackCollectionemb>> &);

  std::map<std::string, std::vector<edm::EDGetTokenT<TrackCollectionemb>>> inputs_;
  std::map<std::string, std::vector<edm::EDGetTokenT<edm::ValueMap<reco::MuonQuality>>>> inputs_qual_;

  typedef edm::ValueMap<reco::TrackRefVector> TrackToTrackMapnew;
  typedef edm::ValueMap<reco::GsfTrackRefVector> GsfTrackToTrackMapnew;

  edm::ESGetToken<GlobalTrackingGeometry, GlobalTrackingGeometryRecord> globalGeomToken_;
  const GlobalTrackingGeometry *geometry_ = nullptr;

  edm::EDGetTokenT<TrackToTrackMapnew> inputs_fixtrackrefs_;
  edm::EDGetTokenT<reco::TrackCollection> inputs_fixtrackcol_;

  edm::EDGetTokenT<GsfTrackToTrackMapnew> inputs_fixgsftrackrefs_;
  edm::EDGetTokenT<reco::GsfTrackCollection> inputs_fixgsftrackcol_;

  edm::EDGetTokenT<reco::MuonToMuonMap> inputs_fixmurefs_;
  edm::EDGetTokenT<reco::MuonCollection> inputs_fixmucol_;

  edm::EDGetTokenT<reco::SuperClusterCollection> inputs_SC_;
  edm::EDGetTokenT<reco::ElectronSeedCollection> inputs_rElectronMergedSeeds_;
  edm::EDGetTokenT<edm::View<TrajectorySeed>> inputs_rElectronMergedSeedViews_;

  std::string alias;
};

#endif