/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
/*
 * Copyright (C) Mellanox Technologies Ltd. 2001-2011.  ALL RIGHTS RESERVED.
 * Copyright (c) 2015      Los Alamos National Security, LLC. All rights
 *                         reserved.
 * Copyright (c) 2015 Cisco Systems, Inc.  All rights reserved.
 * $COPYRIGHT$
 *
 * Additional copyrights may follow
 *
 * $HEADER$
 */

#include "pml_yalla.h"

#include "ompi/runtime/mpiruntime.h"


static int mca_pml_yalla_component_register(void);
static int mca_pml_yalla_component_open(void);
static int mca_pml_yalla_component_close(void);

static  mca_pml_base_module_t*
mca_pml_yalla_component_init(int* priority, bool enable_progress_threads,
                             bool enable_mpi_threads);
static int mca_pml_yalla_component_fini(void);


mca_pml_base_component_2_0_0_t mca_pml_yalla_component = {

    /* First, the mca_base_component_t struct containing meta
     * information about the component itself */
    .pmlm_version = {
        MCA_PML_BASE_VERSION_2_0_0,

        .mca_component_name = "yalla",
        MCA_BASE_MAKE_VERSION(component, OMPI_MAJOR_VERSION, OMPI_MINOR_VERSION,
                              OMPI_RELEASE_VERSION),
        .mca_open_component = mca_pml_yalla_component_open,
        .mca_close_component = mca_pml_yalla_component_close,
        .mca_register_component_params = mca_pml_yalla_component_register,
    },
    .pmlm_data = {
        /* This component is not checkpoint ready */
        MCA_BASE_METADATA_PARAM_NONE
    },

    .pmlm_init = mca_pml_yalla_component_init,
    .pmlm_finalize = mca_pml_yalla_component_fini,
};

static int mca_pml_yalla_component_register(void)
{
    ompi_pml_yalla.verbose = 0;
    (void) mca_base_component_var_register(&mca_pml_yalla_component.pmlm_version, "verbose",
                                           "Verbose level of the yalla component",
                                           MCA_BASE_VAR_TYPE_INT, NULL, 0, 0,
                                           OPAL_INFO_LVL_9,
                                           MCA_BASE_VAR_SCOPE_LOCAL,
                                           &ompi_pml_yalla.verbose);

    ompi_pml_yalla.priority = 50;
    (void) mca_base_component_var_register(&mca_pml_yalla_component.pmlm_version, "priority",
                                           "Priority of the yalla component",
                                           MCA_BASE_VAR_TYPE_INT, NULL, 0, 0,
                                           OPAL_INFO_LVL_3,
                                           MCA_BASE_VAR_SCOPE_LOCAL,
                                           &ompi_pml_yalla.priority);
    return 0;
}

static int mca_pml_yalla_component_open(void)
{
    ompi_pml_yalla.output = opal_output_open(NULL);
    opal_output_set_verbosity(ompi_pml_yalla.output, ompi_pml_yalla.verbose);
    return mca_pml_yalla_open();
}

static int mca_pml_yalla_component_close(void)
{
    int rc;

    rc = mca_pml_yalla_close();
    if (rc != 0) {
        return rc;
    }

    opal_output_close(ompi_pml_yalla.output);
    return 0;
}

static mca_pml_base_module_t*
mca_pml_yalla_component_init(int* priority, bool enable_progress_threads,
                             bool enable_mpi_threads)
{
    int ret;

    if ( (ret = mca_pml_yalla_init()) != 0) {
        return NULL;
    }

    ompi_mpi_dynamics_disable("the Yalla (MXM) PML does not support MPI dynamic process functionality");

    *priority = ompi_pml_yalla.priority;
    return &ompi_pml_yalla.super;
}

static int mca_pml_yalla_component_fini(void)
{
    return mca_pml_yalla_cleanup();
}