Home | History | Annotate | Download | only in visitor 
      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.visitor;
     22 
     23 import proguard.classfile.*;
     24 
     25 
     26 /**
     27  * This ClassVisitor delegates all visits to each ClassVisitor
     28  * in a given list.
     29  *
     30  * @author Eric Lafortune
     31  */
     32 public class MultiClassVisitor implements ClassVisitor
     33 {
     34     private static final int ARRAY_SIZE_INCREMENT = 5;
     35 
     36     private ClassVisitor[] classVisitors;
     37     private int            classVisitorCount;
     38 
     39 
     40     public MultiClassVisitor()
     41     {
     42     }
     43 
     44 
     45     public MultiClassVisitor(ClassVisitor[] classVisitors)
     46     {
     47         this.classVisitors     = classVisitors;
     48         this.classVisitorCount = classVisitors.length;
     49     }
     50 
     51 
     52     public void addClassVisitor(ClassVisitor classVisitor)
     53     {
     54         ensureArraySize();
     55 
     56         classVisitors[classVisitorCount++] = classVisitor;
     57     }
     58 
     59 
     60     private void ensureArraySize()
     61     {
     62         if (classVisitors == null)
     63         {
     64             classVisitors = new ClassVisitor[ARRAY_SIZE_INCREMENT];
     65         }
     66         else if (classVisitors.length == classVisitorCount)
     67         {
     68             ClassVisitor[] newClassVisitors =
     69                 new ClassVisitor[classVisitorCount +
     70                                      ARRAY_SIZE_INCREMENT];
     71             System.arraycopy(classVisitors, 0,
     72                              newClassVisitors, 0,
     73                              classVisitorCount);
     74             classVisitors = newClassVisitors;
     75         }
     76     }
     77 
     78 
     79     // Implementations for ClassVisitor.
     80 
     81     public void visitProgramClass(ProgramClass programClass)
     82     {
     83         for (int index = 0; index < classVisitorCount; index++)
     84         {
     85             classVisitors[index].visitProgramClass(programClass);
     86         }
     87     }
     88 
     89 
     90     public void visitLibraryClass(LibraryClass libraryClass)
     91     {
     92         for (int index = 0; index < classVisitorCount; index++)
     93         {
     94             classVisitors[index].visitLibraryClass(libraryClass);
     95         }
     96     }
     97 }
     98