/*******************************************************************/

/*  u16ports: u16 variants of wide character string functions.     */

/*  Copyright (C) 2017  Z. Gilboa                                  */

/*  Released under the Standard MIT License; see COPYING.U16PORTS. */

/*******************************************************************/

#include <stdint.h>

#include <u16ports/u16ports.h>

/**

 *  scalar	nickname	utf-16		utf-8[0]  utf-8[1]  utf-8[2]  utf-8[3]

 *  ------	--------	--------	--------  --------  --------  --------

 *  00000000	7x		00000000	0xxxxxxx

 *  0xxxxxxx			0xxxxxxx

 *

 *  00000yyy	5y6x		00000yyy	110yyyyy  10xxxxxx

 *  yyxxxxxx			yyxxxxxx

 *

 *  zzzzyyyy	4z6y6x		zzzzyyyy	1110zzzz  10yyyyyy  10xxxxxx

 *  yyxxxxxx			yyxxxxxx

 *

 *  000uuuuu	5u4z6y6x	110110ww	11110uuu  10uuzzzz  10yyyyyy  10xxxxxx

 *  zzzzyyyy			wwzzzzyy

 *  yyxxxxxx			110111yy

 *				yyxxxxxx        (where wwww = uuuuu - 1)

 *

**/

size_t u16_wcstombs(char * s, const uint16_t * src, size_t n)

{

	size_t		ret;

	ssize_t		len;

	char *		dst;

	uint16_t	wx;

	uint16_t	wy;

	uint16_t	wz;

	uint16_t	ww;

	uint16_t	wy_low;

	uint16_t	wy_high;

	uint16_t	u_low;

	uint16_t	u_high;

	uint16_t	uuuuu;

	if (!s) {

		for (ret=0; *src; src++) {

			if (*src <= 0x7F)

				ret += 1;

			else if (*src <= 0x7FF)

				ret += 2;

			else if ((*src < 0xD800) || (*src >= 0xE000))

				ret += 3;

			else if (*src++ >= 0xDC00)

				return -1;

			else if (*src < 0xDC00)

				return -1;

			else if (*src >= 0xE000)

				return -1;

			else

				ret += 4;

		}

		return ret;

	}

	if ((len = (ssize_t)n) < 0)

		return -1;

	for (dst=s; *src; src++) {

		if (*src <= 0x7F) {

			*dst++ = (char)*src;

			len--;

		} else if (*src <= 0x7FF) {

			if ((len = len - 2) < 0)

				return -1;

			wy  = *src;

			wy >>= 6;

			wx  = *src;

			wx <<= 10;

			wx >>= 10;

			*dst++ = (char)(0xC0 | wy);

			*dst++ = (char)(0x80 | wx);

		} else if ((*src < 0xD800) || (*src >= 0xE000)) {

			if ((len = len - 3) < 0)

				return -1;

			wz  = *src;

			wz >>= 12;

			wy  = *src;

			wy <<= 4;

			wy >>= 10;

			wx  = *src;

			wx <<= 10;

			wx >>= 10;

			*dst++ = (char)(0xE0 | wz);

			*dst++ = (char)(0x80 | wy);

			*dst++ = (char)(0x80 | wx);

		} else if (src[0] >= 0xDC00) {

			return -1;

		} else if (src[1] < 0xDC00) {

			return -1;

		} else if (src[1] >= 0xE000) {

			return -1;

		} else {

			if ((len = len - 4) < 0)

				return -1;

			/* low two bytes */

			wy_high   = *src;

			wy_high <<= 14;

			wy_high >>= 10;

			wz   = *src;

			wz <<= 10;

			wz >>= 12;

			ww   = *src;

			ww <<= 6;

			ww >>= 12;

			/* (surrogate pair) */

			src++;

			/* high two bytes */

			wx   = *src;

			wx <<= 10;

			wx >>= 10;

			wy_low   = *src;

			wy_low <<= 6;

			wy_low >>= 12;

			/* uuuuu */

			uuuuu    = ww + 1;

			u_low    = uuuuu;

			u_low  >>= 2;

			u_high  = uuuuu;

			u_high <<= 14;

			u_high >>= 10;

			*dst++ = (char)(0xF0 | u_low);

			*dst++ = (char)(0x80 | u_high | wz);

			*dst++ = (char)(0x80 | wy_low | wy_high);

			*dst++ = (char)(0x80 | wx);

		}

		if (len == 0)

			return dst - s;

	}

	if (len)

		*dst = 0;

	return dst - s;

}