001/* 002 * Licensed to the Apache Software Foundation (ASF) under one or more 003 * contributor license agreements. See the NOTICE file distributed with 004 * this work for additional information regarding copyright ownership. 005 * The ASF licenses this file to You under the Apache License, Version 2.0 006 * (the "License"); you may not use this file except in compliance with 007 * the License. You may obtain a copy of the License at 008 * 009 * http://www.apache.org/licenses/LICENSE-2.0 010 * 011 * Unless required by applicable law or agreed to in writing, software 012 * distributed under the License is distributed on an "AS IS" BASIS, 013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 014 * See the License for the specific language governing permissions and 015 * limitations under the License. 016 * 017 */ 018package org.apache.bcel.util; 019 020import java.util.HashMap; 021import java.util.Map; 022 023import org.apache.bcel.classfile.JavaClass; 024import org.apache.commons.lang3.ArrayUtils; 025 026/** 027 * Utility class implementing a (type-safe) set of JavaClass objects. Since JavaClass has no equals() method, the name of the class is used for comparison. 028 * 029 * @see ClassStack 030 */ 031public class ClassSet { 032 033 private final Map<String, JavaClass> map = new HashMap<>(); 034 035 public boolean add(final JavaClass clazz) { 036 return map.putIfAbsent(clazz.getClassName(), clazz) != null; 037 } 038 039 public boolean empty() { 040 return map.isEmpty(); 041 } 042 043 public String[] getClassNames() { 044 return map.keySet().toArray(ArrayUtils.EMPTY_STRING_ARRAY); 045 } 046 047 public void remove(final JavaClass clazz) { 048 map.remove(clazz.getClassName()); 049 } 050 051 public JavaClass[] toArray() { 052 return map.values().toArray(JavaClass.EMPTY_ARRAY); 053 } 054}