HLTExoticaSubAnalysis.h

Go to the documentation of this file.

#ifndef HLTriggerOffline_Exotica_HLTExoticaSubAnalysis_H
#define HLTriggerOffline_Exotica_HLTExoticaSubAnalysis_H

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "CommonTools/Utils/interface/StringCutObjectSelector.h"
#include "DQMServices/Core/interface/DQMStore.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "DataFormats/Candidate/interface/LeafCandidate.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectronFwd.h"
#include "DataFormats/EgammaCandidates/interface/Photon.h"
#include "DataFormats/EgammaCandidates/interface/PhotonFwd.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/JetReco/interface/CaloJet.h"
#include "DataFormats/JetReco/interface/CaloJetCollection.h"
#include "DataFormats/JetReco/interface/PFJet.h"
#include "DataFormats/JetReco/interface/PFJetCollection.h"
#include "DataFormats/METReco/interface/CaloMET.h"
#include "DataFormats/METReco/interface/CaloMETCollection.h"
#include "DataFormats/METReco/interface/GenMET.h"
#include "DataFormats/METReco/interface/GenMETCollection.h"
#include "DataFormats/METReco/interface/PFMET.h"
#include "DataFormats/METReco/interface/PFMETCollection.h"
#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "DataFormats/TauReco/interface/PFTau.h"
#include "DataFormats/TauReco/interface/PFTauFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "HLTrigger/HLTcore/interface/HLTConfigProvider.h"

#include "HLTriggerOffline/Exotica/interface/HLTExoticaPlotter.h"

#include <cstring>
#include <map>
#include <set>
#include <vector>

struct EVTColContainer;

class HLTExoticaSubAnalysis {
public:
  typedef dqm::legacy::DQMStore DQMStore;
  typedef dqm::legacy::MonitorElement MonitorElement;

  enum class Level { GEN = 98, RECO = 99 };

  HLTExoticaSubAnalysis(const edm::ParameterSet &pset,
                        const std::string &analysisname,
                        edm::ConsumesCollector &&consCollector);
  ~HLTExoticaSubAnalysis();
  void beginJob();
  void beginRun(const edm::Run &iRun, const edm::EventSetup &iEventSetup);
  void endRun();

  void subAnalysisBookHistos(DQMStore::IBooker &iBooker, const edm::Run &iRun, const edm::EventSetup &iSetup);

  void analyze(const edm::Event &iEvent, const edm::EventSetup &iEventSetup, EVTColContainer *cols);

private:
  const std::vector<unsigned int> getObjectsType(const std::string &hltpath) const;

  void getNamesOfObjects(const edm::ParameterSet &anpset);
  void registerConsumes(edm::ConsumesCollector &consCollector);
  void getHandlesToObjects(const edm::Event &iEvent, EVTColContainer *col);
  void initSelector(const unsigned int &objtype);
  void insertCandidates(const unsigned int &objtype,
                        const EVTColContainer *col,
                        std::vector<reco::LeafCandidate> *matches,
                        std::map<int, double> &theSumEt,
                        std::map<int, std::vector<const reco::Track *>> &trkObjs);

  void bookHist(DQMStore::IBooker &iBooker,
                const std::string &source,
                const std::string &objType,
                const std::string &variable);
  void fillHist(const std::string &source, const std::string &objType, const std::string &variable, const float &value);

  edm::ParameterSet _pset;

  std::string _analysisname;

  unsigned int _minCandidates;

  std::vector<std::string> _hltPathsToCheck;
  std::set<std::string> _hltPaths;
  std::map<std::string, std::string> _shortpath2long;

  std::string _hltProcessName;
  edm::InputTag _genParticleLabel;
  edm::InputTag _trigResultsLabel;
  edm::InputTag _beamSpotLabel;
  std::map<unsigned int, edm::InputTag> _recLabels;
  edm::EDGetTokenT<reco::GenParticleCollection> _genParticleToken;
  edm::EDGetTokenT<edm::TriggerResults> _trigResultsToken;
  edm::EDGetTokenT<reco::BeamSpot> _bsToken;
  std::map<unsigned int, edm::EDGetToken> _tokens;

  std::vector<double> _parametersEta;
  std::vector<double> _parametersPhi;
  std::vector<double> _parametersTurnOn;
  std::vector<double> _parametersTurnOnSumEt;
  std::vector<double> _parametersDxy;

  // flag to switch off
  bool _drop_pt2;
  bool _drop_pt3;

  std::map<unsigned int, std::string> _genCut;
  std::map<unsigned int, std::string> _recCut;
  std::map<unsigned int, std::string> _genCut_leading;
  std::map<unsigned int, std::string> _recCut_leading;

  std::map<unsigned int, StringCutObjectSelector<reco::GenParticle> *> _genSelectorMap;
  StringCutObjectSelector<reco::Muon> *_recMuonSelector;
  StringCutObjectSelector<reco::Track> *_recMuonTrkSelector;
  StringCutObjectSelector<reco::Track> *_recTrackSelector;
  StringCutObjectSelector<reco::GsfElectron> *_recElecSelector;
  StringCutObjectSelector<reco::MET> *_recMETSelector;
  StringCutObjectSelector<reco::PFMET> *_recPFMETSelector;
  StringCutObjectSelector<reco::PFMET> *_recPFMHTSelector;
  StringCutObjectSelector<reco::GenMET> *_genMETSelector;
  StringCutObjectSelector<reco::CaloMET> *_recCaloMETSelector;
  StringCutObjectSelector<reco::CaloMET> *_recCaloMHTSelector;
  StringCutObjectSelector<reco::PFTau> *_recPFTauSelector;
  StringCutObjectSelector<reco::Photon> *_recPhotonSelector;
  StringCutObjectSelector<reco::PFJet> *_recPFJetSelector;
  StringCutObjectSelector<reco::CaloJet> *_recCaloJetSelector;

  std::vector<HLTExoticaPlotter> _plotters;

  std::map<std::string, int> _triggerCounter;

  HLTConfigProvider _hltConfig;

  std::map<std::string, MonitorElement *> _elements;
};

#endif