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

/*
 * @test
 * @bug 6700100
 * @summary small instance clone as loads/stores
 * @run main/othervm -XX:-BackgroundCompilation -XX:-UseOnStackReplacement -XX:CompileCommand=dontinline,TestInstanceCloneAsLoadsStores::m* TestInstanceCloneAsLoadsStores
 * @run main/othervm -XX:-BackgroundCompilation -XX:-UseOnStackReplacement -XX:CompileCommand=dontinline,TestInstanceCloneAsLoadsStores::m* -XX:+IgnoreUnrecognizedVMOptions -XX:+StressArrayCopyMacroNode TestInstanceCloneAsLoadsStores
 *
 */

import java.lang.reflect.*;
import java.util.*;

public class TestInstanceCloneAsLoadsStores {
    static class Base implements Cloneable {
        void initialize(Class c, int i) {
            for (Field f : c.getDeclaredFields()) {
                setVal(f, i);
                i++;
            }
            if (c != Base.class) {
                initialize(c.getSuperclass(), i);
            }
        }

        Base() {
            initialize(getClass(), 0);
        }

        void setVal(Field f, int i) {
            try {
                if (f.getType() == int.class) {
                    f.setInt(this, i);
                    return;
                } else if (f.getType() == short.class) {
                    f.setShort(this, (short)i);
                    return;
                } else if (f.getType() == byte.class) {
                    f.setByte(this, (byte)i);
                    return;
                } else if (f.getType() == long.class) {
                    f.setLong(this, i);
                    return;
                }
            } catch(IllegalAccessException iae) {
                throw new RuntimeException("Getting fields failed");
            }
            throw new RuntimeException("unexpected field type");
        }

        int getVal(Field f) {
            try {
                if (f.getType() == int.class) {
                    return f.getInt(this);
                } else if (f.getType() == short.class) {
                    return (int)f.getShort(this);
                } else if (f.getType() == byte.class) {
                    return (int)f.getByte(this);
                } else if (f.getType() == long.class) {
                    return (int)f.getLong(this);
                }
            } catch(IllegalAccessException iae) {
                throw new RuntimeException("Setting fields failed");
            }
            throw new RuntimeException("unexpected field type");
        }

        boolean fields_equal(Class c, Base o) {
            for (Field f : c.getDeclaredFields()) {
                if (getVal(f) != o.getVal(f)) {
                    return false;
                }
            }
            if (c != Base.class) {
                return fields_equal(c.getSuperclass(), o);
            }
            return true;
        }

        public boolean equals(Object obj) {
            return fields_equal(getClass(), (Base)obj);
        }
        
        String print_fields(Class c, String s) {
            for (Field f : c.getDeclaredFields()) {
                if (s != "") {
                    s += "\n";
                }
                s = s + f + " = " + getVal(f);
            }
            if (c != Base.class) {
                return print_fields(c.getSuperclass(), s);
            }
            return s;
        }

        public String toString() {
            return print_fields(getClass(), "");
        }

        int fields_sum(Class c, int s) {
            for (Field f : c.getDeclaredFields()) {
                s += getVal(f);
            }
            if (c != Base.class) {
                return fields_sum(c.getSuperclass(), s);
            }
            return s;
        }

        public int sum() {
            return fields_sum(getClass(), 0);
        }

    }

    static class A extends Base {
        int i1;
        int i2;
        int i3;
        int i4;
        int i5;

        public Object clone() throws CloneNotSupportedException {
            return super.clone();
        }
    }

    static class B extends A {
        int i6;
    }

    static final class D extends Base {
        byte  i1;
        short i2;
        long  i3;
        int   i4;
        int   i5;
        
        public Object clone() throws CloneNotSupportedException {
            return super.clone();
        }
    }

    static final class E extends Base {
        int i1;
        int i2;
        int i3;
        int i4;
        int i5;
        int i6;
        int i7;
        int i8;
        int i9;

        public Object clone() throws CloneNotSupportedException {
            return super.clone();
        }
    }

    static final class F extends Base {
        public Object clone() throws CloneNotSupportedException {
            return super.clone();
        }
    }

    static class G extends Base {
        int i1;
        int i2;
        int i3;
        
        public Object myclone() throws CloneNotSupportedException {
            return clone();
        }
    }

    static class H extends G {
        int i4;
        int i5;
        
        public Object clone() throws CloneNotSupportedException {
            return super.clone();
        }
    }

    static class J extends Base  {
        int i1;
        int i2;
        int i3;

        public Object myclone() throws CloneNotSupportedException {
            return clone();
        }
    }

    static class K extends J {
        int i4;
        int i5;
    }

    // Should be compiled as loads/stores
    static Object m1(D src) throws CloneNotSupportedException {
        return src.clone();
    }

    // Should be compiled as adds of src (dest allocation eliminated)
    static int m2(D src) throws CloneNotSupportedException {
        D dest = (D)src.clone();
        return dest.i1 + dest.i2 + ((int)dest.i3) + dest.i4 + dest.i5;
    }

    // Should be compiled as arraycopy stub call (object too large)
    static int m3(E src) throws CloneNotSupportedException {
        E dest = (E)src.clone();
        return dest.i1 + dest.i2 + dest.i3 + dest.i4 + dest.i5 +
            dest.i6 + dest.i7 + dest.i8 + dest.i9;
    }

    // Need profiling on src's type to be able to know number of
    // fields. Cannot clone as loads/stores if compile doesn't use it.
    static Object m4(A src) throws CloneNotSupportedException {
        return src.clone();
    }

    // Same as above but should optimize out dest allocation
    static int m5(A src) throws CloneNotSupportedException {
        A dest = (A)src.clone();
        return dest.i1 + dest.i2 + dest.i3 + dest.i4 + dest.i5;
    }

    // Check that if we have no fields to clone we do fine
    static Object m6(F src) throws CloneNotSupportedException {
        return src.clone();
    }

    // With virtual call to clone: clone inlined from profling which
    // gives us exact type of src so we can clone it with
    // loads/stores.
    static G m7(G src) throws CloneNotSupportedException {
        return (G)src.myclone();
    }

    // Virtual call to clone but single target: exact type unknown,
    // clone intrinsic uses profiling to determine exact type and
    // clone with loads/stores.
    static J m8(J src) throws CloneNotSupportedException {
        return (J)src.myclone();
    }

    final HashMap<String,Method> tests = new HashMap<>();
    {
        for (Method m : this.getClass().getDeclaredMethods()) {
            if (m.getName().matches("m[0-9]+")) {
                assert(Modifier.isStatic(m.getModifiers())) : m;
                tests.put(m.getName(), m);
            }
        }
    }

    boolean success = true;

    void doTest(Base src, String name) throws Exception {
        Method m = tests.get(name);

        for (int i = 0; i < 20000; i++) {
            boolean failure = false;
            Base res = null;
            int s = 0;
            if (m.getReturnType().isPrimitive()) {
                s = (int)m.invoke(null, src);
                failure = (s != src.sum());
            } else {
                res = (Base)m.invoke(null, src);
                failure = !res.equals(src);
            }
            if (failure) {
                System.out.println("Test " + name + " failed");
                System.out.println("source: ");
                System.out.println(src);
                System.out.println("result: ");
                if (m.getReturnType().isPrimitive()) {
                    System.out.println(s);
                } else {
                    System.out.println(res);
                }
                success = false;
                break;
            }
        }
    }

    public static void main(String[] args) throws Exception {

        TestInstanceCloneAsLoadsStores test = new TestInstanceCloneAsLoadsStores();

        A a = new A();
        B b = new B();
        D d = new D();
        E e = new E();
        F f = new F();
        G g = new G();
        H h = new H();
        J j = new J();
        K k = new K();

        test.doTest(d, "m1");
        test.doTest(d, "m2");
        test.doTest(e, "m3");
        test.doTest(a, "m4");
        test.doTest(a, "m5");
        test.doTest(f, "m6");
        test.doTest(g, "m7");
        test.doTest(k, "m8");

        if (!test.success) {
            throw new RuntimeException("some tests failed");
        }

    }
}