Home | History | Annotate | Download | only in info 
      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.optimize.info;
     22 
     23 import proguard.classfile.*;
     24 import proguard.classfile.visitor.ClassVisitor;
     25 
     26 /**
     27  * This ClassVisitor marks all program classes that it visits with a given
     28  * flag for simple enums.
     29  *
     30  * @author Eric Lafortune
     31  */
     32 public class SimpleEnumMarker
     33 implements   ClassVisitor
     34 {
     35     private final boolean simple;
     36 
     37 
     38     /**
     39      * Creates a new SimpleEnumMarker that marks visited classes with the
     40      * given flag.
     41      */
     42     public SimpleEnumMarker(boolean simple)
     43     {
     44         this.simple = simple;
     45     }
     46 
     47 
     48     // Implementations for ClassVisitor.
     49 
     50     public void visitLibraryClass(LibraryClass libraryClass) {}
     51 
     52     public void visitProgramClass(ProgramClass programClass)
     53     {
     54         setSimpleEnum(programClass);
     55     }
     56 
     57 
     58     // Small utility methods.
     59 
     60     private void setSimpleEnum(Clazz clazz)
     61     {
     62         ClassOptimizationInfo info = ClassOptimizationInfo.getClassOptimizationInfo(clazz);
     63         if (info != null)
     64         {
     65             info.setSimpleEnum(simple);
     66         }
     67     }
     68 
     69 
     70     public static boolean isSimpleEnum(Clazz clazz)
     71     {
     72         ClassOptimizationInfo info = ClassOptimizationInfo.getClassOptimizationInfo(clazz);
     73         return info != null && info.isSimpleEnum();
     74     }
     75 }