import java.util.Random;

public class TestIntMatrixMultiplication {

    public static void main (String...args) throws Exception {
        final int n = args.length > 0 ? Integer.parseInt(args[0]) : 128;
        final int seed = args.length > 1 ? Integer.parseInt(args[1]) : 128;
        final Random random = new Random(seed);

        int[][] a, b, c;

        a = new int[n][n];
        b = new int[n][n];
        c = new int[n][n];

        for(int i=0; i<n; i++) {
            for(int j=0; j<n; j++) {
                a[i][j] = random.nextInt(16);
                b[i][j] = random.nextInt(16);
            }
        }

        IntMatrixMultiplication classical = new ClassicMatrixMultiplication();
        IntMatrixMultiplication strassen1 = new Strassen1MatrixMultiplication();
        IntMatrixMultiplication strassen2 = new Strassen2MatrixMultiplication();

        System.out.println("n = " + n);

        if (a.length < 64) {
            System.out.println("A");
            dumpMatrix(a);
            System.out.println("B");
            dumpMatrix(b);
            System.out.println("classic");
            classical.mult(c, a, b);
            dumpMatrix(c);
            System.out.println("strassen");
            strassen2.mult(c, a, b);
            dumpMatrix(c);

            return;
        }

        for (int i = 0; i < 7; ++i) {
            timeMultiplies(classical, a, b, c);
            if (n <= 256)
                timeMultiplies(strassen1, a, b, c);
            timeMultiplies(strassen2, a, b, c);
        }
    }

    static void timeMultiplies (IntMatrixMultiplication algorithm, int[][] a, int[][] b, int[][] c) {
        final long start = System.currentTimeMillis();
        algorithm.mult(c, a, b);
        final long finish = System.currentTimeMillis();
        System.out.println(algorithm.getName() + " took " + (finish - start) + "ms");
    }

    static void dumpMatrix (int[][] m) {
        for (int[] row : m) {
            System.out.print("[\t");
            for (int val : row) {
                System.out.print(val);
                System.out.print('\t');
            }
            System.out.println(']');
        }
    }
}