MultiClassVisitor.java revision 9f606f95f03a75961498803e24bee6799a7c0885 
1/*
2 * ProGuard -- shrinking, optimization, obfuscation, and preverification
3 *             of Java bytecode.
4 *
5 * Copyright (c) 2002-2009 Eric Lafortune (eric@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 */
21package proguard.classfile.visitor;
22
23import 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 */
32public 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