diff options
Diffstat (limited to 'extern/quadriflow/src/serialize.hpp')
-rw-r--r-- | extern/quadriflow/src/serialize.hpp | 127 |
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 |