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();

   }

 }