1/*
2 * ProGuard -- shrinking, optimization, obfuscation, and preverification
3 *             of Java bytecode.
4 *
5 * Copyright (c) 2002-2014 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;
22
23import proguard.classfile.visitor.*;
24
25import java.util.*;
26
27/**
28 * This is a set of representations of classes. They can be enumerated or
29 * retrieved by name. They can also be accessed by means of class visitors.
30 *
31 * @author Eric Lafortune
32 */
33public class ClassPool
34{
35    // We're using a sorted tree map instead of a hash map to store the classes,
36    // in order to make the processing more deterministic.
37    private final Map classes = new TreeMap();
38
39
40    /**
41     * Clears the class pool.
42     */
43    public void clear()
44    {
45        classes.clear();
46    }
47
48
49    /**
50     * Adds the given Clazz to the class pool.
51     */
52    public void addClass(Clazz clazz)
53    {
54        classes.put(clazz.getName(), clazz);
55    }
56
57
58    /**
59     * Removes the given Clazz from the class pool.
60     */
61    public void removeClass(Clazz clazz)
62    {
63        removeClass(clazz.getName());
64    }
65
66
67    /**
68     * Removes the specified Clazz from the class pool.
69     */
70    public void removeClass(String className)
71    {
72        classes.remove(className);
73    }
74
75
76    /**
77     * Returns a Clazz from the class pool based on its name. Returns
78     * <code>null</code> if the class with the given name is not in the class
79     * pool.
80     */
81    public Clazz getClass(String className)
82    {
83        return (Clazz)classes.get(className);
84    }
85
86
87    /**
88     * Returns an Iterator of all class names in the class pool.
89     */
90    public Iterator classNames()
91    {
92        return classes.keySet().iterator();
93    }
94
95
96    /**
97     * Returns the number of classes in the class pool.
98     */
99    public int size()
100    {
101        return classes.size();
102    }
103
104
105    /**
106     * Applies the given ClassPoolVisitor to the class pool.
107     */
108    public void accept(ClassPoolVisitor classPoolVisitor)
109    {
110        classPoolVisitor.visitClassPool(this);
111    }
112
113
114    /**
115     * Applies the given ClassVisitor to all classes in the class pool,
116     * in random order.
117     */
118    public void classesAccept(ClassVisitor classVisitor)
119    {
120        Iterator iterator = classes.values().iterator();
121        while (iterator.hasNext())
122        {
123            Clazz clazz = (Clazz)iterator.next();
124            clazz.accept(classVisitor);
125        }
126    }
127
128
129    /**
130     * Applies the given ClassVisitor to all classes in the class pool,
131     * in sorted order.
132     */
133    public void classesAcceptAlphabetically(ClassVisitor classVisitor)
134    {
135        // We're already using a tree map.
136        //TreeMap sortedClasses = new TreeMap(classes);
137        //Iterator iterator = sortedClasses.values().iterator();
138
139        Iterator iterator = classes.values().iterator();
140        while (iterator.hasNext())
141        {
142            Clazz clazz = (Clazz)iterator.next();
143            clazz.accept(classVisitor);
144        }
145    }
146
147
148    /**
149     * Applies the given ClassVisitor to the class with the given name,
150     * if it is present in the class pool.
151     */
152    public void classAccept(String className, ClassVisitor classVisitor)
153    {
154        Clazz clazz = getClass(className);
155        if (clazz != null)
156        {
157            clazz.accept(classVisitor);
158        }
159    }
160}
161