mirror of
https://github.com/openmc-dev/openmc.git
synced 2026-07-21 22:45:33 -04:00
607f6babe5
Co-authored-by: Patrick Shriwise <pshriwise@gmail.com> Co-authored-by: Paul Romano <paul.k.romano@gmail.com>
121 lines
3.4 KiB
C++
121 lines
3.4 KiB
C++
#include "openmc/capi.h"
|
|
#include "openmc/cross_sections.h"
|
|
#include "openmc/dagmc.h"
|
|
#include "openmc/error.h"
|
|
#include "openmc/geometry.h"
|
|
#include "openmc/geometry_aux.h"
|
|
#include "openmc/material.h"
|
|
#include "openmc/message_passing.h"
|
|
#include "openmc/nuclide.h"
|
|
#include <iostream>
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
using namespace openmc;
|
|
int openmc_err;
|
|
|
|
// Initialise OpenMC
|
|
#ifdef OPENMC_MPI
|
|
MPI_Comm world = MPI_COMM_WORLD;
|
|
openmc_err = openmc_init(argc, argv, &world);
|
|
#else
|
|
openmc_err = openmc_init(argc, argv, nullptr);
|
|
#endif
|
|
if (openmc_err == -1) {
|
|
// This happens for the -h and -v flags
|
|
return EXIT_SUCCESS;
|
|
} else if (openmc_err) {
|
|
fatal_error(openmc_err_msg);
|
|
}
|
|
|
|
// Create DAGMC ptr
|
|
std::string filename = "dagmc.h5m";
|
|
std::shared_ptr<moab::DagMC> dag_ptr = std::make_shared<moab::DagMC>();
|
|
moab::ErrorCode rval = dag_ptr->load_file(filename.c_str());
|
|
if (rval != moab::MB_SUCCESS) {
|
|
fatal_error("Failed to load file");
|
|
}
|
|
|
|
// Initialize acceleration data structures
|
|
rval = dag_ptr->init_OBBTree();
|
|
if (rval != moab::MB_SUCCESS) {
|
|
fatal_error("Failed to initialize OBB tree");
|
|
}
|
|
|
|
// Get rid of existing geometry
|
|
std::unordered_map<std::string, int> nuclide_map_copy =
|
|
openmc::data::nuclide_map;
|
|
openmc::data::nuclides.clear();
|
|
openmc::data::nuclide_map = nuclide_map_copy;
|
|
openmc::model::surfaces.clear();
|
|
openmc::model::surface_map.clear();
|
|
openmc::model::cells.clear();
|
|
openmc::model::cell_map.clear();
|
|
openmc::model::universes.clear();
|
|
openmc::model::universe_map.clear();
|
|
|
|
// Update materials (emulate an external program)
|
|
for (auto& mat_ptr : openmc::model::materials) {
|
|
mat_ptr->set_temperature(300);
|
|
}
|
|
|
|
// Create new DAGMC universe
|
|
openmc::model::universes.push_back(
|
|
std::make_unique<openmc::DAGUniverse>(dag_ptr));
|
|
model::universe_map[model::universes.back()->id_] =
|
|
model::universes.size() - 1;
|
|
|
|
// Add cells to universes
|
|
openmc::populate_universes();
|
|
|
|
// Make sure implicit complement appears last
|
|
auto dag_univ = dynamic_cast<DAGUniverse*>(model::universes.back().get());
|
|
int n = dag_univ->cells_.size();
|
|
for (int i = 0; i < n - 1; ++i) {
|
|
if (dag_univ->cells_[i] == dag_univ->implicit_complement_idx()) {
|
|
fatal_error("Implicit complement cell should appear last in vector of "
|
|
"cells for DAGMC universe.");
|
|
}
|
|
}
|
|
if (dag_univ->cells_.back() != dag_univ->implicit_complement_idx()) {
|
|
fatal_error(
|
|
"Last cell in DAGMC universe is not an implicit complement cell.");
|
|
}
|
|
|
|
// Set root universe
|
|
openmc::model::root_universe = openmc::find_root_universe();
|
|
openmc::check_dagmc_root_univ();
|
|
|
|
// Final geometry setup and assign temperatures
|
|
openmc::finalize_geometry();
|
|
|
|
// Finalize cross sections having assigned temperatures
|
|
openmc::finalize_cross_sections();
|
|
|
|
// Check that we correctly assigned cell temperatures with non-void fill
|
|
for (auto& cell_ptr : openmc::model::cells) {
|
|
if (cell_ptr->material_.front() != openmc::C_NONE &&
|
|
cell_ptr->temperature() != 300) {
|
|
fatal_error("Failed to set cell temperature");
|
|
}
|
|
}
|
|
|
|
// Finalize cell densities
|
|
openmc::finalize_cell_densities();
|
|
|
|
// Run OpenMC
|
|
openmc_err = openmc_run();
|
|
if (openmc_err)
|
|
fatal_error(openmc_err_msg);
|
|
|
|
// Deallocate memory
|
|
openmc_err = openmc_finalize();
|
|
if (openmc_err)
|
|
fatal_error(openmc_err_msg);
|
|
|
|
#ifdef OPENMC_MPI
|
|
MPI_Finalize();
|
|
#endif
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|