Examples with BeastMCMC beast.app.BeastMCMC used on opensource projects

Example 1 with BeastMCMC

use of beast.app.BeastMCMC in project beast2 by CompEvol.

the class BeastMain method main.

// Main method
public static void main(final String[] args) throws java.io.IOException {
    final List<String> MCMCargs = new ArrayList<>();
    // Utils.loadUIManager();
    final Arguments arguments = new Arguments(new Arguments.Option[] { // new Arguments.Option("strict", "Fail on non-conforming BEAST XML file"),
    new Arguments.Option("window", "Provide a console window"), new Arguments.Option("options", "Display an options dialog"), new Arguments.Option("working", "Change working directory to input file's directory"), new Arguments.LongOption("seed", "Specify a random number generator seed"), new Arguments.StringOption("prefix", "PREFIX", "Specify a prefix for all output log filenames"), new Arguments.StringOption("statefile", "STATEFILE", "Specify the filename for storing/restoring the state"), new Arguments.Option("overwrite", "Allow overwriting of log files"), new Arguments.Option("resume", "Allow appending of log files"), new Arguments.Option("validate", "Parse the XML, but do not run -- useful for debugging XML"), // RRB: not sure what effect this option has
    new Arguments.IntegerOption("errors", "Specify maximum number of numerical errors before stopping"), new Arguments.IntegerOption("threads", "The number of computational threads to use (default 1), -1 for number of cores"), new Arguments.Option("java", "Use Java only, no native implementations"), new Arguments.Option("noerr", "Suppress all output to standard error"), new Arguments.StringOption("loglevel", "LEVEL", "error,warning,info,debug,trace"), new Arguments.IntegerOption("instances", "divide site patterns amongst number of threads (use with -threads option)"), new Arguments.Option("beagle", "Use beagle library if available"), new Arguments.Option("beagle_info", "BEAGLE: show information on available resources"), new Arguments.StringOption("beagle_order", "order", "BEAGLE: set order of resource use"), new Arguments.Option("beagle_CPU", "BEAGLE: use CPU instance"), new Arguments.Option("beagle_GPU", "BEAGLE: use GPU instance if available"), new Arguments.Option("beagle_SSE", "BEAGLE: use SSE extensions if available"), new Arguments.Option("beagle_single", "BEAGLE: use single precision if available"), new Arguments.Option("beagle_double", "BEAGLE: use double precision if available"), new Arguments.StringOption("beagle_scaling", new String[] { "default", "none", "dynamic", "always" }, false, "BEAGLE: specify scaling scheme to use"), new Arguments.Option("help", "Print this information and stop"), new Arguments.Option("version", "Print version and stop"), new Arguments.Option("strictversions", "Use only package versions as specified in the 'required' attribute"), new Arguments.StringOption("D", "DEFINITIONS", "attribute-value pairs to be replaced in the XML, e.g., -D \"arg1=10,arg2=20\"") });
    try {
        arguments.parseArguments(args);
    } catch (Arguments.ArgumentException ae) {
        Log.info.println();
        Log.info.println(ae.getMessage());
        Log.info.println();
        printUsage(arguments);
        System.exit(1);
    }
    if (arguments.hasOption("version")) {
        Log.info.println((new BEASTVersion2()).getVersionString());
        System.exit(0);
    }
    if (arguments.hasOption("help")) {
        printUsage(arguments);
        System.exit(0);
    }
    // final boolean verbose = arguments.hasOption("verbose");
    // final boolean parserWarning = arguments.hasOption("warnings"); // if dev, then auto turn on, otherwise default to turn off
    // final boolean strictXML = arguments.hasOption("strict");
    final boolean window = arguments.hasOption("window");
    final boolean options = arguments.hasOption("options");
    final boolean working = arguments.hasOption("working");
    final boolean doNotRun = arguments.hasOption("validate");
    String fileNamePrefix = null;
    String stateFileName = null;
    // boolean allowOverwrite = arguments.hasOption("overwrite");
    long seed = Randomizer.getSeed();
    boolean useJava = false;
    int threadCount = 1;
    if (arguments.hasOption("java")) {
        useJava = true;
    }
    if (arguments.hasOption("prefix")) {
        fileNamePrefix = arguments.getStringOption("prefix");
    }
    if (arguments.hasOption("statefile")) {
        stateFileName = arguments.getStringOption("statefile");
    }
    long beagleFlags = 0;
    boolean useBeagle = arguments.hasOption("beagle") || arguments.hasOption("beagle_CPU") || arguments.hasOption("beagle_GPU") || arguments.hasOption("beagle_SSE") || arguments.hasOption("beagle_double") || arguments.hasOption("beagle_single") || arguments.hasOption("beagle_order");
    if (arguments.hasOption("beagle_scaling")) {
        System.setProperty("beagle.scaling", arguments.getStringOption("beagle_scaling"));
    }
    boolean beagleShowInfo = arguments.hasOption("beagle_info");
    boolean useSSE = true;
    if (arguments.hasOption("beagle_CPU")) {
        beagleFlags |= BeagleFlag.PROCESSOR_CPU.getMask();
        useSSE = false;
    }
    if (arguments.hasOption("beagle_GPU")) {
        beagleFlags |= BeagleFlag.PROCESSOR_GPU.getMask();
        useSSE = false;
    }
    if (arguments.hasOption("beagle_SSE")) {
        beagleFlags |= BeagleFlag.PROCESSOR_CPU.getMask();
        useSSE = true;
    }
    if (useSSE) {
        beagleFlags |= BeagleFlag.VECTOR_SSE.getMask();
    }
    if (arguments.hasOption("beagle_double")) {
        beagleFlags |= BeagleFlag.PRECISION_DOUBLE.getMask();
    }
    if (arguments.hasOption("beagle_single")) {
        beagleFlags |= BeagleFlag.PRECISION_SINGLE.getMask();
    }
    if (arguments.hasOption("noerr")) {
        System.setErr(new PrintStream(new OutputStream() {

            @Override
            public void write(int b) {
            }
        }));
    }
    if (arguments.hasOption("beagle_order")) {
        System.setProperty("beagle.resource.order", arguments.getStringOption("beagle_order"));
    }
    if (arguments.hasOption("instances")) {
        System.setProperty("beast.instance.count", Integer.toString(arguments.getIntegerOption("instances")));
    }
    if (arguments.hasOption("beagle_scaling")) {
        System.setProperty("beagle.scaling", arguments.getStringOption("beagle_scaling"));
    }
    if (arguments.hasOption("threads")) {
        threadCount = arguments.getIntegerOption("threads");
    }
    if (threadCount <= 0) {
        threadCount = Runtime.getRuntime().availableProcessors();
        Log.warning.println("Setting number of threads to " + threadCount);
    }
    if (arguments.hasOption("seed")) {
        seed = arguments.getLongOption("seed");
        if (seed <= 0) {
            printTitle();
            Log.err.println("The random number seed should be > 0");
            System.exit(1);
        }
    }
    int maxErrorCount = 0;
    if (arguments.hasOption("errors")) {
        maxErrorCount = arguments.getIntegerOption("errors");
        if (maxErrorCount < 0) {
            maxErrorCount = 0;
        }
    }
    BeastConsoleApp consoleApp = null;
    final String nameString = "BEAST " + version.getVersionString();
    if (window) {
        Utils.loadUIManager();
        System.setProperty("com.apple.macos.useScreenMenuBar", "true");
        System.setProperty("apple.laf.useScreenMenuBar", "true");
        System.setProperty("apple.awt.showGrowBox", "true");
        System.setProperty("beast.useWindow", "true");
        final javax.swing.Icon icon = IconUtils.getIcon(BeastMain.class, "images/beast.png");
        final String aboutString = "<html><div style=\"font-family:sans-serif;\"><center>" + "<div style=\"font-size:12;\"><p>Bayesian Evolutionary Analysis Sampling Trees<br>" + "Version " + version.getVersionString() + ", " + version.getDateString() + "</p>" + version.getHTMLCredits() + "</div></center></div></html>";
        consoleApp = new BeastConsoleApp(nameString, aboutString, icon);
        // ensure error and info information is shown in console
        // but not warning, debug or trace, since that typically just
        // results in a lot of clutter
        Log.err = System.err;
        Log.info = System.out;
    }
    printTitle();
    File inputFile = null;
    if (options) {
        Utils.loadUIManager();
        final String titleString = "<html><center><p>Bayesian Evolutionary Analysis Sampling Trees<br>" + "Version " + version.getVersionString() + ", " + version.getDateString() + "</p></center></html>";
        final javax.swing.Icon icon = IconUtils.getIcon(BeastMain.class, "images/beast.png");
        final BeastDialog dialog = new BeastDialog(new JFrame(), titleString, icon);
        if (!dialog.showDialog(nameString, seed)) {
            System.exit(0);
        }
        // }
        switch(dialog.getLogginMode()) {
            case 0:
                /* do not ovewrite */
                break;
            case 1:
                MCMCargs.add("-overwrite");
                break;
            case 2:
                MCMCargs.add("-resume");
                break;
        }
        seed = dialog.getSeed();
        threadCount = dialog.getThreadPoolSize();
        useBeagle = dialog.useBeagle();
        if (useBeagle) {
            beagleShowInfo = dialog.showBeagleInfo();
            if (dialog.preferBeagleCPU()) {
                beagleFlags |= BeagleFlag.PROCESSOR_CPU.getMask();
            }
            if (dialog.preferBeagleSSE()) {
                beagleFlags |= BeagleFlag.VECTOR_SSE.getMask();
            }
            if (dialog.preferBeagleGPU()) {
                beagleFlags |= BeagleFlag.PROCESSOR_GPU.getMask();
            }
            if (dialog.preferBeagleDouble()) {
                beagleFlags |= BeagleFlag.PRECISION_DOUBLE.getMask();
            }
            if (dialog.preferBeagleSingle()) {
                beagleFlags |= BeagleFlag.PRECISION_SINGLE.getMask();
            }
        }
        inputFile = dialog.getInputFile();
        if (!beagleShowInfo && inputFile == null) {
            Log.err.println("No input file specified");
            System.exit(1);
        }
        if (dialog.useStrictVersions()) {
            MCMCargs.add("-strictversions");
        }
    } else {
        if (arguments.hasOption("overwrite")) {
            MCMCargs.add("-overwrite");
        }
        if (arguments.hasOption("resume")) {
            MCMCargs.add("-resume");
        }
    }
    if (arguments.hasOption("strictversions")) {
        MCMCargs.add("-strictversions");
    }
    if (arguments.hasOption("D")) {
        MCMCargs.add("-D");
        MCMCargs.add(arguments.getStringOption("D"));
    }
    if (beagleShowInfo) {
        Log.info.println("\n--- BEAGLE RESOURCES ---\n");
        for (beagle.ResourceDetails details : BeagleFactory.getResourceDetails()) Log.info.println(details.toString());
        if (window)
            return;
        else
            System.exit(0);
    }
    if (inputFile == null) {
        final String[] args2 = arguments.getLeftoverArguments();
        if (args2.length > 1) {
            Log.err.println("Unknown option: " + args2[1]);
            Log.err.println();
            printUsage(arguments);
            System.exit(1);
        }
        String inputFileName = null;
        if (args2.length > 0) {
            inputFileName = args2[0];
            inputFile = new File(inputFileName);
        }
        if (inputFileName == null) {
            // No input file name was given so throw up a dialog box...
            String fileName = getFileNameByDialog("BEAST " + version.getVersionString() + " - Select XML input file");
            if (fileName == null) {
                System.exit(0);
            }
            inputFile = new File(fileName);
        }
    }
    if (inputFile != null && inputFile.getParent() != null && working) {
        System.setProperty("file.name.prefix", inputFile.getParentFile().getAbsolutePath() + File.separator);
    }
    if (window) {
        if (inputFile == null) {
            consoleApp.setTitle("null");
        } else {
            consoleApp.setTitle(inputFile.getName());
        }
    }
    if (useJava) {
        System.setProperty("java.only", "true");
    }
    if (arguments.hasOption("loglevel")) {
        String l = arguments.getStringOption("loglevel");
        switch(l) {
            case "error":
                Log.setLevel(Log.Level.error);
                break;
            case "warning":
                Log.setLevel(Log.Level.warning);
                break;
            case "info":
                Log.setLevel(Log.Level.info);
                break;
            case "debug":
                Log.setLevel(Log.Level.debug);
                break;
            case "trace":
                Log.setLevel(Log.Level.trace);
                break;
        }
    }
    if (fileNamePrefix != null && fileNamePrefix.trim().length() > 0) {
        System.setProperty("file.name.prefix", fileNamePrefix.trim());
    }
    if (stateFileName != null && stateFileName.trim().length() > 0) {
        System.setProperty("state.file.name", stateFileName.trim());
        Log.info.println("Writing state to file " + stateFileName);
    }
    if (beagleFlags != 0) {
        System.setProperty("beagle.preferred.flags", Long.toString(beagleFlags));
    }
    if (threadCount > 0) {
        System.setProperty("thread.count", String.valueOf(threadCount));
        MCMCargs.add("-threads");
        MCMCargs.add(threadCount + "");
    }
    MCMCargs.add("-seed");
    MCMCargs.add(seed + "");
    Randomizer.setSeed(seed);
    Log.info.println("Random number seed: " + seed);
    Log.info.println();
    // Construct the beast object
    final BeastMCMC beastMCMC = new BeastMCMC();
    try {
        // set all the settings...
        MCMCargs.add(inputFile.getAbsolutePath());
        beastMCMC.parseArgs(MCMCargs.toArray(new String[0]));
        if (!doNotRun) {
            new BeastMain(beastMCMC, consoleApp, maxErrorCount);
        } else {
            Log.info.println("Done!");
        }
    } catch (RuntimeException rte) {
        if (window) {
            // appears at the end of the console.
            try {
                Thread.sleep(2000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            Log.info.println();
            Log.info.println("BEAST has terminated with an error. Please select QUIT from the menu.");
        } else {
            rte.printStackTrace();
        }
    // logger.severe will throw a RTE but we want to keep the console visible
    } catch (XMLParserException e) {
        Log.info.println(e.getMessage());
        if (!window) {
            System.exit(1);
        }
    } catch (Exception e) {
        e.printStackTrace();
        if (!window) {
            System.exit(1);
        }
    }
    if (!window) {
        System.exit(0);
    }
}