/*

 * 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 org.graalvm.compiler.hotspot.amd64.test;

import static org.graalvm.compiler.lir.LIRInstruction.OperandFlag.REG;

import org.graalvm.compiler.api.replacements.Snippet;

import org.graalvm.compiler.asm.amd64.AMD64Address;

import org.graalvm.compiler.asm.amd64.AMD64MacroAssembler;

import org.graalvm.compiler.bytecode.BytecodeProvider;

import org.graalvm.compiler.core.common.CompilationIdentifier;

import org.graalvm.compiler.core.common.LIRKind;

import org.graalvm.compiler.core.common.spi.ForeignCallDescriptor;

import org.graalvm.compiler.core.common.type.DataPointerConstant;

import org.graalvm.compiler.debug.DebugContext;

import org.graalvm.compiler.hotspot.HotSpotBackend;

import org.graalvm.compiler.hotspot.HotSpotForeignCallLinkage;

import org.graalvm.compiler.hotspot.meta.HotSpotForeignCallsProviderImpl;

import org.graalvm.compiler.hotspot.meta.HotSpotProviders;

import org.graalvm.compiler.hotspot.stubs.SnippetStub;

import org.graalvm.compiler.lir.LIRInstructionClass;

import org.graalvm.compiler.lir.Variable;

import org.graalvm.compiler.lir.amd64.AMD64LIRInstruction;

import org.graalvm.compiler.lir.asm.ArrayDataPointerConstant;

import org.graalvm.compiler.lir.asm.CompilationResultBuilder;

import org.graalvm.compiler.lir.gen.LIRGeneratorTool;

import org.graalvm.compiler.lir.jtt.LIRTest;

import org.graalvm.compiler.lir.jtt.LIRTestSpecification;

import org.graalvm.compiler.nodes.StructuredGraph;

import org.graalvm.compiler.nodes.extended.ForeignCallNode;

import org.graalvm.compiler.nodes.graphbuilderconf.GraphBuilderConfiguration;

import org.graalvm.compiler.nodes.graphbuilderconf.GraphBuilderContext;

import org.graalvm.compiler.nodes.graphbuilderconf.InvocationPlugin;

import org.graalvm.compiler.nodes.graphbuilderconf.InvocationPlugins;

import org.graalvm.compiler.options.OptionValues;

import org.graalvm.compiler.replacements.ReplacementsImpl;

import org.graalvm.compiler.replacements.classfile.ClassfileBytecodeProvider;

import org.junit.Assume;

import org.junit.Before;

import org.junit.Test;

import jdk.vm.ci.amd64.AMD64;

import jdk.vm.ci.amd64.AMD64.CPUFeature;

import jdk.vm.ci.amd64.AMD64Kind;

import jdk.vm.ci.code.Register;

import jdk.vm.ci.code.ValueUtil;

import jdk.vm.ci.meta.AllocatableValue;

import jdk.vm.ci.meta.MetaAccessProvider;

import jdk.vm.ci.meta.ResolvedJavaMethod;

import jdk.vm.ci.meta.Value;

public class StubAVXTest extends LIRTest {

    @Before

    public void checkAMD64() {

        Assume.assumeTrue("skipping AMD64 specific test", getTarget().arch instanceof AMD64);

        Assume.assumeTrue("skipping AVX test", ((AMD64) getTarget().arch).getFeatures().contains(CPUFeature.AVX));

        if (getBackend() instanceof HotSpotBackend) {

            HotSpotBackend backend = (HotSpotBackend) getBackend();

            Assume.assumeTrue("skipping because of MaxVectorSize", backend.getRuntime().getVMConfig().maxVectorSize >= 32);

        }

    }

    private static final DataPointerConstant avxConstant = new ArrayDataPointerConstant(new float[]{1f, 2f, 3f, 4f, 5f, 6f, 7f, 8f}, 32);

    private static class LoadAVXConstant extends AMD64LIRInstruction {

        public static final LIRInstructionClass<LoadAVXConstant> TYPE = LIRInstructionClass.create(LoadAVXConstant.class);

        @Def({REG}) AllocatableValue result;

        LoadAVXConstant(AllocatableValue result) {

            super(TYPE);

            this.result = result;

        }

        @Override

        public void emitCode(CompilationResultBuilder crb, AMD64MacroAssembler masm) {

            masm.vmovdqu(ValueUtil.asRegister(result), (AMD64Address) crb.recordDataReferenceInCode(avxConstant));

        }

    }

    private static final LIRTestSpecification loadAVXConstant = new LIRTestSpecification() {

        @Override

        public void generate(LIRGeneratorTool gen) {

            Variable ret = gen.newVariable(LIRKind.value(AMD64Kind.V256_SINGLE));

            gen.append(new LoadAVXConstant(ret));

            setResult(ret);

        }

    };

    @LIRIntrinsic

    public static Object loadAVXConstant(@SuppressWarnings("unused") LIRTestSpecification spec) {

        return null;

    }

    private static class CompareAVXRegister extends AMD64LIRInstruction {

        public static final LIRInstructionClass<CompareAVXRegister> TYPE = LIRInstructionClass.create(CompareAVXRegister.class);

        @Def({REG}) AllocatableValue result;

        @Use({REG}) AllocatableValue left;

        @Use({REG}) AllocatableValue right;

        @Temp({REG}) AllocatableValue temp;

        CompareAVXRegister(AllocatableValue result, AllocatableValue left, AllocatableValue right, AllocatableValue temp) {

            super(TYPE);

            this.result = result;

            this.left = left;

            this.right = right;

            this.temp = temp;

        }

        private static int getRXB(Register reg, Register rm) {

            int rxb = (reg.encoding & 0x08) >> 1;

            rxb |= (rm.encoding & 0x08) >> 3;

            return rxb;

        }

        @Override

        public void emitCode(CompilationResultBuilder crb, AMD64MacroAssembler masm) {

            Register res = ValueUtil.asRegister(result);

            Register x = ValueUtil.asRegister(left);

            Register y = ValueUtil.asRegister(right);

            Register tmp = ValueUtil.asRegister(temp);

            // VEX.NDS.256.0F.WIG C2 /r ib(0)

            // VCMPPS tmp, x, y, EQ

            masm.emitByte(0xC4);                                   // VEX 3-byte

            masm.emitByte((~getRXB(tmp, y) & 0x7) << 5 | 0x01);    // RXB m-mmmmm (0F)

            masm.emitByte(((~x.encoding & 0x0f) << 3) | 0b1_00);   // W(0) vvvv L(1) pp(0)

            masm.emitByte(0xC2);

            masm.emitByte(0xC0 | ((tmp.encoding & 0x07) << 3) | (y.encoding & 0x07));

            masm.emitByte(0);

            // VEX.256.0F.WIG 50 /r

            // VMOVMSKPS res, tmp

            masm.emitByte(0xC4);                                   // VEX 3-byte

            masm.emitByte((~getRXB(res, tmp) & 0x7) << 5 | 0x01);  // RXB m-mmmmm (0F)

            masm.emitByte(0b0_1111_1_00);                          // W(0) vvvv L(1) pp(0)

            masm.emitByte(0x50);

            masm.emitByte(0xC0 | ((res.encoding & 0x07) << 3) | (tmp.encoding & 0x07));

        }

    }

    private static final LIRTestSpecification compareAVXRegister = new LIRTestSpecification() {

        @Override

        public void generate(LIRGeneratorTool gen, Value arg0, Value arg1) {

            Variable ret = gen.newVariable(LIRKind.value(AMD64Kind.DWORD));

            gen.append(new CompareAVXRegister(ret, gen.asAllocatable(arg0), gen.asAllocatable(arg1), gen.newVariable(LIRKind.value(AMD64Kind.V256_QWORD))));

            setResult(ret);

        }

    };

    private static class TestStub extends SnippetStub {

        TestStub(OptionValues options, HotSpotProviders providers, HotSpotForeignCallLinkage linkage) {

            super("testStub", options, providers, linkage);

        }

        @Snippet

        static void testStub() {

        }

        @Override

        protected void registerSnippet() {

        }

        @Override

        protected StructuredGraph buildInitialGraph(DebugContext debug, CompilationIdentifier compilationId, Object[] args) {

            // Build the snippet graph directly since snippet registration is closed at this point.

            ReplacementsImpl d = (ReplacementsImpl) providers.getReplacements();

            MetaAccessProvider metaAccess = d.getProviders().getMetaAccess();

            BytecodeProvider bytecodes = new ClassfileBytecodeProvider(metaAccess, d.snippetReflection, ClassLoader.getSystemClassLoader());

            return d.makeGraph(debug, bytecodes, method, args, null, false, null);

        }

    }

    public static final ForeignCallDescriptor TEST_STUB = new ForeignCallDescriptor("test_stub", void.class);

    @LIRIntrinsic

    public static int compareAVXRegister(@SuppressWarnings("unused") LIRTestSpecification spec, Object left, Object right) {

        return left == right ? 0xff : 0;

    }

    @Override

    protected GraphBuilderConfiguration editGraphBuilderConfiguration(GraphBuilderConfiguration conf) {

        InvocationPlugins invocationPlugins = conf.getPlugins().getInvocationPlugins();

        InvocationPlugins.Registration r = new InvocationPlugins.Registration(invocationPlugins, TestStub.class);

        r.register0("testStub", new InvocationPlugin() {

            @Override

            public boolean apply(GraphBuilderContext b, ResolvedJavaMethod targetMethod, InvocationPlugin.Receiver receiver) {

                b.add(new ForeignCallNode(getProviders().getForeignCalls(), TEST_STUB));

                return true;

            }

        });

        return super.editGraphBuilderConfiguration(conf);

    }

    public static int testStub() {

        Object preStub = loadAVXConstant(loadAVXConstant);

        // do something to potentially destroy the value

        TestStub.testStub();

        Object postStub = loadAVXConstant(loadAVXConstant);

        return compareAVXRegister(compareAVXRegister, preStub, postStub);

    }

    @Test

    public void test() {

        HotSpotProviders providers = (HotSpotProviders) getProviders();

        HotSpotForeignCallsProviderImpl foreignCalls = (HotSpotForeignCallsProviderImpl) providers.getForeignCalls();

        linkage.setCompiledStub(new TestStub(getInitialOptions(), providers, linkage));

        runTest("testStub");

    }

}