mirror of
https://github.com/openmc-dev/openmc.git
synced 2026-07-25 12:35:29 -04:00
72 lines
2 KiB
C++
72 lines
2 KiB
C++
//! \file source.h
|
|
//! \brief External source distributions
|
|
|
|
#ifndef OPENMC_SOURCE_H
|
|
#define OPENMC_SOURCE_H
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "pugixml.hpp"
|
|
|
|
#include "openmc/bank.h"
|
|
#include "openmc/distribution_multi.h"
|
|
#include "openmc/distribution_spatial.h"
|
|
#include "openmc/particle.h"
|
|
|
|
namespace openmc {
|
|
|
|
//==============================================================================
|
|
// Global variables
|
|
//==============================================================================
|
|
|
|
class SourceDistribution;
|
|
|
|
namespace model {
|
|
|
|
extern std::vector<SourceDistribution> external_sources;
|
|
|
|
} // namespace model
|
|
|
|
//==============================================================================
|
|
//! External source distribution
|
|
//==============================================================================
|
|
|
|
class SourceDistribution {
|
|
public:
|
|
// Constructors
|
|
SourceDistribution(UPtrSpace space, UPtrAngle angle, UPtrDist energy);
|
|
explicit SourceDistribution(pugi::xml_node node);
|
|
|
|
//! Sample from the external source distribution
|
|
//! \return Sampled site
|
|
Bank sample() const;
|
|
|
|
// Properties
|
|
double strength() const { return strength_; }
|
|
private:
|
|
ParticleType particle_ {ParticleType::neutron}; //!< Type of particle emitted
|
|
double strength_ {1.0}; //!< Source strength
|
|
UPtrSpace space_; //!< Spatial distribution
|
|
UPtrAngle angle_; //!< Angular distribution
|
|
UPtrDist energy_; //!< Energy distribution
|
|
};
|
|
|
|
//==============================================================================
|
|
// Functions
|
|
//==============================================================================
|
|
|
|
//! Initialize source bank from file/distribution
|
|
extern "C" void initialize_source();
|
|
|
|
//! Sample a site from all external source distributions in proportion to their
|
|
//! source strength
|
|
//! \return Sampled source site
|
|
Bank sample_external_source();
|
|
|
|
//! Fill source bank at end of generation for fixed source simulations
|
|
void fill_source_bank_fixedsource();
|
|
|
|
} // namespace openmc
|
|
|
|
#endif // OPENMC_SOURCE_H
|