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() {}
  void reset() {
    PUParticles_.clear();
    PUMinBiasEvts_.clear();
  }

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

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

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

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

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

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

  unsigned nParticles() const { return PUParticles_.size(); }

  unsigned nMinBias() const { return PUMinBiasEvts_.size(); }

private:
  std::vector<PUEvent::PUParticle> PUParticles_;
  std::vector<PUEvent::PUMinBiasEvt> PUMinBiasEvts_;
};

#endif