|
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
|
// -*- C++ -*-
//
// Package: HTXSFilter
// Class: HTXSFilter
//
/**\class HTXSFilter HTXSFilter.cc user/HTXSFilter/plugins/HTXSFilter.cc
Description: [one line class summary]
Implementation:
[Notes on implementation]
*/
//
// Original Author: Janek Bechtel
// Created: Fri, 10 May 2019 14:30:15 GMT
//
//
// system include files
#include <memory>
#include "GeneratorInterface/GenFilters/plugins/HTXSFilter.h"
// 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 "FWCore/Utilities/interface/StreamID.h"
#include "SimDataFormats/HTXS/interface/HiggsTemplateCrossSections.h"
HTXSFilter::HTXSFilter(const edm::ParameterSet& iConfig)
: token_(consumes<HTXS::HiggsClassification>(edm::InputTag("rivetProducerHTXS", "HiggsClassification"))),
htxs_flags(iConfig.getUntrackedParameter("htxs_flags", std::vector<int>())) {}
HTXSFilter::~HTXSFilter() {
// do anything here that needs to be done at destruction time
// (e.g. close files, deallocate resources etc.)
}
//
// member functions
//
// ------------ method called on each new Event ------------
bool HTXSFilter::filter(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const {
using namespace edm;
Handle<HTXS::HiggsClassification> cat;
iEvent.getByToken(token_, cat);
if (htxs_flags.empty()) {
edm::LogInfo("HTXSFilter") << "Selection of HTXS flags to filter is empty. Filtering will not be applied."
<< std::endl;
return true;
}
if (std::find(htxs_flags.begin(), htxs_flags.end(), cat->stage1_1_cat_pTjet30GeV) != htxs_flags.end()) {
return true;
} else {
return false;
}
}
//define this as a plug-in
DEFINE_FWK_MODULE(HTXSFilter);
|