shithub: femtolisp

ref: b09398e98563c23b6988e77b7b5ab84c0e03ceb7
dir: /string.c/

View raw version
/*
  string functions
*/
#include "llt.h"
#include "flisp.h"

BUILTIN("string?", stringp)
{
	argcount(nargs, 1);
	return fl_isstring(args[0]) ? FL_T : FL_F;
}

BUILTIN("string.count", string_count)
{
	size_t start = 0;
	if(nargs < 1 || nargs > 3)
		argcount(nargs, 1);
	if(!fl_isstring(args[0]))
		type_error("string", args[0]);
	size_t len = cv_len((cvalue_t*)ptr(args[0]));
	size_t stop = len;
	if(nargs > 1){
		start = toulong(args[1]);
		if(start > len)
			bounds_error(args[0], args[1]);
		if(nargs > 2){
			stop = toulong(args[2]);
			if(stop > len)
				bounds_error(args[0], args[2]);
			if(stop <= start)
				return fixnum(0);
		}
	}
	char *str = cvalue_data(args[0]);
	return size_wrap(u8_charnum(str+start, stop-start));
}

BUILTIN("string.width", string_width)
{
	argcount(nargs, 1);
	if(iscprim(args[0])){
		cprim_t *cp = ptr(args[0]);
		if(cp_class(cp) == wchartype){
			int w = wcwidth(*(wchar_t*)cp_data(cp));
			return w < 0 ? FL_F : fixnum(w);
		}
	}
	return size_wrap(u8_strwidth(tostring(args[0])));
}

BUILTIN("string.reverse", string_reverse)
{
	argcount(nargs, 1);
	if(!fl_isstring(args[0]))
		type_error("string", args[0]);
	size_t len = cv_len(ptr(args[0]));
	value_t ns = cvalue_string(len);
	u8_reverse(cvalue_data(ns), cvalue_data(args[0]), len);
	return ns;
}

BUILTIN("string.encode", string_encode)
{
	argcount(nargs, 1);
	if(iscvalue(args[0])){
		cvalue_t *cv = ptr(args[0]);
		fltype_t *t = cv_class(cv);
		if(t->eltype == wchartype){
			size_t nc = cv_len(cv) / sizeof(uint32_t);
			uint32_t *ptr = (uint32_t*)cv_data(cv);
			size_t nbytes = u8_codingsize(ptr, nc);
			value_t str = cvalue_string(nbytes);
			ptr = cv_data(ptr(args[0]));  // relocatable pointer
			u8_toutf8(cvalue_data(str), nbytes, ptr, nc);
			return str;
		}
	}
	type_error("wchar array", args[0]);
}

BUILTIN("string.decode", string_decode)
{
	int term = 0;
	if(nargs == 2)
		term = args[1] != FL_F;
	else
		argcount(nargs, 1);
	if(!fl_isstring(args[0]))
		type_error("string", args[0]);
	cvalue_t *cv = ptr(args[0]);
	char *ptr = (char*)cv_data(cv);
	size_t nb = cv_len(cv);
	size_t nc = u8_charnum(ptr, nb);
	size_t newsz = nc*sizeof(uint32_t);
	if(term)
		newsz += sizeof(uint32_t);
	value_t wcstr = cvalue(wcstringtype, newsz);
	ptr = cv_data(ptr(args[0]));  // relocatable pointer
	uint32_t *pwc = cvalue_data(wcstr);
	u8_toucs(pwc, nc, ptr, nb);
	if(term)
		pwc[nc] = 0;
	return wcstr;
}

extern BUILTIN("buffer", buffer);
extern value_t stream_to_string(value_t *ps);

BUILTIN("string", string)
{
	if(nargs == 1 && fl_isstring(args[0]))
		return args[0];
	value_t arg, buf = fn_builtin_buffer(nil, 0);
	fl_gc_handle(&buf);
	ios_t *s = value2c(ios_t*, buf);
	int i;
	value_t oldpr = symbol_value(printreadablysym);
	value_t oldpp = symbol_value(printprettysym);
	set(printreadablysym, FL_F);
	set(printprettysym, FL_F);
	FOR_ARGS(i, 0, arg, args){
		USED(arg);
		fl_print(s, args[i]);
	}
	set(printreadablysym, oldpr);
	set(printprettysym, oldpp);
	value_t outp = stream_to_string(&buf);
	fl_free_gc_handles(1);
	return outp;
}

BUILTIN("string.split", string_split)
{
	argcount(nargs, 2);
	char *s = tostring(args[0]);
	char *delim = tostring(args[1]);
	size_t len = cv_len(ptr(args[0]));
	size_t dlen = cv_len(ptr(args[1]));
	size_t ssz, tokend, tokstart, i = 0;
	value_t first = FL_NIL, c = FL_NIL, last;
	size_t junk;
	fl_gc_handle(&first);
	fl_gc_handle(&last);

	do{
		// find and allocate next token
		tokstart = tokend = i;
		while(i < len && !u8_memchr(delim, u8_nextmemchar(s, &i), dlen, &junk))
			tokend = i;
		ssz = tokend - tokstart;
		last = c; // save previous cons cell
		c = fl_cons(cvalue_string(ssz), FL_NIL);

		// we've done allocation; reload movable pointers
		s = cv_data(ptr(args[0]));
		delim = cv_data(ptr(args[1]));

		if(ssz)
			memmove(cv_data(ptr(car_(c))), &s[tokstart], ssz);

		// link new cell
		if(last == FL_NIL)
			first = c;   // first time, save first cons
		else
			((cons_t*)ptr(last))->cdr = c;

		// note this tricky condition: if the string ends with a
		// delimiter, we need to go around one more time to add an
		// empty string. this happens when (i == len && tokend < i)
	}while(i < len || (i == len && (tokend != i)));
	fl_free_gc_handles(2);
	return first;
}

BUILTIN("string.sub", string_sub)
{
	if(nargs != 2)
		argcount(nargs, 3);
	char *s = tostring(args[0]);
	size_t len = cv_len((cvalue_t*)ptr(args[0]));
	size_t i1, i2;
	i1 = toulong(args[1]);
	if(i1 > len)
		bounds_error(args[0], args[1]);
	if(nargs == 3){
		i2 = toulong(args[2]);
		if(i2 > len)
			bounds_error(args[0], args[2]);
	}else{
		i2 = len;
	}
	if(i2 <= i1)
		return cvalue_string(0);
	value_t ns = cvalue_string(i2-i1);
	memmove(cv_data((cvalue_t*)ptr(ns)), &s[i1], i2-i1);
	return ns;
}

BUILTIN("string.char", string_char)
{
	argcount(nargs, 2);
	char *s = tostring(args[0]);
	size_t len = cv_len(ptr(args[0]));
	size_t i = toulong(args[1]);
	if(i >= len)
		bounds_error(args[0], args[1]);
	size_t sl = u8_seqlen(&s[i]);
	if(sl > len || i > len-sl)
		bounds_error(args[0], args[1]);
	return mk_wchar(u8_nextchar(s, &i));
}

BUILTIN("char.upcase", char_upcase)
{
	argcount(nargs, 1);
	cprim_t *cp = (cprim_t*)ptr(args[0]);
	if(!iscprim(args[0]) || cp_class(cp) != wchartype)
		type_error("wchar", args[0]);
	return mk_wchar(towupper(*(int32_t*)cp_data(cp)));
}

BUILTIN("char.downcase", char_downcase)
{
	argcount(nargs, 1);
	cprim_t *cp = ptr(args[0]);
	if(!iscprim(args[0]) || cp_class(cp) != wchartype)
		type_error("wchar", args[0]);
	return mk_wchar(towlower(*(int32_t*)cp_data(cp)));
}

BUILTIN("char-alphabetic?", char_alphabeticp)
{
	argcount(nargs, 1);
	cprim_t *cp = (cprim_t*)ptr(args[0]);
	if(!iscprim(args[0]) || cp_class(cp) != wchartype)
		type_error("wchar", args[0]);
	return iswalpha(*(int32_t*)cp_data(cp)) ? FL_T : FL_F;
}

static value_t
mem_find_byte(char *s, char c, size_t start, size_t len)
{
	char *p = memchr(s+start, c, len-start);
	if(p == nil)
		return FL_F;
	return size_wrap((size_t)(p - s));
}

BUILTIN("string.find", string_find)
{
	char cbuf[8];
	size_t start = 0;
	if(nargs == 3)
		start = toulong(args[2]);
	else
		argcount(nargs, 2);
	char *s = tostring(args[0]);
	size_t len = cv_len(ptr(args[0]));
	if(start > len)
		bounds_error(args[0], args[2]);
	char *needle; size_t needlesz;

	value_t v = args[1];
	cprim_t *cp = ptr(v);
	if(iscprim(v) && cp_class(cp) == wchartype){
		uint32_t c = *(uint32_t*)cp_data(cp);
		if(c <= 0x7f)
			return mem_find_byte(s, (char)c, start, len);
		needlesz = u8_toutf8(cbuf, sizeof(cbuf), &c, 1);
		needle = cbuf;
	}else if(iscprim(v) && cp_class(cp) == bytetype){
		return mem_find_byte(s, *(char*)cp_data(cp), start, len);
	}else if(fl_isstring(v)){
		cvalue_t *cv = (cvalue_t*)ptr(v);
		needlesz = cv_len(cv);
		needle = (char*)cv_data(cv);
	}else{
		type_error("string", args[1]);
	}
	if(needlesz > len-start)
		return FL_F;
	if(needlesz == 1)
		return mem_find_byte(s, needle[0], start, len);
	if(needlesz == 0)
		return size_wrap(start);
	size_t i;
	for(i = start; i < len-needlesz+1; i++){
		if(s[i] == needle[0] && memcmp(&s[i+1], needle+1, needlesz-1) == 0)
			return size_wrap(i);
	}
	return FL_F;
}

BUILTIN("string.inc", string_inc)
{
	if(nargs < 2 || nargs > 3)
		argcount(nargs, 2);
	char *s = tostring(args[0]);
	size_t len = cv_len((cvalue_t*)ptr(args[0]));
	size_t i = toulong(args[1]);
	size_t cnt = 1;
	if(nargs == 3)
		cnt = toulong(args[2]);
	while(cnt--){
		if(i >= len)
			bounds_error(args[0], args[1]);
		(void)(isutf(s[++i]) || isutf(s[++i]) || isutf(s[++i]) || ++i);
	}
	return size_wrap(i);
}

BUILTIN("string.dec", string_dec)
{
	if(nargs < 2 || nargs > 3)
		argcount(nargs, 2);
	char *s = tostring(args[0]);
	size_t len = cv_len((cvalue_t*)ptr(args[0]));
	size_t i = toulong(args[1]);
	size_t cnt = 1;
	if(nargs == 3)
		cnt = toulong(args[2]);
	// note: i is allowed to start at index len
	if(i > len)
		bounds_error(args[0], args[1]);
	while(cnt--){
		if(i == 0)
			bounds_error(args[0], args[1]);
		(void)(isutf(s[--i]) || isutf(s[--i]) || isutf(s[--i]) || --i);
	}
	return size_wrap(i);
}

static unsigned long
get_radix_arg(value_t arg)
{
	unsigned long radix = toulong(arg);
	if(radix < 2 || radix > 36)
		lerrorf(ArgError, "invalid radix");
	return radix;
}

BUILTIN("number->string", number_2_string)
{
	if(nargs < 1 || nargs > 2)
		argcount(nargs, 2);
	value_t n = args[0];
	int neg = 0;
	uint64_t num;
	if(isfixnum(n))
		num = numval(n);
	else if(!iscprim(n))
		type_error("integer", n);
	else
		num = conv_to_uint64(cp_data(ptr(n)), cp_numtype(ptr(n)));
	if(numval(fl_compare(args[0], fixnum(0))) < 0){
		num = -num;
		neg = 1;
	}
	unsigned long radix = 10;
	if(nargs == 2)
		radix = get_radix_arg(args[1]);
	char buf[128];
	char *str = uint2str(buf, sizeof(buf), num, radix);
	if(neg && str > &buf[0])
		*(--str) = '-';
	return string_from_cstr(str);
}

BUILTIN("string->number", string_2_number)
{
	if(nargs < 1 || nargs > 2)
		argcount(nargs, 2);
	char *str = tostring(args[0]);
	value_t n;
	unsigned long radix = 0;
	if(nargs == 2)
		radix = get_radix_arg(args[1]);
	if(!isnumtok_base(str, &n, (int)radix))
		return FL_F;
	return n;
}

BUILTIN("string.isutf8", string_isutf8)
{
	argcount(nargs, 1);
	char *s = tostring(args[0]);
	size_t len = cv_len((cvalue_t*)ptr(args[0]));
	return u8_isvalid(s, len) ? FL_T : FL_F;
}