331 строка
12 KiB
C
331 строка
12 KiB
C
/*
|
|
* Copyright (c) 2004-2008 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) 2006-2007 Cisco Systems, Inc. All rights reserved.
|
|
*
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "opal_config.h"
|
|
|
|
/* This component will only be compiled on File, where we are
|
|
guaranteed to have <unistd.h> and friends */
|
|
#include <stdio.h>
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#endif /* HAVE_UNISTD_H */
|
|
#ifdef HAVE_STDLIB_H
|
|
#include <stdlib.h>
|
|
#endif /* HAVE_STDLIB_H */
|
|
#ifdef HAVE_STRING_H
|
|
#include <string.h>
|
|
#endif /* HAVE_STRING_H */
|
|
|
|
#include "opal/constants.h"
|
|
#include "opal/mca/base/mca_base_param.h"
|
|
#include "opal/util/argv.h"
|
|
#include "opal/util/show_help.h"
|
|
#include "opal/mca/carto/carto.h"
|
|
#include "opal/mca/carto/base/base.h"
|
|
#include "opal/mca/carto/base/carto_base_graph.h"
|
|
#include "carto_file.h"
|
|
#include "carto_file_lex.h"
|
|
|
|
|
|
|
|
char *carto_file_path;
|
|
|
|
static int opal_carto_file_init(void);
|
|
static int opal_carto_file_finalize(void);
|
|
static int opal_carto_file_parse(const char *cartofile);
|
|
|
|
|
|
/*
|
|
* File carto module
|
|
*/
|
|
static const opal_carto_base_module_1_0_0_t loc_module = {
|
|
opal_carto_file_init,
|
|
opal_carto_base_graph_get_host_graph_fn,
|
|
opal_carto_base_free_graph_fn,
|
|
opal_carto_base_get_nodes_distance_fn,
|
|
opal_carto_base_graph_spf_fn,
|
|
opal_carto_base_graph_find_node_fn,
|
|
opal_carto_file_finalize,
|
|
};
|
|
|
|
|
|
int opal_carto_file_component_query(mca_base_module_t **module, int *priority)
|
|
{
|
|
int param;
|
|
|
|
param = mca_base_param_find("carto", "file", "priority");
|
|
mca_base_param_lookup_int(param, priority);
|
|
|
|
param = mca_base_param_find("carto", "file", "path");
|
|
mca_base_param_lookup_string(param, &carto_file_path);
|
|
|
|
*module = (mca_base_module_t *)&loc_module;
|
|
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
|
|
/**
|
|
* Init the carto file module.
|
|
*
|
|
* @return int success or error
|
|
*/
|
|
static int opal_carto_file_init(void)
|
|
{
|
|
int rc;
|
|
|
|
/* create an empty graph */
|
|
if (NULL == opal_carto_base_common_host_graph) {
|
|
opal_carto_base_graph_create_fn(&opal_carto_base_common_host_graph);
|
|
}
|
|
if (NULL == carto_file_path) {
|
|
return OPAL_ERROR;
|
|
}
|
|
/* parse the carto file and add nodes and connections to the graph */
|
|
rc = opal_carto_file_parse(carto_file_path);
|
|
return rc;
|
|
}
|
|
|
|
/**
|
|
* finalize the carto file module.
|
|
*
|
|
* @return int success or error
|
|
*/
|
|
static int opal_carto_file_finalize(void)
|
|
{
|
|
/* free the host cartography graph. */
|
|
if (NULL != opal_carto_base_common_host_graph) {
|
|
opal_carto_base_free_graph_fn(opal_carto_base_common_host_graph);
|
|
}
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
|
|
/**
|
|
* Parse the carto file and file the host graph with nodes and
|
|
* connections.
|
|
*
|
|
* @param cartofile the path to the carto file.
|
|
*
|
|
* @return int success or error.
|
|
*/
|
|
static int opal_carto_file_parse(const char *cartofile)
|
|
{
|
|
int token;
|
|
opal_carto_base_node_t *node, *end_node;
|
|
uint32_t weight;
|
|
char *node1_name, *node2_name, *value, **argv;
|
|
int cnt, line_number = 1;
|
|
char *token_to_string[] = {
|
|
"New-line",
|
|
"File-error",
|
|
"Node deceleration",
|
|
"Connection deceleration",
|
|
"bi directional connection",
|
|
"Integer",
|
|
"Name",
|
|
"Node connection",
|
|
"Undefined"
|
|
"Undefined"
|
|
"Undefined"
|
|
};
|
|
|
|
/* set the done flag to false. at the end of file the the lexical analyzer will set it to true */
|
|
carto_file_done = false;
|
|
carto_file_in = fopen(cartofile, "r"); /* open the carto file */
|
|
/* if the file not found, return an error */
|
|
if (NULL == carto_file_in) {
|
|
opal_show_help("help-opal-carto-file.txt", "file not found", true, cartofile);
|
|
return OPAL_ERR_NOT_FOUND;
|
|
}
|
|
while (!carto_file_done) {
|
|
token = carto_file_lex();
|
|
switch (token) {
|
|
case OPAL_CARTO_FILE_NEWLINE:
|
|
line_number++;
|
|
break;
|
|
case OPAL_CARTO_FILE_NODE_DECELERATION:
|
|
token = carto_file_lex();
|
|
switch (token) {
|
|
case OPAL_CARTO_FILE_NAME:
|
|
node = (opal_carto_base_node_t *)malloc(sizeof(opal_carto_base_node_t));
|
|
node->node_type = strdup(carto_file_value.sval);
|
|
if (0 == strcmp("socket",node->node_type)) {
|
|
node->is_cpu = true;
|
|
}
|
|
else {
|
|
node->is_cpu = false;
|
|
}
|
|
token = carto_file_lex();
|
|
switch (token) {
|
|
case OPAL_CARTO_FILE_NAME:
|
|
node->node_name = strdup(carto_file_value.sval);
|
|
opal_carto_base_graph_add_node_fn(opal_carto_base_common_host_graph, node);
|
|
break;
|
|
default:
|
|
free(node);
|
|
opal_show_help("help-opal-carto-file.txt", "expected node name",
|
|
true, cartofile, line_number, token_to_string[token]);
|
|
goto error;
|
|
}
|
|
break;
|
|
default:
|
|
opal_show_help("help-opal-carto-file.txt", "expected node type",
|
|
true, cartofile, line_number, token_to_string[token]);
|
|
goto error;
|
|
}
|
|
break;
|
|
case OPAL_CARTO_FILE_CONNECTION_DECELERATION:
|
|
token = carto_file_lex();
|
|
switch (token) {
|
|
case OPAL_CARTO_FILE_NAME:
|
|
node1_name = strdup(carto_file_value.sval);
|
|
while (OPAL_CARTO_FILE_NEWLINE != token) {
|
|
token = carto_file_lex();
|
|
switch (token) {
|
|
case OPAL_CARTO_FILE_NODE_CONNECTION:
|
|
value = carto_file_value.sval;
|
|
argv = opal_argv_split (value, ':');
|
|
cnt = opal_argv_count (argv);
|
|
if (2 == cnt) {
|
|
node2_name = strdup(argv[0]);
|
|
weight = atoi(argv[1]);
|
|
} else {
|
|
opal_show_help("help-opal-carto-file.txt", "incorrect connection", true, cartofile, line_number, value);
|
|
opal_argv_free (argv);
|
|
free(node1_name);
|
|
goto error;
|
|
}
|
|
opal_argv_free (argv);
|
|
/* find the start node of the connection */
|
|
node = opal_carto_base_graph_find_node_fn(opal_carto_base_common_host_graph,node1_name);
|
|
if (NULL == node) {
|
|
opal_show_help("help-opal-carto-file.txt", "vertex not found", true, cartofile, line_number, node1_name);
|
|
free(node1_name);
|
|
free(node2_name);
|
|
goto error;
|
|
}
|
|
/* find the end node of the connection */
|
|
end_node = opal_carto_base_graph_find_node_fn(opal_carto_base_common_host_graph,node2_name);
|
|
if (NULL == end_node) {
|
|
opal_show_help("help-opal-carto-file.txt", "vertex not found", true, cartofile, line_number, node2_name);
|
|
free(node1_name);
|
|
free(node2_name);
|
|
goto error;
|
|
}
|
|
opal_carto_base_connect_nodes_fn(opal_carto_base_common_host_graph, node, end_node, weight);
|
|
free(node2_name);
|
|
break;
|
|
case OPAL_CARTO_FILE_NEWLINE:
|
|
line_number++;
|
|
break;
|
|
default:
|
|
opal_show_help("help-opal-carto-file.txt", "expected Connection",
|
|
true, cartofile, line_number, token_to_string[token]);
|
|
free(node1_name);
|
|
goto error;
|
|
}
|
|
}
|
|
free(node1_name);
|
|
break;
|
|
default:
|
|
opal_show_help("help-opal-carto-file.txt", "expected node name",
|
|
true, cartofile, line_number, token_to_string[token]);
|
|
goto error;
|
|
}
|
|
break;
|
|
case OPAL_CARTO_FILE_BIDIRECTION_CONNECTION:
|
|
token = carto_file_lex();
|
|
switch (token) {
|
|
case OPAL_CARTO_FILE_NAME:
|
|
node1_name = strdup(carto_file_value.sval);
|
|
while (OPAL_CARTO_FILE_NEWLINE != token) {
|
|
token = carto_file_lex();
|
|
switch (token) {
|
|
case OPAL_CARTO_FILE_NODE_CONNECTION:
|
|
value = carto_file_value.sval;
|
|
argv = opal_argv_split (value, ':');
|
|
cnt = opal_argv_count (argv);
|
|
if (2 == cnt) {
|
|
node2_name = strdup(argv[0]);
|
|
weight = atoi(argv[1]);
|
|
} else {
|
|
opal_show_help("help-opal-carto-file.txt", "incorrect connection", true, cartofile, line_number, value);
|
|
opal_argv_free (argv);
|
|
free(node1_name);
|
|
goto error;
|
|
}
|
|
opal_argv_free (argv);
|
|
/* find the start node of the connection */
|
|
node = opal_carto_base_graph_find_node_fn(opal_carto_base_common_host_graph,node1_name);
|
|
if (NULL == node) {
|
|
opal_show_help("help-opal-carto-file.txt", "vertex not found", true, cartofile, line_number, node1_name);
|
|
free(node1_name);
|
|
free(node2_name);
|
|
goto error;
|
|
}
|
|
/* find the end node of the connection */
|
|
end_node = opal_carto_base_graph_find_node_fn(opal_carto_base_common_host_graph,node2_name);
|
|
if (NULL == end_node) {
|
|
opal_show_help("help-opal-carto-file.txt", "vertex not found", true, cartofile, line_number, node2_name);
|
|
free(node1_name);
|
|
free(node2_name);
|
|
goto error;
|
|
}
|
|
opal_carto_base_connect_nodes_fn(opal_carto_base_common_host_graph, node, end_node, weight);
|
|
opal_carto_base_connect_nodes_fn(opal_carto_base_common_host_graph, end_node, node, weight);
|
|
free(node2_name);
|
|
break;
|
|
case OPAL_CARTO_FILE_NEWLINE:
|
|
line_number++;
|
|
break;
|
|
default:
|
|
opal_show_help("help-opal-carto-file.txt", "expected Connection",
|
|
true, cartofile, line_number, token_to_string[token]);
|
|
free(node1_name);
|
|
goto error;
|
|
}
|
|
}
|
|
free(node1_name);
|
|
break;
|
|
default:
|
|
opal_show_help("help-opal-carto-file.txt", "expected node name",
|
|
true, cartofile, line_number, token_to_string[token]);
|
|
goto error;
|
|
}
|
|
break;
|
|
default:
|
|
opal_show_help("help-opal-carto-file.txt", "expected deceleration",
|
|
true, cartofile, line_number, token_to_string[token]);
|
|
goto error;
|
|
|
|
}
|
|
}
|
|
return OPAL_SUCCESS;
|
|
error:
|
|
fclose(carto_file_in);
|
|
return OPAL_ERR_BAD_PARAM;
|
|
}
|
|
|
|
|
|
|
|
|