test/hotspot/jtreg/testlibrary/jittester/src/jdk/test/lib/jittester/Automatic.java

/*
 * Copyright (c) 2005, 2016, 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 jdk.test.lib.jittester;

import jdk.test.lib.util.Pair;
import jdk.test.lib.jittester.factories.IRNodeBuilder;
import jdk.test.lib.jittester.types.TypeKlass;
import jdk.test.lib.jittester.utils.FixedTrees;
import jdk.test.lib.jittester.utils.OptionResolver;
import jdk.test.lib.jittester.utils.OptionResolver.Option;
import jdk.test.lib.jittester.utils.PseudoRandom;
import java.time.LocalTime;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeUnit;
import java.util.function.Function;

public class Automatic {
    public static final int MINUTES_TO_WAIT = Integer.getInteger("jdk.test.lib.jittester", 3);

    private static Pair<IRNode, IRNode> generateIRTree(String name) {
        SymbolTable.removeAll();
        TypeList.removeAll();

        IRNodeBuilder builder = new IRNodeBuilder()
                .setPrefix(name)
                .setName(name)
                .setLevel(0);

        Long complexityLimit = ProductionParams.complexityLimit.value();
        IRNode privateClasses = null;
        if (!ProductionParams.disableClasses.value()) {
            long privateClassComlexity = (long) (complexityLimit * PseudoRandom.random());
            try {
                privateClasses = builder.setComplexityLimit(privateClassComlexity)
                        .getClassDefinitionBlockFactory()
                        .produce();
            } catch (ProductionFailedException ex) {
                ex.printStackTrace(System.out);
            }
        }
        long mainClassComplexity = (long) (complexityLimit * PseudoRandom.random());
        IRNode mainClass = null;
        try {
            mainClass = builder.setComplexityLimit(mainClassComplexity)
                    .getMainKlassFactory()
                    .produce();
            TypeKlass aClass = new TypeKlass(name);
            mainClass.getChild(1).addChild(FixedTrees.generateMainOrExecuteMethod(aClass, true));
            mainClass.getChild(1).addChild(FixedTrees.generateMainOrExecuteMethod(aClass, false));
        } catch (ProductionFailedException ex) {
            ex.printStackTrace(System.out);
        }
        return new Pair<>(mainClass, privateClasses);
    }

    private static void initializeTestGenerator(String[] params) {
        OptionResolver parser = new OptionResolver();
        Option<String> propertyFileOpt = parser.addStringOption('p', "property-file",
                "conf/default.properties", "File to read properties from");
        ProductionParams.register(parser);
        parser.parse(params, propertyFileOpt);
        PseudoRandom.reset(ProductionParams.seed.value());
        TypesParser.parseTypesAndMethods(ProductionParams.classesFile.value(),
                ProductionParams.excludeMethodsFile.value());
        if (ProductionParams.specificSeed.isSet()) {
            PseudoRandom.setCurrentSeed(ProductionParams.specificSeed.value());
        }
    }

    private static List<TestsGenerator> getTestGenerators() {
        List<TestsGenerator> result = new ArrayList<>();
        Class<?> factoryClass;
        Function<String[], List<TestsGenerator>> factory;
        String[] factoryClassNames = ProductionParams.generatorsFactories.value().split(",");
        String[] generatorNames = ProductionParams.generators.value().split(",");
        for (String factoryClassName : factoryClassNames) {
            try {
                factoryClass = Class.forName(factoryClassName);
                factory = (Function<String[], List<TestsGenerator>>) factoryClass.newInstance();
            } catch (ReflectiveOperationException roe) {
                throw new Error("Can't instantiate generators factory", roe);
            }
            result.addAll(factory.apply(generatorNames));
        }
        return result;
    }

    public static void main(String[] args) {
        initializeTestGenerator(args);
        int counter = 0;
        System.out.printf(" %13s | %8s | %8s | %8s |%n", "start time", "count", "generat",
                "running");
        System.out.printf(" %13s | %8s | %8s | %8s |%n", "---", "---", "---", "---");
        List<TestsGenerator> generators = getTestGenerators();
        do {
            double start = System.currentTimeMillis();
            System.out.print("[" + LocalTime.now() + "] |");
            String name = "Test_" + counter;
            Pair<IRNode, IRNode> irTree = generateIRTree(name);
            System.out.printf(" %8d |", counter);
            long maxWaitTime = TimeUnit.MINUTES.toMillis(MINUTES_TO_WAIT);
            double generationTime = System.currentTimeMillis() - start;
            System.out.printf(" %8.0f |", generationTime);
            start = System.currentTimeMillis();
            Thread generatorThread = new Thread(() -> {
                for (TestsGenerator generator : generators) {
                        generator.accept(irTree.first, irTree.second);
                }
            });
            generatorThread.start();
            try {
                generatorThread.join(maxWaitTime);
            } catch (InterruptedException ie) {
                throw new Error("Test generation interrupted: " + ie, ie);
            }
            if (generatorThread.isAlive()) {
                // maxTime reached, so, proceed to next test generation
                generatorThread.interrupt();
            } else {
                double runningTime = System.currentTimeMillis() - start;
                System.out.printf(" %8.0f |%n", runningTime);
                if (runningTime < maxWaitTime) {
                    ++counter;
                }
            }
        } while (counter < ProductionParams.numberOfTests.value());
    }
}