feat(core): Support serializing AdamParams and SEKParams

Author: timniederhausen

core/include/gprat/hyperparameters.hpp

@@ -6,6 +6,7 @@
 #include "gprat/detail/config.hpp"
 
 #include <string>
+#include <memory>
 
 GPRAT_NS_BEGIN
 
@@ -58,6 +59,30 @@ struct AdamParams
     std::string repr() const;
 };
 
+template <class Archive>
+void save_construct_data(Archive &ar, const AdamParams *v, const unsigned int)
+{
+    ar << v->learning_rate;
+    ar << v->beta1;
+    ar << v->beta2;
+    ar << v->epsilon;
+    ar << v->opt_iter;
+}
+
+template <class Archive>
+void load_construct_data(Archive &ar, AdamParams *v, const unsigned int)
+{
+    double learning_rate, beta1, beta2, epsilon;
+    int opt_iter;
+    ar >> learning_rate;
+    ar >> beta1;
+    ar >> beta2;
+    ar >> epsilon;
+    ar >> opt_iter;
+
+    std::construct_at(v, learning_rate, beta1, beta2, epsilon, opt_iter);
+}
+
 GPRAT_NS_END
 
 #endif

core/include/gprat/kernels.hpp

@@ -7,6 +7,7 @@
 
 #include <cstddef>
 #include <vector>
+#include <memory>
 
 GPRAT_NS_BEGIN
 
@@ -79,6 +80,31 @@ struct SEKParams
     const double &get_param(std::size_t index) const;
 };
 
+template <class Archive>
+void save_construct_data(Archive &ar, const SEKParams *v, const unsigned int)
+{
+    ar << v->lengthscale;
+    ar << v->vertical_lengthscale;
+    ar << v->noise_variance;
+}
+
+template <class Archive>
+void load_construct_data(Archive &ar, SEKParams *v, const unsigned int)
+{
+    double lengthscale, vertical_lengthscale, noise_variance;
+    ar >> lengthscale;
+    ar >> vertical_lengthscale;
+    ar >> noise_variance;
+
+    std::construct_at(v, lengthscale, vertical_lengthscale, noise_variance);
+}
+
+template <typename Archive>
+void serialize(Archive &ar, SEKParams &pt, const unsigned int)
+{
+    ar & pt.m_T & pt.w_T;
+}
+
 GPRAT_NS_END
 
 #endif