/*

  * Copyright (c) 2005, 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.  Oracle designates this

  * particular file as subject to the "Classpath" exception as provided

  * by Oracle in the LICENSE file that accompanied this code.

  *

  * 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.apidesign.javap;

 import java.io.*;

 import static org.apidesign.javap.RuntimeConstants.*;

 /* represents one entry of StackMapTable attribute

  */

 abstract class StackMapTableData {

     final int frameType;

     int offsetDelta;

     StackMapTableData(int frameType) {

         this.frameType = frameType;

     }

     abstract void applyTo(TypeArray localTypes, TypeArray stackTypes);

     protected static String toString(

             final String frameType,

             final int offset,

             final int[] localTypes,

             final int[] stackTypes) {

         final StringBuilder sb = new StringBuilder(frameType);

         sb.append("(off: +").append(offset);

         if (localTypes != null) {

             sb.append(", locals: ");

             appendTypes(sb, localTypes);

         }

         if (stackTypes != null) {

             sb.append(", stack: ");

             appendTypes(sb, stackTypes);

         }

         sb.append(')');

         return sb.toString();

     }

     private static void appendTypes(final StringBuilder sb, final int[] types) {

         sb.append('[');

         if (types.length > 0) {

             sb.append(TypeArray.typeString(types[0]));

             for (int i = 1; i < types.length; ++i) {

                 sb.append(", ");

                 sb.append(TypeArray.typeString(types[i]));

             }

         }

         sb.append(']');

     }

     static class SameFrame extends StackMapTableData {

         SameFrame(int frameType, int offsetDelta) {

             super(frameType);

             this.offsetDelta = offsetDelta;

         }

         @Override

         void applyTo(TypeArray localTypes, TypeArray stackTypes) {

             stackTypes.clear();

         }

         @Override

         public String toString() {

             return toString("SAME" + ((frameType == SAME_FRAME_EXTENDED)

                                           ? "_FRAME_EXTENDED" : ""),

                             offsetDelta,

                             null, null);

         }

     }

     static class SameLocals1StackItem extends StackMapTableData {

         final int[] stack;

         SameLocals1StackItem(int frameType, int offsetDelta, int[] stack) {

             super(frameType);

             this.offsetDelta = offsetDelta;

             this.stack = stack;

         }

         @Override

         void applyTo(TypeArray localTypes, TypeArray stackTypes) {

             stackTypes.setAll(stack);

         }

         @Override

         public String toString() {

             return toString(

                        "SAME_LOCALS_1_STACK_ITEM"

                            + ((frameType == SAME_LOCALS_1_STACK_ITEM_EXTENDED)

                                   ? "_EXTENDED" : ""),

                        offsetDelta,

                        null, stack);

         }

     }

     static class ChopFrame extends StackMapTableData {

         ChopFrame(int frameType, int offsetDelta) {

             super(frameType);

             this.offsetDelta = offsetDelta;

         }

         @Override

         void applyTo(TypeArray localTypes, TypeArray stackTypes) {

             localTypes.setSize(localTypes.getSize()

                                    - (SAME_FRAME_EXTENDED - frameType));

             stackTypes.clear();

         }

         @Override

         public String toString() {

             return toString("CHOP", offsetDelta, null, null);

         }

     }

     static class AppendFrame extends StackMapTableData {

         final int[] locals;

         AppendFrame(int frameType, int offsetDelta, int[] locals) {

             super(frameType);

             this.offsetDelta = offsetDelta;

             this.locals = locals;

         }

         @Override

         void applyTo(TypeArray localTypes, TypeArray stackTypes) {

             localTypes.addAll(locals);

             stackTypes.clear();

         }

         @Override

         public String toString() {

             return toString("APPEND", offsetDelta, locals, null);

         }

     }

     static class FullFrame extends StackMapTableData {

         final int[] locals;

         final int[] stack;

         FullFrame(int offsetDelta, int[] locals, int[] stack) {

             super(FULL_FRAME);

             this.offsetDelta = offsetDelta;

             this.locals = locals;

             this.stack = stack;

         }

         @Override

         void applyTo(TypeArray localTypes, TypeArray stackTypes) {

             localTypes.setAll(locals);

             stackTypes.setAll(stack);

         }

         @Override

         public String toString() {

             return toString("FULL", offsetDelta, locals, stack);

         }

     }

     static StackMapTableData getInstance(DataInputStream in, MethodData method)

                   throws IOException {

         int frameType = in.readUnsignedByte();

         if (frameType < SAME_FRAME_BOUND) {

             // same_frame

             return new SameFrame(frameType, frameType);

         } else if (SAME_FRAME_BOUND <= frameType && frameType < SAME_LOCALS_1_STACK_ITEM_BOUND) {

             // same_locals_1_stack_item_frame

             // read additional single stack element

             return new SameLocals1StackItem(frameType,

                                             (frameType - SAME_FRAME_BOUND),

                                             StackMapData.readTypeArray(in, 1, method));

         } else if (frameType == SAME_LOCALS_1_STACK_ITEM_EXTENDED) {

             // same_locals_1_stack_item_extended

             return new SameLocals1StackItem(frameType,

                                             in.readUnsignedShort(),

                                             StackMapData.readTypeArray(in, 1, method));

         } else if (SAME_LOCALS_1_STACK_ITEM_EXTENDED < frameType  && frameType < SAME_FRAME_EXTENDED) {

             // chop_frame or same_frame_extended

             return new ChopFrame(frameType, in.readUnsignedShort());

         } else if (frameType == SAME_FRAME_EXTENDED) {

             // chop_frame or same_frame_extended

             return new SameFrame(frameType, in.readUnsignedShort());

         } else if (SAME_FRAME_EXTENDED < frameType  && frameType < FULL_FRAME) {

             // append_frame

             return new AppendFrame(frameType, in.readUnsignedShort(),

                                    StackMapData.readTypeArray(in, frameType - SAME_FRAME_EXTENDED, method));

         } else if (frameType == FULL_FRAME) {

             // full_frame

             int offsetDelta = in.readUnsignedShort();

             int locals_size = in.readUnsignedShort();

             int[] locals = StackMapData.readTypeArray(in, locals_size, method);

             int stack_size = in.readUnsignedShort();

             int[] stack = StackMapData.readTypeArray(in, stack_size, method);

             return new FullFrame(offsetDelta, locals, stack);

         } else {

             throw new ClassFormatError("unrecognized frame_type in StackMapTable");

         }

     }

 }