RecHitEnergyFilter

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
// -*- C++ -*-
//
// Package:    RecHitEnergyFilter
// Class:      RecHitEnergyFilter
//
/**\class RecHitEnergyFilter RecHitEnergyFilter.cc JacksonJ/RecHitEnergyFilter/src/RecHitEnergyFilter.cc

 Description: <one line class summary>

 Implementation:
     <Notes on implementation>
*/
//
// Original Author:  James Jackson
//         Created:  Mon Jan 11 09:57:58 CET 2010
// $Id: RecHitEnergyFilter.cc,v 1.1 2010/03/06 22:12:30 malgeri Exp $
//
//

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDFilter.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"

//
// class declaration
//

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

private:
  bool filter(edm::Event&, const edm::EventSetup&) override;

  // RecHit input tags
  edm::InputTag ebRecHitsTag_;
  edm::InputTag eeRecHitsTag_;

  // RecHit handles
  edm::Handle<EcalRecHitCollection> ebRecHitsH_;
  edm::Handle<EcalRecHitCollection> eeRecHitsH_;
  const EcalRecHitCollection* ebRecHits_;
  const EcalRecHitCollection* eeRecHits_;

  // Procesing control
  bool doEb_;
  bool doEe_;
  double ebThresh_;
  double eeThresh_;
};

//
// constructors and destructor
//
RecHitEnergyFilter::RecHitEnergyFilter(const edm::ParameterSet& iConfig) {
  doEb_ = iConfig.getParameter<bool>("DoEB");
  doEe_ = iConfig.getParameter<bool>("DoEE");
  ebRecHitsTag_ = iConfig.getParameter<edm::InputTag>("EBRecHits");
  eeRecHitsTag_ = iConfig.getParameter<edm::InputTag>("EERecHits");
  ebThresh_ = iConfig.getParameter<double>("EBThresh");
  eeThresh_ = iConfig.getParameter<double>("EEThresh");
}

RecHitEnergyFilter::~RecHitEnergyFilter() {}

//
// member functions
//

// ------------ method called on each new Event  ------------
bool RecHitEnergyFilter::filter(edm::Event& evt, const edm::EventSetup& iSetup) {
  if (doEb_) {
    evt.getByLabel(ebRecHitsTag_, ebRecHitsH_);
    ebRecHits_ = ebRecHitsH_.product();
    for (EcalRecHitCollection::const_iterator it = ebRecHits_->begin(); it != ebRecHits_->end(); ++it) {
      double hitE = it->energy();
      if (hitE > ebThresh_) {
        return true;
      }
    }
  }
  if (doEe_) {
    evt.getByLabel(eeRecHitsTag_, eeRecHitsH_);
    eeRecHits_ = eeRecHitsH_.product();
    for (EcalRecHitCollection::const_iterator it = eeRecHits_->begin(); it != eeRecHits_->end(); ++it) {
      double hitE = it->energy();
      if (hitE > eeThresh_) {
        return true;
      }
    }
  }

  return false;
}

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