PUEvent.h

Go to the documentation of this file.

#ifndef FastSimulation_PileUpProducer_PUEvent_h
#define FastSimulation_PileUpProducer_PUEvent_h

#include <vector>

class PUEvent {
  
  
 public:
  PUEvent() : NParticles_(0), NMinBias_(0) {}
  virtual ~PUEvent() {}
  void reset() {
    PUParticles_.clear();
    PUMinBiasEvts_.clear();
    NParticles_ = 0;
    NMinBias_ = 0;  
  }

  class PUParticle {
  public:
    PUParticle() : px(0.), py(0.), pz(0.),mass(0.),id(0) {}
    virtual ~PUParticle() {}
    float px; 
    float py;
    float pz;
    float mass;
    int id;
  };   


  class PUMinBiasEvt {
  public:
    PUMinBiasEvt() : first(0), size(0) {}
    virtual ~PUMinBiasEvt() {}
    unsigned first; 
    unsigned size;
  };   


  void addPUParticle( const PUParticle& ptc ) {
    PUParticles_.push_back(ptc);
    ++NParticles_;
  }

  void addPUMinBiasEvt( const PUMinBiasEvt& idx ) {
    PUMinBiasEvts_.push_back(idx);
    ++NMinBias_;
  }

  const std::vector<PUEvent::PUParticle>& thePUParticles() 
    {return PUParticles_;}

  const std::vector<PUEvent::PUMinBiasEvt>& thePUMinBiasEvts() 
    {return PUMinBiasEvts_;}

  const unsigned nParticles() const { return NParticles_; }

  const unsigned nMinBias() const { return NMinBias_; }

 private:

  std::vector<PUEvent::PUParticle> PUParticles_;
  std::vector<PUEvent::PUMinBiasEvt> PUMinBiasEvts_;
  unsigned NParticles_;
  unsigned NMinBias_;
  
};

#endif