ref: a69ab620c009be7b32f8c881d0068ce0c52c52fb
dir: /string.c/
/*
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;
}