/*
 * Copyright (c) 2019, 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.
 *
 */

import java.util.ArrayList;
import jdk.internal.vm.annotation.Contended;

public class FieldLayoutApp {
    public static void main(String args[]) {
        ArrayList<TestObject> list = new ArrayList<>();

        for (int i=0; i<400; i++) {
            list.add(new Base1());
            list.add(new Child1());
            list.add(new Base2());
            list.add(new Child2());
        }

        verifyAll(list);

        // Make sure the oopmaps are laid out correctly.
        System.gc();
        verifyAll(list);
    }

    static void verifyAll(ArrayList<TestObject> list) {
        for (TestObject obj : list) {
            obj.verify();
        }
    }

    static long lastUID = 0;
    synchronized static long makeUID() {
        return ++lastUID;
    }

    synchronized static void verifyUID(long uid) {
        if (uid <= 0 || uid > lastUID) {
            error("Unexpected UID " + uid + ", must be > 0 and <= " + lastUID);
        }
    }

    static void error(String s) {
        throw new RuntimeException(s);
    }

    static void ensure(boolean b) {
        if (!b) {
            error("Assertion failed");
        }
    }

    static String makeString(long n) {
        return Long.toString(n);
    }

    static class TestObject {
        void verify() {}
    }

    static class Base1 extends TestObject {
        byte b1, b2;
        String s1;
        byte b3, b4;
        String s2;
        long uid;
        long l1;
        int i1;
        long l2;
        int i2;

        Base1() {
            uid = makeUID();
            b1 = 1;
            b2 = 2;
            b3 = 3;
            b4 = 4;
            s1 = makeString(uid + 1);
            s2 = makeString(uid + 2);
            i1 = 101;
            i2 = 102;
            l1 = 1001;
            l2 = 1002;
        }

        void verify() {
            super.verify();
            ensure(b1 == 1);
            ensure(b2 == 2);
            ensure(b3 == 3);
            ensure(b4 == 4);
            verifyUID(uid);
            ensure(s1.equals(makeString(uid + 1)));
            ensure(s2.equals(makeString(uid + 2)));
            ensure(i1 == 101);
            ensure(i2 == 102);
            ensure(l1 == 1001);
            ensure(l2 == 1002);
        }
    }

    // Base1 is archived but Child1 is loaded dynamically at runtime. Base1 may be
    // archived with different field layout options that those used during runtime.
    static class Child1 extends Base1 {
        byte cb1, cb2;
        String cs1;
        byte cb3, cb4;
        String cs2;
        long cuid;
        long cl1;
        int ci1;
        long cl2;
        int ci2;

        Child1() {
            cuid = makeUID();
            cb1 = 1;
            cb2 = 2;
            cb3 = 3;
            cb4 = 4;
            cs1 = makeString(cuid + 1);
            cs2 = makeString(cuid + 2);
            ci1 = 101;
            ci2 = 102;
            cl1 = 1001;
            cl2 = 1002;
        }

        void verify() {
            super.verify();
            ensure(cb1 == 1);
            ensure(cb2 == 2);
            ensure(cb3 == 3);
            ensure(cb4 == 4);
            verifyUID(uid);
            ensure(cs1.equals(makeString(cuid + 1)));
            ensure(cs2.equals(makeString(cuid + 2)));
            ensure(ci1 == 101);
            ensure(ci2 == 102);
            ensure(cl1 == 1001);
            ensure(cl2 == 1002);

            // Check the fields declared by the super class:
            ensure(b1 == 1); // javac should generate a FieldRef of FieldLayoutApp$Child1.b1:B,
                             // even though b1 is declared in the super class.
            ensure(b2 == 2);
            ensure(b3 == 3);
            ensure(b4 == 4);
            verifyUID(uid);
            ensure(s1.equals(makeString(uid + 1)));
            ensure(s2.equals(makeString(uid + 2)));

            ensure(i1 == 101);
            ensure(i2 == 102);
            ensure(l1 == 1001);
            ensure(l2 == 1002);
        }
    }

    // Same as Base1 - minus the i1, i2, l1, l2 fields, plus some @Contended annotations
    static class Base2 extends TestObject {
        byte b1, b2;
        String s1;
        @Contended byte b3, b4;
        @Contended String s2;
        long uid;

        Base2() {
            uid = makeUID();
            b1 = 1;
            b2 = 2;
            b3 = 3;
            b4 = 4;
            s1 = makeString(uid + 1);
            s2 = makeString(uid + 2);
        }

        void verify() {
            super.verify();
            ensure(b1 == 1);
            ensure(b2 == 2);
            ensure(b3 == 3);
            ensure(b4 == 4);
            verifyUID(uid);
            ensure(s1.equals(makeString(uid + 1)));
            ensure(s2.equals(makeString(uid + 2)));
        }
    }

    // Same as Child2 - minus the ci1, ci2, cl1, cl2 fields, plus some @Contended annotations
    //
    // Base2 is archived but Child2 is loaded dynamically at runtime. Base2 may be
    // archived with different field layout options that those used during runtime.
    static class Child2 extends Base2 {
        byte cb1, cb2;
        @Contended String cs1;
        byte cb3, cb4;
        String cs2;
        @Contended long cuid;

        Child2() {
            cuid = makeUID();
            cb1 = 1;
            cb2 = 2;
            cb3 = 3;
            cb4 = 4;
            cs1 = makeString(cuid + 1);
            cs2 = makeString(cuid + 2);
        }

        void verify() {
            super.verify();
            ensure(cb1 == 1);
            ensure(cb2 == 2);
            ensure(cb3 == 3);
            ensure(cb4 == 4);
            verifyUID(uid);
            ensure(cs1.equals(makeString(cuid + 1)));
            ensure(cs2.equals(makeString(cuid + 2)));

            // Check the fields declared by the super class:
            ensure(b1 == 1); // javac should generate a FieldRef of FieldLayoutApp$Child2.b1:B,
                             // even though b1 is declared in the super class.
            ensure(b2 == 2);
            ensure(b3 == 3);
            ensure(b4 == 4);
            verifyUID(uid);
            ensure(s1.equals(makeString(uid + 1)));
            ensure(s2.equals(makeString(uid + 2)));
        }
    }
}