5b9fa7e998
the RML/OOB merge in slightly easier This commit was SVN r15530. The following SVN revision numbers were found above: r15517 --> open-mpi/ompi@41977fcc95 r15520 --> open-mpi/ompi@9cbc9df1b8 r15527 --> open-mpi/ompi@2d17dd9516
143 строки
4.4 KiB
C
143 строки
4.4 KiB
C
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2005 The University of Tennessee and The University
|
|
* of Tennessee Research Foundation. 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 "orte_config.h"
|
|
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "orte/orte_constants.h"
|
|
#include "opal/util/output.h"
|
|
#include "orte/class/orte_proc_table.h"
|
|
#include "orte/mca/ns/ns.h"
|
|
#include "orte/mca/ns/ns_types.h"
|
|
|
|
/*
|
|
* orte_process_name_hash_node_t
|
|
*/
|
|
|
|
struct orte_proc_hash_node_t
|
|
{
|
|
opal_list_item_t super;
|
|
orte_process_name_t hn_key;
|
|
void *hn_value;
|
|
};
|
|
typedef struct orte_proc_hash_node_t orte_proc_hash_node_t;
|
|
|
|
static OBJ_CLASS_INSTANCE(
|
|
orte_proc_hash_node_t,
|
|
opal_list_item_t,
|
|
NULL,
|
|
NULL);
|
|
|
|
#define GET_KEY(proc) \
|
|
( (((uint32_t) proc->jobid) << 24) + ((uint32_t) proc->vpid) )
|
|
|
|
void* orte_hash_table_get_proc(opal_hash_table_t* ht,
|
|
const orte_process_name_t* proc)
|
|
{
|
|
uint32_t key = GET_KEY(proc);
|
|
opal_list_t* list = ht->ht_table + (key & ht->ht_mask);
|
|
orte_proc_hash_node_t *node;
|
|
|
|
#if OMPI_ENABLE_DEBUG
|
|
if(ht->ht_table_size == 0) {
|
|
opal_output(0, "opal_hash_table_get_proc:"
|
|
"opal_hash_table_init() has not been called");
|
|
return NULL;
|
|
}
|
|
#endif
|
|
for(node = (orte_proc_hash_node_t*)opal_list_get_first(list);
|
|
node != (orte_proc_hash_node_t*)opal_list_get_end(list);
|
|
node = (orte_proc_hash_node_t*)opal_list_get_next(node)) {
|
|
if (0 == orte_ns.compare_fields(ORTE_NS_CMP_ALL, &node->hn_key, proc)) {
|
|
return node->hn_value;
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
|
|
int orte_hash_table_set_proc(
|
|
opal_hash_table_t* ht,
|
|
const orte_process_name_t* proc,
|
|
void* value)
|
|
{
|
|
uint32_t key = GET_KEY(proc);
|
|
opal_list_t* list = ht->ht_table + (key & ht->ht_mask);
|
|
orte_proc_hash_node_t *node;
|
|
|
|
#if OMPI_ENABLE_DEBUG
|
|
if(ht->ht_table_size == 0) {
|
|
opal_output(0, "opal_hash_table_set_value_proc:"
|
|
"opal_hash_table_init() has not been called");
|
|
return ORTE_ERR_BAD_PARAM;
|
|
}
|
|
#endif
|
|
for(node = (orte_proc_hash_node_t*)opal_list_get_first(list);
|
|
node != (orte_proc_hash_node_t*)opal_list_get_end(list);
|
|
node = (orte_proc_hash_node_t*)opal_list_get_next(node)) {
|
|
if (0 == orte_ns.compare_fields(ORTE_NS_CMP_ALL, &node->hn_key, proc)) {
|
|
node->hn_value = value;
|
|
return ORTE_SUCCESS;
|
|
}
|
|
}
|
|
|
|
node = (orte_proc_hash_node_t*)opal_list_remove_first(&ht->ht_nodes);
|
|
if(NULL == node) {
|
|
node = OBJ_NEW(orte_proc_hash_node_t);
|
|
if(NULL == node)
|
|
return ORTE_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
node->hn_key = *proc;
|
|
node->hn_value = value;
|
|
opal_list_append(list, (opal_list_item_t*)node);
|
|
ht->ht_size++;
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
|
|
int orte_hash_table_remove_proc(
|
|
opal_hash_table_t* ht,
|
|
const orte_process_name_t* proc)
|
|
{
|
|
uint32_t key = GET_KEY(proc);
|
|
opal_list_t* list = ht->ht_table + (key & ht->ht_mask);
|
|
orte_proc_hash_node_t *node;
|
|
|
|
#if OMPI_ENABLE_DEBUG
|
|
if(ht->ht_table_size == 0) {
|
|
opal_output(0, "opal_hash_table_remove_value_proc:"
|
|
"opal_hash_table_init() has not been called");
|
|
return ORTE_ERR_BAD_PARAM;
|
|
}
|
|
#endif
|
|
for(node = (orte_proc_hash_node_t*)opal_list_get_first(list);
|
|
node != (orte_proc_hash_node_t*)opal_list_get_end(list);
|
|
node = (orte_proc_hash_node_t*)opal_list_get_next(node)) {
|
|
if (0 == orte_ns.compare_fields(ORTE_NS_CMP_ALL, &node->hn_key, proc)) {
|
|
opal_list_remove_item(list, (opal_list_item_t*)node);
|
|
opal_list_append(&ht->ht_nodes, (opal_list_item_t*)node);
|
|
ht->ht_size--;
|
|
return ORTE_SUCCESS;
|
|
}
|
|
}
|
|
return ORTE_ERR_NOT_FOUND;
|
|
}
|
|
|