crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-26 21:25:36 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Implement SourceFile derived from SourceDistribution

Browse source
This commit is contained in:
Paul Romano 2020-10-23 06:56:14 -05:00
parent 3f62fab565
commit 5a60133d5b
3 changed files with 58 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

82
src/source.cpp
View file

@ -72,35 +72,6 @@ IndependentSourceDistribution::IndependentSourceDistribution(pugi::xml_node node
// Check for external source file
if (check_for_node(node, "file")) {
// Copy path of source file
auto path_source = get_node_value(node, "file", false, true);
// Check if source file exists
if (!file_exists(path_source)) {
fatal_error(fmt::format("Source file '{}' does not exist.", path_source));
}
// Read the source from a binary file instead of sampling from some
// assumed source distribution
write_message(6, "Reading source file from {}...", path_source);
// Open the binary file
hid_t file_id = file_open(path_source, 'r', true);
// Read the file type
std::string filetype;
read_attribute(file_id, "filetype", filetype);
// Check to make sure this is a source file
if (filetype != "source" && filetype != "statepoint") {
fatal_error("Specified starting source file not a source file type.");
}
// Read in the source bank
read_source_bank(file_id, sites_);
// Close file
file_close(file_id);
} else {
@ -182,16 +153,11 @@ Particle::Bank IndependentSourceDistribution::sample(uint64_t* seed)
int n_reject = 0;
static int n_accept = 0;
while (!found) {
if (!sites_.empty()) {
size_t i_site = sites_.size()*prn(seed);
site = sites_[i_site];
} else {
// Set particle type
site.particle = particle_;
// Set particle type
site.particle = particle_;
// Sample spatial distribution
site.r = space_->sample(seed);
}
// Sample spatial distribution
site.r = space_->sample(seed);
// Now search to see if location exists in geometry
int32_t cell_index, instance;
@ -232,8 +198,6 @@ Particle::Bank IndependentSourceDistribution::sample(uint64_t* seed)
// Increment number of accepted samples
++n_accept;
if (!sites_.empty()) return site;
// Sample angle
site.u = angle_->sample(seed);
@ -265,6 +229,44 @@ Particle::Bank IndependentSourceDistribution::sample(uint64_t* seed)
return site;
}
//==============================================================================
// SourceFile implementation
//==============================================================================
SourceFile::SourceFile(std::string path)
{
// Check if source file exists
if (!file_exists(path)) {
fatal_error(fmt::format("Source file '{}' does not exist.", path));
}
// Read the source from a binary file instead of sampling from some
// assumed source distribution
write_message(6, "Reading source file from {}...", path);
// Open the binary file
hid_t file_id = file_open(path, 'r', true);
// Check to make sure this is a source file
std::string filetype;
read_attribute(file_id, "filetype", filetype);
if (filetype != "source" && filetype != "statepoint") {
fatal_error("Specified starting source file not a source file type.");
}
// Read in the source bank
read_source_bank(file_id, sites_);
// Close file
file_close(file_id);
}
Particle::Bank SourceFile::sample(uint64_t* seed)
{
size_t i_site = sites_.size()*prn(seed);
return sites_[i_site];
}
//==============================================================================
// CustomSourceWrapper implementation
//==============================================================================