114 строки
4.0 KiB
C
114 строки
4.0 KiB
C
/*
|
|
* Copyright (c) 2014 Intel, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "orte_config.h"
|
|
#include "orte/constants.h"
|
|
|
|
#include <string.h>
|
|
|
|
#include "orte/mca/mca.h"
|
|
#include "opal/mca/base/base.h"
|
|
|
|
#include "orte/mca/rtc/base/base.h"
|
|
|
|
static bool selected = false;
|
|
|
|
/*
|
|
* Function for selecting one component from all those that are
|
|
* available.
|
|
*/
|
|
int orte_rtc_base_select(void)
|
|
{
|
|
mca_base_component_list_item_t *cli = NULL;
|
|
mca_base_component_t *component = NULL;
|
|
mca_base_module_t *module = NULL;
|
|
orte_rtc_base_module_t *nmodule;
|
|
orte_rtc_base_selected_module_t *newmodule, *mod;
|
|
int rc, priority;
|
|
bool inserted;
|
|
|
|
if (selected) {
|
|
/* ensure we don't do this twice */
|
|
return ORTE_SUCCESS;
|
|
}
|
|
selected = true;
|
|
|
|
/* Query all available components and ask if they have a module */
|
|
OPAL_LIST_FOREACH(cli, &orte_rtc_base_framework.framework_components, mca_base_component_list_item_t) {
|
|
component = (mca_base_component_t *) cli->cli_component;
|
|
|
|
opal_output_verbose(5, orte_rtc_base_framework.framework_output,
|
|
"mca:rtc:select: checking available component %s", component->mca_component_name);
|
|
|
|
/* If there's no query function, skip it */
|
|
if (NULL == component->mca_query_component) {
|
|
opal_output_verbose(5, orte_rtc_base_framework.framework_output,
|
|
"mca:rtc:select: Skipping component [%s]. It does not implement a query function",
|
|
component->mca_component_name );
|
|
continue;
|
|
}
|
|
|
|
/* Query the component */
|
|
opal_output_verbose(5, orte_rtc_base_framework.framework_output,
|
|
"mca:rtc:select: Querying component [%s]",
|
|
component->mca_component_name);
|
|
rc = component->mca_query_component(&module, &priority);
|
|
|
|
/* If no module was returned, then skip component */
|
|
if (ORTE_SUCCESS != rc || NULL == module) {
|
|
opal_output_verbose(5, orte_rtc_base_framework.framework_output,
|
|
"mca:rtc:select: Skipping component [%s]. Query failed to return a module",
|
|
component->mca_component_name );
|
|
continue;
|
|
}
|
|
nmodule = (orte_rtc_base_module_t*) module;
|
|
|
|
/* give the module a chance to init */
|
|
if (NULL != nmodule->init) {
|
|
if (ORTE_SUCCESS != (rc = nmodule->init())) {
|
|
opal_output_verbose(5, orte_rtc_base_framework.framework_output,
|
|
"mca:rtc:select: Skipping component [%s]. Failed to init",
|
|
component->mca_component_name );
|
|
continue;
|
|
}
|
|
}
|
|
|
|
/* add to the list of selected modules */
|
|
newmodule = OBJ_NEW(orte_rtc_base_selected_module_t);
|
|
newmodule->pri = priority;
|
|
newmodule->module = nmodule;
|
|
newmodule->component = component;
|
|
|
|
/* maintain priority order */
|
|
inserted = false;
|
|
OPAL_LIST_FOREACH(mod, &orte_rtc_base.actives, orte_rtc_base_selected_module_t) {
|
|
if (priority > mod->pri) {
|
|
opal_list_insert_pos(&orte_rtc_base.actives,
|
|
(opal_list_item_t*)mod, &newmodule->super);
|
|
inserted = true;
|
|
break;
|
|
}
|
|
}
|
|
if (!inserted) {
|
|
/* must be lowest priority - add to end */
|
|
opal_list_append(&orte_rtc_base.actives, &newmodule->super);
|
|
}
|
|
}
|
|
|
|
if (4 < opal_output_get_verbosity(orte_rtc_base_framework.framework_output)) {
|
|
opal_output(0, "%s: Final RTC priorities", ORTE_NAME_PRINT(ORTE_PROC_MY_NAME));
|
|
/* show the prioritized list */
|
|
OPAL_LIST_FOREACH(mod, &orte_rtc_base.actives, orte_rtc_base_selected_module_t) {
|
|
opal_output(0, "\tModule: %s Priority: %d", mod->component->mca_component_name, mod->pri);
|
|
}
|
|
}
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|