/*
 * Copyright (c) 2015      Mellanox Technologies, Inc.
 *                         All rights reserved.
 * $COPYRIGHT$
 * 
 * Additional copyrights may follow
 * 
 * $HEADER$
 */

#include "oshmem_config.h"

#include "oshmem/constants.h"
#include "oshmem/mca/atomic/atomic.h"
#include "oshmem/mca/atomic/base/base.h"
#include "oshmem/mca/spml/base/base.h"

#include "atomic_ucx.h"


/*
 * Public string showing the scoll ucx component version number
 */
const char *mca_atomic_ucx_component_version_string =
"Open SHMEM ucx atomic MCA component version " OSHMEM_VERSION;

/*
 * Global variable
 */
mca_spml_ucx_t *mca_spml_self = NULL;

/*
 * Local function
 */
static int ucx_register(void);
static int ucx_open(void);

/*
 * Instantiate the public struct with all of our public information
 * and pointers to our public functions in it
 */

mca_atomic_base_component_t mca_atomic_ucx_component = {

    /* First, the mca_component_t struct containing meta information
       about the component itself */

    {
        MCA_ATOMIC_BASE_VERSION_2_0_0,

        /* Component name and version */
        "ucx",
        OSHMEM_MAJOR_VERSION,
        OSHMEM_MINOR_VERSION,
        OSHMEM_RELEASE_VERSION,

        /* component open */
        ucx_open,
        /* component close */
        NULL,
        /* component query */
        NULL,
        /* component register */
        ucx_register
    },
    {
        /* The component is checkpoint ready */
        MCA_BASE_METADATA_PARAM_CHECKPOINT
    },

    /* Initialization / querying functions */

    mca_atomic_ucx_startup,
    mca_atomic_ucx_finalize,
    mca_atomic_ucx_query
};

static int ucx_register(void)
{
    mca_atomic_ucx_component.priority = 100;
    mca_base_component_var_register (&mca_atomic_ucx_component.atomic_version,
                                     "priority", "Priority of the atomic:ucx "
                                     "component (default: 100)", MCA_BASE_VAR_TYPE_INT,
                                     NULL, 0, MCA_BASE_VAR_FLAG_SETTABLE,
                                     OPAL_INFO_LVL_3,
                                     MCA_BASE_VAR_SCOPE_ALL_EQ,
                                     &mca_atomic_ucx_component.priority);

    opal_common_ucx_mca_var_register(&mca_atomic_ucx_component.atomic_version);

    return OSHMEM_SUCCESS;
}

static int ucx_open(void)
{
    /*
     * This component is able to work using spml:ikrit component only
     * (this check is added instead of !mca_spml_ikrit.enabled)
     */
    if (strcmp(mca_spml_base_selected_component.spmlm_version.mca_component_name, "ucx")) {
        ATOMIC_VERBOSE(5,
                       "Can not use atomic/ucx because spml ucx component disabled");
        return OSHMEM_ERR_NOT_AVAILABLE;
    }
    mca_spml_self = (mca_spml_ucx_t *)mca_spml.self;

    return OSHMEM_SUCCESS;
}

OBJ_CLASS_INSTANCE(mca_atomic_ucx_module_t,
                   mca_atomic_base_module_t,
                   NULL,
                   NULL);