HBHENoiseFilterResultProducer

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175
// -*- C++ -*-
//
// Package:    HBHENoiseFilterResultProducer
// Class:      HBHENoiseFilterResultProducer
//
/**\class HBHENoiseFilterResultProducer

 Description: Produces the result from the HBENoiseFilter

 Implementation:
              Use the HcalNoiseSummary to make cuts on an event-by-event basis
*/
//
// Original Author:  John Paul Chou (Brown)
//
//

#include <iostream>

// system include files
#include <memory>
#include <map>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Utilities/interface/Exception.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/METReco/interface/HcalNoiseSummary.h"
#include "DataFormats/JetReco/interface/PFJetCollection.h"

//
// class declaration
//

class HBHENoiseFilterResultProducer : public edm::stream::EDProducer<> {
public:
  explicit HBHENoiseFilterResultProducer(const edm::ParameterSet&);
  ~HBHENoiseFilterResultProducer() override;

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

  // ----------member data ---------------------------

  // parameters
  edm::EDGetTokenT<HcalNoiseSummary> noisetoken_;
  int minHPDHits_;
  int minHPDNoOtherHits_;
  int minZeros_;

  bool IgnoreTS4TS5ifJetInLowBVRegion_;
  std::string defaultDecision_;

  int minNumIsolatedNoiseChannels_;
  double minIsolatedNoiseSumE_;
  double minIsolatedNoiseSumEt_;

  edm::EDGetTokenT<unsigned int> bunchSpacing_;
  bool useBunchSpacingProducer_;

  // other members
  std::map<std::string, bool> decisionMap_;
};

//
// constructors and destructor
//

HBHENoiseFilterResultProducer::HBHENoiseFilterResultProducer(const edm::ParameterSet& iConfig) {
  //now do what ever initialization is needed
  noisetoken_ = consumes<HcalNoiseSummary>(iConfig.getParameter<edm::InputTag>("noiselabel"));
  minHPDHits_ = iConfig.getParameter<int>("minHPDHits");
  minHPDNoOtherHits_ = iConfig.getParameter<int>("minHPDNoOtherHits");
  minZeros_ = iConfig.getParameter<int>("minZeros");
  IgnoreTS4TS5ifJetInLowBVRegion_ = iConfig.getParameter<bool>("IgnoreTS4TS5ifJetInLowBVRegion");
  defaultDecision_ = iConfig.getParameter<std::string>("defaultDecision");

  // parameters related to isolation filter
  minNumIsolatedNoiseChannels_ = iConfig.getParameter<int>("minNumIsolatedNoiseChannels");
  minIsolatedNoiseSumE_ = iConfig.getParameter<double>("minIsolatedNoiseSumE");
  minIsolatedNoiseSumEt_ = iConfig.getParameter<double>("minIsolatedNoiseSumEt");

  // parameters needed for bunch-spacing check
  bunchSpacing_ = consumes<unsigned int>(edm::InputTag("bunchSpacingProducer"));
  useBunchSpacingProducer_ = iConfig.getParameter<bool>("useBunchSpacingProducer");

  produces<bool>("HBHENoiseFilterResult");
  produces<bool>("HBHENoiseFilterResultRun1");
  produces<bool>("HBHENoiseFilterResultRun2Loose");
  produces<bool>("HBHENoiseFilterResultRun2Tight");
  produces<bool>("HBHEIsoNoiseFilterResult");
}

HBHENoiseFilterResultProducer::~HBHENoiseFilterResultProducer() {}

//
// member functions
//

// ------------ method called on each new Event  ------------
void HBHENoiseFilterResultProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  using namespace edm;

  // get the Noise summary object
  edm::Handle<HcalNoiseSummary> summary_h;
  iEvent.getByToken(noisetoken_, summary_h);
  if (!summary_h.isValid()) {
    throw edm::Exception(edm::errors::ProductNotFound) << " could not find HcalNoiseSummary.\n";
    return;
  }
  const HcalNoiseSummary& summary(*summary_h);

  bool goodJetFoundInLowBVRegion = false;
  if (IgnoreTS4TS5ifJetInLowBVRegion_)
    goodJetFoundInLowBVRegion = summary.goodJetFoundInLowBVRegion();

  const bool failCommon = summary.maxHPDHits() >= minHPDHits_ || summary.maxHPDNoOtherHits() >= minHPDNoOtherHits_ ||
                          summary.maxZeros() >= minZeros_;

  const bool failRun1 = failCommon || (summary.HasBadRBXTS4TS5() && !goodJetFoundInLowBVRegion);
  decisionMap_["HBHENoiseFilterResultRun1"] = failRun1;

  const bool failRun2Loose = failCommon || (summary.HasBadRBXRechitR45Loose() && !goodJetFoundInLowBVRegion);
  decisionMap_["HBHENoiseFilterResultRun2Loose"] = failRun2Loose;

  const bool failRun2Tight = failCommon || (summary.HasBadRBXRechitR45Tight() && !goodJetFoundInLowBVRegion);
  decisionMap_["HBHENoiseFilterResultRun2Tight"] = failRun2Tight;

  // Write out the standard flags
  std::unique_ptr<bool> pOut;
  for (std::map<std::string, bool>::const_iterator it = decisionMap_.begin(); it != decisionMap_.end(); ++it) {
    pOut = std::make_unique<bool>(!it->second);
    iEvent.put(std::move(pOut), it->first);
  }

  // Overwrite defaultDecision_ dynamically based on bunchSpacingProducer
  if (useBunchSpacingProducer_) {
    edm::Handle<unsigned int> bunchSpacingH;
    iEvent.getByToken(bunchSpacing_, bunchSpacingH);
    unsigned int bunchspacing = 0;
    if (bunchSpacingH.isValid()) {
      bunchspacing = *bunchSpacingH;
      if (bunchspacing == 50) {
        defaultDecision_ = "HBHENoiseFilterResultRun1";
      } else {
        defaultDecision_ = "HBHENoiseFilterResultRun2Loose";
      }
    }
  }

  // Write out the default flag
  std::map<std::string, bool>::const_iterator it = decisionMap_.find(defaultDecision_);
  if (it == decisionMap_.end())
    throw cms::Exception("Invalid HBHENoiseFilterResultProducer parameter \"defaultDecision\"");
  pOut = std::make_unique<bool>(!it->second);
  iEvent.put(std::move(pOut), "HBHENoiseFilterResult");

  // Check isolation requirements
  const bool failIsolation = summary.numIsolatedNoiseChannels() >= minNumIsolatedNoiseChannels_ ||
                             summary.isolatedNoiseSumE() >= minIsolatedNoiseSumE_ ||
                             summary.isolatedNoiseSumEt() >= minIsolatedNoiseSumEt_;
  pOut = std::make_unique<bool>(!failIsolation);
  iEvent.put(std::move(pOut), "HBHEIsoNoiseFilterResult");

  return;
}

//define this as a plug-in
DEFINE_FWK_MODULE(HBHENoiseFilterResultProducer);