/*
 * Copyright (c) 2012, Oracle and/or its affiliates. All rights reserved.
 * 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */
package com.oracle.graal.hotspot.stubs;

import static java.lang.reflect.Modifier.*;

import java.lang.reflect.*;
import java.util.concurrent.*;

import com.oracle.graal.api.code.*;
import com.oracle.graal.api.code.RuntimeCallTarget.Descriptor;
import com.oracle.graal.api.meta.*;
import com.oracle.graal.compiler.*;
import com.oracle.graal.debug.*;
import com.oracle.graal.graph.*;
import com.oracle.graal.hotspot.*;
import com.oracle.graal.hotspot.meta.*;
import com.oracle.graal.java.*;
import com.oracle.graal.nodes.*;
import com.oracle.graal.phases.*;
import com.oracle.graal.phases.PhasePlan.PhasePosition;
import com.oracle.graal.snippets.*;
import com.oracle.graal.snippets.Snippet.ConstantParameter;
import com.oracle.graal.snippets.SnippetTemplate.AbstractTemplates;
import com.oracle.graal.snippets.SnippetTemplate.Key;

/**
 * Base class for implementing some low level code providing the out-of-line slow path for a
 * snippet. A concrete stub is defined a subclass of this class.
 * <p>
 * Implementation detail: The stub classes re-use some of the functionality for {@link Snippet}s
 * purely for convenience (e.g., can re-use the {@link SnippetInstaller}).
 */
public abstract class Stub extends AbstractTemplates implements SnippetsInterface {

    /**
     * The method implementing the stub.
     */
    protected final HotSpotResolvedJavaMethod stubMethod;

    /**
     * The linkage information for the stub.
     */
    protected final HotSpotRuntimeCallTarget linkage;

    /**
     * The code installed for the stub.
     */
    protected InstalledCode stubCode;

    /**
     * Creates a new stub container. The new stub still needs to be
     * {@linkplain #install(GraalCompiler) installed}.
     * 
     * @param descriptor linkage details for a call to the stub
     */
    @SuppressWarnings("unchecked")
    public Stub(HotSpotRuntime runtime, Assumptions assumptions, TargetDescription target, Descriptor descriptor) {
        super(runtime, assumptions, target, null);
        stubMethod = findStubMethod(runtime, getClass());
        linkage = runtime.registerStub(descriptor, this);
        assert linkage != null;
    }

    /**
     * Adds the {@linkplain ConstantParameter constant} arguments of this stub.
     */
    protected abstract void populateKey(Key key);

    protected HotSpotRuntime runtime() {
        return (HotSpotRuntime) runtime;
    }

    /**
     * Gets the method implementing this stub.
     */
    public ResolvedJavaMethod getMethod() {
        return stubMethod;
    }

    public HotSpotRuntimeCallTarget getLinkage() {
        return linkage;
    }

    /**
     * Compiles the code for this stub, installs it and initializes the address used for calls to
     * it.
     */
    public void install(GraalCompiler compiler) {
        StructuredGraph graph = (StructuredGraph) stubMethod.getCompilerStorage().get(Graph.class);

        Key key = new Key(stubMethod);
        populateKey(key);
        SnippetTemplate template = cache.get(key, assumptions);
        graph = template.copySpecializedGraph();

        PhasePlan phasePlan = new PhasePlan();
        GraphBuilderPhase graphBuilderPhase = new GraphBuilderPhase(runtime, GraphBuilderConfiguration.getDefault(), OptimisticOptimizations.ALL);
        phasePlan.addPhase(PhasePosition.AFTER_PARSING, graphBuilderPhase);
        final CompilationResult compResult = compiler.compileMethod(stubMethod, graph, null, phasePlan, OptimisticOptimizations.ALL);

        final CodeInfo[] info = new CodeInfo[1];
        stubCode = Debug.scope("CodeInstall", new Object[]{compiler, stubMethod}, new Callable<InstalledCode>() {

            @Override
            public InstalledCode call() {
                InstalledCode installedCode = runtime().addMethod(stubMethod, compResult, info);
                assert installedCode != null : "error installing stub " + stubMethod;
                if (Debug.isDumpEnabled()) {
                    Debug.dump(new Object[]{compResult, info[0]}, "After code installation");
                }
                return installedCode;
            }
        });

        assert stubCode != null : "error installing stub " + stubMethod;
        linkage.setAddress(info[0].getStart());
    }

    /**
     * Finds the static method annotated with {@link Snippet} in a given class of which there must
     * be exactly one.
     */
    private static HotSpotResolvedJavaMethod findStubMethod(HotSpotRuntime runtime, Class<?> stubClass) {
        HotSpotResolvedJavaMethod m = null;
        for (Method candidate : stubClass.getDeclaredMethods()) {
            if (isStatic(candidate.getModifiers()) && candidate.getAnnotation(Snippet.class) != null) {
                assert m == null : "more than one method annotated with @" + Snippet.class.getSimpleName() + " in " + stubClass;
                m = (HotSpotResolvedJavaMethod) runtime.lookupJavaMethod(candidate);
            }
        }
        assert m != null : "no static method annotated with @" + Snippet.class.getSimpleName() + " in " + stubClass;
        return m;
    }
}