package org.paneris.bibliomania.loadtest;
   
   import java.io.Serializable;
   import java.util.Enumeration;
   import java.util.Vector;
   
   class DiscreteDistribution implements Serializable {
     private static final long serialVersionUID = -3646912334575385881L;
   
    private Vector<Object> elements = new Vector<Object>();
    private Vector<Object> cumulative = new Vector<Object>();
  
    public DiscreteDistribution() {}
  
    public void add(Object o, int n) {
      for (int i = 0; i < n; ++i)
        cumulative.addElement(o);
      elements.addElement(o);
    }
   
    public void add(Object o) {
      add(o, 1);
    }
  
    public Object sample() {
      return cumulative.elementAt(LoadTest.random.nextInt(cumulative.size()));
    }
  
    public Enumeration<Object> elements() {
      return elements.elements();
    }
  }