Skims/src/SecondaryVertexFilter.cc

0001 // -*- C++ -*-
0002 //
0003 // Package:    SecondaryVertexFilter
0004 // Class:      SecondaryVertexFilter
0005 //
0006 /**\class SecondaryVertexFilter SecondaryVertexFilter.cc DPGAnalysis/SecondaryVertexFilter/src/SecondaryVertexFilter.cc
0007
0008  Description: <one line class summary>
0009
0010  Implementation:
0011      <Notes on implementation>
0012 */
0013 //
0014 // Original Author:  Andrea RIZZI
0015 //         Created:  Mon Dec  7 18:02:10 CET 2009
0016 // $Id: SecondaryVertexFilter.cc,v 1.2 2010/02/17 23:39:10 wmtan Exp $
0017 //
0018 //
0019
0020 // system include files
0021 #include <memory>
0022
0023 // user include files
0024 #include "FWCore/Framework/interface/Frameworkfwd.h"
0025 #include "FWCore/Framework/interface/stream/EDFilter.h"
0026
0027 #include "FWCore/Framework/interface/Event.h"
0028 #include "FWCore/Framework/interface/MakerMacros.h"
0029
0030 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0031 #include "FWCore/Utilities/interface/InputTag.h"
0032 #include "DataFormats/VertexReco/interface/Vertex.h"
0033 #include "DataFormats/VertexReco/interface/VertexFwd.h"
0034 #include "DataFormats/BTauReco/interface/SecondaryVertexTagInfo.h"
0035 //
0036 // class declaration
0037 //
0038
0039 class SecondaryVertexFilter : public edm::stream::EDFilter<> {
0040 public:
0041   explicit SecondaryVertexFilter(const edm::ParameterSet&);
0042   ~SecondaryVertexFilter() override;
0043
0044 private:
0045   bool filter(edm::Event&, const edm::EventSetup&) override;
0046   edm::InputTag vertexSrc;
0047   unsigned int minNumTracks;
0048   double maxAbsZ;
0049   double maxd0;
0050   // ----------member data ---------------------------
0051 };
0052
0053 SecondaryVertexFilter::SecondaryVertexFilter(const edm::ParameterSet& iConfig) {
0054   vertexSrc = iConfig.getParameter<edm::InputTag>("vertexCollection");
0055   minNumTracks = iConfig.getParameter<unsigned int>("minimumNumberOfTracks");
0056   maxAbsZ = iConfig.getParameter<double>("maxAbsZ");
0057   maxd0 = iConfig.getParameter<double>("maxd0");
0058 }
0059
0060 SecondaryVertexFilter::~SecondaryVertexFilter() {}
0061
0062 bool SecondaryVertexFilter::filter(edm::Event& iEvent, const edm::EventSetup& iSetup) {
0063   bool result = false;
0064   edm::Handle<reco::SecondaryVertexTagInfoCollection> pvHandle;
0065   iEvent.getByLabel(vertexSrc, pvHandle);
0066   const reco::SecondaryVertexTagInfoCollection& vertices = *pvHandle.product();
0067   for (reco::SecondaryVertexTagInfoCollection::const_iterator it = vertices.begin(); it != vertices.end(); ++it) {
0068     if (it->nVertices() > 0)
0069       result = true;
0070     //   if(it->tracksSize() > minNumTracks &&
0071     //      ( (maxAbsZ <=0 ) || fabs(it->z()) <= maxAbsZ ) &&
0072     //    ( (maxd0 <=0 ) || fabs(it->position().rho()) <= maxd0 )
0073     //) result = true;
0074   }
0075
0076   return result;
0077 }
0078
0079 //define this as a plug-in
0080 DEFINE_FWK_MODULE(SecondaryVertexFilter);