Loading...
Searching...
No Matches
py_engine.h
Go to the documentation of this file.
#pragma once
#include "gridfire/engine/engine.h"
#include "gridfire/expectations/expected_engine.h"
#include "fourdst/composition/atomicSpecies.h"
#include <vector>
#include <expected>
class PyEngine final : public gridfire::Engine {
public:
const std::vector<fourdst::atomic::Species>& getNetworkSpecies() const override;
std::expected<gridfire::StepDerivatives<double>,gridfire::expectations::StaleEngineError> calculateRHSAndEnergy(const std::vector<double> &Y, double T9, double rho) const override;
private:
mutable std::vector<fourdst::atomic::Species> m_species_cache;
};
class PyDynamicEngine final : public gridfire::DynamicEngine {
public:
const std::vector<fourdst::atomic::Species>& getNetworkSpecies() const override;
std::expected<gridfire::StepDerivatives<double>,gridfire::expectations::StaleEngineError> calculateRHSAndEnergy(const std::vector<double> &Y, double T9, double rho) const override;
void generateJacobianMatrix(const std::vector<double> &Y_dynamic, double T9, double rho) const override;
void generateJacobianMatrix(const std::vector<double> &Y_dynamic, double T9, double rho, const gridfire::SparsityPattern &sparsityPattern) const override;
double getJacobianMatrixEntry(int i, int j) const override;
void generateStoichiometryMatrix() override;
int getStoichiometryMatrixEntry(int speciesIndex, int reactionIndex) const override;
double calculateMolarReactionFlow(const gridfire::reaction::Reaction &reaction, const std::vector<double> &Y, double T9, double rho) const override;
const gridfire::reaction::LogicalReactionSet& getNetworkReactions() const override;
void setNetworkReactions(const gridfire::reaction::LogicalReactionSet& reactions) override;
std::expected<std::unordered_map<fourdst::atomic::Species, double>, gridfire::expectations::StaleEngineError> getSpeciesTimescales(const std::vector<double> &Y, double T9, double rho) const override;
std::expected<std::unordered_map<fourdst::atomic::Species, double>, gridfire::expectations::StaleEngineError> getSpeciesDestructionTimescales(const std::vector<double> &Y, double T9, double rho) const override;
fourdst::composition::Composition update(const gridfire::NetIn &netIn) override;
bool isStale(const gridfire::NetIn &netIn) override;
void setScreeningModel(gridfire::screening::ScreeningType model) override;
gridfire::screening::ScreeningType getScreeningModel() const override;
int getSpeciesIndex(const fourdst::atomic::Species &species) const override;
std::vector<double> mapNetInToMolarAbundanceVector(const gridfire::NetIn &netIn) const override;
gridfire::PrimingReport primeEngine(const gridfire::NetIn &netIn) override;
gridfire::BuildDepthType getDepth() const override {
throw std::logic_error("Network depth not supported by this engine.");
}
void rebuild(const fourdst::composition::Composition& comp, gridfire::BuildDepthType depth) override {
throw std::logic_error("Setting network depth not supported by this engine.");
}
private:
mutable std::vector<fourdst::atomic::Species> m_species_cache;
};
class PyEngineView final : public gridfire::EngineView<gridfire::Engine> {
const gridfire::Engine& getBaseEngine() const override;
};
class PyDynamicEngineView final : public gridfire::EngineView<gridfire::DynamicEngine> {
const gridfire::DynamicEngine& getBaseEngine() const override;
};
PyDynamicEngine
Definition py_engine.h:20
PyDynamicEngine::getSpeciesDestructionTimescales
std::expected< std::unordered_map< fourdst::atomic::Species, double >, gridfire::expectations::StaleEngineError > getSpeciesDestructionTimescales(const std::vector< double > &Y, double T9, double rho) const override
Definition py_engine.cpp:158
PyDynamicEngine::getSpeciesTimescales
std::expected< std::unordered_map< fourdst::atomic::Species, double >, gridfire::expectations::StaleEngineError > getSpeciesTimescales(const std::vector< double > &Y, double T9, double rho) const override
Compute timescales for all species in the network.
Definition py_engine.cpp:149
PyDynamicEngine::getJacobianMatrixEntry
double getJacobianMatrixEntry(int i, int j) const override
Get an entry from the previously generated Jacobian matrix.
Definition py_engine.cpp:97
PyDynamicEngine::generateStoichiometryMatrix
void generateStoichiometryMatrix() override
Generate the stoichiometry matrix for the network.
Definition py_engine.cpp:106
PyDynamicEngine::m_species_cache
std::vector< fourdst::atomic::Species > m_species_cache
Definition py_engine.h:48
PyDynamicEngine::getSpeciesIndex
int getSpeciesIndex(const fourdst::atomic::Species &species) const override
Definition py_engine.cpp:202
PyDynamicEngine::rebuild
void rebuild(const fourdst::composition::Composition &comp, gridfire::BuildDepthType depth) override
Definition py_engine.h:44
PyDynamicEngine::isStale
bool isStale(const gridfire::NetIn &netIn) override
Definition py_engine.cpp:176
PyDynamicEngine::getNetworkReactions
const gridfire::reaction::LogicalReactionSet & getNetworkReactions() const override
Get the set of logical reactions in the network.
Definition py_engine.cpp:132
PyDynamicEngine::calculateRHSAndEnergy
std::expected< gridfire::StepDerivatives< double >, gridfire::expectations::StaleEngineError > calculateRHSAndEnergy(const std::vector< double > &Y, double T9, double rho) const override
Calculate the right-hand side (dY/dt) and energy generation.
Definition py_engine.cpp:70
PyDynamicEngine::generateJacobianMatrix
void generateJacobianMatrix(const std::vector< double > &Y_dynamic, double T9, double rho) const override
Generate the Jacobian matrix for the current state.
Definition py_engine.cpp:79
PyDynamicEngine::mapNetInToMolarAbundanceVector
std::vector< double > mapNetInToMolarAbundanceVector(const gridfire::NetIn &netIn) const override
Definition py_engine.cpp:211
PyDynamicEngine::calculateMolarReactionFlow
double calculateMolarReactionFlow(const gridfire::reaction::Reaction &reaction, const std::vector< double > &Y, double T9, double rho) const override
Calculate the molar reaction flow for a given reaction.
Definition py_engine.cpp:123
PyDynamicEngine::getStoichiometryMatrixEntry
int getStoichiometryMatrixEntry(int speciesIndex, int reactionIndex) const override
Get an entry from the stoichiometry matrix.
Definition py_engine.cpp:114
PyDynamicEngine::getScreeningModel
gridfire::screening::ScreeningType getScreeningModel() const override
Get the current electron screening model.
Definition py_engine.cpp:194
PyDynamicEngine::primeEngine
gridfire::PrimingReport primeEngine(const gridfire::NetIn &netIn) override
Definition py_engine.cpp:220
PyDynamicEngine::getDepth
gridfire::BuildDepthType getDepth() const override
Definition py_engine.h:41
PyDynamicEngine::update
fourdst::composition::Composition update(const gridfire::NetIn &netIn) override
Update the internal state of the engine.
Definition py_engine.cpp:167
PyDynamicEngine::setScreeningModel
void setScreeningModel(gridfire::screening::ScreeningType model) override
Set the electron screening model.
Definition py_engine.cpp:185
PyDynamicEngine::getNetworkSpecies
const std::vector< fourdst::atomic::Species > & getNetworkSpecies() const override
PyDynamicEngine Implementation ///.
Definition py_engine.cpp:50
PyDynamicEngine::setNetworkReactions
void setNetworkReactions(const gridfire::reaction::LogicalReactionSet &reactions) override
Definition py_engine.cpp:140
PyDynamicEngineView
Definition py_engine.h:57
PyDynamicEngineView::getBaseEngine
const gridfire::DynamicEngine & getBaseEngine() const override
Access the underlying engine instance.
Definition py_engine.cpp:237
PyEngine
Definition py_engine.h:12
PyEngine::getNetworkSpecies
const std::vector< fourdst::atomic::Species > & getNetworkSpecies() const override
Get the list of species in the network.
Definition py_engine.cpp:16
PyEngine::calculateRHSAndEnergy
std::expected< gridfire::StepDerivatives< double >, gridfire::expectations::StaleEngineError > calculateRHSAndEnergy(const std::vector< double > &Y, double T9, double rho) const override
Calculate the right-hand side (dY/dt) and energy generation.
Definition py_engine.cpp:37
PyEngine::m_species_cache
std::vector< fourdst::atomic::Species > m_species_cache
Definition py_engine.h:17
PyEngineView
Definition py_engine.h:53
PyEngineView::getBaseEngine
const gridfire::Engine & getBaseEngine() const override
Access the underlying engine instance.
Definition py_engine.cpp:229
gridfire::DynamicEngine
Abstract class for engines supporting Jacobian and stoichiometry operations.
Definition engine_abstract.h:130
gridfire::Engine
Abstract base class for a reaction network engine.
Definition engine_abstract.h:84
gridfire::EngineView
Abstract base class for a "view" of a reaction network engine.
Definition engine_view_abstract.h:74
gridfire::reaction::Reaction
Represents a single nuclear reaction from a specific data source.
Definition reaction.h:72
engine.h
Core header for the GridFire reaction network engine module.
expected_engine.h
gridfire::reaction::LogicalReactionSet
TemplatedReactionSet< LogicalReaction > LogicalReactionSet
A set of logical reactions.
Definition reaction.h:563
gridfire::screening::ScreeningType
ScreeningType
Enumerates the available plasma screening models.
Definition screening_types.h:15
gridfire::BuildDepthType
std::variant< NetworkBuildDepth, int > BuildDepthType
Variant specifying either a predefined NetworkBuildDepth or a custom integer depth.
Definition building.h:37
gridfire::SparsityPattern
std::vector< std::pair< size_t, size_t > > SparsityPattern
Definition engine_abstract.h:65
gridfire::NetIn
Definition network.h:53
gridfire::PrimingReport
Captures the result of a network priming operation.
Definition reporting.h:67
gridfire::expectations::StaleEngineError
Definition expected_engine.h:44