1 files changed, 127 insertions, 0 deletions
diff --git a/extern/quadriflow/src/serialize.hpp b/extern/quadriflow/src/serialize.hpp
new file mode 100644
index 00000000000..3e670e02fa5
--- /dev/null
+++ b/extern/quadriflow/src/serialize.hpp
@@ -0,0 +1,127 @@
+#ifndef SERIALIZE_H_
+#define SERIALIZE_H_
+
+#include <Eigen/Core>
+#include <fstream>
+#include <map>
+#include <set>
+#include <vector>
+#include "adjacent-matrix.hpp"
+
+namespace qflow {
+
+template <typename T, int A, int B>
+inline void Save(FILE* fp, const Eigen::Matrix<T, A, B>& m) {
+    int r = m.rows(), c = m.cols();
+    fwrite(&r, sizeof(int), 1, fp);
+    fwrite(&c, sizeof(int), 1, fp);
+    std::vector<T> buffer(r * c);
+    for (int i = 0; i < r; ++i) {
+        for (int j = 0; j < c; ++j) {
+            buffer[i * c + j] = m(i, j);
+        }
+    }
+    fwrite(buffer.data(), sizeof(T), r * c, fp);
+}
+
+template <typename T, int A, int B>
+inline void Read(FILE* fp, Eigen::Matrix<T, A, B>& m) {
+    int r, c;
+    fread(&r, sizeof(int), 1, fp);
+    fread(&c, sizeof(int), 1, fp);
+    std::vector<T> buffer(r * c);
+    fread(buffer.data(), sizeof(T), r * c, fp);
+    m.resize(r, c);
+    for (int i = 0; i < r; ++i) {
+        for (int j = 0; j < c; ++j) {
+            m(i, j) = buffer[i * c + j];
+        }
+    }
+}
+
+inline void Save(FILE* fp, const Link& p) { fwrite(&p, sizeof(Link), 1, fp); }
+
+inline void Read(FILE* fp, Link& p) { fread(&p, sizeof(Link), 1, fp); }
+
+inline void Save(FILE* fp, const TaggedLink& p) { fwrite(&p, sizeof(TaggedLink), 1, fp); }
+
+inline void Read(FILE* fp, TaggedLink& p) { fread(&p, sizeof(TaggedLink), 1, fp); }
+
+inline void Save(FILE* fp, double p) { fwrite(&p, sizeof(double), 1, fp); }
+
+inline void Read(FILE* fp, double& p) { fread(&p, sizeof(double), 1, fp); }
+
+inline void Save(FILE* fp, int p) { fwrite(&p, sizeof(int), 1, fp); }
+
+inline void Read(FILE* fp, int& p) { fread(&p, sizeof(int), 1, fp); }
+
+template <class T, class F>
+inline void Save(FILE* fp, const std::pair<T, F>& p) {
+    fwrite(&p.first, sizeof(T), 1, fp);
+    fwrite(&p.second, sizeof(F), 1, fp);
+}
+
+template <class T, class F>
+inline void Read(FILE* fp, std::pair<T, F>& p) {
+    fread(&p.first, sizeof(T), 1, fp);
+    fread(&p.second, sizeof(F), 1, fp);
+}
+
+template <class T, class F>
+inline void Save(FILE* fp, const std::map<T, F>& p) {
+    int num = p.size();
+    fwrite(&num, sizeof(int), 1, fp);
+    for (auto& s : p) {
+        fwrite(&s, sizeof(s), 1, fp);
+    }
+}
+
+template <class T, class F>
+inline void Read(FILE* fp, std::map<T, F>& p) {
+    int num;
+    p.clear();
+    fread(&num, sizeof(int), 1, fp);
+    for (int i = 0; i < num; ++i) {
+        std::pair<T, F> m;
+        fread(&m, sizeof(m), 1, fp);
+        p.insert(m);
+    }
+}
+
+template <class T>
+void Save(FILE* fp, const std::vector<T>& p) {
+    int num = p.size();
+    fwrite(&num, sizeof(int), 1, fp);
+    for (auto& q : p) {
+        Save(fp, q);
+    }
+}
+
+template <class T>
+void Read(FILE* fp, std::vector<T>& p) {
+    int num;
+    fread(&num, sizeof(int), 1, fp);
+    p.resize(num);
+    for (auto& q : p) {
+        Read(fp, q);
+    }
+}
+
+template <class T>
+void Save(FILE* fp, const std::set<T>& p) {
+    std::vector<T> buffer;
+    buffer.insert(buffer.end(), p.begin(), p.end());
+    Save(fp, buffer);
+}
+
+template <class T>
+void Read(FILE* fp, std::set<T>& p) {
+    std::vector<T> buffer;
+    Read(fp, buffer);
+    p.clear();
+    for (auto& q : buffer) p.insert(q);
+}
+
+} // namespace qflow
+
+#endif