JLangOptions.java

//Copyright (C) 2018 Cornell University

package jlang;

import polyglot.ext.jl5.JL5Options;
import polyglot.main.OptFlag;
import polyglot.main.UsageError;

import java.util.Set;

public class JLangOptions extends JL5Options {

    public String entryPointClass;
    public boolean entryPointEmitted;
    public boolean printDesugar;
    public int maxPasses;

    public JLangOptions(JLangExtensionInfo extension) {
        super(extension);
        this.setOutputExtension("ll");
    }

    @Override
    protected void populateFlags(Set<OptFlag<?>> flags) {
        //Add the ox flag before the super call to ensure the correct default ("ll") is used.
        flags.add(new OptFlag<String>("-ox", "<ext>", "set output extension") {
            @Override
            public Arg<String> handle(String[] args, int index) {
                return createArg(index + 1, args[index]);
            }

            @Override
            public Arg<String> defaultArg() {
                return createDefault("ll");
            }
        });

        super.populateFlags(flags);
        flags.add(new OptFlag<String>(
                "-entry-point", "<classname>",
                "Fully qualified main class to call") {
            @Override
            public Arg<String> handle(String[] args, int index) {
                return createArg(index + 1, args[index]);
            }

            @Override
            public Arg<String> defaultArg() {
                return createDefault(this.defaultValue);
            }
        });

        flags.add(new OptFlag.Switch(
                "-dump-desugared", "Print the desugared AST to stderr", true));
        
        flags.add(new OptFlag.IntFlag("-max-runs", "<number of total compiler runs>",
        		"An upper bound on the number of total compiler runs"
        		+ " which ~ 7 * num_input_files") {
        	@Override
        	public Arg<Integer> defaultArg() {
        		return createDefault(-1);
        	}
        });
    }

    @Override
    protected void handleArg(OptFlag.Arg<?> arg) throws UsageError {
        if (arg.flag().ids().contains("-entry-point")) {
            this.entryPointClass = (String) arg.value();
        }
        else if (arg.flag().ids().contains("-dump-desugared")) {
            this.printDesugar = (Boolean) arg.value();
        }
        else if (arg.flag().ids().contains("-max-runs")) {
        	this.maxPasses = (Integer) arg.value();
        }
        else super.handleArg(arg);
    }

}