1/*
2 * Copyright (c) 2005, 2015, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24package jdk.test.lib.jittester.classes;
25
26import jdk.test.lib.jittester.IRNode;
27import jdk.test.lib.jittester.types.TypeKlass;
28import jdk.test.lib.jittester.visitors.Visitor;
29
30public class MainKlass extends IRNode {
31    public enum MainKlassPart {
32        DATA_MEMBERS,
33        MEMBER_FUNCTIONS,
34        TEST_FUNCTION,
35        PRINT_VARIABLES,
36    }
37
38    private final String name;
39    private final TypeKlass thisKlass;
40
41    public MainKlass(String name, TypeKlass thisKlass, IRNode variableDeclarations,
42            IRNode functionDefinitions, IRNode testFunction, IRNode printVariables) {
43        super(thisKlass);
44        addChild(variableDeclarations);
45        addChild(functionDefinitions);
46        addChild(testFunction);
47        addChild(printVariables);
48        this.name = name;
49        this.thisKlass = thisKlass;
50    }
51
52    @Override
53    public long complexity() {
54        IRNode dataMembers = getChild(MainKlassPart.DATA_MEMBERS.ordinal());
55        IRNode testFunction = getChild(MainKlassPart.TEST_FUNCTION.ordinal());
56        return dataMembers.complexity() + testFunction.complexity();
57    }
58
59    @Override
60    public<T> T accept(Visitor<T> v) {
61        return v.visit(this);
62    }
63
64    @Override
65    public String getName() {
66        return name;
67    }
68
69    public TypeKlass getThisKlass() {
70        return thisKlass;
71    }
72}
73