msgpack/cpp/type/integer.hpp

//
// MessagePack for C++ static resolution routine
//
// Copyright (C) 2008 FURUHASHI Sadayuki
//
//    Licensed under the Apache License, Version 2.0 (the "License");
//    you may not use this file except in compliance with the License.
//    You may obtain a copy of the License at
//
//        http://www.apache.org/licenses/LICENSE-2.0
//
//    Unless required by applicable law or agreed to in writing, software
//    distributed under the License is distributed on an "AS IS" BASIS,
//    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//    See the License for the specific language governing permissions and
//    limitations under the License.
//
#ifndef MSGPACK_TYPE_INTEGER_HPP__
#define MSGPACK_TYPE_INTEGER_HPP__

#include "msgpack/object.hpp"
#include <limits>

namespace msgpack {


namespace type {
namespace detail {
	template <typename T, bool Signed>
	struct convert_integer_sign;

	template <typename T>
	struct convert_integer_sign<T, true> {
		static inline T convert(object o) {
			if(o.type == type::POSITIVE_INTEGER) {
				if(o.via.u64 > (uint64_t)std::numeric_limits<T>::max())
					{ throw type_error(); }
				return o.via.u64;
			} else if(o.type == type::NEGATIVE_INTEGER) {
				if(o.via.i64 < (int64_t)-std::numeric_limits<T>::max())
					{ throw type_error(); }
				return o.via.i64;
			}
			throw type_error();
		}
	};
	
	template <typename T>
	struct convert_integer_sign<T, false> {
		static inline T convert(object o) {
			if(o.type == type::POSITIVE_INTEGER) {
				if(o.via.u64 > (uint64_t)std::numeric_limits<T>::max())
					{ throw type_error(); }
				return o.via.u64;
			}
			throw type_error();
		}
	};
	
	template <typename T>
	static inline T convert_integer(object o)
	{
		return detail::convert_integer_sign<T, std::numeric_limits<T>::is_signed>::convert(o);
	}



	template <typename T, typename Stream, int Size, bool Signed>
	struct pack_integer_size_sign;

	template <typename T, typename Stream>
	struct pack_integer_size_sign<T, Stream, 1, true> {
		static inline void pack(T v, packer<Stream>& o)
			{ o.pack_int8(v); }
	};

	template <typename T, typename Stream>
	struct pack_integer_size_sign<T, Stream, 1, false> {
		static inline void pack(T v, packer<Stream>& o)
			{ o.pack_uint8(v); }
	};

	template <typename T, typename Stream>
	struct pack_integer_size_sign<T, Stream, 2, true> {
		static inline void pack(T v, packer<Stream>& o) {
			if( (int16_t)v <= (int16_t)std::numeric_limits<int8_t>::max() &&
				(int16_t)v >= (int16_t)std::numeric_limits<int8_t>::min())
				{ o.pack_int8(v); }
			else { o.pack_int16(v); }
		}
	};

	template <typename T, typename Stream>
	struct pack_integer_size_sign<T, Stream, 2, false> {
		static inline void pack(T v, packer<Stream>& o) {
			if( (uint16_t)v <= (uint16_t)std::numeric_limits<uint8_t>::max())
				{ o.pack_uint8(v); }
			else { o.pack_uint16(v); }
		}
	};

	template <typename T, typename Stream>
	struct pack_integer_size_sign<T, Stream, 4, true> {
		static inline void pack(T v, packer<Stream>& o) {
			if( (int32_t)v <= (int32_t)std::numeric_limits<int8_t>::max() &&
				(int32_t)v >= (int32_t)std::numeric_limits<int8_t>::min())
				{ o.pack_int8(v); }
			else if( (int32_t)v <= (int32_t)std::numeric_limits<int16_t>::max() &&
					 (int32_t)v >= (int32_t)std::numeric_limits<int16_t>::min())
				{ o.pack_int16(v); }
			else { o.pack_int32(v); }
		}
	};

	template <typename T, typename Stream>
	struct pack_integer_size_sign<T, Stream, 4, false> {
		static inline void pack(T v, packer<Stream>& o) {
			if( (uint32_t)v <= (uint32_t)std::numeric_limits<uint8_t>::max())
				{ o.pack_uint8(v); }
			else if( (uint32_t)v <= (uint32_t)std::numeric_limits<uint16_t>::max())
				{ o.pack_uint16(v); }
			else { o.pack_uint32(v); }
		}
	};

	template <typename T, typename Stream>
	struct pack_integer_size_sign<T, Stream, 8, true> {
		static inline void pack(T v, packer<Stream>& o) {
			if( (int64_t)v <= (int64_t)std::numeric_limits<int8_t>::max() &&
				(int64_t)v >= (int64_t)std::numeric_limits<int8_t>::min())
				{ o.pack_int8(v); }
			else if( (int64_t)v <= (int64_t)std::numeric_limits<int16_t>::max() &&
					 (int64_t)v >= (int64_t)std::numeric_limits<int16_t>::min())
				{ o.pack_int16(v); }
			else if( (int64_t)v <= (int64_t)std::numeric_limits<int32_t>::max() &&
					 (int64_t)v >= (int64_t)std::numeric_limits<int32_t>::min())
				{ o.pack_int32(v); }
			else { o.pack_int64(v); }
		}
	};

	template <typename T, typename Stream>
	struct pack_integer_size_sign<T, Stream, 8, false> {
		static inline void pack(T v, packer<Stream>& o) {
			if( (uint64_t)v <= (uint64_t)std::numeric_limits<uint8_t>::max())
				{ o.pack_uint8(v); }
			else if( (uint64_t)v <= (uint64_t)std::numeric_limits<uint16_t>::max())
				{ o.pack_uint16(v); }
			else if( (uint64_t)v <= (uint64_t)std::numeric_limits<uint32_t>::max())
				{ o.pack_uint32(v); }
			else { o.pack_uint64(v); }
		}
	};


	template <typename T, typename Stream>
	static inline void pack_integer(T v, packer<Stream>& o)
	{
		pack_integer_size_sign<T, Stream, sizeof(T), std::numeric_limits<T>::is_signed>::pack(v, o);
	}

}  // namespace detail
}  // namespace type


inline signed char& operator>> (object o, signed char& v)
	{ v = type::detail::convert_integer<signed char>(o); return v; }

inline signed short& operator>> (object o, signed short& v)
	{ v = type::detail::convert_integer<signed short>(o); return v; }

inline signed int& operator>> (object o, signed int& v)
	{ v = type::detail::convert_integer<signed int>(o); return v; }

inline signed long& operator>> (object o, signed long& v)
	{ v = type::detail::convert_integer<signed long>(o); return v; }

inline signed long long& operator>> (object o, signed long long& v)
	{ v = type::detail::convert_integer<signed long long>(o); return v; }


inline unsigned char& operator>> (object o, unsigned char& v)
	{ v = type::detail::convert_integer<unsigned char>(o); return v; }

inline unsigned short& operator>> (object o, unsigned short& v)
	{ v = type::detail::convert_integer<unsigned short>(o); return v; }

inline unsigned int& operator>> (object o, unsigned int& v)
	{ v = type::detail::convert_integer<unsigned int>(o); return v; }

inline unsigned long& operator>> (object o, unsigned long& v)
	{ v = type::detail::convert_integer<unsigned long>(o); return v; }

inline unsigned long long& operator>> (object o, unsigned long long& v)
	{ v = type::detail::convert_integer<unsigned long long>(o); return v; }


template <typename Stream>
inline packer<Stream>& operator<< (packer<Stream>& o, const signed char& v)
	{ type::detail::pack_integer<signed char>(v, o); return o; }

template <typename Stream>
inline packer<Stream>& operator<< (packer<Stream>& o, const signed short& v)
	{ type::detail::pack_integer<signed short>(v, o); return o; }

template <typename Stream>
inline packer<Stream>& operator<< (packer<Stream>& o, const signed int& v)
	{ type::detail::pack_integer<signed int>(v, o); return o; }

template <typename Stream>
inline packer<Stream>& operator<< (packer<Stream>& o, const signed long& v)
	{ type::detail::pack_integer<signed long>(v, o); return o; }

template <typename Stream>
inline packer<Stream>& operator<< (packer<Stream>& o, const signed long long& v)
	{ type::detail::pack_integer<signed long long>(v, o); return o; }


template <typename Stream>
inline packer<Stream>& operator<< (packer<Stream>& o, const unsigned char& v)
	{ type::detail::pack_integer<unsigned char>(v, o); return o; }

template <typename Stream>
inline packer<Stream>& operator<< (packer<Stream>& o, const unsigned short& v)
	{ type::detail::pack_integer<unsigned short>(v, o); return o; }

template <typename Stream>
inline packer<Stream>& operator<< (packer<Stream>& o, const unsigned int& v)
	{ type::detail::pack_integer<unsigned int>(v, o); return o; }

template <typename Stream>
inline packer<Stream>& operator<< (packer<Stream>& o, const unsigned long& v)
	{ type::detail::pack_integer<unsigned long>(v, o); return o; }

template <typename Stream>
inline packer<Stream>& operator<< (packer<Stream>& o, const unsigned long long& v)
	{ type::detail::pack_integer<unsigned long long>(v, o); return o; }


}  // namespace msgpack

#endif /* msgpack/type/integer.hpp */