Differential D5594 Diff 18116 extern/quadriflow/src/serialize.hpp

Changeset View

Standalone View

extern/quadriflow/src/serialize.hpp

This file was added.

				#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