GenFilters/plugins/PythiaProbeFilter.h

0001 #ifndef PYTHIAPROBEFILTER_h
0002 #define PYTHIAPROBEFILTER_h
0003 // -*- C++ -*-
0004 //
0005 // Package:    PythiaProbeFilter
0006 // Class:      PythiaProbeFilter
0007 //
0008 /**\class PythiaProbeFilter PythiaProbeFilter.cc
0009
0010  Description: Filter to exclude selected particles from passing pT,eta cuts etc. Usefull when we are interested in a decay that its daughters should not pass any cuts, but another particle of the same flavour should e.g B+B- production with B+->K+mumu forcing (probe side) and we want mu to come from B- (tag side)
0011
0012  Implementation:
0013      <Notes on implementation>
0014 */
0015 //
0016 // Original Author:  Georgios Karathanasis
0017 //         Created:  Mar 14 2019
0018 //
0019 //
0020
0021 // system include files
0022 #include <memory>
0023
0024 // user include files
0025 #include "FWCore/Framework/interface/Frameworkfwd.h"
0026 #include "FWCore/Framework/interface/global/EDFilter.h"
0027
0028 #include "FWCore/Framework/interface/Event.h"
0029 #include "FWCore/Framework/interface/MakerMacros.h"
0030
0031 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0032
0033 #include "Pythia8/Pythia.h"
0034
0035 //
0036 // class decleration
0037 //
0038 namespace edm {
0039   class HepMCProduct;
0040 }
0041
0042 class PythiaProbeFilter : public edm::global::EDFilter<> {
0043 public:
0044   explicit PythiaProbeFilter(const edm::ParameterSet&);
0045   ~PythiaProbeFilter() override;
0046
0047   bool filter(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
0048   bool AlreadyExcludedCheck(std::vector<unsigned int> excludedList, unsigned int current_part) const;
0049
0050 private:
0051   // ----------memeber function----------------------
0052
0053   // ----------member data ---------------------------
0054
0055   const edm::EDGetTokenT<edm::HepMCProduct> token_;
0056   std::vector<int> exclsisIDs;
0057   std::vector<int> exclauntIDs;
0058   const int particleID;
0059   const int MomID;
0060   const int GrandMomID;
0061   const bool chargeconju;
0062   const int nsisters;
0063   const int naunts;
0064   const double minptcut;
0065   const double maxptcut;
0066   const double minetacut;
0067   const double maxetacut;
0068   const bool countQEDCorPhotons;
0069   bool identicalParticle;
0070   std::unique_ptr<Pythia8::Pythia> fLookupGen;  // this instance is for accessing particleData information
0071 };
0072 #endif