TokenUtil.java

/*
 * Copyright (c) 2014 tabletoptool.com team.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Public License v3.0
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/gpl.html
 * 
 * Contributors:
 *     rptools.com team - initial implementation
 *     tabletoptool.com team - further development
 */
package com.t3.util;

import java.awt.Dimension;
import java.awt.Image;
import java.awt.image.BufferedImage;
import java.awt.image.ImageObserver;
import java.awt.image.PixelGrabber;

import org.apache.log4j.Logger;

import com.t3.model.Token;

public class TokenUtil {
	private static final Logger log = Logger.getLogger(TokenUtil.class);

	public static int getIndexNearestTo(int[] array, int value) {
		int delta = -1;
		int closest = -1;
		for (int i = 0; i < array.length; i++) {
			int currDelta = Math.abs(value - array[i]);
			if (delta < 0 || currDelta < delta) {
				closest = i;
				delta = currDelta;
			}
		}
		return closest;
	}

	public static Token.TokenShape guessTokenType(Image image) {
		if (image instanceof BufferedImage) {
			return guessTokenType((BufferedImage) image);
		}
		int pixelCount = 0;
		int width = image.getWidth(null);
		int height = image.getHeight(null);
		int[] pixelArray = new int[width * height];
		PixelGrabber pg = new PixelGrabber(image, 0, 0, width, height, pixelArray, 0, width);
		try {
			pg.grabPixels();
		} catch (InterruptedException e) {
			String msg = "interrupted waiting for pixels!";
			log.warn(msg);
			return Token.TokenShape.TOP_DOWN;
		}
		if ((pg.getStatus() & ImageObserver.ABORT) != 0) {
			String msg = "image fetch aborted or errored";
			log.warn(msg);
			return Token.TokenShape.TOP_DOWN;
		}
		for (int y = 0; y < height; y++) {
			for (int x = 0; x < width; x++) {
				// Get the next pixel
				int pixel = pixelArray[y * width + x];
				if ((pixel & 0xff000000) != 0) {
					pixelCount++;
				}
			}
		}
		return guessTokenType(new Dimension(image.getWidth(null), image.getHeight(null)), pixelCount);
	}

	public static Token.TokenShape guessTokenType(BufferedImage image) {
		int pixelCount = 0;
		for (int row = 0; row < image.getHeight(); row++) {
			for (int col = 0; col < image.getWidth(); col++) {
				int pixel = image.getRGB(col, row);
				if ((pixel & 0xff000000) != 0) {
					pixelCount++;
				}
			}
		}
		return guessTokenType(new Dimension(image.getWidth(), image.getHeight()), pixelCount);
	}

	private static Token.TokenShape guessTokenType(Dimension size, int pixelCount) {
		double circlePixelCount = (int) (Math.PI * (size.width / 2) * (size.height / 2));
		double squarePixelCount = size.width * size.height;
		double topDownPixelCount = circlePixelCount * 3 / 4; // arbitrary
		double circleResult = Math.abs(1 - (pixelCount / circlePixelCount));
		double squareResult = Math.abs(1 - (pixelCount / squarePixelCount));
		double topDownResult = Math.abs(1 - (pixelCount / topDownPixelCount));

		if (circleResult < squareResult && circleResult < topDownResult) {
			return Token.TokenShape.CIRCLE;
		}
		if (squareResult < circleResult && squareResult < topDownResult) {
			return Token.TokenShape.SQUARE;
		}
		return Token.TokenShape.TOP_DOWN;
	}
}