#ifndef FILE_STORAGE_H
#define FILE_STORAGE_H

#include <vector>
#include <iostream>
#include <opencv2/opencv.hpp>

template<class T> inline void WriteVariable(std::ostream& out, const T& var) {
	out.write(reinterpret_cast<const char*>(&var), sizeof(T));
}

template<class T> inline void ReadVariable(std::istream& in, T& var) {
	in.read(reinterpret_cast<char*>(&var), sizeof(T));
}

template<class T> inline T ReadVariable(std::istream& in) {
	T val;
	in.read(reinterpret_cast<char*>(&val), sizeof(T));
	return val;
}

template<> void WriteVariable<cv::Mat>(std::ostream& out, const cv::Mat& mat);
template<> void ReadVariable<cv::Mat>(std::istream& in, cv::Mat& mat);
template<> void ReadVariable<CvMat*>(std::istream& in, CvMat*& mat);
template<> void WriteVariable<cv::PCA>(std::ostream& out, const cv::PCA& pca);
template<> void ReadVariable<cv::PCA>(std::istream& in, cv::PCA& pca);

/// Vectors of vectors not supported
template<class T> void WriteVec(std::ostream& out, const std::vector<T>& vec) {
	WriteVariable(out, vec.size());
	for (size_t i = 0; i < vec.size(); i++)
		WriteVariable(out, vec[i]);
}

/// Vectors of vectors not supported
template<class T> void ReadVec(std::istream& in, std::vector<T>& vec) {
	size_t size;
	ReadVariable(in, size);
	vec.resize(size);
	for (size_t i = 0; i < size; i++) {
		T var;
		ReadVariable(in, var);
		vec[i] = var;
	}
}

/// Vectors of vectors not supported
template<typename _Tp, int cn>
void WriteVec(std::ostream& out, const cv::Vec<_Tp, cn>& vec) {
	for (int i = 0; i < cn; ++i)
		WriteVariable(out, vec[i]);
}

/// Vectors of vectors not supported
template<typename _Tp, int cn>
void ReadVec(std::istream& in, cv::Vec<_Tp, cn>& vec) {
	for (int i = 0; i < cn; ++i)
		ReadVariable(in, vec[i]);
}

#endif /* FILE_STORAGE_H */