mirror of
https://github.com/OPM/opm-simulators.git
synced 2024-12-25 08:41:00 -06:00
Reformatting with clang-format.
This commit is contained in:
parent
76bacff3cf
commit
430d8995eb
@ -22,8 +22,8 @@
|
||||
#define XSD_CXX11_TEMPLATE_ALIAS 1
|
||||
|
||||
#include <damaris/model/ModifyModel.hpp>
|
||||
#include <opm/simulators/utils/DamarisOutputModule.hpp>
|
||||
#include <opm/simulators/utils/DamarisKeywords.hpp>
|
||||
#include <opm/simulators/utils/DamarisOutputModule.hpp>
|
||||
#include <opm/simulators/utils/ParallelCommunication.hpp>
|
||||
|
||||
namespace Opm::DamarisOutput
|
||||
@ -32,99 +32,106 @@ namespace Opm::DamarisOutput
|
||||
std::string initDamarisXmlFile(); // Defined in initDamarisXMLFile.cpp, to avoid messing up this file.
|
||||
|
||||
|
||||
void initializeDamaris(MPI_Comm comm, int mpiRank, std::string OutputDir, bool enableAsyncDamarisOutput)
|
||||
void
|
||||
initializeDamaris(MPI_Comm comm, int mpiRank, std::string outputDir, bool enableAsyncDamarisOutput)
|
||||
{
|
||||
// Prepare the XML file
|
||||
std::string damaris_config_xml = initDamarisXmlFile();
|
||||
damaris::model::ModifyModel myMod = damaris::model::ModifyModel(damaris_config_xml);
|
||||
// The map will make it precise the output directory and FileMode (either FilePerCore or Collective storage)
|
||||
// The map file find all occurences of the string in position 1 and repalce it/them with string in position 2
|
||||
std::map<std::string,std::string> find_replace_map = DamarisKeywords(OutputDir, enableAsyncDamarisOutput);
|
||||
std::map<std::string, std::string> find_replace_map = DamarisKeywords(outputDir, enableAsyncDamarisOutput);
|
||||
myMod.RepalceWithRegEx(find_replace_map);
|
||||
std::string damaris_xml_filename_str = OutputDir +"/damaris_config.xml" ;
|
||||
std::string damaris_xml_filename_str = outputDir + "/damaris_config.xml";
|
||||
|
||||
if (mpiRank == 0) {
|
||||
myMod.SaveXMLStringToFile(damaris_xml_filename_str);
|
||||
myMod.SaveXMLStringToFile(damaris_xml_filename_str);
|
||||
}
|
||||
|
||||
int damaris_err;
|
||||
|
||||
/* Get the name of the Damaris input file from an environment variable if available */
|
||||
const char *cs_damaris_xml_file = getenv("FLOW_DAMARIS_XML_FILE");
|
||||
if (cs_damaris_xml_file != NULL)
|
||||
{
|
||||
std::cout << "INFO: initializing Damaris from environment variable FLOW_DAMARIS_XML_FILE: " << cs_damaris_xml_file << std::endl;
|
||||
damaris_err = damaris_initialize(cs_damaris_xml_file, MPI_COMM_WORLD);
|
||||
if (damaris_err != DAMARIS_OK ) {
|
||||
std::cerr << "ERROR: damaris_initialize() error via FLOW_DAMARIS_XML_FILE=" << cs_damaris_xml_file << std::endl;
|
||||
}
|
||||
const char* cs_damaris_xml_file = getenv("FLOW_DAMARIS_XML_FILE");
|
||||
if (cs_damaris_xml_file != NULL) {
|
||||
std::cout << "INFO: initializing Damaris from environment variable FLOW_DAMARIS_XML_FILE: "
|
||||
<< cs_damaris_xml_file << std::endl;
|
||||
damaris_err = damaris_initialize(cs_damaris_xml_file, MPI_COMM_WORLD);
|
||||
if (damaris_err != DAMARIS_OK) {
|
||||
std::cerr << "ERROR: damaris_initialize() error via FLOW_DAMARIS_XML_FILE=" << cs_damaris_xml_file
|
||||
<< std::endl;
|
||||
}
|
||||
} else {
|
||||
std::cout << "INFO: initializing Damaris using internally built file:" << damaris_xml_filename_str << std::endl;
|
||||
damaris_err = damaris_initialize(damaris_xml_filename_str.c_str(), comm);
|
||||
if (damaris_err != DAMARIS_OK) {
|
||||
std::cerr << "ERROR: damaris_initialize() error via built file:" << std::endl << myMod.GetConfigString();
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
std::cout << "INFO: initializing Damaris using internally built file:" << damaris_xml_filename_str << std::endl;
|
||||
damaris_err = damaris_initialize(damaris_xml_filename_str.c_str() , comm);
|
||||
if (damaris_err != DAMARIS_OK ) {
|
||||
std::cerr << "ERROR: damaris_initialize() error via built file:" << std::endl << myMod.GetConfigString();
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
void setupDamarisWritingPars(Parallel::Communication comm, const int n_elements_local_grid)
|
||||
void
|
||||
setupDamarisWritingPars(Parallel::Communication comm, const int n_elements_local_grid)
|
||||
{
|
||||
int damaris_err = DAMARIS_OK;
|
||||
|
||||
const int nranks = comm.size();
|
||||
const int rank = comm.rank();
|
||||
const int rank = comm.rank();
|
||||
|
||||
std::vector<unsigned long long> elements_rank_sizes(nranks); // one for each rank -- to be gathered from each client rank
|
||||
std::vector<unsigned long long> elements_rank_offsets(nranks); // one for each rank, first one 0 -- to be computed - Probably could use MPI_Scan()!
|
||||
std::vector<unsigned long long> elements_rank_sizes(nranks); // one for each rank -- to be gathered from each client rank
|
||||
std::vector<unsigned long long> elements_rank_offsets(nranks); // one for each rank, first one 0 -- to be computed - Probably could use MPI_Scan()!
|
||||
|
||||
// n_elements_local_grid should be the full model size
|
||||
const unsigned long long n_elements_local = n_elements_local_grid ;
|
||||
const unsigned long long n_elements_local = n_elements_local_grid;
|
||||
|
||||
std::cout << "INFO (" << rank << "): n_elements_local_grid = " << n_elements_local_grid << std::endl ;
|
||||
std::cout << "INFO (" << rank << "): n_elements_local_grid = " << n_elements_local_grid << std::endl;
|
||||
|
||||
// This gets the n_elements_local from all ranks and copies them to a std::vector of all the values on all ranks (elements_rank_sizes[]).
|
||||
// This gets the n_elements_local from all ranks and copies them to a std::vector of all the values on all ranks
|
||||
// (elements_rank_sizes[]).
|
||||
comm.allgather(&n_elements_local, 1, elements_rank_sizes.data());
|
||||
elements_rank_offsets[0] = 0ULL ;
|
||||
// This scan makes the offsets to the start of each ranks grid section if each local grid data was concatenated (in rank order)
|
||||
for (int t1 = 1 ; t1 < nranks; t1++) {
|
||||
elements_rank_offsets[t1] = elements_rank_offsets[t1-1] + elements_rank_sizes[t1-1];
|
||||
elements_rank_offsets[0] = 0ULL;
|
||||
// This scan makes the offsets to the start of each ranks grid section if each local grid data was concatenated (in
|
||||
// rank order)
|
||||
for (int t1 = 1; t1 < nranks; t1++) {
|
||||
elements_rank_offsets[t1] = elements_rank_offsets[t1 - 1] + elements_rank_sizes[t1 - 1];
|
||||
}
|
||||
|
||||
// find the global/total size
|
||||
unsigned long long n_elements_global_max = elements_rank_offsets[nranks-1] ;
|
||||
n_elements_global_max += elements_rank_sizes[nranks-1] ; // add the last ranks size to the already accumulated offset values
|
||||
unsigned long long n_elements_global_max = elements_rank_offsets[nranks - 1];
|
||||
n_elements_global_max += elements_rank_sizes[nranks - 1]; // add the last ranks size to the already accumulated offset values
|
||||
|
||||
if (rank == 0 ) {
|
||||
std::cout << "INFO (" << rank << "): n_elements_global_max = " << n_elements_global_max << std::endl ;
|
||||
if (rank == 0) {
|
||||
std::cout << "INFO (" << rank << "): n_elements_global_max = " << n_elements_global_max << std::endl;
|
||||
}
|
||||
|
||||
|
||||
// Set the paramater so that the Damaris servers can allocate the correct amount of memory for the variabe
|
||||
// Damaris parameters only support int data types. This will limit models to be under size of 2^32-1 elements
|
||||
// ToDo: Do we need to check that local ranks are 0 based ?
|
||||
int temp_int = static_cast<int>(elements_rank_sizes[rank]) ;
|
||||
damaris_err = damaris_parameter_set("n_elements_local",&temp_int, sizeof(int));
|
||||
if (damaris_err != DAMARIS_OK ) {
|
||||
std::cerr << "ERROR: Damaris library produced an error result for damaris_parameter_set(n_elements_local,&temp_int, sizeof(int));" << std::endl ;
|
||||
int temp_int = static_cast<int>(elements_rank_sizes[rank]);
|
||||
damaris_err = damaris_parameter_set("n_elements_local", &temp_int, sizeof(int));
|
||||
if (damaris_err != DAMARIS_OK) {
|
||||
std::cerr << "ERROR: Damaris library produced an error result for "
|
||||
"damaris_parameter_set(\"n_elements_local\", &temp_int, sizeof(int));"
|
||||
<< std::endl;
|
||||
}
|
||||
// Damaris parameters only support int data types. This will limit models to be under size of 2^32-1 elements
|
||||
// ToDo: Do we need to check that n_elements_global_max will fit in a C int type (INT_MAX)
|
||||
temp_int = static_cast<int>(n_elements_global_max) ;
|
||||
damaris_err = damaris_parameter_set("n_elements_total",&temp_int, sizeof(int));
|
||||
if (damaris_err != DAMARIS_OK ) {
|
||||
std::cerr << "ERROR: Damaris library produced an error result for damaris_parameter_set(n_elements_local,&temp_int, sizeof(int));" << std::endl ;
|
||||
temp_int = static_cast<int>(n_elements_global_max);
|
||||
damaris_err = damaris_parameter_set("n_elements_total", &temp_int, sizeof(int));
|
||||
if (damaris_err != DAMARIS_OK) {
|
||||
std::cerr << "ERROR: Damaris library produced an error result for "
|
||||
"damaris_parameter_set(\"n_elements_local\", &temp_int, sizeof(int));"
|
||||
<< std::endl;
|
||||
}
|
||||
|
||||
// Use damaris_set_position to set the offset in the global size of the array.
|
||||
// This is used so that output functionality (e.g. HDF5Store) knows global offsets of the data of the ranks
|
||||
int64_t temp_int64_t[1] ;
|
||||
temp_int64_t[0] = static_cast<int64_t>(elements_rank_offsets[rank]) ;
|
||||
damaris_err = damaris_set_position("PRESSURE",temp_int64_t) ;
|
||||
if (damaris_err != DAMARIS_OK ) {
|
||||
std::cerr << "ERROR: Damaris library produced an error result for damaris_set_position(\"PRESSURE\",temp_int64_t);" << std::endl ;
|
||||
int64_t temp_int64_t[1];
|
||||
temp_int64_t[0] = static_cast<int64_t>(elements_rank_offsets[rank]);
|
||||
damaris_err = damaris_set_position("PRESSURE", temp_int64_t);
|
||||
if (damaris_err != DAMARIS_OK) {
|
||||
std::cerr << "ERROR: Damaris library produced an error result for "
|
||||
"damaris_set_position(\"PRESSURE\", temp_int64_t);"
|
||||
<< std::endl;
|
||||
}
|
||||
}
|
||||
}
|
||||
} // namespace Opm::DamarisOutput
|
||||
|
Loading…
Reference in New Issue
Block a user