Move base into lib
This involves reverting dd267e8fc3
,
but that gets lost in the move.
This builds on Ubuntu and Windows at this time.
This commit is contained in:
303
lib/base/dict.c
Normal file
303
lib/base/dict.c
Normal file
@@ -0,0 +1,303 @@
|
||||
/*
|
||||
* Copyright (c) 2002, 1997 Kungliga Tekniska Högskolan
|
||||
* (Royal Institute of Technology, Stockholm, Sweden).
|
||||
* All rights reserved.
|
||||
*
|
||||
* Portions Copyright (c) 2010 Apple Inc. All rights reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions
|
||||
* are met:
|
||||
*
|
||||
* 1. Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
*
|
||||
* 2. Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in the
|
||||
* documentation and/or other materials provided with the distribution.
|
||||
*
|
||||
* 3. Neither the name of the Institute nor the names of its contributors
|
||||
* may be used to endorse or promote products derived from this software
|
||||
* without specific prior written permission.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
|
||||
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
* ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
|
||||
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
||||
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
||||
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
||||
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
||||
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
||||
* SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#include "baselocl.h"
|
||||
|
||||
struct hashentry {
|
||||
struct hashentry **prev;
|
||||
struct hashentry *next;
|
||||
heim_object_t key;
|
||||
heim_object_t value;
|
||||
};
|
||||
|
||||
struct heim_dict_data {
|
||||
size_t size;
|
||||
struct hashentry **tab;
|
||||
};
|
||||
|
||||
static void
|
||||
dict_dealloc(void *ptr)
|
||||
{
|
||||
heim_dict_t dict = ptr;
|
||||
struct hashentry **h, *g, *i;
|
||||
|
||||
for (h = dict->tab; h < &dict->tab[dict->size]; ++h) {
|
||||
for (g = h[0]; g; g = i) {
|
||||
i = g->next;
|
||||
heim_release(g->key);
|
||||
heim_release(g->value);
|
||||
free(g);
|
||||
}
|
||||
}
|
||||
free(dict->tab);
|
||||
}
|
||||
|
||||
struct heim_type_data dict_object = {
|
||||
HEIM_TID_DICT,
|
||||
"dict-object",
|
||||
NULL,
|
||||
dict_dealloc,
|
||||
NULL,
|
||||
NULL,
|
||||
NULL,
|
||||
NULL
|
||||
};
|
||||
|
||||
static size_t
|
||||
isprime(size_t p)
|
||||
{
|
||||
size_t q, i;
|
||||
|
||||
for(i = 2 ; i < p; i++) {
|
||||
q = p / i;
|
||||
|
||||
if (i * q == p)
|
||||
return 0;
|
||||
if (i * i > p)
|
||||
return 1;
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
static size_t
|
||||
findprime(size_t p)
|
||||
{
|
||||
if (p % 2 == 0)
|
||||
p++;
|
||||
|
||||
while (isprime(p) == 0)
|
||||
p += 2;
|
||||
|
||||
return p;
|
||||
}
|
||||
|
||||
/**
|
||||
* Allocate an array
|
||||
*
|
||||
* @return A new allocated array, free with heim_release()
|
||||
*/
|
||||
|
||||
heim_dict_t
|
||||
heim_dict_create(size_t size)
|
||||
{
|
||||
heim_dict_t dict;
|
||||
|
||||
dict = _heim_alloc_object(&dict_object, sizeof(*dict));
|
||||
|
||||
dict->size = findprime(size);
|
||||
if (dict->size == 0) {
|
||||
heim_release(dict);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
dict->tab = calloc(dict->size, sizeof(dict->tab[0]));
|
||||
if (dict->tab == NULL) {
|
||||
dict->size = 0;
|
||||
heim_release(dict);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return dict;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get type id of an dict
|
||||
*
|
||||
* @return the type id
|
||||
*/
|
||||
|
||||
heim_tid_t
|
||||
heim_dict_get_type_id(void)
|
||||
{
|
||||
return HEIM_TID_DICT;
|
||||
}
|
||||
|
||||
/* Intern search function */
|
||||
|
||||
static struct hashentry *
|
||||
_search(heim_dict_t dict, heim_object_t ptr)
|
||||
{
|
||||
unsigned long v = heim_get_hash(ptr);
|
||||
struct hashentry *p;
|
||||
|
||||
for (p = dict->tab[v % dict->size]; p != NULL; p = p->next)
|
||||
if (heim_cmp(ptr, p->key) == 0)
|
||||
return p;
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
/**
|
||||
* Search for element in hash table
|
||||
*
|
||||
* @value dict the dict to search in
|
||||
* @value key the key to search for
|
||||
*
|
||||
* @return a not-retained copy of the value for key or NULL if not found
|
||||
*/
|
||||
|
||||
heim_object_t
|
||||
heim_dict_get_value(heim_dict_t dict, heim_object_t key)
|
||||
{
|
||||
struct hashentry *p;
|
||||
p = _search(dict, key);
|
||||
if (p == NULL)
|
||||
return NULL;
|
||||
|
||||
return p->value;
|
||||
}
|
||||
|
||||
/**
|
||||
* Search for element in hash table
|
||||
*
|
||||
* @value dict the dict to search in
|
||||
* @value key the key to search for
|
||||
*
|
||||
* @return a retained copy of the value for key or NULL if not found
|
||||
*/
|
||||
|
||||
heim_object_t
|
||||
heim_dict_copy_value(heim_dict_t dict, heim_object_t key)
|
||||
{
|
||||
struct hashentry *p;
|
||||
p = _search(dict, key);
|
||||
if (p == NULL)
|
||||
return NULL;
|
||||
|
||||
return heim_retain(p->value);
|
||||
}
|
||||
|
||||
/**
|
||||
* Add key and value to dict
|
||||
*
|
||||
* @value dict the dict to add too
|
||||
* @value key the key to add
|
||||
* @value value the value to add
|
||||
*
|
||||
* @return 0 if added, errno if not
|
||||
*/
|
||||
|
||||
int
|
||||
heim_dict_set_value(heim_dict_t dict, heim_object_t key, heim_object_t value)
|
||||
{
|
||||
struct hashentry **tabptr, *h;
|
||||
|
||||
h = _search(dict, key);
|
||||
if (h) {
|
||||
heim_release(h->value);
|
||||
h->value = heim_retain(value);
|
||||
} else {
|
||||
unsigned long v;
|
||||
|
||||
h = malloc(sizeof(*h));
|
||||
if (h == NULL)
|
||||
return ENOMEM;
|
||||
|
||||
h->key = heim_retain(key);
|
||||
h->value = heim_retain(value);
|
||||
|
||||
v = heim_get_hash(key);
|
||||
|
||||
tabptr = &dict->tab[v % dict->size];
|
||||
h->next = *tabptr;
|
||||
*tabptr = h;
|
||||
h->prev = tabptr;
|
||||
if (h->next)
|
||||
h->next->prev = &h->next;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* Delete element with key key
|
||||
*
|
||||
* @value dict the dict to delete from
|
||||
* @value key the key to delete
|
||||
*/
|
||||
|
||||
void
|
||||
heim_dict_delete_key(heim_dict_t dict, heim_object_t key)
|
||||
{
|
||||
struct hashentry *h = _search(dict, key);
|
||||
|
||||
if (h == NULL)
|
||||
return;
|
||||
|
||||
heim_release(h->key);
|
||||
heim_release(h->value);
|
||||
|
||||
if ((*(h->prev) = h->next) != NULL)
|
||||
h->next->prev = h->prev;
|
||||
|
||||
free(h);
|
||||
}
|
||||
|
||||
/**
|
||||
* Do something for each element
|
||||
*
|
||||
* @value dict the dict to interate over
|
||||
* @value func the function to search for
|
||||
* @value arg argument to func
|
||||
*/
|
||||
|
||||
void
|
||||
heim_dict_iterate_f(heim_dict_t dict, void *arg, heim_dict_iterator_f_t func)
|
||||
{
|
||||
struct hashentry **h, *g;
|
||||
|
||||
for (h = dict->tab; h < &dict->tab[dict->size]; ++h)
|
||||
for (g = *h; g; g = g->next)
|
||||
func(g->key, g->value, arg);
|
||||
}
|
||||
|
||||
#ifdef __BLOCKS__
|
||||
/**
|
||||
* Do something for each element
|
||||
*
|
||||
* @value dict the dict to interate over
|
||||
* @value func the function to search for
|
||||
*/
|
||||
|
||||
void
|
||||
heim_dict_iterate(heim_dict_t dict, void (^func)(heim_object_t, heim_object_t))
|
||||
{
|
||||
struct hashentry **h, *g;
|
||||
|
||||
for (h = dict->tab; h < &dict->tab[dict->size]; ++h)
|
||||
for (g = *h; g; g = g->next)
|
||||
func(g->key, g->value);
|
||||
}
|
||||
#endif
|
Reference in New Issue
Block a user