Diff: string.cpp

Revision:

0:4ab1392a0142

diff -r 000000000000 -r 4ab1392a0142 string.cpp
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/string.cpp	Sun Oct 07 11:49:09 2012 +0000
@@ -0,0 +1,165 @@
+#include "string.h"
+
+tp_obj tp_string_t(TP, int n) {
+    tp_obj r = tp_string_n(0,n);
+    r.string.info = (_tp_string*)tp_malloc(sizeof(_tp_string)+n);
+    r.string.val = r.string.info->s;
+    return r;
+}
+
+tp_obj tp_printf(TP, char const *fmt,...) {
+    int l;
+    tp_obj r;
+    char *s;
+    va_list arg;
+    va_start(arg, fmt);
+    l = vsnprintf(NULL, 0, fmt,arg);
+    r = tp_string_t(tp,l);
+    s = r.string.info->s;
+    va_end(arg);
+    va_start(arg, fmt);
+    vsprintf(s,fmt,arg);
+    va_end(arg);
+    return tp_track(tp,r);
+}
+
+int _tp_str_index(tp_obj s, tp_obj k) {
+    int i=0;
+    while ((s.string.len - i) >= k.string.len) {
+        if (memcmp(s.string.val+i,k.string.val,k.string.len) == 0) {
+            return i;
+        }
+        i += 1;
+    }
+    return -1;
+}
+
+tp_obj tp_join(TP) {
+    tp_obj delim = TP_OBJ();
+    tp_obj val = TP_OBJ();
+    int l=0,i;
+    tp_obj r;
+    char *s;
+    for (i=0; i<val.list.val->len; i++) {
+        if (i!=0) { l += delim.string.len; }
+        l += tp_str(tp,val.list.val->items[i]).string.len;
+    }
+    r = tp_string_t(tp,l);
+    s = r.string.info->s;
+    l = 0;
+    for (i=0; i<val.list.val->len; i++) {
+        tp_obj e;
+        if (i!=0) {
+            memcpy(s+l,delim.string.val,delim.string.len); l += delim.string.len;
+        }
+        e = tp_str(tp,val.list.val->items[i]);
+        memcpy(s+l,e.string.val,e.string.len); l += e.string.len;
+    }
+    return tp_track(tp,r);
+}
+
+tp_obj tp_string_slice(TP,tp_obj s, int a, int b) {
+    tp_obj r = tp_string_t(tp,b-a);
+    char *m = r.string.info->s;
+    memcpy(m,s.string.val+a,b-a);
+    return tp_track(tp,r);
+}
+
+tp_obj tp_split(TP) 
+{
+    tp_obj v = TP_OBJ();
+    tp_obj d = TP_OBJ();
+    tp_obj r = tp_list(tp);
+
+    int i;
+    while ((i=_tp_str_index(v,d))!=-1) {
+        _tp_list_append(tp,r.list.val,tp_string_slice(tp,v,0,i));
+        v.string.val += i + d.string.len; v.string.len -= i + d.string.len;
+/*         tp_grey(tp,r); // should stop gc or something instead*/
+    }
+    _tp_list_append(tp,r.list.val,tp_string_slice(tp,v,0,v.string.len));
+/*     tp_grey(tp,r); // should stop gc or something instead*/
+    return r;
+}
+
+
+tp_obj tp_find(TP) {
+    tp_obj s = TP_OBJ();
+    tp_obj v = TP_OBJ();
+    return tp_number(_tp_str_index(s,v));
+}
+
+tp_obj tp_str_index(TP) {
+    tp_obj s = TP_OBJ();
+    tp_obj v = TP_OBJ();
+    int n = _tp_str_index(s,v);
+    if (n >= 0) { return tp_number(n); }
+    tp_raise(tp_None,"tp_str_index(%s,%s)",s,v);
+}
+
+tp_obj tp_str2(TP) {
+    tp_obj v = TP_OBJ();
+    return tp_str(tp,v);
+}
+
+tp_obj tp_chr(TP) {
+    int v = TP_NUM();
+    return tp_string_n(tp->chars[(unsigned char)v],1);
+}
+tp_obj tp_ord(TP) {
+    char const *s = TP_STR();
+    return tp_number((unsigned char)s[0]);
+}
+
+tp_obj tp_strip(TP) {
+    char const *v = TP_STR();
+    int i, l = strlen(v); int a = l, b = 0;
+    tp_obj r;
+    char *s;
+    for (i=0; i<l; i++) {
+        if (v[i] != ' ' && v[i] != '\n' && v[i] != '\t' && v[i] != '\r') {
+            a = _tp_min(a,i); b = _tp_max(b,i+1);
+        }
+    }
+    if ((b-a) < 0) { return tp_string(""); }
+    r = tp_string_t(tp,b-a);
+    s = r.string.info->s;
+    memcpy(s,v+a,b-a);
+    return tp_track(tp,r);
+}
+
+
+tp_obj tp_replace(TP) {
+    tp_obj s = TP_OBJ();
+    tp_obj k = TP_OBJ();
+    tp_obj v = TP_OBJ();
+    tp_obj p = s;
+    int i,n = 0;
+    int c;
+    int l;
+    tp_obj rr;
+    char *r;
+    char *d;
+    tp_obj z;
+    while ((i = _tp_str_index(p,k)) != -1) {
+        n += 1;
+        p.string.val += i + k.string.len; p.string.len -= i + k.string.len;
+    }
+/*     fprintf(stderr,"ns: %d\n",n); */
+    l = s.string.len + n * (v.string.len-k.string.len);
+    rr = tp_string_t(tp,l);
+    r = rr.string.info->s;
+    d = r;
+    z = p = s;
+    while ((i = _tp_str_index(p,k)) != -1) {
+        p.string.val += i; p.string.len -= i;
+        memcpy(d,z.string.val,c=(p.string.val-z.string.val)); d += c;
+        p.string.val += k.string.len; p.string.len -= k.string.len;
+        memcpy(d,v.string.val,v.string.len); d += v.string.len;
+        z = p;
+    }
+    memcpy(d,z.string.val,(s.string.val + s.string.len) - z.string.val);
+
+    return tp_track(tp,rr);
+}
+