codes.derive.foldem.tool.EvaluationBenchmarker Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of foldem Show documentation

A Texas Hold 'em Poker Analysis Framework for Java.

The newest version!

package codes.derive.foldem.tool;

import java.util.concurrent.Callable;
import java.util.concurrent.TimeUnit;

import codes.derive.foldem.Hand;
import codes.derive.foldem.board.Board;
import codes.derive.foldem.board.Boards;
import codes.derive.foldem.eval.Evaluator;

/**
 * A type that can be used to measure the performance of an evaluation in
 * evaluations/second.
 */
public class EvaluationBenchmarker implements Callable {

	/* The evaluator to use. */
	private final Evaluator evaluator;

	/* The number of evaluations to perform */
	private final int runs;

	/**
	 * Constructs a new {@link EvaluationBenchmarker} for the specified
	 * {@link Evaluator}.
	 * 
	 * @param evaluator
	 *            The evaluator to use.
	 * @param runs
	 *            The number of evaluations to perform for a measurement.
	 */
	public EvaluationBenchmarker(Evaluator evaluator, int runs) {
		this.evaluator = evaluator;
		this.runs = runs;
	}

	@Override
	public Integer call() {

		// Create a hand and board to run our evaluations on.
		Hand hand = new Hand("AcAs");
		Board board = Boards.board("AdAhKsKcKd");

		// Record the start time.
		long start = System.nanoTime();

		// Begin running evaluations.
		for (int i = 0; i < runs; i++) {
			evaluator.rank(hand, board);
		}

		// Measure how much time passed.
		long elapsed = TimeUnit.NANOSECONDS.toSeconds(System.nanoTime() - start);

		// Return the number of evaluations per second.
		return (int) (runs / elapsed);
	}

}