/*
 * Copyright (c) 2025 IBM Corporation. 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.
 */

/*
 * @test
 * @bug 8373396
 * @summary Verify that min/max add ideal optimizations get applied correctly
 * @modules java.base/jdk.internal.misc
 * @library /test/lib /
 * @run driver compiler.c2.irTests.TestMinMaxIdeal
 */

package compiler.c2.irTests;

import compiler.lib.compile_framework.CompileFramework;
import compiler.lib.template_framework.Template;
import compiler.lib.template_framework.TemplateToken;
import compiler.lib.template_framework.library.CodeGenerationDataNameType;
import compiler.lib.template_framework.library.PrimitiveType;
import compiler.lib.template_framework.library.TestFrameworkClass;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import static compiler.lib.template_framework.Template.let;
import static compiler.lib.template_framework.Template.scope;

public class TestMinMaxIdeal {
    public static void main(String[] args) {
        // Create a new CompileFramework instance.
        CompileFramework comp = new CompileFramework();

        String templatedPackage ="compiler.c2.templated";
        String templatedClassName ="MinMaxIdeal";
        String templatedFQN = "%s.%s".formatted(templatedPackage, templatedClassName);

        // Add a java source file.
        comp.addJavaSourceCode(templatedFQN, generate(templatedClassName, templatedPackage, comp));

        // Compile the source file.
        comp.compile();

        comp.invoke(templatedFQN, "main", new Object[] {new String[] {}});
    }

    private static String generate(String templatedClassName, String templatedPackage, CompileFramework comp) {
        // Create a list to collect all tests.
        List<TemplateToken> testTemplateTokens = new ArrayList<>();

        testTemplateTokens.add(new TestGenerator(Op.MAX_I).generate());
        testTemplateTokens.add(new TestGenerator(Op.MAX_L).generate());

        // Create the test class, which runs all testTemplateTokens.
        return TestFrameworkClass.render(
            // package and class name.
            templatedPackage, templatedClassName,
            // List of imports.
            Collections.emptySet(),
            // classpath, so the Test VM has access to the compiled class files.
            comp.getEscapedClassPathOfCompiledClasses(),
            // The list of tests.
            testTemplateTokens);
    }

    enum Op {
        MAX_I("max", CodeGenerationDataNameType.ints()),
        MAX_L("max", CodeGenerationDataNameType.longs());

        final String functionName;
        final PrimitiveType type;

        Op(String functionName, PrimitiveType type) {
            this.functionName = functionName;
            this.type = type;
        }
    }

    record TestGenerator(Op op) {
        TemplateToken generate() {
            var template = Template.make(() -> scope(
                let("irNodeName", op.name()),
                let("boxedTypeName", op.type.boxedTypeName()),
                let("op", op.name()),
                let("type", op.type.name()),
                let("functionName", op.functionName),
                """
                @Test
                @IR(counts = {IRNode.#irNodeName, "= 0"})
                @Arguments(values = {Argument.NUMBER_42, Argument.NUMBER_42})
                public #type test_commute_#op(#type a, #type b) {
                    return #boxedTypeName.#functionName(a, b) + #boxedTypeName.#functionName(b, a);
                }

                @Test
                @IR(counts = {IRNode.#irNodeName, "= 0"})
                @Arguments(values = {Argument.NUMBER_42})
                public #type test_flatten_#op(#type a) {
                    return #boxedTypeName.#functionName(#boxedTypeName.#functionName(a, 1), 2);
                }

                @Test
                @IR(counts = {IRNode.#irNodeName, "= 0"})
                @Arguments(values = {Argument.NUMBER_42, Argument.NUMBER_42})
                public #type test_push_constant_left_#op(#type a, #type b) {
                    return #boxedTypeName.#functionName(#boxedTypeName.#functionName(a, 1), b) + #boxedTypeName.#functionName(b, a);
                }

                @Test
                @IR(counts = {IRNode.#irNodeName, "= 0"})
                @Arguments(values = {Argument.NUMBER_42, Argument.NUMBER_42})
                public #type test_push_constant_right_#op(#type a, #type b) {
                    return #boxedTypeName.#functionName(a, #boxedTypeName.#functionName(b, 1)) + #boxedTypeName.#functionName(b, a);
                }
                """
            ));
            return template.asToken();
        }
    }
}