/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

 * CA 95054 USA or visit www.sun.com if you need additional information or

 * have any questions.

 */

/*

 *

 */

package bench.serial;

import bench.ConfigFormatException;

import bench.Harness;

import bench.HtmlReporter;

import bench.Reporter;

import bench.TextReporter;

import bench.XmlReporter;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.io.InputStream;

import java.io.IOException;

import java.io.OutputStream;

import java.io.PrintStream;

import java.util.Timer;

import java.util.TimerTask;

/**

 * Object serialization benchmark mainline.

 */

public class Main {

    static final String CONFFILE = "/bench/serial/config";

    static final String VERSION = "1.3";

    static final int TEXT = 0;

    static final int HTML = 1;

    static final int XML = 2;

    static boolean verbose;

    static boolean list;

    static boolean exitOnTimer;

    static int testDurationSeconds;

    static volatile boolean exitRequested;

    static Timer timer;

    static int format = TEXT;

    static InputStream confstr;

    static OutputStream repstr;

    static Harness harness;

    static Reporter reporter;

    /**

     * Print help message.

     */

    static void usage() {

        PrintStream p = System.err;

        p.println("\nUsage: java -jar serialbench.jar [-options]");

        p.println("\nwhere options are:");

        p.println("  -h              print this message");

        p.println("  -v              verbose mode");

        p.println("  -l              list configuration file");

        p.println("  -t <num hours>  repeat benchmarks for specified number of hours");

        p.println("  -o <file>       specify output file");

        p.println("  -c <file>       specify (non-default) configuration file");

        p.println("  -html           format output as html (default is text)");

        p.println("  -xml            format output as xml");

    }

    /**

     * Print error message and exit.

     */

    static void die(String mesg) {

        System.err.println(mesg);

        System.exit(1);

    }

    /**

     * Mainline parses command line, then hands off to benchmark harness.

     */

    public static void main(String[] args) {

        parseArgs(args);

        setupStreams();

        if (list) {

            listConfig();

        } else {

            setupHarness();

            setupReporter();

            if (exitOnTimer) {

                setupTimer(testDurationSeconds);

                while (true) {

                    runBenchmarks();

                    if (exitRequested) {

                        System.exit(0);

                    }

                }

            } else {

                runBenchmarks();

                System.exit(0);

            }

        }

    }

    /**

     * Parse command-line arguments.

     */

    static void parseArgs(String[] args) {

        for (int i = 0; i < args.length; i++) {

            if (args[i].equals("-h")) {

                usage();

                System.exit(0);

            } else if (args[i].equals("-v")) {

                verbose = true;

            } else if (args[i].equals("-l")) {

                list = true;

            } else if (args[i].equals("-t")) {

                if (++i >= args.length)

                    die("Error: no timeout value specified");

                try {

                    exitOnTimer = true;

                    testDurationSeconds = Integer.parseInt(args[i]) * 3600;

                } catch (Exception e) {

                    die("Error: unable to determine timeout value");

                }

            } else if (args[i].equals("-o")) {

                if (++i >= args.length)

                    die("Error: no output file specified");

                try {

                    repstr = new FileOutputStream(args[i]);

                } catch (IOException e) {

                    die("Error: unable to open \"" + args[i] + "\"");

                }

            } else if (args[i].equals("-c")) {

                if (++i >= args.length)

                    die("Error: no config file specified");

                try {

                    confstr = new FileInputStream(args[i]);

                } catch (IOException e) {

                    die("Error: unable to open \"" + args[i] + "\"");

                }

            } else if (args[i].equals("-html")) {

                if (format != TEXT)

                    die("Error: conflicting formats");

                format = HTML;

            } else if (args[i].equals("-xml")) {

                if (format != TEXT)

                    die("Error: conflicting formats");

                format = XML;

            } else {

                System.err.println("Illegal option: \"" + args[i] + "\"");

                usage();

                System.exit(1);

            }

        }

    }

    /**

     * Set up configuration file and report streams, if not set already.

     */

    static void setupStreams() {

        if (repstr == null)

            repstr = System.out;

        if (confstr == null)

            confstr = (new Main()).getClass().getResourceAsStream(CONFFILE);

        if (confstr == null)

            die("Error: unable to find default config file");

    }

    /**

     * Print contents of configuration file to selected output stream.

     */

    static void listConfig() {

        try {

            byte[] buf = new byte[256];

            int len;

            while ((len = confstr.read(buf)) != -1)

                repstr.write(buf, 0, len);

        } catch (IOException e) {

            die("Error: failed to list config file");

        }

    }

    /**

     * Set up the timer to end the test.

     *

     * @param delay the amount of delay, in seconds, before requesting

     * the process exit

     */

    static void setupTimer(int delay) {

        timer = new Timer(true);

        timer.schedule(

            new TimerTask() {

                public void run() {

                    exitRequested = true;

                }

            },

            delay * 1000);

    }

    /**

     * Set up benchmark harness.

     */

    static void setupHarness() {

        try {

            harness = new Harness(confstr);

        } catch (ConfigFormatException e) {

            String errmsg = e.getMessage();

            if (errmsg != null) {

                die("Error parsing config file: " + errmsg);

            } else {

                die("Error: illegal config file syntax");

            }

        } catch (IOException e) {

            die("Error: failed to read config file");

        }

    }

    /**

     * Setup benchmark reporter.

     */

    static void setupReporter() {

        String title = "Object Serialization Benchmark, v" + VERSION;

        switch (format) {

            case TEXT:

                reporter = new TextReporter(repstr, title);

                break;

            case HTML:

                reporter = new HtmlReporter(repstr, title);

                break;

            case XML:

                reporter = new XmlReporter(repstr, title);

                break;

            default:

                die("Error: unrecognized format type");

        }

    }

    /**

     * Run benchmarks.

     */

    static void runBenchmarks() {

        harness.runBenchmarks(reporter, verbose);

    }

}