Muon/test/MuEnrichType1Filter.h

0001 #ifndef MuEnrichFltr_h
0002 #define MuEnrichFltr_h
0003 //
0004 // class declaration
0005 //
0006 #include "FWCore/Framework/interface/global/EDFilter.h"
0007 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0008 #include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
0009
0010 #include <atomic>
0011
0012 class MuEnrichType1Filter : public edm::global::EDFilter<> {
0013 public:
0014   explicit MuEnrichType1Filter(const edm::ParameterSet&);
0015   ~MuEnrichType1Filter();
0016
0017 private:
0018   void beginJob() override;
0019   bool filter(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
0020   void endJob() override;
0021   edm::EDGetTokenT<edm::HepMCProduct> theGenToken;
0022   mutable std::atomic<int> nrejected;
0023   mutable std::atomic<int> naccepted;
0024   int type;
0025   // ----------member data ---------------------------
0026 };
0027
0028 #endif