pool_user_def.c

Go to the documentation of this file.

/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
/*
 * See COPYRIGHT in top-level directory.
 */
 
#include "abti.h"
 
#include <stdlib.h>
#include <stdarg.h>
#include <string.h>
 
int ABT_pool_user_def_create(ABT_pool_user_create_unit_fn p_create_unit,
                             ABT_pool_user_free_unit_fn p_free_unit,
                             ABT_pool_user_is_empty_fn p_is_empty,
                             ABT_pool_user_pop_fn p_pop,
                             ABT_pool_user_push_fn p_push,
                             ABT_pool_user_def *newdef)
{
    ABTI_UB_ASSERT(ABTI_initialized());
    ABTI_UB_ASSERT(newdef);
    ABTI_UB_ASSERT(p_create_unit);
    ABTI_UB_ASSERT(p_free_unit);
    ABTI_UB_ASSERT(p_is_empty);
    ABTI_UB_ASSERT(p_pop);
    ABTI_UB_ASSERT(p_push);
 
    ABTI_pool_user_def *p_newdef;
    int abt_errno =
        ABTU_calloc(1, sizeof(ABTI_pool_user_def), (void **)&p_newdef);
    ABTI_CHECK_ERROR(abt_errno);
 
    ABTI_UB_ASSERT(p_newdef->symbol == NULL); /* This value must be NULL. */
    /* Set values */
    p_newdef->required_def.p_create_unit = p_create_unit;
    p_newdef->required_def.p_free_unit = p_free_unit;
    p_newdef->required_def.p_is_empty = p_is_empty;
    p_newdef->required_def.p_pop = p_pop;
    p_newdef->required_def.p_push = p_push;
 
    *newdef = ABTI_pool_user_def_get_handle(p_newdef);
    return ABT_SUCCESS;
}
 
int ABT_pool_user_def_free(ABT_pool_user_def *def)
{
    ABTI_UB_ASSERT(ABTI_initialized());
    ABTI_UB_ASSERT(def);
 
    ABT_pool_user_def h_def = *def;
    ABTI_pool_user_def *p_def = ABTI_pool_user_def_get_ptr(h_def);
    ABTI_CHECK_NULL_POOL_USER_DEF_PTR(p_def);
 
    /* Free the memory */
    ABTU_free(p_def);
    *def = ABT_POOL_USER_DEF_NULL;
    return ABT_SUCCESS;
}
 
int ABT_pool_user_def_set_init(ABT_pool_user_def def,
                               ABT_pool_user_init_fn p_init)
{
    ABTI_UB_ASSERT(ABTI_initialized());
 
    ABTI_pool_user_def *p_def = ABTI_pool_user_def_get_ptr(def);
    ABTI_CHECK_NULL_POOL_USER_DEF_PTR(p_def);
 
    p_def->optional_def.p_init = p_init;
    return ABT_SUCCESS;
}
 
int ABT_pool_user_def_set_free(ABT_pool_user_def def,
                               ABT_pool_user_free_fn p_free)
{
    ABTI_UB_ASSERT(ABTI_initialized());
 
    ABTI_pool_user_def *p_def = ABTI_pool_user_def_get_ptr(def);
    ABTI_CHECK_NULL_POOL_USER_DEF_PTR(p_def);
 
    p_def->optional_def.p_free = p_free;
    return ABT_SUCCESS;
}
 
int ABT_pool_user_def_set_get_size(ABT_pool_user_def def,
                                   ABT_pool_user_get_size_fn p_get_size)
{
    ABTI_UB_ASSERT(ABTI_initialized());
 
    ABTI_pool_user_def *p_def = ABTI_pool_user_def_get_ptr(def);
    ABTI_CHECK_NULL_POOL_USER_DEF_PTR(p_def);
 
    p_def->optional_def.p_get_size = p_get_size;
    return ABT_SUCCESS;
}
 
int ABT_pool_user_def_set_pop_wait(ABT_pool_user_def def,
                                   ABT_pool_user_pop_wait_fn p_pop_wait)
{
    ABTI_UB_ASSERT(ABTI_initialized());
 
    ABTI_pool_user_def *p_def = ABTI_pool_user_def_get_ptr(def);
    ABTI_CHECK_NULL_POOL_USER_DEF_PTR(p_def);
 
    p_def->optional_def.p_pop_wait = p_pop_wait;
    return ABT_SUCCESS;
}
 
int ABT_pool_user_def_set_pop_many(ABT_pool_user_def def,
                                   ABT_pool_user_pop_many_fn p_pop_many)
{
    ABTI_UB_ASSERT(ABTI_initialized());
 
    ABTI_pool_user_def *p_def = ABTI_pool_user_def_get_ptr(def);
    ABTI_CHECK_NULL_POOL_USER_DEF_PTR(p_def);
 
    p_def->optional_def.p_pop_many = p_pop_many;
    return ABT_SUCCESS;
}
 
int ABT_pool_user_def_set_push_many(ABT_pool_user_def def,
                                    ABT_pool_user_push_many_fn p_push_many)
{
    ABTI_UB_ASSERT(ABTI_initialized());
 
    ABTI_pool_user_def *p_def = ABTI_pool_user_def_get_ptr(def);
    ABTI_CHECK_NULL_POOL_USER_DEF_PTR(p_def);
 
    p_def->optional_def.p_push_many = p_push_many;
    return ABT_SUCCESS;
}
 
int ABT_pool_user_def_set_print_all(ABT_pool_user_def def,
                                    ABT_pool_user_print_all_fn p_print_all)
{
    ABTI_UB_ASSERT(ABTI_initialized());
 
    ABTI_pool_user_def *p_def = ABTI_pool_user_def_get_ptr(def);
    ABTI_CHECK_NULL_POOL_USER_DEF_PTR(p_def);
 
    p_def->optional_def.p_print_all = p_print_all;
    return ABT_SUCCESS;
}
 
/*****************************************************************************/
/* Private APIs                                                              */
/*****************************************************************************/
 
ABT_bool ABTI_pool_user_def_is_new(const ABT_pool_user_def def)
{
    /* If def points to ABT_pool_def, this u_create_from_thread not be NULL.
     * Otherwise, it is "symbol", so it must be NULL. */
    return def->u_create_from_thread ? ABT_FALSE : ABT_TRUE;
}