/*! \file pibinarystream.h
 * \ingroup Core
 * \~\brief
 * \~english Binary serialization interface
 * \~russian Интерфейс бинарной сериализации
*/
/*
	PIP - Platform Independent Primitives
	Binary serialization interface
	Ivan Pelipenko peri4ko@yandex.ru

	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU Lesser General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU Lesser General Public License for more details.

	You should have received a copy of the GNU Lesser General Public License
	along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef PIBINARYSTREAM_H
#define PIBINARYSTREAM_H

#include "pibitarray.h"
#include "pimap.h"
#include "pivector2d.h"


//! \ingroup Core
//! \~\brief
//! \~english Binary serialization interface.
//! \~russian Интерфейс бинарной сериализации.
template<typename P>
class PIBinaryStream {
public:
	// one should implements next methods:
	//
	// bool binaryStreamAppend(const void * d, size_t s);
	// bool binaryStreamTake(void * d, size_t s);

	bool append(const void * d, size_t s) {
		if (!static_cast<P*>(this)->binaryStreamAppend(d, s)) {
			return false;
			printf("[PIBinaryStream] append() error\n");
		}
		return true;
	}
	bool take(void * d, size_t s) {
		if (!static_cast<P*>(this)->binaryStreamTake(d, s)) {
			return false;
			printf("[PIBinaryStream] take() error\n");
		}
		return true;
	}
	template<typename T>
	void append(T v) {append(&v, sizeof(v));}
	uchar takeByte() {uchar r = 0; take(&r, sizeof(r)); return r;}
	int takeInt() {int r = 0; take(&r, sizeof(r)); return r;}
};


// helper class to detect default operators
template<typename P>
class PIBinaryStreamRef {
public:
	PIBinaryStreamRef(PIBinaryStream<P> & s): p(s) {}
	PIBinaryStream<P> & p;
};


template<typename P, typename T> inline PIBinaryStream<P> & operator <<(PIBinaryStreamRef<P> s, const T & v) {s.p << v; return s.p;}
template<typename P, typename T> inline PIBinaryStream<P> & operator >>(PIBinaryStreamRef<P> s,       T & v) {s.p >> v; return s.p;}


// small types
template<typename P> inline PIBinaryStreamRef<P> operator <<(PIBinaryStream<P> & s, const bool    v) {s.append((uchar)v); return s;}
template<typename P> inline PIBinaryStreamRef<P> operator <<(PIBinaryStream<P> & s, const char    v) {s.append((uchar)v); return s;}
template<typename P> inline PIBinaryStreamRef<P> operator <<(PIBinaryStream<P> & s, const uchar   v) {s.append((uchar)v); return s;}
template<typename P> inline PIBinaryStreamRef<P> operator >>(PIBinaryStream<P> & s,       bool  & v) {v = s.takeByte();   return s;}
template<typename P> inline PIBinaryStreamRef<P> operator >>(PIBinaryStream<P> & s,       char  & v) {v = s.takeByte();   return s;}
template<typename P> inline PIBinaryStreamRef<P> operator >>(PIBinaryStream<P> & s,       uchar & v) {v = s.takeByte();   return s;}




// store simple types


template<typename P, typename T, typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0>
inline PIBinaryStreamRef<P> operator <<(PIBinaryStream<P> & s, const T & v) {
	s.append(&v, sizeof(v));
	return s;
}


//! \~english Store operator for PIVector of any trivial copyable type
//! \~russian Оператор сохранения для PIVector тривиальных типов
template<typename P, typename T,
typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0,
typename std::enable_if< std::is_same<decltype(std::declval<PIBinaryStream<P>&>() << std::declval<const T &>()), PIBinaryStreamRef<P>>::value, int>::type = 0>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const PIVector<T> & v) {
	piCout << "<< vector trivial default";
	s.append((int)v.size());
	s.append(v.data(), v.size() * sizeof(T));
	return s;
}
template<typename P, typename T,
typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0,
typename std::enable_if<!std::is_same<decltype(std::declval<PIBinaryStream<P>&>() << std::declval<const T &>()), PIBinaryStreamRef<P>>::value, int>::type = 0>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const PIVector<T> & v) {
	piCout << "<< vector trivial custom";
	s.append((int)v.size());
	for (size_t i = 0; i < v.size(); ++i) s << v[i];
	return s;
}


//! \~english Store operator for PIDeque of any trivial copyable type
//! \~russian Оператор сохранения для PIDeque тривиальных типов
template<typename P, typename T,
typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0,
typename std::enable_if< std::is_same<decltype(std::declval<PIBinaryStream<P>&>() << std::declval<const T &>()), PIBinaryStreamRef<P>>::value, int>::type = 0>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const PIDeque<T> & v) {
	piCout << "<< deque trivial default";
	s.append((int)v.size());
	s.append(v.data(), v.size() * sizeof(T));
	return s;
}
template<typename P, typename T,
typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0,
typename std::enable_if<!std::is_same<decltype(std::declval<PIBinaryStream<P>&>() << std::declval<const T &>()), PIBinaryStreamRef<P>>::value, int>::type = 0>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const PIDeque<T> & v) {
	piCout << "<< deque trivial custom";
	s.append((int)v.size());
	for (size_t i = 0; i < v.size(); ++i) s << v[i];
	return s;
}


//! \~english Store operator for PIVector2D of any trivial copyable type
//! \~russian Оператор сохранения для PIVector2D тривиальных типов
template<typename P, typename T,
typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0,
typename std::enable_if< std::is_same<decltype(std::declval<PIBinaryStream<P>&>() << std::declval<const T &>()), PIBinaryStreamRef<P>>::value, int>::type = 0>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const PIVector2D<T> & v) {
	piCout << "<< vector2d trivial default";
	s.append((int)v.rows());
	s.append((int)v.cols());
	s.append(v.data(), v.size() * sizeof(T));
	return s;
}
template<typename P, typename T,
typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0,
typename std::enable_if<!std::is_same<decltype(std::declval<PIBinaryStream<P>&>() << std::declval<const T &>()), PIBinaryStreamRef<P>>::value, int>::type = 0>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const PIVector2D<T> & v) {
	piCout << "<< vector2d trivial custom";
	s.append((int)v.rows());
	s.append((int)v.cols());
	s << v.toPlainVector();
	return s;
}


//! \~english Store operator
//! \~russian Оператор сохранения
template<typename P>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const PIBitArray & v) {s << v.size_ << v.data_; return s;}


//! \~english Store operator
//! \~russian Оператор сохранения
template<typename P, typename Type0, typename Type1>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const PIPair<Type0, Type1> & v) {s << v.first << v.second; return s;}




// restore simple types


template<typename P, typename T, typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0>
inline PIBinaryStreamRef<P> operator >>(PIBinaryStream<P> & s, T & v) {
	if (!s.take(&v, sizeof(v))) {
		printf("error with %s\n", __PIP_TYPENAME__(T));
		assert(false);
	}
	return s;
}


//! \~english Restore operator for PIVector of any trivial copyable type
//! \~russian Оператор извлечения для PIVector тривиальных типов
template<typename P, typename T,
typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0,
typename std::enable_if< std::is_same<decltype(std::declval<PIBinaryStream<P>&>() >> std::declval<T &>()), PIBinaryStreamRef<P>>::value, int>::type = 0>
inline PIBinaryStream<P> & operator >>(PIBinaryStream<P> & s, PIVector<T> & v) {
	piCout << ">> vector trivial default";
	int sz = s.takeInt();
	v._resizeRaw(sz);
	if (!s.take(v.data(), sz * sizeof(T))) {
		printf("error with PIVector<%s>\n", __PIP_TYPENAME__(T));
		assert(false);
	}
	return s;
}
template<typename P, typename T,
typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0,
typename std::enable_if<!std::is_same<decltype(std::declval<PIBinaryStream<P>&>() >> std::declval<T &>()), PIBinaryStreamRef<P>>::value, int>::type = 0>
inline PIBinaryStream<P> & operator >>(PIBinaryStream<P> & s, PIVector<T> & v) {
	piCout << ">> vector trivial custom";
	int sz = s.takeInt();
	v._resizeRaw(sz);
	for (int i = 0; i < sz; ++i) s >> v[i];
	return s;
}


//! \~english Restore operator for PIDeque of any trivial copyable type
//! \~russian Оператор извлечения для PIDeque тривиальных типов
template<typename P, typename T,
typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0,
typename std::enable_if< std::is_same<decltype(std::declval<PIBinaryStream<P>&>() >> std::declval<T &>()), PIBinaryStreamRef<P>>::value, int>::type = 0>
inline PIBinaryStream<P> & operator >>(PIBinaryStream<P> & s, PIDeque<T> & v) {
	piCout << ">> deque trivial default";
	int sz = s.takeInt();
	v._resizeRaw(sz);
	if (!s.take(v.data(), sz * sizeof(T))) {
		printf("error with PIDeque<%s>\n", __PIP_TYPENAME__(T));
		assert(false);
	}
	return s;
}
template<typename P, typename T,
typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0,
typename std::enable_if<!std::is_same<decltype(std::declval<PIBinaryStream<P>&>() >> std::declval<T &>()), PIBinaryStreamRef<P>>::value, int>::type = 0>
inline PIBinaryStream<P> & operator >>(PIBinaryStream<P> & s, PIDeque<T> & v) {
	piCout << ">> deque trivial custom";
	int sz = s.takeInt();
	v._resizeRaw(sz);
	for (int i = 0; i < sz; ++i) s >> v[i];
	return s;
}


//! \~english Restore operator for PIVector2D of any trivial copyable type
//! \~russian Оператор извлечения для PIVector2D тривиальных типов
template<typename P, typename T,
typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0,
typename std::enable_if< std::is_same<decltype(std::declval<PIBinaryStream<P>&>() >> std::declval<T &>()), PIBinaryStreamRef<P>>::value, int>::type = 0>
inline PIBinaryStream<P> & operator >>(PIBinaryStream<P> & s, PIVector2D<T> & v) {
	piCout << ">> vector2d trivial default";
	int r, c;
	r = s.takeInt();
	c = s.takeInt();
	v._resizeRaw(r, c);
	if (!s.take(v.data(), v.size() * sizeof(T))) {
		printf("error with PIVector2D<%s>\n", __PIP_TYPENAME__(T));
		assert(false);
	}
	return s;
}
template<typename P, typename T,
typename std::enable_if< std::is_trivially_copyable<T>::value, int>::type = 0,
typename std::enable_if<!std::is_same<decltype(std::declval<PIBinaryStream<P>&>() >> std::declval<T &>()), PIBinaryStreamRef<P>>::value, int>::type = 0>
inline PIBinaryStream<P> & operator >>(PIBinaryStream<P> & s, PIVector2D<T> & v) {
	piCout << ">> vector2d trivial custom";
	int r, c;
	PIVector<T> tmp;
	r = s.takeInt();
	c = s.takeInt();
	s >> tmp;
	v = PIVector2D<T>(r, c, tmp);
	return s;
}
;




// store complex types


//! \~english Store operator for PIVector of any compound type
//! \~russian Оператор сохранения для PIVector сложных типов
template<typename P, typename T, typename std::enable_if<!std::is_trivially_copyable<T>::value, int>::type = 0>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const PIVector<T> & v) {
	piCout << "<< vector complex";
	s.append(int(v.size_s()));
	for (size_t i = 0; i < v.size(); ++i) s << v[i];
	return s;
}


//! \~english Store operator for PIDeque of any compound type
//! \~russian Оператор сохранения для PIDeque сложных типов
template<typename P, typename T, typename std::enable_if<!std::is_trivially_copyable<T>::value, int>::type = 0>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const PIDeque<T> & v) {
	piCout << "<< deque complex";
	s.append(int(v.size_s()));
	for (size_t i = 0; i < v.size(); ++i) s << v[i];
	return s;
}


//! \~english Store operator for PIVector2D of any compound type
//! \~russian Оператор сохранения для PIVector2D сложных типов
template<typename P, typename T, typename std::enable_if<!std::is_trivially_copyable<T>::value, int>::type = 0>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const PIVector2D<T> & v) {
	piCout << "<< vector2d complex";
	s.append(int(v.rows()));
	s.append(int(v.cols()));
	s << v.toPlainVector();
	return s;
}




// restore complex types


//! \~english Restore operator for PIVector of any compound type
//! \~russian Оператор извлечения для PIVector сложных типов
template<typename P, typename T, typename std::enable_if<!std::is_trivially_copyable<T>::value, int>::type = 0>
inline PIBinaryStream<P> & operator >>(PIBinaryStream<P> & s, PIVector<T> & v) {
	piCout << ">> vector complex";
	/*if (s.size_s() < 4) {
		printf("error with PIVector<%s>\n", __PIP_TYPENAME__(T));
		assert(s.size_s() >= 4);
	}*/
	v.resize(s.takeInt());
	for (size_t i = 0; i < v.size(); ++i) s >> v[i];
	return s;
}


//! \~english Restore operator for PIDeque of any compound type
//! \~russian Оператор извлечения для PIDeque сложных типов
template<typename P, typename T, typename std::enable_if<!std::is_trivially_copyable<T>::value, int>::type = 0>
inline PIBinaryStream<P> & operator >>(PIBinaryStream<P> & s, PIDeque<T> & v) {
	piCout << ">> deque complex";
	/*if (s.size_s() < 4) {
		printf("error with PIDeque<%s>\n", __PIP_TYPENAME__(T));
		assert(s.size_s() >= 4);
	}*/
	v.resize(s.takeInt());
	for (size_t i = 0; i < v.size(); ++i) s >> v[i];
	return s;
}


//! \~english Restore operator for PIVector2D of any compound type
//! \~russian Оператор извлечения для PIVector2D сложных типов
template<typename P, typename T, typename std::enable_if<!std::is_trivially_copyable<T>::value, int>::type = 0>
inline PIBinaryStream<P> & operator >>(PIBinaryStream<P> & s, PIVector2D<T> & v) {
	piCout << ">> vector2d complex";
	/*if (s.size_s() < 8) {
		printf("error with PIVecto2Dr<%s>\n", __PIP_TYPENAME__(T));
		assert(s.size_s() >= 8);
	}*/
	int r, c;
	PIVector<T> tmp;
	r = s.takeInt();
	c = s.takeInt();
	s >> tmp;
	v = PIVector2D<T>(r, c, tmp);
	return s;
}




// other types


//! \~english Store operator
//! \~russian Оператор сохранения
template <typename P, typename Key, typename T>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const PIMap<Key, T> & v) {
	s.append((int)v.pim_index.size_s());
	for (uint i = 0; i < v.size(); ++i) {
		s.append((int)v.pim_index[i].index);
		s << v.pim_index[i].key;
	}
	s << v.pim_content;
	return s;
}


//! \~english Restore operator
//! \~russian Оператор извлечения
template <typename P, typename Key, typename T>
inline PIBinaryStream<P> & operator >>(PIBinaryStream<P> & s, PIMap<Key, T> & v) {
	/*if (s.size_s() < 4) {
		printf("error with PIMap<%s, %s>\n", __PIP_TYPENAME__(Key), __PIP_TYPENAME__(T));
		assert(s.size_s() >= 4);
	}*/
	int sz = s.takeInt(); v.pim_index.resize(sz);
	int ind = 0;
	for (int i = 0; i < sz; ++i) {
		ind = s.takeInt();
		s >> v.pim_index[i].key;
		v.pim_index[i].index = ind;
	}
	s >> v.pim_content;
	if (v.pim_content.size_s() != v.pim_index.size_s()) {
		piCout << "Warning: loaded invalid PIMap, clear";
		v.clear();
	}
	return s;
}




// non-defined complex types


template<typename P, typename T, typename std::enable_if<!std::is_trivially_copyable<T>::value, int>::type = 0>
inline PIBinaryStream<P> & operator <<(PIBinaryStream<P> & s, const T & ) {
	static_assert(std::is_trivially_copyable<T>::value, "[PIBinaryStream] Error: using undeclared operator << for complex type!");
	return s;
}

template<typename P, typename T, typename std::enable_if<!std::is_trivially_copyable<T>::value, int>::type = 0>
inline PIBinaryStream<P> & operator >>(PIBinaryStream<P> & s, T & ) {
	static_assert(std::is_trivially_copyable<T>::value, "[PIBinaryStream] Error: using undeclared operator >> for complex type!");
	return s;
}




#endif