test/analyserhepmc/JetInputHepMC.h

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