BoxTrialCase.java

/*
 * Encog(tm) Java Examples v3.4
 * http://www.heatonresearch.com/encog/
 * https://github.com/encog/encog-java-examples
 *
 * Copyright 2008-2016 Heaton Research, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *   
 * For more information on Heaton Research copyrights, licenses 
 * and trademarks visit:
 * http://www.heatonresearch.com/copyright
 */
package org.encog.examples.neural.neat.boxes;

import java.util.Random;

import org.encog.mathutil.IntPair;

/**
 * One case in the boxes score.  Position the boxes at random locations.
 */
public class BoxTrialCase {

	public static final int BASE_RESOLUTION = 11;
	public static final int BOUNDS = BASE_RESOLUTION - 1;

	private IntPair smallBoxTopLeft;
	private IntPair largeBoxTopLeft;

	private Random rnd;

	public BoxTrialCase(Random theRnd) {
		this.rnd = theRnd;
	}

	public IntPair initTestCase(int largeBoxRelativePos) {
		IntPair[] loc = generateRandomTestCase(largeBoxRelativePos);
		smallBoxTopLeft = loc[0];
		largeBoxTopLeft = (IntPair)loc[1].clone();
		largeBoxTopLeft.add(-1);
		return loc[1];
	}

	public double getPixel(double x, double y) {
		int pixelX = (int) (((x + 1.0) * BoxTrialCase.BASE_RESOLUTION) / 2.0);
		int pixelY = (int) (((y + 1.0) * BoxTrialCase.BASE_RESOLUTION) / 2.0);

		if (smallBoxTopLeft.getX() == pixelX
				&& smallBoxTopLeft.getY() == pixelY) {
			return 1.0;
		}

		int deltaX = pixelX - largeBoxTopLeft.getX();
		int deltaY = pixelY - largeBoxTopLeft.getY();
		return (deltaX > -1 && deltaX < 3 && deltaY > -1 && deltaY < 3) ? 1.0
				: 0.0;
	}

	private IntPair[] generateRandomTestCase(int largeBoxRelativePos) {
		IntPair smallBoxPos = new IntPair(rnd.nextInt(BoxTrialCase.BASE_RESOLUTION),
				rnd.nextInt(BoxTrialCase.BASE_RESOLUTION));

		IntPair largeBoxPos = (IntPair) smallBoxPos.clone();
		switch (largeBoxRelativePos) {
		case 0:
			largeBoxPos.addX(5);
			break;
		case 1:
			largeBoxPos.addY(5);
			break;
		case 2:
			if (rnd.nextBoolean()) {
				largeBoxPos.add(3, 4);
			} else {
				largeBoxPos.add(4, 3);
			}
			break;
		}

		if (largeBoxPos.getX() > BoxTrialCase.BOUNDS) {
			largeBoxPos.addX(-BoxTrialCase.BASE_RESOLUTION);

			if (0 == largeBoxPos.getX()) {
				largeBoxPos.add(1);
			}
		} else if (BoxTrialCase.BOUNDS == largeBoxPos.getX()) {
			largeBoxPos.addX(-1);
		} else if (largeBoxPos.getX() == 0) {
			largeBoxPos.addX(1);
		}

		if (largeBoxPos.getY() > BoxTrialCase.BOUNDS) {
			largeBoxPos.addY(-BoxTrialCase.BASE_RESOLUTION);

			if (0 == largeBoxPos.getY()) {
				largeBoxPos.addY(1);
			}
		} else if (BoxTrialCase.BOUNDS == largeBoxPos.getY()) {
			largeBoxPos.addY(-1);
		} else if (0 == largeBoxPos.getY()) {
			largeBoxPos.addY(1);
		}
		return new IntPair[] { smallBoxPos, largeBoxPos };
	}
}