/*

  * Copyright (c) 2012, 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;

 public final class StackMapIterator {

     private static final StackMapTableData INITIAL_FRAME =

             new StackMapTableData(-1) {

                 @Override

                 void applyTo(TypeArray localTypes, TypeArray stackTypes) {

                     localTypes.clear();

                     stackTypes.clear();

                 }

                 @Override

                 public String toString() {

                     return toString("INITIAL", 0, null, null);

                 }

             };

     private final StackMapTableData[] stackMapTable;

     private final TypeArray localTypes;

     private final TypeArray stackTypes;

     private int nextFrameIndex;

     private int lastFrameByteCodeOffset;

     private int byteCodeOffset;

     StackMapIterator(final StackMapTableData[] stackMapTable) {

         this.stackMapTable = (stackMapTable != null)

                                  ? stackMapTable

                                  : new StackMapTableData[0];

         localTypes = new TypeArray();

         stackTypes = new TypeArray();

         lastFrameByteCodeOffset = -1;

         advanceBy(0);

     }

     public String getFrameAsString() {

         return getCurrentFrame().toString();

     }

     public int getFrameIndex() {

         return nextFrameIndex;

     }

     public TypeArray getFrameStack() {

         return stackTypes;

     }

     public void advanceBy(final int numByteCodes) {

         if (numByteCodes < 0) {

             throw new IllegalStateException("Forward only iterator");

         }

         byteCodeOffset += numByteCodes;

         while ((nextFrameIndex < stackMapTable.length)

                     && ((byteCodeOffset - lastFrameByteCodeOffset)

                             >= (stackMapTable[nextFrameIndex].offsetDelta

                                     + 1))) {

             final StackMapTableData nextFrame = stackMapTable[nextFrameIndex];

             lastFrameByteCodeOffset += nextFrame.offsetDelta + 1;

             nextFrame.applyTo(localTypes, stackTypes);

             ++nextFrameIndex;

         }

     }

     public void advanceTo(final int nextByteCodeOffset) {

         advanceBy(nextByteCodeOffset - byteCodeOffset);

     }

     private StackMapTableData getCurrentFrame() {

         return (nextFrameIndex == 0)

                 ? INITIAL_FRAME

                 : stackMapTable[nextFrameIndex - 1];

     }

 }