#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 */