com.carrotsearch.randomizedtesting.generators.RandomPicks Maven / Gradle / Ivy

Go to download
package com.carrotsearch.randomizedtesting.generators;

import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Random;

/**
 * Random selections of objects.
 */
public final class RandomPicks {
  public static byte randomFrom(Random r, byte [] array) { checkZeroLength(array.length); return array[r.nextInt(array.length)]; }
  public static short randomFrom(Random r, short [] array) { checkZeroLength(array.length); return array[r.nextInt(array.length)]; }
  public static int randomFrom(Random r, int [] array) { checkZeroLength(array.length); return array[r.nextInt(array.length)]; }
  public static char randomFrom(Random r, char [] array) { checkZeroLength(array.length); return array[r.nextInt(array.length)]; }
  public static float randomFrom(Random r, float [] array) { checkZeroLength(array.length); return array[r.nextInt(array.length)]; }
  public static long randomFrom(Random r, long [] array) { checkZeroLength(array.length); return array[r.nextInt(array.length)]; }
  public static double randomFrom(Random r, double [] array) { checkZeroLength(array.length); return array[r.nextInt(array.length)]; }

  private static void checkZeroLength(int length) {
    if (length == 0) {
      throw new IllegalArgumentException("Can't pick a random object from an empty array.");
    }
  }

  /**
   * Pick a random object from the given array.
   */
  public static  T randomFrom(Random r, T [] array) {
    checkZeroLength(array.length);
    return array[r.nextInt(array.length)];
  }

  /**
   * Pick a random object from the given list.
   */
  public static  T randomFrom(Random r, List list) {
    if (list.size() == 0) {
      throw new IllegalArgumentException("Can't pick a random object from an empty list.");
    }
    return list.get(r.nextInt(list.size()));
  }

  /**
   * Pick a random object from the collection. Requires linear scanning.
   */
  public static  T randomFrom(Random r, Collection collection) {
    final int size = collection.size();
    if (size == 0) {
      throw new IllegalArgumentException("Can't pick a random object from an empty collection.");
    }
    int pick = r.nextInt(size);
    T value = null;
    for (Iterator i = collection.iterator();; pick--) {
      value = i.next();
      if (pick == 0) {
        break;
      }
    }
    return value;
  }
}