root/trunk/ProjectFortress/src/com/sun/fortress/interpreter/Driver.java @ 3252

/*******************************************************************************
    Copyright 2008 Sun Microsystems, Inc.,
    4150 Network Circle, Santa Clara, California 95054, U.S.A.
    All rights reserved.

    U.S. Government Rights - Commercial software.
    Government users are subject to the Sun Microsystems, Inc. standard
    license agreement and applicable provisions of the FAR and its supplements.

    Use is subject to license terms.

    This distribution may include materials developed by third parties.

    Sun, Sun Microsystems, the Sun logo and Java are trademarks or registered
    trademarks of Sun Microsystems, Inc. in the U.S. and other countries.
 ******************************************************************************/

package com.sun.fortress.interpreter;

import static com.sun.fortress.exceptions.InterpreterBug.bug;
import static com.sun.fortress.exceptions.ProgramError.error;
import static com.sun.fortress.exceptions.ProgramError.errorMsg;

import com.sun.fortress.interpreter.env.APIWrapper;
import com.sun.fortress.interpreter.env.CUWrapper;
import com.sun.fortress.interpreter.env.ComponentWrapper;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Hashtable;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
import java.util.Stack;

import com.sun.fortress.repository.DerivedFiles;
import com.sun.fortress.repository.FortressRepository;
import com.sun.fortress.repository.IOAst;
import com.sun.fortress.compiler.index.ApiIndex;
import com.sun.fortress.compiler.index.ComponentIndex;
import com.sun.fortress.exceptions.RedefinitionError;
import com.sun.fortress.interpreter.evaluator.CollectTests;
import com.sun.fortress.interpreter.evaluator.Environment;
import com.sun.fortress.interpreter.evaluator.EvaluatorBase;
import com.sun.fortress.interpreter.evaluator.Init;
import com.sun.fortress.interpreter.evaluator.tasks.EvaluatorTask;
import com.sun.fortress.interpreter.evaluator.tasks.FortressTaskRunnerGroup;
import com.sun.fortress.interpreter.evaluator.types.FTraitOrObject;
import com.sun.fortress.interpreter.evaluator.types.FTraitOrObjectOrGeneric;
import com.sun.fortress.interpreter.evaluator.types.FType;
import com.sun.fortress.interpreter.evaluator.values.FunctionClosure;
import com.sun.fortress.interpreter.evaluator.values.FString;
import com.sun.fortress.interpreter.evaluator.values.FValue;
import com.sun.fortress.interpreter.evaluator.values.OverloadedFunction;
import com.sun.fortress.interpreter.glue.WellKnownNames;
import com.sun.fortress.nodes.APIName;
import com.sun.fortress.nodes.AliasedAPIName;
import com.sun.fortress.nodes.AliasedSimpleName;
import com.sun.fortress.nodes.Api;
import com.sun.fortress.nodes.CompilationUnit;
import com.sun.fortress.nodes.Component;
import com.sun.fortress.nodes.Decl;
import com.sun.fortress.nodes.GrammarDecl;
import com.sun.fortress.nodes.Id;
import com.sun.fortress.nodes.IdOrOpOrAnonymousName;
import com.sun.fortress.nodes.Import;
import com.sun.fortress.nodes.ImportApi;
import com.sun.fortress.nodes.ImportNames;
import com.sun.fortress.nodes.ImportStar;
import com.sun.fortress.nodes.ImportedNames;
import com.sun.fortress.nodes.NodeAbstractVisitor_void;
import com.sun.fortress.nodes.NodeVisitor_void;
import com.sun.fortress.nodes.ObjectDecl;
import com.sun.fortress.nodes.SyntaxDecl;
import com.sun.fortress.nodes.SyntaxDef;
import com.sun.fortress.nodes.TestDecl;
import com.sun.fortress.nodes.TraitDecl;
import com.sun.fortress.nodes_util.NodeFactory;
import com.sun.fortress.nodes_util.NodeUtil;
import com.sun.fortress.useful.BASet;
import com.sun.fortress.useful.CheckedNullPointerException;
import com.sun.fortress.useful.Fn;
import com.sun.fortress.useful.HasAt;
import com.sun.fortress.useful.NI;
import com.sun.fortress.useful.StringComparer;
import com.sun.fortress.useful.Useful;
import com.sun.fortress.useful.Visitor1;
import com.sun.fortress.useful.Visitor2;
import edu.rice.cs.plt.tuple.Option;

import static com.sun.fortress.interpreter.glue.WellKnownNames.*;

public class Driver {

    private static boolean _libraryTest = false;

    private Driver() {};

    static public void runTests() {
    }

    /**
     * Native code sometimes needs access to the library component wrapper.
     */
    static ComponentWrapper libraryComponentWrapper = null;
    public static Environment getFortressLibrary() {
        return libraryComponentWrapper.getEnvironment();
    }

    public static ArrayList<ComponentWrapper> components;

    public static void reset() {
        components = null;
        libraryComponentWrapper = null;
    }

    public static Environment evalComponent(ComponentIndex p,
                                            FortressRepository fr)
        throws IOException {

        Init.initializeEverything();

        /*
         * Begin "linker" -- to be replaced with the real thing, when more
         * infrastructure is present.
         */

        HashMap<String, ComponentWrapper> linker = new HashMap<String, ComponentWrapper>();

        Stack<ComponentWrapper> pile = new Stack<ComponentWrapper>();
        // ArrayList<ComponentWrapper>
        components = new ArrayList<ComponentWrapper>();

        /*
         * This looks like gratuitous and useless error checking that
         * interferes with the "test" flag.
         *
        for(String defaultLib : WellKnownNames.defaultLibrary()) {
            APIName libAPI = NodeFactory.makeAPIName(defaultLib);
            if (p.getName().equals(libAPI)) {
                error("Fortress built-in library " + defaultLib + " does not export executable.");
            }
        }
        */

        ComponentWrapper comp = commandLineComponent(fr, linker, pile, p);

        /*
         * This "linker" implements a one-to-one, same-name correspondence
         * between APIs and components.
         *
         * phase 0.5: starting with the main component, identify other
         * components that will be needed (the fortress will contain the map
         * from component C's imported APIs to their implementing components).
         * Each component will be paired with its top level environment (TLE).
         *
         * phase 1: for each component, prepare its TLE by creating all name
         * slots (nothing in the slots will is initialized AT ALL -- it's all
         * thunks, empty mutables, and uninitialized types).
         *
         * phase 1.5: for each component C, add to C's TLE all the names that C
         * imports (using the API->component map from the fortress to figure out
         * the values/types associated with each of those names).
         *
         */
        comp.touchExports(false);
        // linker.put("Executable", comp);
        // Don't ever use the string "Executable" directly!
        // Instead look in WellKnownNames.fss
        //pile.push(comp);


        /*
         * This is a patch; eventually, it will all be done explicitly.
         */

        /*
         * Notice that builtins is used ONLY to satisfy the interface of the
         * importer for purposes of injecting primitives &c into other components.
         */
        ComponentWrapper builtins = new ComponentWrapper(readTreeOrSourceComponent(anyTypeLibrary(), anyTypeLibrary(), fr), linker, WellKnownNames.defaultLibrary());
        builtins.getEnvironment().installPrimitives();
        linker.put(anyTypeLibrary(), builtins);

        APIWrapper lib = null;

        libraryComponentWrapper = ensureApiImplemented(fr, linker, pile,
                                                       NodeFactory.makeAPIName(NodeFactory.interpreterSpan,
                                                                               fortressLibrary()));
        lib = libraryComponentWrapper.getExportedCW(fortressLibrary());

        ComponentWrapper nativescomp =
            ensureApiImplemented(fr, linker, pile,
                                 NodeFactory.makeAPIName(NodeFactory.interpreterSpan,
                                                         fortressBuiltin()));
        APIWrapper natives = nativescomp.getExportedCW(fortressBuiltin());

        /*
         * This performs closure over APIs and components, ensuring that all are
         * initialized through phase one of building environments.
         */
        while (!pile.isEmpty()) {

            ComponentWrapper cw = pile.pop();
            components.add(cw);

            CompilationUnit c = cw.getCompilationUnit();
            List<Import> imports = c.getImports();

            ensureImportsImplemented(fr, linker, pile, imports);
        }

        // Desugarer needs to know about trait members.
        for (CUWrapper cw : components) {
            cw.preloadTopLevel();
        }

        // Iterate to a fixed point, pushing trait info as far as necessary.
        // This is not the same as pushing information into environments, which
        // occurs below the call(s) to populateOne.
        boolean change = true;
        while (change) {
            change = false;
            for (ComponentWrapper cw : components) {
                change |= injectTraitMembersForDesugaring(linker, cw);
            }

            change |= injectLibraryTraits(components, lib);
            change |= injectLibraryTraits(components, natives);
        }

        /*
         * After all apis etc have been imported, populate their environments.
         * First the component is populated, then (recursively, inside that
         * call to populate) the exported apis are populated.
         */
        for (int i = components.size() - 1; i >= 0; i--) {
            ComponentWrapper cw = components.get(i);
            // System.err.println("populating " + cw);
            cw.populateOne();
        }

        for (CUWrapper cw : components) {
            cw.initTypes();
        }
        for (CUWrapper cw : components) {
            scanAllFunctionalMethods(cw.getEnvironment());
        }
        for (CUWrapper cw : components) {
            cw.initFuncs();
        }

        for (CUWrapper cw : components) {
            finishAllFunctionalMethods(cw.getEnvironment());
        }

        for (CUWrapper cw : components) {
            cw.reset();
        }

         for (CUWrapper cw : components) {
            cw.initVars();
        }

        return comp.getEnvironment();
    }


    /**
     * Copies rewriting information (traits, what members they define) from apis into
     * components.  This scaffolding is necessary to get the simplification right
     * for member (field, method) references.
     *
     * @param linker
     * @param cw
     */
    private static boolean injectTraitMembersForDesugaring(
            HashMap<String, ComponentWrapper> linker, ComponentWrapper cw) {
        CompilationUnit c = cw.getCompilationUnit();
        List<Import> imports = c.getImports();
        boolean change = false;

        for (Import i : imports) {
            if (i instanceof ImportApi) {

                    /*
                     * Not-yet-implemented because of issues with selectors.
                     */
                bug(errorMsg("NYI 'Import api APIName'; ",
                             "try 'import APIName.{...}' instead."));
            } else if (i instanceof ImportedNames) {
                ImportedNames ix = (ImportedNames) i;
                APIName source = ix.getApiName();
                String from_apiname = NodeUtil.nameString(source);

                ComponentWrapper from_cw = linker.get(from_apiname);
                APIWrapper api_cw = from_cw.getExportedCW(from_apiname);

                /* Pull in names, UNqualified */

                if (ix instanceof ImportNames) {
                    /* A set of names */
                    List<AliasedSimpleName> names = ((ImportNames) ix).getAliasedNames();
                    for (AliasedSimpleName an : names) {
                        IdOrOpOrAnonymousName name = an.getName();
                        Option<IdOrOpOrAnonymousName> alias = an.getAlias();
                        /*
                         * If alias exists, associate the binding from
                         * component_wrapper with alias, otherwise associate
                         * it with plain old name.
                         */
                        /* probable bug: need to insert into ownNonFunction names */
                        change |= cw.injectAtTopLevel(alias.unwrap(name).stringName(),
                                                                name.stringName(),
                                                                api_cw,
                                                                cw.excludedImportNames);
                    }

                } else if (ix instanceof ImportStar) {
                    /* All names BUT excepts, as they are listed. */
                    final List<IdOrOpOrAnonymousName> excepts = ((ImportStar) ix)
                            .getExceptNames();
                    final Set<String> except_names = Useful.applyToAllInserting(
                            excepts,
                            new Fn<IdOrOpOrAnonymousName, String>() {
                                public String apply(IdOrOpOrAnonymousName n) {
                                    return NodeUtil.nameString(n);
                                }
                            },
                            new HashSet<String>());

                    for (String s : api_cw.getTopLevelRewriteNames()) {
                        if (! except_names.contains(s) &&
                                !cw.isOwnName(s) &&
                                !cw.excludedImportNames.contains(s)) {
                            change |= cw.injectAtTopLevel(s, s, api_cw, cw.excludedImportNames);
                        }
                    }
                    for (String s : api_cw.getFunctionals()) {
                        if (! except_names.contains(s) &&
                                !cw.excludedImportNames.contains(s)) {
                            change |= cw.injectAtTopLevel(s, s, api_cw, cw.excludedImportNames);
                        }
                    }
                }
            } else {
                bug(errorMsg("NYI Import " + i));
            }
        }


        return change;
    }

    private static boolean injectLibraryTraits(List<ComponentWrapper> components,
            APIWrapper lib) {
        boolean change = false;
        for (ComponentWrapper cw : components) {
            for (String s : lib.getTopLevelRewriteNames()) {
                if (cw.isOwnName(s)) {
                    continue;
                }
                change |= cw.injectAtTopLevel(s, s, lib, cw.excludedImportNames);
            }
            for (String s : lib.getFunctionals()) {
                change |= cw.injectAtTopLevel(s, s, lib, cw.excludedImportNames);
            }
        }
        return change;
    }

    private static void scanAllFunctionalMethods(final Environment e) {
        Visitor2<String, FType> vt = new Visitor2<String, FType>() {
            public void visit(String s, FType o) {
                if (o instanceof FTraitOrObjectOrGeneric) {
                    FTraitOrObjectOrGeneric tooog = (FTraitOrObjectOrGeneric) o;
                    tooog.initializeFunctionalMethods(e);
                    if (tooog instanceof FTraitOrObject) {
                        for (FType t : ((FTraitOrObject)tooog).getProperTransitiveExtends())
                            if (t instanceof FTraitOrObjectOrGeneric)
                                ((FTraitOrObject)t).initializeFunctionalMethods(e);
                    } else {

                    }
                }
            }
        };
        e.visit(vt, null, null, null, null);
    }

    private static void finishAllFunctionalMethods(final Environment e) {
        Visitor2<String, FType> vt = new Visitor2<String, FType>() {
            public void visit(String s, FType o) {
                if (o instanceof FTraitOrObjectOrGeneric) {
                    FTraitOrObjectOrGeneric tooog = (FTraitOrObjectOrGeneric) o;
                    tooog.finishFunctionalMethods(e);
                    if (tooog instanceof FTraitOrObject) {
                        for (FType t : ((FTraitOrObject)tooog).getProperTransitiveExtends())
                            if (t instanceof FTraitOrObjectOrGeneric)
                                ((FTraitOrObject)t).finishFunctionalMethods(e);
                    }
                }
            }
        };
        e.visit(vt, null, null, null, null);
    }

    /**
     * For each imported API, checks to see if it is already "linked". If not,
     * "finds" it (looks for apiname.fss/apiname.tfs) and reads it in and sticks
     * it in the "pile".
     *
     * For each component, also reads in the corresponding APIs so that we know
     * what to export.
     *
     * @param linker
     * @param pile
     * @param imports
     */
    private static void ensureImportsImplemented (
            FortressRepository fr,
            HashMap<String, ComponentWrapper> linker,
            Stack<ComponentWrapper> pile,
            List<Import> imports
        )
        throws IOException
    {

        for (Import i : imports) {
            if (i instanceof ImportApi) {
                ImportApi ix = (ImportApi) i;
                List<AliasedAPIName> apis = ix.getApis();
                for (AliasedAPIName adi : apis) {
                    APIName id = adi.getApiName();
                    ensureApiImplemented(fr, linker, pile, id);
                }
            }
            else if (i instanceof ImportedNames) {
                ImportedNames ix = (ImportedNames) i;
                APIName source = ix.getApiName();
                ensureApiImplemented(fr, linker, pile, source);
            }
            else {
                bug(i, "Unrecognized import");
            }
        }
    }

    /**
     * Returns the component wrapper (for the component) that exports this API.
     * Needs to generalize to sets of APIs in the future, perhaps.
     *
     * @param linker
     * @param pile
     * @param id
     */
    private static ComponentWrapper ensureApiImplemented(
            FortressRepository fr,
            HashMap<String, ComponentWrapper> linker,
            Stack<ComponentWrapper> pile, APIName name) throws IOException {
        String apiname = NodeUtil.nameString(name);
        ComponentWrapper newwrapper = linker.get(apiname);
        if (newwrapper == null) {
            /*
             * Here, the linker prototype takes the extreme shortcut of assuming
             * that Api Foo is implemented by Component Foo, dots and all.
             *
             * These few lines are what needs to be replaced by a real linker.
             */
            Api newapi = readTreeOrSourceApi(apiname, apiname, fr);
            ComponentIndex newcomp = readTreeOrSourceComponent(apiname, apiname, fr) ;

            APIWrapper apicw = new APIWrapper(newapi, linker, WellKnownNames.defaultLibrary());
            newwrapper = new ComponentWrapper(newcomp, apicw, linker, WellKnownNames.defaultLibrary());
            newwrapper.touchExports(true);
            linker.put(apiname, newwrapper);
            pile.push(newwrapper);

            List<Import> imports = newapi.getImports();
            ensureImportsImplemented(fr, linker, pile, imports);
        }
        return newwrapper;
    }

    private static ComponentWrapper commandLineComponent(FortressRepository fr,
            HashMap<String, ComponentWrapper> linker,
            Stack<ComponentWrapper> pile, ComponentIndex comp_index) throws IOException {

            Component comp = (Component) (comp_index.ast());
            APIName name = comp.getName();
            String apiname = NodeUtil.nameString(name);
            ComponentWrapper comp_wrapper;

            List<APIWrapper> exports_list = new ArrayList<APIWrapper>(1);
            for (APIName ex_apiname : comp.getExports()) {
                String ex_name = NodeUtil.nameString(ex_apiname);
                Api newapi = readTreeOrSourceApi(ex_name, ex_name, fr);
                exports_list.add( new APIWrapper(newapi, linker, WellKnownNames.defaultLibrary()) );
            }

            comp_wrapper = new ComponentWrapper(comp_index, exports_list, linker, WellKnownNames.defaultLibrary());
            fr.forgetComponent(name);
            comp_wrapper.touchExports(true);
            linker.put(apiname, comp_wrapper);
            pile.push(comp_wrapper);

            for (CUWrapper apicw : exports_list) {
                List<Import> imports  = apicw.getImports();
                linker.put(apicw.name(), comp_wrapper);
                ensureImportsImplemented(fr, linker, pile, imports);
            }
        return comp_wrapper;
    }

    // This runs the program from inside a task.
    public static FValue
        runProgramTask(ComponentIndex p,
                       List<String> args, String toBeRun,
                       FortressRepository fr)
        throws IOException
    {

        Environment e = evalComponent(p, fr);

        FunctionClosure run_fn = e.getClosure(toBeRun);
        Toplevel toplevel = new Toplevel();
        ArrayList<FValue> fvalueArgs = new ArrayList<FValue>();
        for (String s : args) {
            fvalueArgs.add(FString.make(s));
        }
        FValue ret = run_fn.functionInvocation(fvalueArgs, toplevel);
        // try {
        // e.dump(System.out);
        // } catch (IOException ioe) {
        // System.out.println("io exception" + ioe);
        // }
       return ret;
    }

    static FortressTaskRunnerGroup group;

    static int getNumThreads() {
        int numThreads;

        String numThreadsString = System.getenv("FORTRESS_THREADS");
        if (numThreadsString != null)
            numThreads = Integer.parseInt(numThreadsString);
        else {
            int availThreads = Runtime.getRuntime().availableProcessors();
            if (availThreads <= 2)
                numThreads = availThreads;
            else
                numThreads = (int) Math.floor((double) availThreads/2.0);
        }
        return numThreads;
    }


    public static void runTests(FortressRepository fr, ComponentIndex p, boolean verbose)
        throws Throwable {

        CollectTests bte = new CollectTests ();
        bte.visit(p.ast());
        List<String> tests = CollectTests.getTests();

        if (group == null)
            group = new FortressTaskRunnerGroup(getNumThreads());

        for (String s : tests) {
            List<String> args = new ArrayList<String>();
            if ( verbose )
                System.err.print("starting " + s + "... ");
            EvaluatorTask evTask = new EvaluatorTask(fr, p, s, args);
            group.invoke(evTask);
            if (evTask.causedException()) {
                throw evTask.taskException();
            }
            if ( verbose )
                System.err.println("finishing " + s);
        }
    }

    // This creates the parallel context
    public static FValue runProgram(FortressRepository fr, ComponentIndex p,
                                    List<String> args)
        throws Throwable {

        if (group == null)
            group = new FortressTaskRunnerGroup(getNumThreads());

        EvaluatorTask evTask = new EvaluatorTask(fr, p, "run", args);
        try {
            group.invoke(evTask);
            if (evTask.causedException()) {
                throw evTask.taskException();
            }
            return evTask.result();
        }
        finally {
            // group.interruptAll();
        }
    }


    private static class Toplevel implements HasAt {
        public String at() {
            return "toplevel";
        }

        public String stringName() {
            return "driver";
        }
    }

    private static ComponentIndex readTreeOrSourceComponent(String key, String basename, FortressRepository p) throws IOException {

        String name  = key;
        APIName apiname = NodeFactory.makeAPIName(NodeFactory.interpreterSpan, name);

        ComponentIndex ci = p.getComponent(apiname);
        return ci;

    }


    private static Api readTreeOrSourceApi(String key, String basename, FortressRepository p) throws IOException {
        String name  = key;
        APIName apiname = NodeFactory.makeAPIName(NodeFactory.interpreterSpan, name);
        ApiIndex ci = p.getApi(apiname);
        CompilationUnit c = ci.ast();
        return (Api) c;

    }

    static Hashtable<String, CompilationUnit> libraryCache = new Hashtable<String, CompilationUnit>();

}