/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
/*
 * 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 (c) 2009      Cisco Systems, Inc.  All rights reserved.
 * Copyright (c) 2016      Research Organization for Information Science
 *                         and Technology (RIST). All rights reserved.
 * Copyright (c) 2016      Los Alamos National Security, LLC. All rights
 *                         reserved.
 * $COPYRIGHT$
 *
 * Additional copyrights may follow
 *
 * $HEADER$
 */

#include "opal_config.h"

#include "opal/constants.h"
#include "opal/mca/base/base.h"
#include "opal/mca/mca.h"
#include "opal/mca/memory/base/base.h"
#include "opal/mca/memory/base/empty.h"
#include "opal/mca/memory/memory.h"

/*
 * The following file was created by configure.  It contains extern
 * statements and the definition of an array of pointers to each
 * component's public mca_base_component_t struct.
 */
#include "opal/mca/memory/base/static-components.h"

static int empty_process(void)
{
    return OPAL_SUCCESS;
}

static int empty_query(int *priority)
{
    *priority = 0;
    return OPAL_SUCCESS;
}

/*
 * Local variables
 */
static opal_memory_base_component_2_0_0_t empty_component = {
    /* Empty / safe functions to call if no memory component is selected */
    .memoryc_query = empty_query,
    .memoryc_process = empty_process,
    .memoryc_register = opal_memory_base_component_register_empty,
    .memoryc_deregister = opal_memory_base_component_deregister_empty,
    .memoryc_set_alignment = opal_memory_base_component_set_alignment_empty,
};

/*
 * Globals
 */
opal_memory_base_component_2_0_0_t *opal_memory = &empty_component;

void opal_memory_base_malloc_init_hook(void)
{
    if (opal_memory->memoryc_init_hook) {
        opal_memory->memoryc_init_hook();
    }
}

/*
 * Function for finding and opening either all MCA components, or the one
 * that was specifically requested via a MCA parameter.
 */
static int opal_memory_base_open(mca_base_open_flag_t flags)
{
    mca_base_component_list_item_t *item, *next;
    opal_memory_base_component_2_0_0_t *tmp;
    int priority, highest_priority = 0;
    int ret;

    /* can only be zero or one */
    OPAL_LIST_FOREACH (item, &opal_memory_base_framework.framework_components,
                       mca_base_component_list_item_t) {
        tmp = (opal_memory_base_component_2_0_0_t *) item->cli_component;
        ret = tmp->memoryc_query(&priority);
        if (OPAL_SUCCESS != ret || priority < highest_priority) {
            continue;
        }

        highest_priority = priority;
        opal_memory = tmp;
    }

    OPAL_LIST_FOREACH_SAFE (item, next, &opal_memory_base_framework.framework_components,
                            mca_base_component_list_item_t) {
        if ((void *) opal_memory != (void *) item->cli_component) {
            mca_base_component_unload(item->cli_component,
                                      opal_memory_base_framework.framework_output);
            opal_list_remove_item(&opal_memory_base_framework.framework_components, &item->super);
        }
    }

    /* open remaining component */
    ret = mca_base_framework_components_open(&opal_memory_base_framework, flags);
    if (ret != OPAL_SUCCESS) {
        return ret;
    }

    /* All done */
    return OPAL_SUCCESS;
}

/* Use default register/close functions */
MCA_BASE_FRAMEWORK_DECLARE(opal, memory, "memory hooks", NULL, opal_memory_base_open, NULL,
                           mca_memory_base_static_components, 0);