/*
 * Copyright (c) 2013, 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.compiler.ptx;

import com.oracle.graal.api.code.*;
import com.oracle.graal.api.meta.*;
import com.oracle.graal.asm.*;
import com.oracle.graal.hotspot.*;
import com.oracle.graal.hotspot.bridge.*;
import com.oracle.graal.lir.*;
import com.oracle.graal.lir.asm.*;
import com.oracle.graal.nodes.*;

public class PTXTargetMethodAssembler extends TargetMethodAssembler {

    private static CompilerToGPU toGPU = HotSpotGraalRuntime.graalRuntime().getCompilerToGPU();

    private static boolean validDevice = toGPU.deviceInit();

    private static final int totalProcessors = (validDevice ? toGPU.availableProcessors() : 0);

    public static int getAvailableProcessors() {
        return totalProcessors;
    }

    // detach ??

    public PTXTargetMethodAssembler(TargetDescription target, CodeCacheProvider codeCache, ForeignCallsProvider foreignCalls, FrameMap frameMap, AbstractAssembler asm, FrameContext frameContext,
                    CompilationResult compilationResult) {
        super(target, codeCache, foreignCalls, frameMap, asm, frameContext, compilationResult);
    }

    @Override
    public CompilationResult finishTargetMethod(StructuredGraph graph) {
        ResolvedJavaMethod method = graph.method();
        assert method != null : graph + " is not associated wth a method";

        ExternalCompilationResult graalCompile = (ExternalCompilationResult) super.finishTargetMethod(graph);

        try {
            if ((validDevice) && (graalCompile.getTargetCode() != null)) {
                long kernel = toGPU.generateKernel(graalCompile.getTargetCode(), method.getName());
                graalCompile.setEntryPoint(kernel);
            }
        } catch (Throwable th) {
            th.printStackTrace();
        }

        return graalCompile;
    }
}