16f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org/*
26f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org******************************************************************************
36f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org*
46f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org*   Copyright (C) 2012, International Business Machines
56f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org*   Corporation and others.  All Rights Reserved.
66f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org*
76f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org******************************************************************************
86f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org*/
96f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org
106f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org/**
116f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org * \file
126f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org * \brief C++: internal template EnumSet<>
136f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org */
146f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org
156f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org#ifndef ENUMSET_H
166f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org#define ENUMSET_H
176f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org
186f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org#include "unicode/utypes.h"
196f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org
206f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org#if U_SHOW_CPLUSPLUS_API
216f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org
226f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.orgU_NAMESPACE_BEGIN
236f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org
246f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org/**
256f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org * enum bitset for boolean fields. Similar to Java EnumSet<>.
266f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org * Needs to range check.
276f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org * @internal
286f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org */
296f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.orgtemplate<typename T, uint32_t minValue, uint32_t limitValue>
306f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.orgclass EnumSet {
316f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.orgpublic:
326f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline EnumSet() : fBools(0) {}
336f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline EnumSet(const EnumSet<T,minValue,limitValue>& other) : fBools(other.fBools) {}
346f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline ~EnumSet() {}
356f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline void clear() { fBools=0; }
366f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline void add(T toAdd) { set(toAdd, 1); }
376f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline void remove(T toRemove) { set(toRemove, 0); }
386f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline int32_t contains(T toCheck) const { return get(toCheck); }
396f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline void set(T toSet, int32_t v) { fBools=(fBools&(~flag(toSet)))|(v?(flag(toSet)):0); }
406f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline int32_t get(T toCheck) const { return (fBools & flag(toCheck))?1:0; }
416f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline UBool isValidEnum(T toCheck) const {  return (toCheck>=minValue&&toCheck<limitValue); }
426f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline UBool isValidValue(int32_t v) const { return (v==0||v==1); }
436f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline const EnumSet<T,minValue,limitValue>& operator=(const EnumSet<T,minValue,limitValue>& other) {
446f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org        fBools = other.fBools;
456f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org        return *this;
466f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    }
476f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org
486f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline uint32_t getAll() const {
496f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org        return fBools;
506f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    }
516f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org
526f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.orgprivate:
536f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    inline uint32_t flag(T toCheck) const { return (1<<(toCheck-minValue)); }
546f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.orgprivate:
556f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org    uint32_t fBools;
566f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org};
576f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org
586f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.orgU_NAMESPACE_END
596f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org
606f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org#endif /* U_SHOW_CPLUSPLUS_API */
616f31ac30b9092fd02a8c97e5216cf53f3e4fae4jshin@chromium.org#endif /* ENUMSET_H */
62