Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-02-14 13:07:27

0001 // JetInputHepMC declaration
0002 
0003 #ifndef JetInputHepMC_h
0004 #define JetInputHepMC_h
0005 
0006 #include <vector>
0007 
0008 #include <HepMC/GenEvent.h>
0009 #include <HepMC/GenParticle.h>
0010 #include <HepMC/GenVertex.h>
0011 
0012 #include <iostream>
0013 
0014 class JetInputHepMC {
0015 public:
0016   typedef std::vector<bool> ParticleBitmap;
0017   typedef std::vector<const HepMC::GenParticle *> ParticleVector;
0018   JetInputHepMC();
0019   ~JetInputHepMC();
0020 
0021   ParticleVector operator()(const HepMC::GenEvent *event) const;
0022 
0023   double getPtMin() const { return ptMin; }
0024   void setPtMin(double ptMin) { this->ptMin = ptMin; }
0025 
0026   void setIgnoredParticles(const std::vector<unsigned int> &particleIDs);
0027 
0028   bool isIgnored(int pdgId) const;
0029 
0030 private:
0031   std::vector<unsigned int> ignoreParticleIDs;
0032   double ptMin;
0033 };
0034 
0035 #endif  // JetInputHepMC_h