|
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
|
// -*- C++ -*-
//
// Package: SecondaryVertexFilter
// Class: SecondaryVertexFilter
//
/**\class SecondaryVertexFilter SecondaryVertexFilter.cc DPGAnalysis/SecondaryVertexFilter/src/SecondaryVertexFilter.cc
Description: <one line class summary>
Implementation:
<Notes on implementation>
*/
//
// Original Author: Andrea RIZZI
// Created: Mon Dec 7 18:02:10 CET 2009
// $Id: SecondaryVertexFilter.cc,v 1.2 2010/02/17 23:39:10 wmtan 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 "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/BTauReco/interface/SecondaryVertexTagInfo.h"
//
// class declaration
//
class SecondaryVertexFilter : public edm::stream::EDFilter<> {
public:
explicit SecondaryVertexFilter(const edm::ParameterSet&);
~SecondaryVertexFilter() override;
private:
bool filter(edm::Event&, const edm::EventSetup&) override;
edm::InputTag vertexSrc;
unsigned int minNumTracks;
double maxAbsZ;
double maxd0;
// ----------member data ---------------------------
};
SecondaryVertexFilter::SecondaryVertexFilter(const edm::ParameterSet& iConfig) {
vertexSrc = iConfig.getParameter<edm::InputTag>("vertexCollection");
minNumTracks = iConfig.getParameter<unsigned int>("minimumNumberOfTracks");
maxAbsZ = iConfig.getParameter<double>("maxAbsZ");
maxd0 = iConfig.getParameter<double>("maxd0");
}
SecondaryVertexFilter::~SecondaryVertexFilter() {}
bool SecondaryVertexFilter::filter(edm::Event& iEvent, const edm::EventSetup& iSetup) {
bool result = false;
edm::Handle<reco::SecondaryVertexTagInfoCollection> pvHandle;
iEvent.getByLabel(vertexSrc, pvHandle);
const reco::SecondaryVertexTagInfoCollection& vertices = *pvHandle.product();
for (reco::SecondaryVertexTagInfoCollection::const_iterator it = vertices.begin(); it != vertices.end(); ++it) {
if (it->nVertices() > 0)
result = true;
// if(it->tracksSize() > minNumTracks &&
// ( (maxAbsZ <=0 ) || fabs(it->z()) <= maxAbsZ ) &&
// ( (maxd0 <=0 ) || fabs(it->position().rho()) <= maxd0 )
//) result = true;
}
return result;
}
//define this as a plug-in
DEFINE_FWK_MODULE(SecondaryVertexFilter);
|