/*
 * Copyright (c) 2004-2005 The Trustees of Indiana University.
 *                         All rights reserved.
 * Copyright (c) 2004-2005 The Trustees of the University of Tennessee.
 *                         All rights reserved.
 * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart, 
 *                         University of Stuttgart.  All rights reserved.
 * Copyright (c) 2004-2005 The Regents of the University of California.
 *                         All rights reserved.
 * $COPYRIGHT$
 * 
 * Additional copyrights may follow
 * 
 * $HEADER$
 */

#include "ompi_config.h"

#include <string.h>
#include <stdlib.h>

#include "include/constants.h"
#include "util/output.h"
#include "class/ompi_list.h"
#include "class/ompi_hash_table.h"

/*
 * ompi_hash_table_t
 */

#define HASH_MULTIPLIER 31

static void ompi_hash_table_construct(ompi_hash_table_t* ht);
static void ompi_hash_table_destruct(ompi_hash_table_t* ht);


OBJ_CLASS_INSTANCE(
    ompi_hash_table_t, 
    ompi_object_t,
    ompi_hash_table_construct,
    ompi_hash_table_destruct
);


static void ompi_hash_table_construct(ompi_hash_table_t* ht)
{
    OBJ_CONSTRUCT(&ht->ht_nodes, ompi_list_t);
    ht->ht_table = NULL;
    ht->ht_table_size = 0;
    ht->ht_size = 0;
}


static void ompi_hash_table_destruct(ompi_hash_table_t* ht)
{
    size_t i;
    ompi_hash_table_remove_all(ht);
    for(i=0; i<ht->ht_table_size; i++) {
        OBJ_DESTRUCT(ht->ht_table+i);
    }
    if(NULL != ht->ht_table) {
        free(ht->ht_table);
    }
    OBJ_DESTRUCT(&ht->ht_nodes);
}


int ompi_hash_table_init(ompi_hash_table_t* ht, size_t table_size)
{
    size_t i;
    size_t power2 = 1;
    size_t tmp = table_size;
    while(tmp) {
       tmp >>= 1;
       power2 <<= 1;
    }

    ht->ht_mask = power2-1;
    ht->ht_table = (ompi_list_t *)malloc(power2 * sizeof(ompi_list_t));
    if(NULL == ht->ht_table) {
        return OMPI_ERR_OUT_OF_RESOURCE;
    }
    for(i=ht->ht_table_size; i<power2; i++) {
        ompi_list_t* list = ht->ht_table+i;
        OBJ_CONSTRUCT(list, ompi_list_t);
    }
    ht->ht_table_size = power2;
    return OMPI_SUCCESS;
}

int ompi_hash_table_remove_all(ompi_hash_table_t* ht)
{
    size_t i;
    for(i=0; i<ht->ht_table_size; i++) {
        ompi_list_t* list = ht->ht_table+i;
        while(ompi_list_get_size(list)) {
            ompi_list_item_t *item = ompi_list_remove_first(list);
            OBJ_RELEASE(item);
        }
    }

    while(ompi_list_get_size(&ht->ht_nodes)) {
        ompi_list_item_t* item = ompi_list_remove_first(&ht->ht_nodes);
        OBJ_RELEASE(item);
    }
    ht->ht_size = 0;
    return OMPI_SUCCESS;
}
 
/***************************************************************************/

/*
 *  ompi_uint32_hash_node_t
 */

struct ompi_uint32_hash_node_t
{
    ompi_list_item_t super;
    uint32_t hn_key;
    void *hn_value;
};
typedef struct ompi_uint32_hash_node_t ompi_uint32_hash_node_t;

static OBJ_CLASS_INSTANCE(ompi_uint32_hash_node_t,
                          ompi_list_item_t,
                          NULL,
                          NULL);


int ompi_hash_table_get_value_uint32(ompi_hash_table_t* ht, uint32_t key,
				     void **ptr)
{
    ompi_list_t* list = ht->ht_table + (key & ht->ht_mask);
    ompi_uint32_hash_node_t *node;

#if OMPI_ENABLE_DEBUG
    if(ht->ht_table_size == 0) {
        ompi_output(0, "ompi_hash_table_get_value_uint32:"
		   "ompi_hash_table_init() has not been called");
        return OMPI_ERROR;
    }
#endif
    for(node =  (ompi_uint32_hash_node_t*)ompi_list_get_first(list);
        node != (ompi_uint32_hash_node_t*)ompi_list_get_end(list);
        node =  (ompi_uint32_hash_node_t*)ompi_list_get_next(node)) {
        if (node->hn_key == key) {
	    *ptr = node->hn_value;
            return OMPI_SUCCESS;
        }
    } 
    return OMPI_ERR_NOT_FOUND;
}


int ompi_hash_table_set_value_uint32(ompi_hash_table_t* ht,
				    uint32_t key, void* value)
{
    ompi_list_t* list = ht->ht_table + (key & ht->ht_mask);
    ompi_uint32_hash_node_t *node;

#if OMPI_ENABLE_DEBUG
    if(ht->ht_table_size == 0) {
        ompi_output(0, "ompi_hash_table_set_value_uint32:"
		   "ompi_hash_table_init() has not been called");
        return OMPI_ERR_BAD_PARAM;
    }
#endif
    for(node =  (ompi_uint32_hash_node_t*)ompi_list_get_first(list);
        node != (ompi_uint32_hash_node_t*)ompi_list_get_end(list);
        node =  (ompi_uint32_hash_node_t*)ompi_list_get_next(node)) {
        if (node->hn_key == key) {
            node->hn_value = value;
            return OMPI_SUCCESS;
        }
    } 

    node = (ompi_uint32_hash_node_t*)ompi_list_remove_first(&ht->ht_nodes); 
    if(NULL == node) {
        node = OBJ_NEW(ompi_uint32_hash_node_t);
        if(NULL == node)
            return OMPI_ERR_OUT_OF_RESOURCE;
    }
    node->hn_key = key;
    node->hn_value = value;
    ompi_list_append(list, (ompi_list_item_t*)node);
    ht->ht_size++;
    return OMPI_SUCCESS;
}


int ompi_hash_table_remove_value_uint32(ompi_hash_table_t* ht, uint32_t key)
{
    ompi_list_t* list = ht->ht_table + (key & ht->ht_mask);
    ompi_uint32_hash_node_t *node;

#if OMPI_ENABLE_DEBUG
    if(ht->ht_table_size == 0) {
        ompi_output(0, "ompi_hash_table_remove_value_uint32:"
		   "ompi_hash_table_init() has not been called");
        return OMPI_ERR_BAD_PARAM;
    }
#endif
    for(node =  (ompi_uint32_hash_node_t*)ompi_list_get_first(list);
        node != (ompi_uint32_hash_node_t*)ompi_list_get_end(list);
        node =  (ompi_uint32_hash_node_t*)ompi_list_get_next(node)) {
        if (node->hn_key == key) {
            ompi_list_remove_item(list, (ompi_list_item_t*)node);
            ompi_list_append(&ht->ht_nodes, (ompi_list_item_t*)node);
            ht->ht_size--;
            return OMPI_SUCCESS;
        }
    } 
    return OMPI_ERR_NOT_FOUND;
}

/***************************************************************************/

/*
 *  ompi_uint64_hash_node_t
 */

struct ompi_uint64_hash_node_t
{
    ompi_list_item_t super;
    uint64_t hn_key;
    void* hn_value;
};
typedef struct ompi_uint64_hash_node_t ompi_uint64_hash_node_t;

static OBJ_CLASS_INSTANCE(ompi_uint64_hash_node_t,
                          ompi_list_item_t,
                          NULL,
                          NULL);


int ompi_hash_table_get_value_uint64(ompi_hash_table_t* ht, uint64_t key,
				     void **ptr)
{
    ompi_list_t* list = ht->ht_table + (key & ht->ht_mask);
    ompi_uint64_hash_node_t *node;

#if OMPI_ENABLE_DEBUG
    if(ht->ht_table_size == 0) {
        ompi_output(0, "ompi_hash_table_get_value_uint64:"
		   "ompi_hash_table_init() has not been called");
        return OMPI_ERROR;
    }
#endif
    for(node =  (ompi_uint64_hash_node_t*)ompi_list_get_first(list);
        node != (ompi_uint64_hash_node_t*)ompi_list_get_end(list);
        node =  (ompi_uint64_hash_node_t*)ompi_list_get_next(node)) {
        if (node->hn_key == key) {
            *ptr = node->hn_value;
	    return OMPI_SUCCESS;
        }
    } 
    return OMPI_ERR_NOT_FOUND;
}


int ompi_hash_table_set_value_uint64(ompi_hash_table_t* ht,
				    uint64_t key, void* value)
{
    ompi_list_t* list = ht->ht_table + (key & ht->ht_mask);
    ompi_uint64_hash_node_t *node;

#if OMPI_ENABLE_DEBUG
    if(ht->ht_table_size == 0) {
        ompi_output(0, "ompi_hash_table_set_value_uint64:"
		   "ompi_hash_table_init() has not been called");
        return OMPI_ERR_BAD_PARAM;
    }
#endif
    for(node =  (ompi_uint64_hash_node_t*)ompi_list_get_first(list);
        node != (ompi_uint64_hash_node_t*)ompi_list_get_end(list);
        node =  (ompi_uint64_hash_node_t*)ompi_list_get_next(node)) {
        if (node->hn_key == key) {
            node->hn_value = value;
            return OMPI_SUCCESS;
        }
    } 

    node = (ompi_uint64_hash_node_t*)ompi_list_remove_first(&ht->ht_nodes); 
    if(NULL == node) {
        node = OBJ_NEW(ompi_uint64_hash_node_t);
        if(NULL == node) {
            return OMPI_ERR_OUT_OF_RESOURCE;
        }
    }
    node->hn_key = key;
    node->hn_value = value;
    ompi_list_append(list, (ompi_list_item_t*)node);
    ht->ht_size++;
    return OMPI_SUCCESS;
}


int ompi_hash_table_remove_value_uint64(ompi_hash_table_t* ht, uint64_t key)
{
    ompi_list_t* list = ht->ht_table + (key & ht->ht_mask);
    ompi_uint64_hash_node_t *node;

#if OMPI_ENABLE_DEBUG
    if(ht->ht_table_size == 0) {
        ompi_output(0, "ompi_hash_table_remove_value_uint64:"
		   "ompi_hash_table_init() has not been called");
        return OMPI_ERR_BAD_PARAM;
    }
#endif
    for(node =  (ompi_uint64_hash_node_t*)ompi_list_get_first(list);
        node != (ompi_uint64_hash_node_t*)ompi_list_get_end(list);
        node =  (ompi_uint64_hash_node_t*)ompi_list_get_next(node)) {
        if (node->hn_key == key) {
            ompi_list_remove_item(list, (ompi_list_item_t*)node);
            ompi_list_append(&ht->ht_nodes, (ompi_list_item_t*)node);
            ht->ht_size--;
            return OMPI_SUCCESS;
        }
    } 
    return OMPI_ERR_NOT_FOUND;
}

/***************************************************************************/

/*
 *  ompi_ptr_hash_node_t
 */

struct ompi_ptr_hash_node_t
{
    ompi_list_item_t super;
    void*  hn_key;
    size_t hn_key_size;
    void*  hn_value;
};
typedef struct ompi_ptr_hash_node_t ompi_ptr_hash_node_t;

static void ompi_ptr_hash_node_construct(ompi_ptr_hash_node_t* hn)
{
    hn->hn_key_size = 0;
    hn->hn_key = NULL;
    hn->hn_value = NULL;
}

static void ompi_ptr_hash_node_destruct(ompi_ptr_hash_node_t* hn)
{
    if(NULL != hn->hn_key) {
        free(hn->hn_key);
    }
}

static OBJ_CLASS_INSTANCE(ompi_ptr_hash_node_t,
                          ompi_list_item_t,
                          ompi_ptr_hash_node_construct,
                          ompi_ptr_hash_node_destruct);


static inline uint32_t ompi_hash_value(size_t mask, const void *key,
                                       uint32_t keysize)
{
    uint32_t h, i;
    const unsigned char *p;
    
    h = 0;
    p = (const unsigned char *)key;
    for (i = 0; i < keysize; i++, p++)
        h = HASH_MULTIPLIER*h + *p;
    return (h & mask);
}

int ompi_hash_table_get_value_ptr(ompi_hash_table_t* ht, const void* key,
				  size_t key_size, void **ptr)
{
    ompi_list_t* list = ht->ht_table + ompi_hash_value(ht->ht_mask, key, 
                                                       key_size);
    ompi_ptr_hash_node_t *node;

#if OMPI_ENABLE_DEBUG
    if(ht->ht_table_size == 0) {
        ompi_output(0, "ompi_hash_table_get_value_ptr:"
		   "ompi_hash_table_init() has not been called");
        return OMPI_ERROR;
    }
#endif
    for(node =  (ompi_ptr_hash_node_t*)ompi_list_get_first(list);
        node != (ompi_ptr_hash_node_t*)ompi_list_get_end(list);
        node =  (ompi_ptr_hash_node_t*)ompi_list_get_next(node)) {
        if (node->hn_key_size == key_size &&
            memcmp(node->hn_key, key, key_size) == 0) {
            *ptr = node->hn_value;
	    return OMPI_SUCCESS;
        }
    } 
    return OMPI_ERR_NOT_FOUND;
}


int ompi_hash_table_set_value_ptr(ompi_hash_table_t* ht, const void* key,
                                  size_t key_size, void* value)
{
    ompi_list_t* list = ht->ht_table + ompi_hash_value(ht->ht_mask, key,
                                                       key_size);
    ompi_ptr_hash_node_t *node;

#if OMPI_ENABLE_DEBUG
    if(ht->ht_table_size == 0) {
        ompi_output(0, "ompi_hash_table_set_value_ptr:"
		   "ompi_hash_table_init() has not been called");
        return OMPI_ERR_BAD_PARAM;
    }
#endif
    for(node =  (ompi_ptr_hash_node_t*)ompi_list_get_first(list);
        node != (ompi_ptr_hash_node_t*)ompi_list_get_end(list);
        node =  (ompi_ptr_hash_node_t*)ompi_list_get_next(node)) {
        if (node->hn_key_size == key_size &&
            memcmp(node->hn_key, key, key_size) == 0) {
            node->hn_value = value;
            return OMPI_SUCCESS;
        }
    } 

    node = (ompi_ptr_hash_node_t*)ompi_list_remove_first(&ht->ht_nodes); 
    if(NULL == node) {
        node = OBJ_NEW(ompi_ptr_hash_node_t);
        if(NULL == node) {
            return OMPI_ERR_OUT_OF_RESOURCE;
        }
    }
    node->hn_key = malloc(key_size);
    node->hn_key_size = key_size;
    node->hn_value = value;
    memcpy(node->hn_key, key, key_size);
    ompi_list_append(list, (ompi_list_item_t*)node);
    ht->ht_size++;
    return OMPI_SUCCESS;
}


int ompi_hash_table_remove_value_ptr(ompi_hash_table_t* ht,
                                     const void* key, size_t key_size)
{
    ompi_list_t* list = ht->ht_table + ompi_hash_value(ht->ht_mask,
                                                       key, key_size);
    ompi_ptr_hash_node_t *node;

#if OMPI_ENABLE_DEBUG
    if(ht->ht_table_size == 0) {
        ompi_output(0, "ompi_hash_table_remove_value_ptr: "
		   "ompi_hash_table_init() has not been called");
        return OMPI_ERR_BAD_PARAM;
    }
#endif
    for(node =  (ompi_ptr_hash_node_t*)ompi_list_get_first(list);
        node != (ompi_ptr_hash_node_t*)ompi_list_get_end(list);
        node =  (ompi_ptr_hash_node_t*)ompi_list_get_next(node)) {
        if (node->hn_key_size == key_size &&
            memcmp(node->hn_key, key, key_size) == 0) {
            free(node->hn_key);
            node->hn_key = NULL;
            node->hn_key_size = 0;
            ompi_list_remove_item(list, (ompi_list_item_t*)node);
            ompi_list_append(&ht->ht_nodes, (ompi_list_item_t*)node);
            ht->ht_size--;
            return OMPI_SUCCESS;
        }
    } 
 return OMPI_ERR_NOT_FOUND;
}


int 
ompi_hash_table_get_first_key_uint32(ompi_hash_table_t *ht, uint32_t *key, 
                                     void **value, void **node)
{
    size_t i;
    ompi_uint32_hash_node_t *list_node;

    /* Go through all the lists and return the first element off the
       first non-empty list */
    
    for (i = 0; i < ht->ht_table_size; ++i) {
        if (ompi_list_get_size(ht->ht_table + i) > 0) {
            list_node = (ompi_uint32_hash_node_t*)
                ompi_list_get_first(ht->ht_table + i);
            *node = list_node;
            *key = list_node->hn_key;
            *value = list_node->hn_value;
            return OMPI_SUCCESS;
        }
    }

    /* The hash table is empty */

    return OMPI_ERROR;
}


int 
ompi_hash_table_get_next_key_uint32(ompi_hash_table_t *ht, uint32_t *key,
                                    void **value, void *in_node, 
                                    void **out_node)
{
    size_t i;
    ompi_list_t *list;
    ompi_list_item_t *item;
    ompi_uint32_hash_node_t *next;

    /* Try to simply get the next value in the list.  If there isn't
       one, find the next non-empty list and take the first value */

    next = (ompi_uint32_hash_node_t*) in_node;
    list = ht->ht_table + (next->hn_key & ht->ht_mask);
    item = ompi_list_get_next(next);
    if (ompi_list_get_end(list) == item) {
        item = NULL;
        for (i = (list - ht->ht_table) + 1; i < ht->ht_table_size; ++i) {
            if (ompi_list_get_size(ht->ht_table + i) > 0) {
                item = ompi_list_get_first(ht->ht_table + i);
                break;
            }
        }

        /* If we didn't find another non-empty list after this one,
           then we're at the end of the hash table */

        if (NULL == item) {
            return OMPI_ERROR;
        }
    }

    /* We found it.  Save the values (use "next" to avoid some
       typecasting) */

    *out_node = (void *) item;
    next = (ompi_uint32_hash_node_t *) *out_node;
    *key = next->hn_key;
    *value = next->hn_value;

    return OMPI_SUCCESS;
}


int 
ompi_hash_table_get_first_key_uint64(ompi_hash_table_t *ht, uint64_t *key,
                                     void **value, void **node)
{
    size_t i;
    ompi_uint64_hash_node_t *list_node;

    /* Go through all the lists and return the first element off the
       first non-empty list */
    
    for (i = 0; i < ht->ht_table_size; ++i) {
        if (ompi_list_get_size(ht->ht_table + i) > 0) {
            list_node = (ompi_uint64_hash_node_t*)
                ompi_list_get_first(ht->ht_table + i);
            *node = list_node;
            *key = list_node->hn_key;
            *value = list_node->hn_value;
            return OMPI_SUCCESS;
        }
    }

    /* The hash table is empty */

    return OMPI_ERROR;
}


int 
ompi_hash_table_get_next_key_uint64(ompi_hash_table_t *ht, uint64_t *key,
                                    void **value, void *in_node, 
                                    void **out_node)
{
    size_t i;
    ompi_list_t *list;
    ompi_list_item_t *item;
    ompi_uint64_hash_node_t *next;

    /* Try to simply get the next value in the list.  If there isn't
       one, find the next non-empty list and take the first value */

    next = (ompi_uint64_hash_node_t*) in_node;
    list = ht->ht_table + (next->hn_key & ht->ht_mask);
    item = ompi_list_get_next(next);
    if (ompi_list_get_end(list) == item) {
        item = NULL;
        for (i = (list - ht->ht_table) + 1; i < ht->ht_table_size; ++i) {
            if (ompi_list_get_size(ht->ht_table + i) > 0) {
                item = ompi_list_get_first(ht->ht_table + i);
                break;
            }
        }

        /* If we didn't find another non-empty list after this one,
           then we're at the end of the hash table */

        if (NULL == item) {
            return OMPI_ERROR;
        }
    }

    /* We found it.  Save the values (use "next" to avoid some
       typecasting) */

    *out_node = (void *) item;
    next = (ompi_uint64_hash_node_t *) *out_node;
    *key = next->hn_key;
    *value = next->hn_value;

    return OMPI_SUCCESS;
}