mirror of
https://github.com/openmc-dev/openmc.git
synced 2026-07-24 12:05:32 -04:00
21e8d91e90
Rename sample_source -> sample. Use std::string as type of argument to openmc_create_source. Use reinterpret_cast when accessing dlsym. Also moves check for dlerror to be before the attempt to create the source (avoids null reference if fails) and capture error message before dlclose, so that the error message is still available. Formatting updates and some refactoring for examples and tests.
66 lines
2.2 KiB
C++
66 lines
2.2 KiB
C++
#include <cmath> // for M_PI
|
|
#include <memory> // for unique_ptr
|
|
#include <unordered_map>
|
|
|
|
#include "openmc/random_lcg.h"
|
|
#include "openmc/source.h"
|
|
#include "openmc/particle.h"
|
|
|
|
class Source : public openmc::CustomSource
|
|
{
|
|
public:
|
|
Source(double radius, double energy) : radius_(radius), energy_(energy) { }
|
|
|
|
// Defines a function that can create a unique pointer to a new instance of this class
|
|
// by extracting the parameters from the provided string.
|
|
static std::unique_ptr<Source> from_string(std::string parameters)
|
|
{
|
|
std::unordered_map<std::string, std::string> parameter_mapping;
|
|
|
|
std::stringstream ss(parameters);
|
|
std::string parameter;
|
|
while (std::getline(ss, parameter, ',')) {
|
|
parameter.erase(0, parameter.find_first_not_of(' '));
|
|
std::string key = parameter.substr(0, parameter.find_first_of('='));
|
|
std::string value = parameter.substr(parameter.find_first_of('=') + 1, parameter.length());
|
|
parameter_mapping[key] = value;
|
|
}
|
|
|
|
double radius = std::stod(parameter_mapping["radius"]);
|
|
double energy = std::stod(parameter_mapping["energy"]);
|
|
return std::make_unique<Source>(radius, energy);
|
|
}
|
|
|
|
// Samples from an instance of this class.
|
|
openmc::Particle::Bank sample(uint64_t* seed)
|
|
{
|
|
openmc::Particle::Bank particle;
|
|
// wgt
|
|
particle.particle = openmc::Particle::Type::neutron;
|
|
particle.wgt = 1.0;
|
|
// position
|
|
double angle = 2.0 * M_PI * openmc::prn(seed);
|
|
double radius = this->radius_;
|
|
particle.r.x = radius * std::cos(angle);
|
|
particle.r.y = radius * std::sin(angle);
|
|
particle.r.z = 0.0;
|
|
// angle
|
|
particle.u = {1.0, 0.0, 0.0};
|
|
particle.E = this->energy_;
|
|
particle.delayed_group = 0;
|
|
|
|
return particle;
|
|
}
|
|
|
|
private:
|
|
double radius_;
|
|
double energy_;
|
|
};
|
|
|
|
// A function to create a unique pointer to an instance of this class when generated
|
|
// via a plugin call using dlopen/dlsym.
|
|
// You must have external C linkage here otherwise dlopen will not find the file
|
|
extern "C" std::unique_ptr<Source> openmc_create_source(std::string parameters)
|
|
{
|
|
return Source::from_string(parameters);
|
|
}
|