/*
 * Copyright (c) 2022, 2023, 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 org.openjdk.bench.jdk.classfile;

import java.lang.reflect.AccessFlag;
import jdk.internal.classfile.ClassElement;
import jdk.internal.classfile.ClassModel;
import jdk.internal.classfile.Classfile;
import jdk.internal.classfile.FieldModel;
import jdk.internal.org.objectweb.asm.*;
import jdk.internal.org.objectweb.asm.tree.*;
import org.openjdk.jmh.annotations.*;
import org.openjdk.jmh.infra.Blackhole;

public class ReadMetadata extends AbstractCorpusBenchmark {

    @Benchmark
    @BenchmarkMode(Mode.Throughput)
    public void asmStreamReadName(Blackhole bh) {
        for (byte[] bytes : classes) {
            ClassReader cr = new ClassReader(bytes);
            var  visitor = new ClassVisitor(Opcodes.ASM9) {
                String theName;

                @Override
                public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {
                    theName = name;
                }
            };
            cr.accept(visitor, 0);
            bh.consume(visitor.theName);
        }
    }

    @Benchmark
    @BenchmarkMode(Mode.Throughput)
    public void asmTreeReadName(Blackhole bh) {
        for (byte[] bytes : classes) {
            ClassNode node = new ClassNode();
            ClassReader cr = new ClassReader(bytes);
            cr.accept(node, 0);
            bh.consume(node.name);
        }
    }

    @Benchmark
    @BenchmarkMode(Mode.Throughput)
    public void jdkReadName(Blackhole bh) {
        var cc = Classfile.of();
        for (byte[] bytes : classes) {
            bh.consume(cc.parse(bytes).thisClass().asInternalName());
        }
    }

    @Benchmark
    @BenchmarkMode(Mode.Throughput)
    public void jdkReadMemberNames(Blackhole bh) {
        var cc = Classfile.of();
        for (byte[] bytes : classes) {
            var cm = cc.parse(bytes);
            bh.consume(cm.thisClass().asInternalName());
            for (var f : cm.fields()) {
                bh.consume(f.fieldName().stringValue());
                bh.consume(f.fieldType().stringValue());
            }
            for (var m : cm.methods()) {
                bh.consume(m.methodName().stringValue());
                bh.consume(m.methodType().stringValue());
            }
        }
    }

    @Benchmark
    @BenchmarkMode(Mode.Throughput)
    public void asmStreamCountFields(Blackhole bh) {
        for (byte[] bytes : classes) {
            ClassReader cr = new ClassReader(bytes);
            var visitor = new ClassVisitor(Opcodes.ASM9) {
                int count;

                @Override
                public FieldVisitor visitField(int access, String name, String descriptor, String signature, Object value) {
                    if ((access & Opcodes.ACC_PUBLIC) != 1) {
                        ++count;
                    }
                    return null;
                }
            };
            cr.accept(visitor, 0);
            bh.consume(visitor.count);
        }
    }

    @Benchmark
    @BenchmarkMode(Mode.Throughput)
    public void asmTreeCountFields(Blackhole bh) {
        for (byte[] bytes : classes) {
            int count = 0;
            ClassNode node = new ClassNode();
            ClassReader cr = new ClassReader(bytes);
            cr.accept(node, 0);
            for (FieldNode fn : node.fields)
                if ((fn.access & Opcodes.ACC_PUBLIC) != 1) {
                    ++count;
                }
            bh.consume(count);
        }
    }

    @Benchmark
    @BenchmarkMode(Mode.Throughput)
    public void jdkTreeCountFields(Blackhole bh) {
        var cc = Classfile.of();
        for (byte[] bytes : classes) {
            int count = 0;
            ClassModel cm = cc.parse(bytes);
            for (FieldModel fm : cm.fields())
                if (!fm.flags().has(AccessFlag.PUBLIC)) {
                    ++count;
                }
            bh.consume(count);
        }
    }

    @Benchmark
    @BenchmarkMode(Mode.Throughput)
    public void jdkCountFields(Blackhole bh) {
        var cc = Classfile.of();
        for (byte[] bytes : classes) {
            int count = 0;
            ClassModel cm = cc.parse(bytes);
            for (ClassElement ce : cm) {
                if (ce instanceof FieldModel fm) {
                    if (!fm.flags().has(AccessFlag.PUBLIC)) {
                        ++count;
                    }
                }
            }
            bh.consume(count);
        }
    }
}
