feat: remove header dependency to c++20 features

This commit is contained in:
Max Lübke 2024-09-10 09:10:33 +02:00
parent 0bf0826c9b
commit 1f586f1748
19 changed files with 103 additions and 183 deletions

View File

@ -1,49 +1,35 @@
#pragma once
#include "EquilibriumWrapper.hpp"
#include "ExchangeWrapper.hpp"
#include "KineticWrapper.hpp"
#include "POETInit.hpp"
#include "SolutionWrapper.hpp"
#include "SurfaceWrapper.hpp"
#include <IPhreeqc.hpp>
#include <Exchange.h>
#include <PPassemblage.h>
#include <Phreeqc.h>
#include <Solution.h>
#include <Surface.h>
#include <cstddef>
#include <cxxKinetics.h>
#include <memory>
#include <span>
#include <string>
#include <vector>
/**
* @brief Class for running Phreeqc wrappped in POET
*
* Direct interface to Phreeqc, without utilizing *_MODIFY keywords/scripts to
* Direct interface to Phreeqc, without utilizing *_MODIFY keywords/scripts
to
* set new values. Use with already initialized Phreeqc config.
*
*/
class PhreeqcEngine : public IPhreeqc {
class PhreeqcEngine {
public:
/**
* @brief Construct a new Phreeqc Engine object
*
* Construct a new Phreeqc Engine object by previously initialized POETConfig.
* Construct a new Phreeqc Engine object by previously initialized
POETConfig.
*
* @param config Holds the configuration for the Phreeqc engine.
*/
PhreeqcEngine(const POETConfig &config) : IPhreeqc() {
this->LoadDatabaseString(config.database.c_str());
this->RunString(config.input_script.c_str());
PhreeqcEngine(const POETConfig &config);
this->init_wrappers(config.cell);
}
/**
* @brief Destroy the Phreeqc Engine object
*
*/
~PhreeqcEngine();
/**
* @brief Siimulate a cell for a given time step
@ -56,42 +42,6 @@ public:
void runCell(std::vector<double> &cell_values, double time_step);
private:
void init_wrappers(const POETInitCell &cell);
void run(double time_step);
cxxSolution *Get_solution(std::size_t n) {
return Utilities::Rxn_find(this->PhreeqcPtr->Get_Rxn_solution_map(), n);
}
cxxExchange *Get_exchange(std::size_t n) {
return Utilities::Rxn_find(this->PhreeqcPtr->Get_Rxn_exchange_map(), n);
}
cxxKinetics *Get_kinetic(std::size_t n) {
return Utilities::Rxn_find(this->PhreeqcPtr->Get_Rxn_kinetics_map(), n);
}
cxxPPassemblage *Get_equilibrium(std::size_t n) {
return Utilities::Rxn_find(this->PhreeqcPtr->Get_Rxn_pp_assemblage_map(),
n);
}
cxxSurface *Get_surface(std::size_t n) {
return Utilities::Rxn_find(this->PhreeqcPtr->Get_Rxn_surface_map(), n);
}
void get_essential_values(std::span<double> &data);
void set_essential_values(const std::span<double> &data);
std::unique_ptr<SolutionWrapper> solutionWrapperPtr;
std::unique_ptr<ExchangeWrapper> exchangeWrapperPtr;
std::unique_ptr<KineticWrapper> kineticsWrapperPtr;
std::unique_ptr<EquilibriumWrapper> equilibriumWrapperPtr;
std::unique_ptr<SurfaceWrapper> surfaceWrapperPtr;
bool has_exchange = false;
bool has_kinetics = false;
bool has_equilibrium = false;
bool has_surface = false;
class Impl;
std::unique_ptr<Impl> impl;
};

View File

@ -4,7 +4,89 @@
#include <span>
#include <sstream>
void PhreeqcEngine::init_wrappers(const POETInitCell &cell) {
#include <IPhreeqc.hpp>
#include <Phreeqc.h>
#include "Wrapper/EquilibriumWrapper.hpp"
#include "Wrapper/ExchangeWrapper.hpp"
#include "Wrapper/KineticWrapper.hpp"
#include "Wrapper/SolutionWrapper.hpp"
#include "Wrapper/SurfaceWrapper.hpp"
class PhreeqcEngine::Impl : public IPhreeqc {
public:
Impl(const POETConfig &config);
void init_wrappers(const POETInitCell &cell);
void run(double time_step);
cxxSolution *Get_solution(std::size_t n) {
return Utilities::Rxn_find(this->PhreeqcPtr->Get_Rxn_solution_map(), n);
}
cxxExchange *Get_exchange(std::size_t n) {
return Utilities::Rxn_find(this->PhreeqcPtr->Get_Rxn_exchange_map(), n);
}
cxxKinetics *Get_kinetic(std::size_t n) {
return Utilities::Rxn_find(this->PhreeqcPtr->Get_Rxn_kinetics_map(), n);
}
cxxPPassemblage *Get_equilibrium(std::size_t n) {
return Utilities::Rxn_find(this->PhreeqcPtr->Get_Rxn_pp_assemblage_map(),
n);
}
cxxSurface *Get_surface(std::size_t n) {
return Utilities::Rxn_find(this->PhreeqcPtr->Get_Rxn_surface_map(), n);
}
void get_essential_values(std::span<double> &data);
void set_essential_values(const std::span<double> &data);
std::unique_ptr<SolutionWrapper> solutionWrapperPtr;
std::unique_ptr<ExchangeWrapper> exchangeWrapperPtr;
std::unique_ptr<KineticWrapper> kineticsWrapperPtr;
std::unique_ptr<EquilibriumWrapper> equilibriumWrapperPtr;
std::unique_ptr<SurfaceWrapper> surfaceWrapperPtr;
bool has_exchange = false;
bool has_kinetics = false;
bool has_equilibrium = false;
bool has_surface = false;
};
PhreeqcEngine::PhreeqcEngine(const POETConfig &config)
: impl(std::make_unique<Impl>(config)) {}
PhreeqcEngine::Impl::Impl(const POETConfig &config) {
this->LoadDatabaseString(config.database.c_str());
this->RunString(config.input_script.c_str());
this->init_wrappers(config.cell);
}
PhreeqcEngine::~PhreeqcEngine() = default;
void PhreeqcEngine::runCell(std::vector<double> &cell_values,
double time_step) {
// skip ID
std::span<double> cell_data{cell_values.begin() + 1, cell_values.end()};
this->impl->set_essential_values(cell_data);
this->impl->run(time_step);
this->impl->get_essential_values(cell_data);
}
void PhreeqcEngine::Impl::run(double time_step) {
std::stringstream time_ss;
time_ss << std::fixed << std::setprecision(20) << time_step;
const std::string runs_string =
"RUN_CELLS\n -cells 1\n -time_step " + time_ss.str() + "\nEND\n";
this->RunString(runs_string.c_str());
}
void PhreeqcEngine::Impl::init_wrappers(const POETInitCell &cell) {
// Solutions
this->solutionWrapperPtr =
@ -41,7 +123,7 @@ void PhreeqcEngine::init_wrappers(const POETInitCell &cell) {
}
}
void PhreeqcEngine::get_essential_values(std::span<double> &data) {
void PhreeqcEngine::Impl::get_essential_values(std::span<double> &data) {
this->solutionWrapperPtr->get(data);
@ -76,54 +158,9 @@ void PhreeqcEngine::get_essential_values(std::span<double> &data) {
data.subspan(offset, this->surfaceWrapperPtr->size())};
this->surfaceWrapperPtr->get(surf_span);
}
// // Exchange
// if (this->Get_exchange(cell_number) != NULL) {
// std::vector<double> exch_values(this->exchangeWrapperPtr->size());
// std::span<double> exch_span{exch_values};
// this->exchangeWrapperPtr->get(exch_span);
// // this->Get_exchange(cell_number)->get_essential_values(exch_values);
// essentials.insert(essentials.end(), exch_values.begin(),
// exch_values.end());
// }
// // Kinetics
// if (this->Get_kinetic(cell_number) != NULL) {
// std::vector<double> kin_values;
// this->Get_kinetic(cell_number)->get_essential_values(kin_values);
// essentials.insert(essentials.end(), kin_values.begin(),
// kin_values.end());
// }
// // PPassemblage
// if (this->Get_equilibrium(cell_number) != NULL) {
// std::vector<double> equ_values;
// this->Get_equilibrium(cell_number)->get_essential_values(equ_values);
// essentials.insert(essentials.end(), equ_values.begin(),
// equ_values.end());
// }
// // Surface
// if (this->Get_surface(cell_number) != NULL) {
// std::vector<double> surf_values(this->surfaceWrapperPtr->size());
// std::span<double> surf_span{surf_values};
// this->surfaceWrapperPtr->get(surf_span);
// // this->Get_surface(cell_number)->get_essential_values(surf_values);
// essentials.insert(essentials.end(), surf_values.begin(),
// surf_values.end());
// }
// return essentials;
}
void PhreeqcEngine::set_essential_values(const std::span<double> &data) {
void PhreeqcEngine::Impl::set_essential_values(const std::span<double> &data) {
this->solutionWrapperPtr->set(data);
this->PhreeqcPtr->initial_solutions_poet(1);
@ -159,71 +196,4 @@ void PhreeqcEngine::set_essential_values(const std::span<double> &data) {
data.subspan(offset, this->surfaceWrapperPtr->size())};
this->surfaceWrapperPtr->set(surf_span);
}
// // Solutions
// if (this->Get_solution(cell_number) != NULL) {
// this->Get_solution(cell_number)->set_essential_values(dump_it, order);
// this->PhreeqcPtr->initial_solutions_poet(cell_number);
// }
// // Exchange
// if (this->Get_exchange(cell_number) != NULL) {
// auto &exch_pointer = this->exchangeWrapperPtr;
// std::span<double> exch_span{dump_it, exch_pointer->size()};
// exch_pointer->set(exch_span);
// dump_it += exch_pointer->size();
// // this->Get_exchange(cell_number)->set_essential_values(dump_it);
// // this->PhreeqcPtr->Rxn_new_exchange.insert(cell_number);
// // this->Get_exchange(cell_number)->Set_new_def(true);
// // this->Get_exchange(cell_number)->Set_solution_equilibria(true);
// // this->Get_exchange(cell_number)->Set_n_solution(cell_number);
// // this->PhreeqcPtr->initial_exchangers(FALSE);
// }
// // Kinetics
// if (this->Get_kinetic(cell_number) != NULL) {
// this->Get_kinetic(cell_number)->set_essential_values(dump_it);
// }
// // PPassemblage
// if (this->Get_equilibrium(cell_number) != NULL) {
// this->Get_equilibrium(cell_number)->set_essential_values(dump_it);
// }
// // Surface
// if (this->Get_surface(cell_number) != NULL) {
// auto *tmp = this->Get_surface(cell_number);
// auto &surf_pointer = this->surfaceWrapperPtr;
// std::span<double> surf_span{dump_it, surf_pointer->size()};
// surf_pointer->set(surf_span);
// dump_it += surf_pointer->size();
// // this->PhreeqcPtr->Rxn_new_surface.insert(cell_number);
// // this->Get_surface(cell_number)->set_essential_values(dump_it);
// // this->PhreeqcPtr->Rxn_new_surface.insert(cell_number);
// // this->PhreeqcPtr->initial_surfaces(TRUE);
// }
}
void PhreeqcEngine::runCell(std::vector<double> &cell_values,
double time_step) {
// skip ID
std::span<double> cell_data{cell_values.begin() + 1, cell_values.end()};
this->set_essential_values(cell_data);
this->run(time_step);
this->get_essential_values(cell_data);
}
void PhreeqcEngine::run(double time_step) {
std::stringstream time_ss;
time_ss << std::fixed << std::setprecision(20) << time_step;
const std::string runs_string =
"RUN_CELLS\n -cells 1\n -time_step " + time_ss.str() + "\nEND\n";
this->RunString(runs_string.c_str());
}

View File

@ -1,9 +1,9 @@
#include "EquilibriumWrapper.hpp"
#include "ExchangeWrapper.hpp"
#include "KineticWrapper.hpp"
#include "PhreeqcInit.hpp"
#include "SolutionWrapper.hpp"
#include "SurfaceWrapper.hpp"
#include "Wrapper/EquilibriumWrapper.hpp"
#include "Wrapper/ExchangeWrapper.hpp"
#include "Wrapper/KineticWrapper.hpp"
#include "Wrapper/SolutionWrapper.hpp"
#include "Wrapper/SurfaceWrapper.hpp"
#include "global_structures.h"
#include <cassert>