static int _change_state_bps(char *com, int state)
{
char *host;
int i = 0;
uint16_t pos[params.cluster_dims];
char letter = '.';
bool used = false;
char *c_state = "up";
hostlist_t hl = NULL;
int rc = 1;
if (state == NODE_STATE_DOWN) {
letter = '#';
used = true;
c_state = "down";
}
while (com[i] && (com[i] != '[') &&
((com[i] < '0') || (com[i] > '9')) &&
((com[i] < 'A') || (com[i] > 'Z')))
i++;
if (com[i] == '\0') {
memset(error_string, 0, 255);
sprintf(error_string,
"You didn't specify any nodes to make %s. "
"in statement '%s'",
c_state, com);
return 0;
}
if (!(hl = hostlist_create(com+i))) {
memset(error_string, 0, 255);
sprintf(error_string, "Bad hostlist given '%s'", com+i);
return 0;
}
while ((host = hostlist_shift(hl))) {
ba_mp_t *ba_mp;
smap_node_t *smap_node;
for (i = 0; i < params.cluster_dims; i++)
pos[i] = select_char2coord(host[i]);
if (!(ba_mp = bg_configure_coord2ba_mp(pos))) {
memset(error_string, 0, 255);
sprintf(error_string, "Bad host given '%s'", host);
rc = 0;
break;
}
bg_configure_ba_update_mp_state(ba_mp, state);
smap_node = smap_system_ptr->grid[ba_mp->index];
smap_node->color = 0;
smap_node->letter = letter;
smap_node->used = used;
free(host);
}
hostlist_destroy(hl);
return rc;
}
static void _internal_setup_grid(int level, uint16_t *coords)
{
ba_mp_t *ba_mp;
smap_node_t *smap_node;
if (level > params.cluster_dims)
return;
if (level < params.cluster_dims) {
for (coords[level] = 0;
coords[level] < dim_size[level];
coords[level]++) {
/* handle the outer dims here */
_internal_setup_grid(level+1, coords);
}
return;
}
ba_mp = bg_configure_coord2ba_mp(coords);
if (!ba_mp || ba_mp->index > smap_system_ptr->node_cnt)
return;
smap_node = xmalloc(sizeof(smap_node_t));
smap_node->coord = xmalloc(sizeof(uint16_t) * params.cluster_dims);
memcpy(smap_node->coord, coords,
sizeof(uint16_t) * params.cluster_dims);
smap_node->index = ba_mp->index;
smap_system_ptr->grid[smap_node->index] = smap_node;
}
static int _add_bg_record(select_ba_request_t *blockreq, List allocated_blocks)
{
int rc = 1;
#ifdef HAVE_BG
char *nodes = NULL, *host;
int diff = 0;
int largest_diff = -1;
uint16_t start[params.cluster_dims];
uint16_t end[params.cluster_dims];
uint16_t best_start[params.cluster_dims];
int i, j = 0;
hostlist_t hl = NULL;
bitstr_t *mark_bitmap = NULL;
char tmp_char[params.cluster_dims+1],
tmp_char2[params.cluster_dims+1];
memset(tmp_char, 0, sizeof(tmp_char));
memset(tmp_char2, 0, sizeof(tmp_char2));
for (i = 0; i < params.cluster_dims; i++) {
best_start[0] = 0;
blockreq->geometry[i] = 0;
end[i] = (int16_t)-1;
}
nodes = blockreq->save_name;
if (!nodes)
return SLURM_SUCCESS;
while (nodes[j] && (nodes[j] != '[') &&
((nodes[j] < '0') || (nodes[j] > '9')) &&
((nodes[j] < 'A') || (nodes[j] > 'Z')))
j++;
if (nodes[j] == '\0') {
snprintf(error_string, sizeof(error_string),
"This block '%s' for some reason didn't contain "
"any midplanes.",
nodes);
rc = 0;
goto fini;
}
if (!(hl = hostlist_create(nodes+j))) {
snprintf(error_string, sizeof(error_string),
"Bad hostlist given '%s'", nodes+j);
rc = 0;
goto fini;
}
/* figure out the geo and the size */
mark_bitmap = bit_alloc(smap_system_ptr->node_cnt);
while ((host = hostlist_shift(hl))) {
ba_mp_t *ba_mp;
uint16_t pos[params.cluster_dims];
for (i = 0; i < params.cluster_dims; i++)
pos[i] = select_char2coord(host[i]);
free(host);
if (!(ba_mp = bg_configure_coord2ba_mp(pos))) {
memset(error_string, 0, 255);
sprintf(error_string, "Bad host given '%s'", host);
rc = 0;
break;
}
bit_set(mark_bitmap, ba_mp->index);
for (i = 0; i < params.cluster_dims; i++) {
if (ba_mp->coord[i] > (int16_t)end[i]) {
blockreq->geometry[i]++;
end[i] = ba_mp->coord[i];
}
}
}
hostlist_destroy(hl);
if (!rc)
goto fini;
/* figure out the start pos */
while (nodes[j] != '\0') {
int mid = j + params.cluster_dims + 1;
int fin = mid + params.cluster_dims + 1;
if (((nodes[j] == '[') || (nodes[j] == ',')) &&
((nodes[mid] == 'x') || (nodes[mid] == '-')) &&
((nodes[fin] == ']') || (nodes[fin] == ','))) {
j++; /* Skip leading '[' or ',' */
for (i = 0; i < params.cluster_dims; i++, j++)
start[i] = select_char2coord(nodes[j]);
j++; /* Skip middle 'x' or '-' */
for (i = 0; i < params.cluster_dims; i++, j++)
end[i] = select_char2coord(nodes[j]);
diff = end[0] - start[0];
} else if (((nodes[j] >= '0') && (nodes[j] <= '9')) ||
((nodes[j] >= 'A') && (nodes[j] <= 'Z'))) {
for (i = 0; i < params.cluster_dims; i++, j++)
start[i] = select_char2coord(nodes[j]);
diff = 0;
} else {
j++;
continue;
}
if (diff > largest_diff) {
largest_diff = diff;
memcpy(best_start, start, sizeof(best_start));
}
if (nodes[j] != ',')
break;
}
if (largest_diff == -1) {
snprintf(error_string, sizeof(error_string),
"No hostnames given here");
goto fini;
}
memcpy(blockreq->start, best_start, sizeof(blockreq->start));
if(!_full_request(blockreq, mark_bitmap, allocated_blocks))
destroy_select_ba_request(blockreq);
fini:
FREE_NULL_BITMAP(mark_bitmap);
#endif
return rc;
}
