tboudreaux/GridFire
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(GraphNetwork): started templating for auto diff

Browse Source
This commit is contained in:
Emily Boudreaux
2025-06-19 15:20:37 -04:00
parent cd5a6b200b
commit e0704dcafe
2 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/network/private/netgraph.cpp
View File

@@ -20,7 +20,8 @@
namespace serif::network {
GraphNetwork::GraphNetwork(
template <typename GeneralScalarType>
GraphNetwork<GeneralScalarType>::GraphNetwork(
const serif::composition::Composition &composition
):
Network(REACLIB),
@@ -32,7 +33,8 @@ namespace serif::network {
syncInternalMaps();
}
GraphNetwork::GraphNetwork(
template <typename GeneralScalarType>
GraphNetwork<GeneralScalarType>::GraphNetwork(
const serif::composition::Composition &composition,
const double cullingThreshold,
const double T9
@@ -46,14 +48,16 @@ namespace serif::network {
syncInternalMaps();
}
void GraphNetwork::syncInternalMaps() {
template <typename GeneralScalarType>
void GraphNetwork<GeneralScalarType>::syncInternalMaps() {
collectNetworkSpecies();
populateReactionIDMap();
populateSpeciesToIndexMap();
}
// --- Network Graph Construction Methods ---
void GraphNetwork::collectNetworkSpecies() {
template <typename GeneralScalarType>
void GraphNetwork<GeneralScalarType>::collectNetworkSpecies() {
m_networkSpecies.clear();
m_networkSpeciesMap.clear();
@@ -81,7 +85,8 @@ namespace serif::network {
}
void GraphNetwork::populateReactionIDMap() {
template <typename GeneralScalarType>
void GraphNetwork<GeneralScalarType>::populateReactionIDMap() {
LOG_INFO(m_logger, "Populating reaction ID map for REACLIB graph network (serif::network::GraphNetwork)...");
m_reactionIDMap.clear();
for (const auto& reaction: m_reactions) {

11
src/network/public/netgraph.h
View File

@@ -18,6 +18,7 @@
// REACLIBReactions are quite large data structures, so this could be a performance bottleneck.
namespace serif::network {
template <typename GeneralScalarType>
class GraphNetwork final : public Network {
public:
explicit GraphNetwork(const serif::composition::Composition &composition);
@@ -38,15 +39,15 @@ namespace serif::network {
reaclib::REACLIBReactionSet m_reactions; ///< Set of REACLIB reactions for the network.
serif::composition::Composition m_initialComposition;
serif::composition::Composition m_currentComposition;
double m_cullingThreshold; ///< Threshold for culling reactions.
double m_T9; ///< Temperature in T9 units to evaluate culling.
GeneralScalarType m_cullingThreshold; ///< Threshold for culling reactions.
GeneralScalarType m_T9; ///< Temperature in T9 units to evaluate culling.
std::vector<serif::atomic::Species> m_networkSpecies; ///< The species in the network.
std::unordered_map<std::string_view, serif::atomic::Species> m_networkSpeciesMap;
std::unordered_map<std::string_view, const reaclib::REACLIBReaction> m_reactionIDMap; ///< Map of reaction IDs to REACLIB reactions.
boost::numeric::ublas::compressed_matrix<int> m_stoichiometryMatrix; ///< Stoichiometry matrix for the network.
boost::numeric::ublas::compressed_matrix<double> m_jacobianMatrix; ///< Jacobian matrix for the network.
boost::numeric::ublas::compressed_matrix<GeneralScalarType> m_jacobianMatrix; ///< Jacobian matrix for the network.
std::unordered_map<serif::atomic::Species, size_t> m_speciesToIndexMap; ///< Map of species to their index in the stoichiometry matrix.
@@ -62,8 +63,8 @@ namespace serif::network {
void generateStoichiometryMatrix();
void generateJacobianMatrix();
std::vector<double> calculateRHS(const std::vector<double>& Y, const double T9, const double rho) const;
double calculateReactionRate(const reaclib::REACLIBReaction &reaction, const std::vector<double>& Y, const double T9, const double rho) const;
std::vector<GeneralScalarType> calculateRHS(const std::vector<GeneralScalarType>& Y, const GeneralScalarType T9, const GeneralScalarType rho) const;
GeneralScalarType calculateReactionRate(const reaclib::REACLIBReaction &reaction, const std::vector<GeneralScalarType>& Y, const GeneralScalarType T9, const GeneralScalarType rho) const;
void pruneNetworkGraph();
};