--- /dev/null
+/* PSPP - a program for statistical analysis.
+ Copyright (C) 2009, 2010 Free Software Foundation, Inc.
+
+ 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 3 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, see <http://www.gnu.org/licenses/>. */
+
+#include <config.h>
+
+#include "libpspp/stringi-map.h"
+
+#include <stdlib.h>
+#include <string.h>
+
+#include "libpspp/hash-functions.h"
+#include "libpspp/string-set.h"
+#include "libpspp/stringi-set.h"
+
+#include "gl/xalloc.h"
+
+static struct stringi_map_node *stringi_map_find_node__ (
+ const struct stringi_map *, const char *key, unsigned int hash);
+static bool stringi_map_delete__ (struct stringi_map *, const char *key,
+ unsigned int hash);
+static struct stringi_map_node *stringi_map_insert__ (struct stringi_map *,
+ char *key, char *value,
+ unsigned int hash);
+
+/* Sets NODE's value to a copy of NEW_VALUE and returns the node's previous
+ value. The caller is responsible for freeing the returned string (with
+ free()). */
+char *
+stringi_map_node_swap_value (struct stringi_map_node *node,
+ const char *new_value)
+{
+ return stringi_map_node_swap_value_nocopy (node, xstrdup (new_value));
+}
+
+/* Sets NODE's value to NEW_VALUE, which must be a malloc()'d string,
+ transferring ownership of NEW_VALUE to the node. Returns the node's
+ previous value, which the caller is responsible for freeing (with
+ free()). */
+char *
+stringi_map_node_swap_value_nocopy (struct stringi_map_node *node,
+ char *new_value)
+{
+ char *old_value = node->value;
+ node->value = new_value;
+ return old_value;
+}
+
+/* Replaces NODE's value by a copy of VALUE. */
+void
+stringi_map_node_set_value (struct stringi_map_node *node, const char *value)
+{
+ stringi_map_node_set_value_nocopy (node, xstrdup (value));
+}
+
+/* Replaces NODE's value by VALUE, which must be a malloc()'d string,
+ transferring ownership of VALUE to the node.. */
+void
+stringi_map_node_set_value_nocopy (struct stringi_map_node *node, char *value)
+{
+ free (node->value);
+ node->value = value;
+}
+
+/* Frees NODE and and its key and value. Ordinarily nodes are owned by
+ stringi_maps, but this function should only be used by a caller that owns
+ NODE, such as one that has called stringi_map_delete_nofree() for the
+ node. */
+void
+stringi_map_node_destroy (struct stringi_map_node *node)
+{
+ free (node->key);
+ free (node->value);
+ free (node);
+}
+\f
+/* Initializes MAP as an initially empty case-insensitive string map. */
+void
+stringi_map_init (struct stringi_map *map)
+{
+ hmap_init (&map->hmap);
+}
+
+/* Initializes MAP as a new string map that initially contains the same pairs
+ as OLD. */
+void
+stringi_map_clone (struct stringi_map *map, const struct stringi_map *old)
+{
+ const struct stringi_map_node *node;
+ const char *key, *value;
+
+ stringi_map_init (map);
+ hmap_reserve (&map->hmap, stringi_map_count (old));
+ STRINGI_MAP_FOR_EACH_KEY_VALUE (key, value, node, old)
+ stringi_map_insert__ (map, xstrdup (key), xstrdup (value),
+ node->hmap_node.hash);
+}
+
+/* Exchanges the contents of string maps A and B. */
+void
+stringi_map_swap (struct stringi_map *a, struct stringi_map *b)
+{
+ hmap_swap (&a->hmap, &b->hmap);
+}
+
+/* Frees MAP and its nodes and key-value pairs. */
+void
+stringi_map_destroy (struct stringi_map *map)
+{
+ if (map != NULL)
+ {
+ stringi_map_clear (map);
+ hmap_destroy (&map->hmap);
+ }
+}
+
+/* Returns true if MAP contains KEY (or an equivalent with different case) as a
+ key, otherwise false. */
+bool
+stringi_map_contains (const struct stringi_map *map, const char *key)
+{
+ return stringi_map_find_node (map, key) != NULL;
+}
+
+/* If MAP contains KEY (or an equivalent with different case) as a key, returns
+ the corresponding value. Otherwise, returns a null pointer. */
+const char *
+stringi_map_find (const struct stringi_map *map, const char *key)
+{
+ const struct stringi_map_node *node = stringi_map_find_node (map, key);
+ return node != NULL ? node->value : NULL;
+}
+
+/* If MAP contains KEY (or an equivalent with different case) as a key, returns
+ the corresponding node. Otherwise, returns a null pointer. */
+struct stringi_map_node *
+stringi_map_find_node (const struct stringi_map *map, const char *key)
+{
+ return stringi_map_find_node__ (map, key, hash_case_string (key, 0));
+}
+
+/* If MAP contains KEY (or an equivalent with different case) as a key, deletes
+ that key's node and returns its value, which the caller is responsible for
+ freeing (using free()). Otherwise, returns a null pointer. */
+char *
+stringi_map_find_and_delete (struct stringi_map *map, const char *key)
+{
+ struct stringi_map_node *node = stringi_map_find_node (map, key);
+ char *value = NULL;
+ if (node != NULL)
+ {
+ value = node->value;
+ node->value = NULL;
+ stringi_map_delete_node (map, node);
+ }
+ return value;
+}
+
+/* If MAP does not contain KEY (or an equivalent with different case) as a key,
+ inserts a new node containing copies of KEY and VALUE and returns the new
+ node. Otherwise, returns the existing node that contains KEY. */
+struct stringi_map_node *
+stringi_map_insert (struct stringi_map *map, const char *key,
+ const char *value)
+{
+ unsigned int hash = hash_case_string (key, 0);
+ struct stringi_map_node *node = stringi_map_find_node__ (map, key, hash);
+ if (node == NULL)
+ node = stringi_map_insert__ (map, xstrdup (key), xstrdup (value), hash);
+ return node;
+}
+
+/* If MAP does not contain KEY (or an equivalent with different case) as a key,
+ inserts a new node containing KEY and VALUE and returns the new node.
+ Otherwise, returns the existing node that contains KEY. Either way,
+ ownership of KEY and VALUE is transferred to MAP. */
+struct stringi_map_node *
+stringi_map_insert_nocopy (struct stringi_map *map, char *key, char *value)
+{
+ unsigned int hash = hash_case_string (key, 0);
+ struct stringi_map_node *node = stringi_map_find_node__ (map, key, hash);
+ if (node == NULL)
+ node = stringi_map_insert__ (map, key, value, hash);
+ else
+ {
+ free (key);
+ free (value);
+ }
+ return node;
+}
+
+/* If MAP does not contain KEY (or an equivalent with different case) as a key,
+ inserts a new node containing copies of KEY and VALUE. Otherwise, replaces
+ the existing node's value by a copy of VALUE. Returns the node. */
+struct stringi_map_node *
+stringi_map_replace (struct stringi_map *map, const char *key,
+ const char *value)
+{
+ unsigned int hash = hash_case_string (key, 0);
+ struct stringi_map_node *node = stringi_map_find_node__ (map, key, hash);
+ if (node == NULL)
+ node = stringi_map_insert__ (map, xstrdup (key), xstrdup (value), hash);
+ else
+ stringi_map_node_set_value (node, value);
+ return node;
+}
+
+/* If MAP does not contain KEY (or an equivalent with different case) as a key,
+ inserts a new node containing KEY and VALUE. Otherwise, replaces the
+ existing node's value by VALUE. Either way, ownership of KEY and VALUE is
+ transferred to MAP. Returns the node. */
+struct stringi_map_node *
+stringi_map_replace_nocopy (struct stringi_map *map, char *key, char *value)
+{
+ unsigned int hash = hash_case_string (key, 0);
+ struct stringi_map_node *node = stringi_map_find_node__ (map, key, hash);
+ if (node == NULL)
+ node = stringi_map_insert__ (map, key, value, hash);
+ else
+ {
+ free (key);
+ stringi_map_node_set_value_nocopy (node, value);
+ }
+ return node;
+}
+
+/* Searches MAP for a node with KEY (or an equivalent with different case) as
+ its key. If found, deletes the node and its key and value and returns true.
+ Otherwise, returns false without modifying MAP. */
+bool
+stringi_map_delete (struct stringi_map *map, const char *key)
+{
+ return stringi_map_delete__ (map, key, hash_case_string (key, 0));
+}
+
+/* Deletes NODE from MAP and destroys the node and its key and value. */
+void
+stringi_map_delete_node (struct stringi_map *map,
+ struct stringi_map_node *node)
+{
+ stringi_map_delete_nofree (map, node);
+ stringi_map_node_destroy (node);
+}
+
+/* Deletes NODE from MAP. Transfers ownership of NODE to the caller, which
+ becomes responsible for destroying it. */
+void
+stringi_map_delete_nofree (struct stringi_map *map,
+ struct stringi_map_node *node)
+{
+ hmap_delete (&map->hmap, &node->hmap_node);
+}
+
+/* Removes all nodes from MAP and frees them and their keys and values. */
+void
+stringi_map_clear (struct stringi_map *map)
+{
+ struct stringi_map_node *node, *next;
+
+ STRINGI_MAP_FOR_EACH_NODE_SAFE (node, next, map)
+ stringi_map_delete_node (map, node);
+}
+
+/* Inserts a copy of each of the nodes in SRC into DST. When SRC and DST both
+ have a particular key (or keys that differ only in case), the value in DST's
+ node is left unchanged. */
+void
+stringi_map_insert_map (struct stringi_map *dst, const struct stringi_map *src)
+{
+ const struct stringi_map_node *node;
+
+ STRINGI_MAP_FOR_EACH_NODE (node, src)
+ {
+ if (!stringi_map_find_node__ (dst, node->key, node->hmap_node.hash))
+ stringi_map_insert__ (dst, xstrdup (node->key), xstrdup (node->value),
+ node->hmap_node.hash);
+ }
+}
+
+/* Inserts a copy of each of the nodes in SRC into DST. When SRC and DST both
+ have a particular key (or keys that differ only in case), the value in DST's
+ node is replaced by a copy of the value in SRC's node. */
+void
+stringi_map_replace_map (struct stringi_map *dst,
+ const struct stringi_map *src)
+{
+ const struct stringi_map_node *snode;
+
+ STRINGI_MAP_FOR_EACH_NODE (snode, src)
+ {
+ struct stringi_map_node *dnode;
+ dnode = stringi_map_find_node__ (dst, snode->key, snode->hmap_node.hash);
+ if (dnode != NULL)
+ stringi_map_node_set_value (dnode, snode->value);
+ else
+ stringi_map_insert__ (dst,
+ xstrdup (snode->key), xstrdup (snode->value),
+ snode->hmap_node.hash);
+ }
+}
+
+/* Inserts each of the keys in MAP into KEYS. KEYS must already have been
+ initialized (using stringi_set_init()). */
+void
+stringi_map_get_keys (const struct stringi_map *map, struct stringi_set *keys)
+{
+ const struct stringi_map_node *node;
+ const char *key;
+
+ STRINGI_MAP_FOR_EACH_KEY (key, node, map)
+ stringi_set_insert (keys, key);
+}
+
+/* Inserts each of the values in MAP into VALUES. VALUES must already have
+ been initialized (using string_set_init()). */
+void
+stringi_map_get_values (const struct stringi_map *map,
+ struct string_set *values)
+{
+ const struct stringi_map_node *node;
+ const char *value;
+
+ STRINGI_MAP_FOR_EACH_VALUE (value, node, map)
+ string_set_insert (values, value);
+}
+\f
+static struct stringi_map_node *
+stringi_map_find_node__ (const struct stringi_map *map, const char *key,
+ unsigned int hash)
+{
+ struct stringi_map_node *node;
+
+ HMAP_FOR_EACH_WITH_HASH (node, struct stringi_map_node, hmap_node,
+ hash, &map->hmap)
+ if (!strcasecmp (key, node->key))
+ return node;
+
+ return NULL;
+}
+
+static bool
+stringi_map_delete__ (struct stringi_map *map, const char *key,
+ unsigned int hash)
+{
+ struct stringi_map_node *node = stringi_map_find_node__ (map, key, hash);
+ if (node != NULL)
+ {
+ stringi_map_delete_node (map, node);
+ return true;
+ }
+ else
+ return false;
+}
+
+static struct stringi_map_node *
+stringi_map_insert__ (struct stringi_map *map, char *key, char *value,
+ unsigned int hash)
+{
+ struct stringi_map_node *node = xmalloc (sizeof *node);
+ node->key = key;
+ node->value = value;
+ hmap_insert (&map->hmap, &node->hmap_node, hash);
+ return node;
+}
+