/***************************************************
 Assignment 3   - Spring 2004
 Original Code Copied from: Mikhail Afanasyev
 Course: CS 182
 
 Assignment 3: Backpropagation
 
 Modified by Liam Mac Dermed for gradehelper
 **************************************************/

/**
 * Class for testing backpropagation networks.
 */
public class TesterPart2 {
	
	static int[][] data = {{0,0},{0,1},{1,0},{1,1}};
	static int[][] andTargets = {{0},{0},{0},{1}};
	static int[][] orTargets = {{0},{1},{1},{1}};
	static int[][] sameTargets = {{1},{0},{0},{1}};
	
	static int[][] MULTIdata = {
			{0,0,0,0} , {0,0,0,1} , {0,0,1,0} , {0,0,1,1} ,
			{0,1,0,0} , {0,1,0,1} , {0,1,1,0} , {0,1,1,1} ,
			{1,0,0,0} , {1,0,0,1} , {1,0,1,0} , {1,0,1,1} ,
			{1,1,0,0} , {1,1,0,1} , {1,1,1,0} , {1,1,1,1} };
	
	static int[][] MULTItargets = {
			{0,0} , {0,1} , {1,1} , {1,0} ,
			{0,1} , {1,1} , {1,0} , {0,1} ,
			{0,1} , {1,1} , {1,0} , {1,1} ,
			{0,0} , {1,1} , {1,0} , {0,0} };
	
	static public Net createMulti () {
		return new Net(4, 2, 5, 2, MULTIdata, MULTItargets);
	}
	
	/**
	 * Create a network for learning the AND function.
	 */
	static public Net createAnd () {
		return new Net(2, 1, 1, 1, data, andTargets);
	}
	
	// network for leaning OR function
	static public Net createOr () {
		return new Net(2, 1, 1, 1, data, orTargets);
	}
	
	/**
	 * Create a network for learning the SAME function.
	 */
	static public Net createSame (){
		return new Net(2, 1, 2, 1, data, sameTargets);
	}
	
	/**
	 * Create network for leaning N-input, M-hidden node auto encoder
	 */
	
	static public Net createAutoCoder(int inputs, int hidden) {
		int patterns[][]  = new int[inputs] [inputs];
		for (int i=0; i<inputs; i++) {
			patterns[i] = new int[inputs];
			for (int j=0; j<inputs; j++)
				patterns[i][j] = (i==j)?1:0;
		};
		return new Net(inputs, 1, hidden, inputs, patterns, patterns);
	};
	
	/**
	 * The function should be called as:
	 *    % java TestBPAG ftbl ne lr mom ec
	 * where:
	 *    ftbl is the function to be learned ("AND", "OR, "SAME", etc...)
	 *    ne is the number of epochs
	 *    lr is the learning rate
	 *    mom is the momentum
	 *    ec is the error criterion
	 */
	public static void main(String[] argv) {
		//TestBP test = new TestBP();
		//test.testAnd();  // or test.testSame();
		if(argv.length == 0) {
			Net n;
			
			System.out.print("\n   Attempting to train AND:        \n");
			
			n = createAnd();
			n.train();
			System.out.print(n.toString());
			
			System.out.print("\n   Attempting to train SAME:        \n");
			n = createSame();
			n.train();
			n.toString();
			System.out.print(n.toString());
			
			System.out.print("\n   Attempting to train a  4-2-4 autoencoder:        \n");
			n = createAutoCoder(4, 2);
			n.train();
			n.toString();
			System.out.print(n.toString());
			
			
		} else {
			
			if (argv.length != 5) {
				System.err.println("Invalid argument count");
				System.err.println("The function should be called as:");
				System.err.println("    % java TestBPAG ftbl ne lr mom ec");
				System.err.println(" where:");
				System.err.println("    ftbl is the function to be learned ('AND', 'OR', 'SAME', 'AUTO8', 'MULTI')");
				System.err.println("    ne is the number of epochs");
				System.err.println("    lr is the learning rate");
				System.err.println("    mom is the momentum");
				System.err.println("    ec is the error criterion ");
				return;
			};
			
			Net n;
			
			if (argv[0].equalsIgnoreCase("AND")) {
				n = createAnd();
			} else if (argv[0].equalsIgnoreCase("OR")) {
				n = createOr();
			} else if (argv[0].equalsIgnoreCase("SAME")) {
				n = createSame();
			} else if (argv[0].equalsIgnoreCase("AUTO8")) {
				n = createAutoCoder(8,3);
			} else if (argv[0].equalsIgnoreCase("MULTI")) {
				n = createMulti();
			} else {
				System.err.println("Unknown function name");
				return;
			};
			
			n.setTrainingParameters(Integer.parseInt(argv[1]),
					Double.parseDouble(argv[2]),
					Double.parseDouble(argv[3]),
					Double.parseDouble(argv[4]));
			
			n.train();
			
			System.out.print(">>>  Weights for " + argv[0] + ": \n");
			System.out.print(n.toString());
		}
	}
}