/* set.h Set manipulation. Copyright (C) 2012 Bill Currie Author: Bill Currie Date: 2012/8/4 This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to: Free Software Foundation, Inc. 59 Temple Place - Suite 330 Boston, MA 02111-1307, USA */ #ifndef __QF_set_h #define __QF_set_h #include "QF/qtypes.h" #include "QF/darray.h" /** \defgroup set Set handling \ingroup utils */ ///@{ //FIXME other archs #if defined(__x86_64__) || defined(__aarch64__) typedef uint64_t set_bits_t; #else typedef uint32_t set_bits_t; #endif #define SET_DEFMAP_SIZE ((32 - sizeof (struct set_s *) \ - sizeof (set_bits_t *) \ - sizeof (int) - sizeof (unsigned))\ / sizeof (set_bits_t)) #define SET_BITS (sizeof (set_bits_t) * 8) #define SET_DEFMAP_BITS (SET_DEFMAP_SIZE * SET_BITS) //NOTE: x is the element number, so size is x + 1 #define SET_SIZE(x) (((x) + SET_BITS) & ~(SET_BITS - 1)) #define SET_WORDS_STATIC(x) (SET_SIZE (x) / SET_BITS) #define SET_WORDS(s) ((s)->size / SET_BITS) #define SET_ZERO ((set_bits_t) 0) #define SET_ONE ((set_bits_t) 1) #define SET_TEST_MEMBER(s, x) \ (((const byte *)(s)->map)[(x) / 8] & (SET_ONE << ((x) % 8))) #define SET_ADD(s, x) \ (((byte *)(s)->map)[(x) / 8] |= (SET_ONE << ((x) % 8))) #define SET_REMOVE(s, x) \ (((byte *)(s)->map)[(x) / 8] &= ~(SET_ONE << ((x) % 8))) #define SET_LARGE_SET(x) (SET_SIZE (x) > SET_DEFMAP_BITS) #define SET_SAFE_SIZE(x) (SET_LARGE_SET (x) ? SET_SIZE (x) : SET_DEFMAP_BITS) #define SET_STATIC_INIT(x, alloc) { \ .size = SET_SAFE_SIZE (x), \ .map = alloc (SET_SAFE_SIZE (x) / 8), \ } #define SET_STATIC_ARRAY(array) { \ .size = 8 * __builtin_choose_expr ( \ sizeof (array[0]) == sizeof (set_bits_t), \ sizeof (array), \ (void) 0 \ ), \ .map = array, \ } /** Represent a set using a bitmap. When \a inverted is zero, ones in the bitmap represent members, but when \a inverted is non-zero, zeros in the bitmap represent members. However, this really is all private implementation details and it is best to treat set_t as a black box. */ typedef struct set_s { struct set_s *next; ///< private. for ALLOC set_bits_t *map; ///< bitmap of set members int inverted; ///< if true, 0 indicates membership unsigned size; ///< number of representable members set_bits_t defmap[SET_DEFMAP_SIZE];///< backing store for small sets } set_t; /** Represent the state of a scan through a set. Very useful in for-loops: \code set_t *set; set_iter_t *iter; create_and_populate (set); for (iter = set_first (set); iter; iter = set_next (iter)) do_something (iter->element); \endcode */ typedef struct set_iter_s { struct set_iter_s *next; ///< private. for ALLOC const set_t *set; ///< the set to which this iterator belongs /** The result of set_first() or set_next(). set_next() will start at the following element. \note For inverted sets, indicates a non-member. */ unsigned element; } set_iter_t; typedef struct set_pool_s { set_t *set_freelist; set_iter_t *set_iter_freelist; struct DARRAY_TYPE (set_t *) set_blocks; struct DARRAY_TYPE (set_iter_t *) set_iter_blocks; } set_pool_t; void set_pool_init (set_pool_t *set_pool); /** Delete a set iterator that is no longer needed. \param set_iter The set iterator to be deleted. */ void set_del_iter (set_iter_t *set_iter); void set_del_iter_r (set_pool_t *set_pool, set_iter_t *set_iter); /** Create a new set. The set is initialized to be the empty set. \return The newly created, empty set. */ set_t *set_new (void); set_t *set_new_r (set_pool_t *set_pool); /** Create a new set with space pre-allocated for the specified set size. Although sets automatically grow to accommodate new members as necessary, sometimes the maximum set size is known in advance and it can be more efficient to grow the set in advance. The set is initialized to be the empty set. \param size The number of elements for which space is to be allocated. \return The newly created, empty set. \note \a size is the actual amount of elements, not the number of the highest element (ie, for values 0..n, size is n + 1). */ set_t *set_new_size (unsigned size); set_t *set_new_size_r (set_pool_t *set_pool, unsigned size); /** Delete a set that is no longer needed. \param set The set to be deleted. */ void set_delete (set_t *set); void set_delete_r (set_pool_t *set_pool, set_t *set); /** Pre-expand a set with space for the specified element Has no effect if the set is already large enough to hold the specified element. \param set The set to be expanded \param size The minimum number of elements representable by the set \note \a size is the actual amount of elements, not the number of the highest element (ie, for values 0..n, size is n + 1). */ void set_expand (set_t *set, unsigned size); /** Shrink the set's backing memory to the minimum required to hold the set. This does not affect (nor is affected by) whether the set is an infinite set. \param set The set to trim */ void set_trim (set_t *set); /** Add an element to a set. It is not an error to add an element that is already a member of the set. \note \a set is modified. \param set The set to which the element will be added. \param x The element to be added. \return The modified set. */ set_t *set_add (set_t *set, unsigned x); /** Add a range of elements to a set. It is not an error to add elements that are already members of the set. \note \a set is modified. \param set The set to which the element will be added. \param start The first element to be added. \param count The number of elements to be added. \return The modified set. */ set_t *set_add_range (set_t *set, unsigned start, unsigned count); /** Remove an element from a set. It is not an error to remove an element that is not a member of the set. \note \a set is modified. \param set The set from which the element will be removed. \param x The element to be removed. \return The modified set. */ set_t *set_remove (set_t *set, unsigned x); /** Remove a range of elements from a set. It is not an error to remove elements that not members of the set. \note \a set is modified. \param set The set from which the element will be removed. \param start The first element to be removed. \param count The number of elements to be removed. \return The modified set. */ set_t *set_remove_range (set_t *set, unsigned start, unsigned count); /** Compute the inverse of a set. The computation is done as \a set = ~\a set. \note \a set is modified. \param set The set to be inverted. \return The set modified to be ~\a src. */ set_t *set_invert (set_t *set); /** Compute the union of two sets. The computation is done as \a dst = \a dst | \a src. \note \a dst is modified. \param dst The destination set to be modified. \param src The source set. \return The destination set modified to be \a dst | \a src. */ set_t *set_union (set_t *dst, const set_t *src); /** Compute the intersection of two sets. The computation is done as \a dst = \a dst & \a src. \note \a dst is modified. \param dst The destination set to be modified. \param src The source set. \return The destination set modified to be \a dst & \a src. */ set_t *set_intersection (set_t *dst, const set_t *src); /** Compute the diffedrence of two sets. The computation is done as \a dst = \a dst - \a src. \note \a dst is modified. \param dst The destination set to be modified. \param src The source set. \return The destination set modified to be \a dst - \a src. */ set_t *set_difference (set_t *dst, const set_t *src); /** Compute the diffedrence of two sets. The computation is done as \a dst = \a src - \a dst. \note \a dst is modified. \param dst The destination set to be modified. \param src The source set. \return The destination set modified to be \a src - \a dst. */ set_t *set_reverse_difference (set_t *dst, const set_t *src); /** Make a set equivalent to another. \note \a dst is modified. \param dst The destination set to make equivalent. \param src The source set to assign to \a dst. \return The modified destination set. */ set_t *set_assign (set_t *dst, const set_t *src); /** Make a set the empty set. \note \a set is modified. \param set The set to make the empty set. \return \a set. */ set_t *set_empty (set_t *set); /** Make a set the set of everything. \note \a set is modified. \param set The set to make the set of everything. \return \a set. */ set_t *set_everything (set_t *set); /** Test if a set is the empty set. \param set The set to test. \return 1 if \a set is empty (non-inverted). */ int set_is_empty (const set_t *set) __attribute__((pure)); /** Test if a set is the set of everything. \param set The set to test. \return 1 if \a set is the set of everything (empty inverted set). */ int set_is_everything (const set_t *set) __attribute__((pure)); /** Test if two sets are disjoint. \param s1 The first set to test. \param s2 The second set to test. \return 1 if \a s2 is disjoint from \a s1, 0 if not. \note The emtpy set is disjoint with itself. */ int set_is_disjoint (const set_t *s1, const set_t *s2) __attribute__((pure)); /** Test if two sets intersect. \param s1 The first set to test. \param s2 The second set to test. \return 1 if \a s2 intersects \a s1, 0 if not. \note Equivalent non-empty sets are treated as intersecting. */ int set_is_intersecting (const set_t *s1, const set_t *s2) __attribute__((pure)); /** Test if two sets are equivalent. \param s1 The first set to test. \param s2 The second set to test. \return 1 if \a s2 is equivalent to \a s1, 0 if not. */ int set_is_equivalent (const set_t *s1, const set_t *s2) __attribute__((pure)); /** Test if a set is a subset of another set. An equivalent set is considered to be a subset. \param set The potential super-set. \param sub The potential subset. \return 1 if \a sub is a subset of \a set, or if the sets are equivalent. */ int set_is_subset (const set_t *set, const set_t *sub) __attribute__((pure)); /** Test an element for membership in a set. \param set The set to test. \param x The element to test. \return 1 if the element is a member of the set, otherwise 0. */ int set_is_member (const set_t *set, unsigned x) __attribute__((pure)); /** Obtain the number of members (or non-members) of a set. Normal sets return the number of members, inverted sets return the number of non-members. \param set The set from which the number of (non-)members will be obtained. \return The number of (non-)members. Both empty sets and sets of evertything will return 0. */ unsigned set_count (const set_t *set) __attribute__((pure)); /** Find the first "member" of the set. \warning For normal sets, the set iterator represents a member of the set, but for inverted sets, the set iterator represetns a non-member of the set. \param set The set to scan. \return A pointer to a set iterator indicating the first (non-)member of the set, or null if the set is empty or of everything. */ set_iter_t *set_first (const set_t *set); set_iter_t *set_first_r (set_pool_t *set_pool, const set_t *set); /** Find the next "member" of the set. \warning For normal sets, the set iterator represents a member of the set, but for inverted sets, the set iterator represetns a non-member of the set. \param set_iter The set iterator representing the state of the current scan. \return A pointer to a set iterator indicating the next (non-)member of the set, or null if no mor (non-)members are available. \note The set iterator is automatically deleted when the end of the set is reached. */ set_iter_t *set_next (set_iter_t *set_iter); set_iter_t *set_next_r (set_pool_t *set_pool, set_iter_t *set_iter); set_iter_t *set_while (set_iter_t *set_iter); set_iter_t *set_while_r (set_pool_t *set_pool, set_iter_t *set_iter); struct dstring_s; /** Return a human-readable string representing the set. Empty sets will be represented by the string "{}". Sets of everything will be represented by the string "{...}". Inverted sets will have the first implicit member followed by "..." (eg, "256 ..."). \param set The set to be converted to a string. \return The human readable representation of the string. \warning The string is kept in a static variable, so subsequent calls will overwrite the results of preceeding calls. */ const char *set_as_string (const set_t *set); /** Return a human-readable string representing the set. Empty sets will be represented by the string "{}". Sets of everything will be represented by the string "{...}". Inverted sets will have the first implicit member followed by "..." (eg, "256 ..."). \param str dstring to which the representation will be written \param set The set to be converted to a string. \return The string held in str \warning The string is NOT cleared, but rather the set representation is appeneded to the string. This makes it more useful when constructing strings in a threaded environment. */ const char *set_to_dstring (struct dstring_s *str, const set_t *set); const char *set_to_dstring_r (set_pool_t *set_pool, struct dstring_s *str, const set_t *set); ///@} #endif//__QF_set_h