CandIsoDepositProducer.h

Go to the documentation of this file.

#ifndef MuonIsolationProducers_CandIsoDepositProducer_H
#define MuonIsolationProducers_CandIsoDepositProducer_H

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "DataFormats/Common/interface/AssociationVector.h"
#include "DataFormats/Common/interface/RefToBaseProd.h"
#include <DataFormats/RecoCandidate/interface/RecoCandidate.h>

#include "DataFormats/RecoCandidate/interface/IsoDepositFwd.h"
#include "DataFormats/RecoCandidate/interface/IsoDeposit.h"
#include "PhysicsTools/IsolationAlgos/interface/IsoDepositExtractor.h"

#include <string>

namespace edm {
  class Event;
}
namespace edm {
  class EventSetup;
}

class CandIsoDepositProducer : public edm::stream::EDProducer<> {
public:
  CandIsoDepositProducer(const edm::ParameterSet &);

  ~CandIsoDepositProducer() override;

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

private:
  inline const reco::Track *extractTrack(const reco::Candidate &cand, reco::Track *dummyStorage) const;
  enum TrackType { FakeT, BestT, StandAloneMuonT, CombinedMuonT, TrackT, GsfT, CandidateT };
  edm::ParameterSet theConfig;
  edm::EDGetTokenT<edm::View<reco::Candidate> > theCandCollectionToken;
  TrackType theTrackType;
  std::vector<std::string> theDepositNames;
  bool theMultipleDepositsFlag;
  std::unique_ptr<reco::isodeposit::IsoDepositExtractor> theExtractor;
};
#endif