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    package org.apache.commons.lang3;
018    
019    import java.util.ArrayList;
020    import java.util.Arrays;
021    import java.util.LinkedHashMap;
022    import java.util.List;
023    import java.util.Map;
024    import java.util.EnumSet;
025    
026    /**
027     * <p>Utility library to provide helper methods for Java enums.</p>
028     *
029     * <p>#ThreadSafe#</p>
030     *
031     * @since 3.0
032     * @version $Id: EnumUtils.java 1149320 2011-07-21 19:11:47Z mbenson $
033     */
034    public class EnumUtils {
035    
036        /**
037         * This constructor is public to permit tools that require a JavaBean
038         * instance to operate.
039         */
040        public EnumUtils() {
041        }
042    
043        /**
044         * <p>Gets the {@code Map} of enums by name.</p>
045         *
046         * <p>This method is useful when you need a map of enums by name.</p>
047         *
048         * @param <E> the type of the enumeration
049         * @param enumClass  the class of the enum to query, not null
050         * @return the modifiable map of enum names to enums, never null
051         */
052        public static <E extends Enum<E>> Map<String, E> getEnumMap(Class<E> enumClass) {
053            Map<String, E> map = new LinkedHashMap<String, E>();
054            for (E e: enumClass.getEnumConstants()) {
055                map.put(e.name(), e);
056            }
057            return map;
058        }
059    
060        /**
061         * <p>Gets the {@code List} of enums.</p>
062         *
063         * <p>This method is useful when you need a list of enums rather than an array.</p>
064         *
065         * @param <E> the type of the enumeration
066         * @param enumClass  the class of the enum to query, not null
067         * @return the modifiable list of enums, never null
068         */
069        public static <E extends Enum<E>> List<E> getEnumList(Class<E> enumClass) {
070            return new ArrayList<E>(Arrays.asList(enumClass.getEnumConstants()));
071        }
072    
073        /**
074         * <p>Checks if the specified name is a valid enum for the class.</p>
075         *
076         * <p>This method differs from {@link Enum#valueOf} in that checks if the name is
077         * a valid enum without needing to catch the exception.</p>
078         *
079         * @param <E> the type of the enumeration
080         * @param enumClass  the class of the enum to query, not null
081         * @param enumName   the enum name, null returns false
082         * @return true if the enum name is valid, otherwise false
083         */
084        public static <E extends Enum<E>> boolean isValidEnum(Class<E> enumClass, String enumName) {
085            if (enumName == null) {
086                return false;
087            }
088            try {
089                Enum.valueOf(enumClass, enumName);
090                return true;
091            } catch (IllegalArgumentException ex) {
092                return false;
093            }
094        }
095    
096        /**
097         * <p>Gets the enum for the class, returning {@code null} if not found.</p>
098         *
099         * <p>This method differs from {@link Enum#valueOf} in that it does not throw an exception
100         * for an invalid enum name.</p>
101         *
102         * @param <E> the type of the enumeration
103         * @param enumClass  the class of the enum to query, not null
104         * @param enumName   the enum name, null returns null
105         * @return the enum, null if not found
106         */
107        public static <E extends Enum<E>> E getEnum(Class<E> enumClass, String enumName) {
108            if (enumName == null) {
109                return null;
110            }
111            try {
112                return Enum.valueOf(enumClass, enumName);
113            } catch (IllegalArgumentException ex) {
114                return null;
115            }
116        }
117    
118        /**
119         * <p>Creates a long bit vector representation of the given subset of an Enum.</p>
120         *
121         * <p>This generates a value that is usable by {@link EnumUtils#processBitVector}.</p>
122         *
123         * <p>Do not use this method if you have more than 64 values in your Enum, as this
124         * would create a value greater than a long can hold.</p>
125         *
126         * @param enumClass the class of the enum we are working with, not {@code null}
127         * @param values    the values we want to convert, not {@code null}
128         * @param <E>       the type of the enumeration
129         * @return a long whose binary value represents the given set of enum values.
130         * @throws NullPointerException if {@code enumClass} or {@code values} is {@code null}
131         * @throws IllegalArgumentException if {@code enumClass} is not an enum class or has more than 64 values
132         * @since 3.0.1
133         */
134        public static <E extends Enum<E>> long generateBitVector(Class<E> enumClass, Iterable<E> values) {
135            checkBitVectorable(enumClass);
136            Validate.notNull(values);
137            long total = 0;
138            for (E constant : values) {
139                total |= (1 << constant.ordinal());
140            }
141            return total;
142        }
143    
144        /**
145         * <p>Creates a long bit vector representation of the given array of Enum values.</p>
146         *
147         * <p>This generates a value that is usable by {@link EnumUtils#processBitVector}.</p>
148         *
149         * <p>Do not use this method if you have more than 64 values in your Enum, as this
150         * would create a value greater than a long can hold.</p>
151         *
152         * @param enumClass the class of the enum we are working with, not {@code null}
153         * @param values    the values we want to convert, not {@code null}
154         * @param <E>       the type of the enumeration
155         * @return a long whose binary value represents the given set of enum values.
156         * @throws NullPointerException if {@code enumClass} or {@code values} is {@code null}
157         * @throws IllegalArgumentException if {@code enumClass} is not an enum class or has more than 64 values
158         * @since 3.0.1
159         */
160        public static <E extends Enum<E>> long generateBitVector(Class<E> enumClass, E... values) {
161            Validate.noNullElements(values);
162            return generateBitVector(enumClass, Arrays.<E> asList(values));
163        }
164    
165        /**
166         * <p>Convert a long value created by {@link EnumUtils#generateBitVector} into the set of
167         * enum values that it represents.</p>
168         *
169         * <p>If you store this value, beware any changes to the enum that would affect ordinal values.</p>
170         * @param enumClass the class of the enum we are working with, not {@code null}
171         * @param value     the long value representation of a set of enum values
172         * @param <E>       the type of the enumeration
173         * @return a set of enum values
174         * @throws NullPointerException if {@code enumClass} is {@code null}
175         * @throws IllegalArgumentException if {@code enumClass} is not an enum class or has more than 64 values
176         * @since 3.0.1
177         */
178        public static <E extends Enum<E>> EnumSet<E> processBitVector(Class<E> enumClass, long value) {
179            final E[] constants = checkBitVectorable(enumClass).getEnumConstants();
180            final EnumSet<E> results = EnumSet.noneOf(enumClass);
181            for (E constant : constants) {
182                if ((value & (1 << constant.ordinal())) != 0) {
183                    results.add(constant);
184                }
185            }
186            return results;
187        }
188    
189        /**
190         * Validate that {@code enumClass} is compatible with representation in a {@code long}.
191         * @param <E> the type of the enumeration
192         * @param enumClass to check
193         * @return {@code enumClass}
194         * @throws NullPointerException if {@code enumClass} is {@code null}
195         * @throws IllegalArgumentException if {@code enumClass} is not an enum class or has more than 64 values
196         * @since 3.0.1
197         */
198        private static <E extends Enum<E>> Class<E> checkBitVectorable(Class<E> enumClass) {
199            Validate.notNull(enumClass, "EnumClass must be defined.");
200    
201            final E[] constants = enumClass.getEnumConstants();
202            Validate.isTrue(constants != null, "%s does not seem to be an Enum type", enumClass);
203            Validate.isTrue(constants.length <= Long.SIZE, "Cannot store %s %s values in %s bits", constants.length,
204                enumClass.getSimpleName(), Long.SIZE);
205    
206            return enumClass;
207        }
208    }