#include "openmc/xml_interface.h"

#include <fmt/core.h>

#include "openmc/error.h"

#include "openmc/string_utils.h"

#include "openmc/vector.h"

namespace openmc {

std::string get_node_value(

pugi::xml_node node, const char* name, bool lowercase, bool strip)

{

// Search for either an attribute or child tag and get the data as a char*.

const pugi::char_t* value_char;

if (node.attribute(name)) {

value_char = node.attribute(name).value();

} else if (node.child(name)) {

value_char = node.child_value(name);

} else {

fatal_error(fmt::format(

"Node \"{}\" is not a member of the \"{}\" XML node", name, node.name()));

}

std::string value {value_char};

// Convert to lower-case if needed

if (lowercase)

to_lower(value);

// Strip leading/trailing whitespace if needed

if (strip) {

value.erase(0, value.find_first_not_of(" \t\r\n"));

value.erase(value.find_last_not_of(" \t\r\n") + 1);

}

return value;

}

bool get_node_value_bool(pugi::xml_node node, const char* name)

{

if (node.attribute(name)) {

return node.attribute(name).as_bool();

} else if (node.child(name)) {

return node.child(name).text().as_bool();

} else {

fatal_error(fmt::format(

"Node \"{}\" is not a member of the \"{}\" XML node", name, node.name()));

}

return false;

}

vector<Position> get_node_position_array(

pugi::xml_node node, const char* name, bool lowercase)

{

vector<double> coords = get_node_array<double>(node, name, lowercase);

if (coords.size() % 3 != 0) {

fatal_error(fmt::format(

"Incorect number of coordinates in Position array ({}) for \"{}\"",

coords.size(), name));

}

vector<Position> positions;

positions.reserve(coords.size() / 3);

for (size_t i = 0; i < coords.size(); i += 3) {

positions.push_back({coords[i], coords[i + 1], coords[i + 2]});

}

return positions;

}

Position get_node_position(

pugi::xml_node node, const char* name, bool lowercase)

{

vector<double> arr = get_node_array<double>(node, name, lowercase);

return Position(arr);

}

} // namespace openmc