/**************************************************************************** * NCSA HDF * * Software Development Group * * National Center for Supercomputing Applications * * University of Illinois at Urbana-Champaign * * 605 E. Springfield, Champaign IL 61820 * * * * For conditions of distribution and use, see the accompanying * * hdf/COPYING file. * * * ****************************************************************************/ #ifdef RCSID static char RcsId[] = "@(#)$Revision$"; #endif /* $Id$ */ /* FILE atom.c - Internal storage routines for handling "atoms" REMARKS Atoms are just ID's which allow objects (void *'s currently) to be bundled into "groups" for more general storage. DESIGN The groups are stored in an array of pointers to store each group in an element. Each "atomic group" node contains a link to a hash table to manage the atoms in each group. The allowed "atomic groups" are stored in an enum (called group_t) in atom.h. BUGS/LIMITATIONS Can't interate over the atoms in a group. LOCAL ROUTINES H5A_find_atom - Returns a pointer to an atom_info_t from a atom ID H5A_get_atom_node - Gets an atom node (uses the atom free list) H5A_release_atom_node - Releases an atom node (uses the atom free list) EXPORTED ROUTINES Atom Functions: H5A_register - Register an object in a group and get an atom for it H5A_object - Get the object for an atom H5A_group - Get the group for an atom H5A_remove - Remove an atom from a group H5A_search - Search a group for a particular object Atom Group Functions: H5A_init_group - Initialize a group to store atoms in H5A_destroy_group - Destroy an atomic group Atom Group Cleanup: H5Ashutdown - Terminate various static buffers. AUTHOR Quincey Koziol MODIFICATION HISTORY 1/3/96 - Starting writing specs & coding prototype 1/7/96 - Finished coding prototype 6/10/97 - Moved into HDF5 library */ #include #include #include #include #define PABLO_MASK H5A_mask /*-------------------- Locally scoped variables -----------------------------*/ #ifdef ATOMS_ARE_CACHED /* Array of pointers to atomic groups */ static hid_t atom_id_cache[ATOM_CACHE_SIZE] = {-1, -1, -1, -1}; static void *atom_obj_cache[ATOM_CACHE_SIZE]; #endif /* Array of pointers to atomic groups */ static atom_group_t *atom_group_list[MAXGROUP]; /* Pointer to the atom node free list */ static atom_info_t *atom_free_list = NULL; /* Interface initialialization? */ static hbool_t interface_initialize_g = FALSE; #define INTERFACE_INIT H5A_init_interface static herr_t H5A_init_interface(void); /*--------------------- Local function prototypes ---------------------------*/ static atom_info_t *H5A_find_atom(hid_t atm); static atom_info_t *H5A_get_atom_node(void); static herr_t H5A_release_atom_node(atom_info_t *atm); /*-------------------------------------------------------------------------- NAME H5A_init_interface -- Initialize interface-specific information USAGE herr_t H5A_init_interface() RETURNS SUCCEED/FAIL DESCRIPTION Initializes any interface-specific data or routines. --------------------------------------------------------------------------*/ static herr_t H5A_init_interface(void) { herr_t ret_value = SUCCEED; FUNC_ENTER(H5A_init_interface, FAIL); /* Registers the cleanup routine with the exit chain */ ret_value = H5_add_exit(&H5A_term_interface); FUNC_LEAVE(ret_value); } /****************************************************************************** NAME H5A_init_group - Initialize an atomic group DESCRIPTION Creates a global atomic group to store atoms in. If the group has already been initialized, this routine just increments the count of # of initializations and returns without trying to change the size of the hash table. A specific number of group entries may be reserved to enable "constant" values to be handed out which are valid atoms in the group, but which do not map to any data structures and are not allocated dynamicly later. RETURNS Returns SUCCEED if successful and FAIL otherwise ******************************************************************************/ intn H5A_init_group(group_t grp, /* IN: Group to initialize */ intn hash_size, /* IN: Minimum hash table size to use for group */ uintn reserved, /* IN: Number of hash table entries to reserve */ herr_t (*free_func) (void *) /* IN: Function to call when releasing ref counted objects */ ) { atom_group_t *grp_ptr = NULL; /* ptr to the atomic group */ intn ret_value = SUCCEED; FUNC_ENTER(H5A_init_group, FAIL); if ((grp <= BADGROUP || grp >= MAXGROUP) && hash_size > 0) { HGOTO_DONE(FAIL); } #ifdef HASH_SIZE_POWER_2 /* * If anyone knows a faster test for a power of two, please change this * silly code -QAK */ if (!(hash_size == 2 || hash_size == 4 || hash_size == 8 || hash_size == 16 || hash_size == 32 || hash_size == 64 || hash_size == 128 || hash_size == 256 || hash_size == 512 || hash_size == 1024 || hash_size == 2048 || hash_size == 4096 || hash_size == 8192 || hash_size == 16374 || hash_size == 32768 || hash_size == 65536 || hash_size == 131072 || hash_size == 262144 || hash_size == 524288 || hash_size == 1048576 || hash_size == 2097152 || hash_size == 4194304 || hash_size == 8388608 || hash_size == 16777216 || hash_size == 33554432 || hash_size == 67108864 || hash_size == 134217728 || hash_size == 268435456)) HGOTO_DONE(FAIL); #endif /* HASH_SIZE_POWER_2 */ if (atom_group_list[grp] == NULL) { /* Allocate the group information */ grp_ptr = H5MM_xcalloc(1, sizeof(atom_group_t)); atom_group_list[grp] = grp_ptr; } else { /* Get the pointer to the existing group */ grp_ptr = atom_group_list[grp]; } if (grp_ptr->count == 0) { /* Initialize the atom group structure */ grp_ptr->hash_size = hash_size; grp_ptr->reserved = reserved; grp_ptr->wrapped = 0; grp_ptr->atoms = 0; grp_ptr->nextid = reserved; grp_ptr->free_func = free_func; grp_ptr->atom_list = H5MM_xcalloc(hash_size, sizeof(atom_info_t *)); } /* Increment the count of the times this group has been initialized */ grp_ptr->count++; #ifdef QAK printf("%s: group ID=%d, count=%d, current # of active atoms=%d\n", FUNC, grp, grp_ptr->count, grp_ptr->atoms); #endif /* QAK */ done: if (ret_value == FAIL) { /* Error condition cleanup */ if (grp_ptr != NULL) { H5MM_xfree (grp_ptr->atom_list); H5MM_xfree (grp_ptr); } } /* Normal function cleanup */ FUNC_LEAVE(ret_value); } /*------------------------------------------------------------------------- * Function: H5A_destroy_group * * Purpose: Decrements the reference count on an entire group of atoms. * If the group reference count becomes zero then the group is * destroyed along with all atoms in that group regardless of * their reference counts. Destroying atoms involves calling * the free-func for each atom object and then adding the atom * struct to the atom free list. * * Return: Success: SUCCEED * * Failure: FAIL * * Programmer: Unknown * * Modifications: * * Robb Matzke, 25 Feb 1998 * Atoms are freed when a group is destroyed. * *------------------------------------------------------------------------- */ herr_t H5A_destroy_group(group_t grp) { atom_group_t *grp_ptr = NULL; /* ptr to the atomic group */ atom_info_t *cur=NULL, *next=NULL; intn ret_value = SUCCEED; intn i; FUNC_ENTER(H5A_destroy_group, FAIL); if (grp <= BADGROUP || grp >= MAXGROUP) HGOTO_DONE(FAIL); grp_ptr = atom_group_list[grp]; if (grp_ptr == NULL || grp_ptr->count <= 0) HGOTO_DONE(FAIL); #ifdef QAK printf("%s: group ID=%d, count=%d, current # of active atoms=%d\n", FUNC, grp, grp_ptr->count, grp_ptr->atoms); #endif /* QAK */ /* * Decrement the number of users of the atomic group. If this is the * last user of the group then release all atoms from the group. The * free function is invoked for each atom being freed. */ if ((--(grp_ptr->count)) == 0) { #ifdef ATOMS_ARE_CACHED /* * Remove atoms from the global atom cache. */ for (i=0; ifree_func) { for (i=0; ihash_size; i++) { for (cur=grp_ptr->atom_list[i]; cur; cur=next) { /* Free the object */ (grp_ptr->free_func)(cur->obj_ptr); /* Add atom struct to free list */ next = cur->next; cur->next = atom_free_list; atom_free_list = cur; } } } /* Free local cache and reset group */ H5MM_xfree(grp_ptr->atom_list); HDmemset (grp_ptr, 0, sizeof(grp_ptr)); } done: FUNC_LEAVE(ret_value); } /****************************************************************************** NAME H5A_register - Register an object in a group and get an atom for it. DESCRIPTION Registers an object in a group and returns an atom for it. This routine does _not_ check for unique-ness of the objects, if you register an object twice, you will get two different atoms for it. This routine does make certain that each atom in a group is unique. Atoms are created by getting a unique number for the group the atom is in and incorporating the group into the atom which is returned to the user. RETURNS Returns atom if successful and FAIL otherwise *******************************************************************************/ hid_t H5A_register(group_t grp, /* IN: Group to register the object in */ void *object /* IN: Object to attach to atom */ ) { atom_group_t *grp_ptr = NULL; /* ptr to the atomic group */ atom_info_t *atm_ptr = NULL; /* ptr to the new atom */ hid_t atm_id; /* new atom ID */ uintn hash_loc; /* new item's hash table location */ hid_t ret_value = SUCCEED; FUNC_ENTER(H5A_register, FAIL); if (grp <= BADGROUP || grp >= MAXGROUP) HGOTO_DONE(FAIL); grp_ptr = atom_group_list[grp]; if (grp_ptr == NULL || grp_ptr->count <= 0) HGOTO_DONE(FAIL); if ((atm_ptr = H5A_get_atom_node()) == NULL) HGOTO_DONE(FAIL); /* Create the atom & it's ID */ atm_id = MAKE_ATOM(grp, grp_ptr->nextid); atm_ptr->id = atm_id; atm_ptr->count = 1; /*initial reference count*/ atm_ptr->obj_ptr = object; atm_ptr->next = NULL; /* hash bucket already full, prepend to front of chain */ hash_loc = grp_ptr->nextid % (uintn) grp_ptr->hash_size; if (grp_ptr->atom_list[hash_loc] != NULL) atm_ptr->next = grp_ptr->atom_list[hash_loc]; /* Insert into the group */ grp_ptr->atom_list[hash_loc] = atm_ptr; grp_ptr->atoms++; grp_ptr->nextid++; /* * This next section of code checks for the 'nextid' getting too large and * wrapping around, thus necessitating checking for duplicate atoms being * handed out. */ if (grp_ptr->nextid > (uintn) ATOM_MASK || grp_ptr->wrapped != 0) { if (grp_ptr->wrapped == 0) { /* set the "wrapped around" flag if it isn't already */ grp_ptr->wrapped = 1; /* re-start the ID counter */ grp_ptr->nextid = grp_ptr->reserved; } do { /* new atom to check for */ hid_t next_atom = MAKE_ATOM(grp, grp_ptr->nextid); atom_info_t *curr_atm; /* ptr to the current atom */ hash_loc = ATOM_TO_LOC (grp_ptr->nextid, grp_ptr->hash_size); curr_atm = grp_ptr->atom_list[hash_loc]; if (curr_atm == NULL) break; /* Ha! this is not likely... */ while (curr_atm != NULL) { if (curr_atm->id == next_atom) break; curr_atm = curr_atm->next; } if (curr_atm == NULL) break; /* must not have found a match */ grp_ptr->nextid++; } while (grp_ptr->nextid <= (uintn) ATOM_MASK); if (grp_ptr->nextid > (uintn) ATOM_MASK) { /* All the atoms are gone! */ HGOTO_DONE(FAIL); } } ret_value = atm_id; done: if (ret_value == FAIL) { /* Error condition cleanup */ } /* Normal function cleanup */ FUNC_LEAVE(ret_value); } /****************************************************************************** NAME H5A_inc_ref - Adds a reference to a reference counted atom. IN: Atom to increment reference count for DESCRIPTION Increments the number of references outstanding for an atom. This will fail if the group is not a reference counted group. RETURNS atm/FAIL *******************************************************************************/ hid_t H5A_inc_ref(hid_t atm) { group_t grp = ATOM_TO_GROUP(atm); /* object's group */ atom_group_t *grp_ptr = NULL; /* ptr to the atomic group*/ atom_info_t *atm_ptr = NULL; /* ptr to the new atom */ hid_t ret_value = FAIL; FUNC_ENTER(H5A_inc_ref, FAIL); grp_ptr = atom_group_list[grp]; if (grp_ptr == NULL || grp_ptr->count <= 0 || grp_ptr->free_func == NULL) { HRETURN(FAIL); } /* General lookup of the atom */ if (NULL!=(atm_ptr = H5A_find_atom(atm))) { atm_ptr->count++; ret_value = atm; } FUNC_LEAVE(ret_value); } /****************************************************************************** NAME H5A_object - Returns to the object ptr for the atom DESCRIPTION Retrieves the object ptr which is associated with the atom. RETURNS Returns object ptr if successful and NULL otherwise *******************************************************************************/ void * H5A_object(hid_t atm) { #ifdef ATOMS_ARE_CACHED uintn i; /* local counter */ #endif /* ATOMS_ARE_CACHED */ atom_info_t *atm_ptr = NULL; /* ptr to the new atom */ void *ret_value = NULL; FUNC_ENTER(H5A_object, NULL); #ifdef ATOMS_ARE_CACHED /* * Look for the atom in the cache first. Implement a simple "move * forward" caching scheme by swapping the found cache item with the * previous cache item. This gradually migrates used cache items toward * the front of the cache and unused items toward the end. For instance, * finding `e' in the cache results in: * * Before: a b c d e f g h i j * | | | X | | | | | * After: a b c e d f g h i j */ for (i=0; i 0) { hid_t t_atom = atom_id_cache[i-1]; void *t_obj = atom_obj_cache[i-1]; atom_id_cache[i-1] = atom_id_cache[i]; atom_obj_cache[i-1] = atom_obj_cache[i]; atom_id_cache[i] = t_atom; atom_obj_cache[i] = t_obj; } HGOTO_DONE(ret_value); } #endif /* ATOMS_ARE_CACHED */ /* General lookup of the atom */ if ((atm_ptr = H5A_find_atom(atm)) == NULL) HGOTO_DONE(NULL); /* Check if we've found the correct atom */ if (atm_ptr != NULL) ret_value = atm_ptr->obj_ptr; done: FUNC_LEAVE(ret_value); } /****************************************************************************** NAME H5A_group - Returns to the group for the atom DESCRIPTION Retrieves the group which is associated with the atom. RETURNS Returns group if successful and BADGROUP otherwise *******************************************************************************/ group_t H5A_group(hid_t atm) { group_t ret_value = BADGROUP; FUNC_ENTER(H5A_group, BADGROUP); ret_value = ATOM_TO_GROUP(atm); if (ret_value <= BADGROUP || ret_value >= MAXGROUP) { HGOTO_DONE(BADGROUP); } done: FUNC_LEAVE(ret_value); } /****************************************************************************** NAME H5A_remove - Removes an atom from a group DESCRIPTION Removes an atom from a group. RETURNS Returns atom's object if successful and NULL otherwise *******************************************************************************/ void * H5A_remove(hid_t atm) { atom_group_t *grp_ptr = NULL;/* ptr to the atomic group */ atom_info_t *curr_atm, /* ptr to the current atom */ *last_atm; /* ptr to the last atom */ group_t grp; /* atom's atomic group */ uintn hash_loc; /* atom's hash table location */ #ifdef ATOMS_ARE_CACHED uintn i; /* local counting variable */ #endif void * ret_value = NULL; FUNC_ENTER(H5A_remove, NULL); grp = ATOM_TO_GROUP(atm); if (grp <= BADGROUP || grp >= MAXGROUP) HGOTO_DONE(NULL); grp_ptr = atom_group_list[grp]; if (grp_ptr == NULL || grp_ptr->count <= 0) HGOTO_DONE(NULL); /* Get the location in which the atom is located */ hash_loc = (uintn) ATOM_TO_LOC(atm, grp_ptr->hash_size); curr_atm = grp_ptr->atom_list[hash_loc]; if (curr_atm == NULL) HGOTO_DONE(NULL); last_atm = NULL; while (curr_atm != NULL) { if (curr_atm->id == atm) break; last_atm = curr_atm; curr_atm = curr_atm->next; } if (curr_atm != NULL) { if (last_atm == NULL) { /* atom is the first the chain */ grp_ptr->atom_list[hash_loc] = curr_atm->next; } else { last_atm->next = curr_atm->next; } ret_value = curr_atm->obj_ptr; H5A_release_atom_node(curr_atm); } else { /* couldn't find the atom in the proper place */ HGOTO_DONE(NULL); } #ifdef ATOMS_ARE_CACHED /* Delete object from cache */ for (i = 0; i < ATOM_CACHE_SIZE; i++) if (atom_id_cache[i] == atm) { atom_id_cache[i] = (-1); atom_obj_cache[i] = NULL; break; /* we assume there is only one instance in the cache */ } #endif /* ATOMS_ARE_CACHED */ /* Decrement the number of atoms in the group */ (grp_ptr->atoms)--; done: FUNC_LEAVE(ret_value); } /*------------------------------------------------------------------------- * Function: H5A_dec_ref * * Purpose: Decrements the number of references outstanding for an atom. * This will fail if the group is not a reference counted group. * The atom group's 'free' function will be called for the atom * if the reference count for the atom reaches 0 and a free * function has been defined at group creation time. * * Return: Success: New reference count. * * Failure: FAIL * * Programmer: Unknown * * Modifications: * * Robb Matzke, 19 Feb 1998 * It is no longer an error when the reference count of an item reaches * zero and no `free' function has been defined. The object is still * removed from the list. * *------------------------------------------------------------------------- */ intn H5A_dec_ref(hid_t atm) { group_t grp = ATOM_TO_GROUP(atm); /* Group the object is in */ atom_group_t *grp_ptr = NULL; /* ptr to the atomic group */ atom_info_t *atm_ptr = NULL; /* ptr to the new atom */ void * obj; /* object to call 'free' function with */ intn ret_value = FAIL; FUNC_ENTER(H5A_dec_ref, FAIL); grp_ptr = atom_group_list[grp]; if (grp_ptr == NULL || grp_ptr->count <= 0) { HRETURN(FAIL); } /* General lookup of the atom */ if ((atm_ptr = H5A_find_atom(atm)) != NULL) { /* Decrement the reference count */ ret_value = --(atm_ptr->count); /* If the reference count is zero, remove the object from the group */ if (0 == atm_ptr->count && (obj = H5A_remove(atm)) != NULL) { /* * call the user's 'free' function for the atom's information, * otherwise just leak memory. */ if (grp_ptr->free_func) (grp_ptr->free_func)(obj); } ret_value = SUCCEED; } FUNC_LEAVE(ret_value); } /****************************************************************************** NAME H5A_search - Search for an object in a group and get it's pointer. DESCRIPTION Searchs for an object in a group and returns the pointer to it. This routine calls the function pointer passed in for each object in the group until it finds a match. Currently there is no way to resume a search. RETURNS Returns pointer an atom's object if successful and NULL otherwise *******************************************************************************/ void * H5A_search(group_t grp, /* IN: Group to search for the object in */ H5Asearch_func_t func, /* IN: Ptr to the comparison function */ const void *key /* IN: pointer to key to compare against */ ) { atom_group_t *grp_ptr = NULL; /* ptr to the atomic group */ atom_info_t *atm_ptr = NULL; /* ptr to the new atom */ intn i; /* local counting variable */ void * ret_value = NULL; FUNC_ENTER(H5A_search, NULL); if (grp <= BADGROUP || grp >= MAXGROUP) HGOTO_DONE(NULL); grp_ptr = atom_group_list[grp]; if (grp_ptr == NULL || grp_ptr->count <= 0) HGOTO_DONE(NULL); /* Start at the beginning of the array */ for (i = 0; i < grp_ptr->hash_size; i++) { atm_ptr = grp_ptr->atom_list[i]; while (atm_ptr != NULL) { if ((*func) (atm_ptr->obj_ptr, key)) HGOTO_DONE(atm_ptr->obj_ptr); /* found the item we are looking for */ atm_ptr = atm_ptr->next; } } done: FUNC_LEAVE(ret_value); } /****************************************************************************** NAME H5A_find_atom - Finds a atom in a group DESCRIPTION Retrieves the atom ptr which is associated with the atom. RETURNS Returns atom ptr if successful and NULL otherwise *******************************************************************************/ static atom_info_t * H5A_find_atom(hid_t atm) { atom_group_t *grp_ptr = NULL; /* ptr to the atomic group */ atom_info_t *atm_ptr = NULL; /* ptr to the new atom */ group_t grp; /* atom's atomic group */ uintn hash_loc; /* atom's hash table location */ atom_info_t *ret_value = NULL; FUNC_ENTER(H5A_find_atom, NULL); grp = ATOM_TO_GROUP(atm); if (grp <= BADGROUP || grp >= MAXGROUP) HGOTO_DONE(NULL); grp_ptr = atom_group_list[grp]; if (grp_ptr == NULL || grp_ptr->count <= 0) HGOTO_DONE(NULL); /* Get the location in which the atom is located */ hash_loc = (uintn) ATOM_TO_LOC(atm, grp_ptr->hash_size); atm_ptr = grp_ptr->atom_list[hash_loc]; if (atm_ptr == NULL) HGOTO_DONE(NULL); while (atm_ptr != NULL) { if (atm_ptr->id == atm) break; atm_ptr = atm_ptr->next; } ret_value = atm_ptr; #ifdef ATOMS_ARE_CACHED atom_id_cache[ATOM_CACHE_SIZE-1] = atm; atom_obj_cache[ATOM_CACHE_SIZE-1] = atm_ptr->obj_ptr; #endif /* ATOMS_ARE_CACHED */ done: FUNC_LEAVE(ret_value); } /****************************************************************************** NAME H5A_get_atom_node - Gets an atom node DESCRIPTION Either gets an atom node from the free list (if there is one available) or allocate a node. RETURNS Returns atom ptr if successful and NULL otherwise *******************************************************************************/ static atom_info_t * H5A_get_atom_node(void) { atom_info_t *ret_value = NULL; FUNC_ENTER(H5A_get_atom_node, NULL); if (atom_free_list != NULL) { ret_value = atom_free_list; atom_free_list = atom_free_list->next; } else { ret_value = H5MM_xmalloc(sizeof(atom_info_t)); } FUNC_LEAVE(ret_value); } /****************************************************************************** NAME H5A_release_atom_node - Releases an atom node DESCRIPTION Puts an atom node into the free list RETURNS SUCCEED *******************************************************************************/ static herr_t H5A_release_atom_node(atom_info_t *atm) { FUNC_ENTER(H5A_release_atom_node, FAIL); /* Insert the atom at the beginning of the free list */ atm->next = atom_free_list; atom_free_list = atm; FUNC_LEAVE(SUCCEED); } /*-------------------------------------------------------------------------- NAME H5A_term_interface PURPOSE Terminate various static buffers. USAGE intn H5A_term_interface() RETURNS Returns SUCCEED/FAIL DESCRIPTION Free various buffers allocated in the H5A routines. GLOBAL VARIABLES COMMENTS, BUGS, ASSUMPTIONS Should only ever be called by the "atexit" function HDFend EXAMPLES REVISION LOG --------------------------------------------------------------------------*/ void H5A_term_interface(void) { atom_info_t *curr; intn i; /* Release the free-list if it exists */ if (atom_free_list != NULL) { while (atom_free_list != NULL) { curr = atom_free_list; atom_free_list = atom_free_list->next; HDfree(curr); } } /* Release all groups */ for (i = 0; i < (intn) MAXGROUP; i++) { if (atom_group_list[i] != NULL) { HDfree(atom_group_list[i]); atom_group_list[i] = NULL; } } }