Home | History | Annotate | Download | only in editor
      1 /*
      2  * ProGuard -- shrinking, optimization, obfuscation, and preverification
      3  *             of Java bytecode.
      4  *
      5  * Copyright (c) 2002-2014 Eric Lafortune (eric (at) graphics.cornell.edu)
      6  *
      7  * This program is free software; you can redistribute it and/or modify it
      8  * under the terms of the GNU General Public License as published by the Free
      9  * Software Foundation; either version 2 of the License, or (at your option)
     10  * any later version.
     11  *
     12  * This program is distributed in the hope that it will be useful, but WITHOUT
     13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
     14  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
     15  * more details.
     16  *
     17  * You should have received a copy of the GNU General Public License along
     18  * with this program; if not, write to the Free Software Foundation, Inc.,
     19  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
     20  */
     21 package proguard.classfile.editor;
     22 
     23 import proguard.classfile.*;
     24 import proguard.classfile.attribute.*;
     25 import proguard.classfile.attribute.visitor.*;
     26 import proguard.classfile.util.SimplifiedVisitor;
     27 
     28 /**
     29  * This AttributeVisitor computes and updates the maximum stack size of the
     30  * code attributes that it visits.
     31  *
     32  * @author Eric Lafortune
     33  */
     34 public class StackSizeUpdater
     35 extends      SimplifiedVisitor
     36 implements   AttributeVisitor
     37 {
     38     private final StackSizeComputer stackSizeComputer = new StackSizeComputer();
     39 
     40 
     41     // Implementations for AttributeVisitor.
     42 
     43     public void visitAnyAttribute(Clazz clazz, Attribute attribute) {}
     44 
     45 
     46     public void visitCodeAttribute(Clazz clazz, Method method, CodeAttribute codeAttribute)
     47     {
     48         // Compute the stack sizes.
     49         stackSizeComputer.visitCodeAttribute(clazz, method, codeAttribute);
     50 
     51         // Update the maximum stack size.
     52         codeAttribute.u2maxStack = stackSizeComputer.getMaxStackSize();
     53     }
     54 }
     55