#include "file_storage.h"

template<> void WriteVariable<cv::Mat>(std::ostream& out, const cv::Mat& mat) {
	WriteVariable(out, mat.rows);
	WriteVariable(out, mat.cols);
	WriteVariable(out, mat.type());
	const size_t rowBytes = mat.cols * mat.elemSize();
	for (int i = 0; i < mat.rows; i++)
		out.write(reinterpret_cast<const char*>(mat.ptr(i, 0)), rowBytes);
}

void ReadMatData(std::istream& in, cv::Mat& mat) {
	const size_t rowBytes = mat.cols * mat.elemSize();
	for (int i = 0; i < mat.rows; i++)
		in.read(reinterpret_cast<char*>(mat.ptr(i, 0)), rowBytes);
}

template<> void ReadVariable<cv::Mat>(std::istream& in, cv::Mat& mat) {
	int rows, cols, type;
	ReadVariable(in, rows);
	ReadVariable(in, cols);
	ReadVariable(in, type);
	mat.create(rows, cols, type);
	ReadMatData(in, mat);
}

template<> void ReadVariable<CvMat*>(std::istream& in, CvMat*& mat) {
	int rows, cols, type;
	ReadVariable(in, rows);
	ReadVariable(in, cols);
	ReadVariable(in, type);
	mat = cvCreateMat(rows, cols, type);
	cv::Mat wrapper = mat;
	ReadMatData(in, wrapper);
}

template<> void WriteVariable<cv::PCA>(std::ostream& out, const cv::PCA& pca) {
	WriteVariable(out, pca.mean);
	WriteVariable(out, pca.eigenvalues);
	WriteVariable(out, pca.eigenvectors);
}

template<> void ReadVariable<cv::PCA>(std::istream& in, cv::PCA& pca) {
	ReadVariable(in, pca.mean);
	ReadVariable(in, pca.eigenvalues);
	ReadVariable(in, pca.eigenvectors);
}