92 lines
1.7 KiB
C
92 lines
1.7 KiB
C
#include "picrin.h"
|
|
#include "picrin/string.h"
|
|
|
|
void
|
|
pic_str_set(pic_state *pic, pic_str *str, size_t i, char c)
|
|
{
|
|
pic_str *x, *y, *z, *tmp;
|
|
|
|
if (pic_strlen(str) <= i) {
|
|
pic_errorf(pic, "index out of range %d", i);
|
|
}
|
|
|
|
x = pic_substr(pic, str, 0, i);
|
|
y = pic_make_str_fill(pic, 1, c);
|
|
z = pic_substr(pic, str, i + 1, pic_strlen(str));
|
|
|
|
tmp = pic_strcat(pic, x, pic_strcat(pic, y, z));
|
|
|
|
XROPE_INCREF(tmp->rope);
|
|
XROPE_DECREF(str->rope);
|
|
str->rope = tmp->rope;
|
|
}
|
|
|
|
static pic_value
|
|
pic_str_string_set(pic_state *pic)
|
|
{
|
|
pic_str *str;
|
|
char c;
|
|
int k;
|
|
|
|
pic_get_args(pic, "sic", &str, &k, &c);
|
|
|
|
pic_str_set(pic, str, k, c);
|
|
return pic_none_value();
|
|
}
|
|
|
|
static pic_value
|
|
pic_str_string_copy_ip(pic_state *pic)
|
|
{
|
|
pic_str *to, *from;
|
|
int n, at, start, end;
|
|
|
|
n = pic_get_args(pic, "sis|ii", &to, &at, &from, &start, &end);
|
|
|
|
switch (n) {
|
|
case 3:
|
|
start = 0;
|
|
case 4:
|
|
end = pic_strlen(from);
|
|
}
|
|
if (to == from) {
|
|
from = pic_substr(pic, from, 0, end);
|
|
}
|
|
|
|
while (start < end) {
|
|
pic_str_set(pic, to, at++, pic_str_ref(pic, from, start++));
|
|
}
|
|
return pic_none_value();
|
|
}
|
|
|
|
static pic_value
|
|
pic_str_string_fill_ip(pic_state *pic)
|
|
{
|
|
pic_str *str;
|
|
char c;
|
|
int n, start, end;
|
|
|
|
n = pic_get_args(pic, "sc|ii", &str, &c, &start, &end);
|
|
|
|
switch (n) {
|
|
case 2:
|
|
start = 0;
|
|
case 3:
|
|
end = pic_strlen(str);
|
|
}
|
|
|
|
while (start < end) {
|
|
pic_str_set(pic, str, start++, c);
|
|
}
|
|
return pic_none_value();
|
|
}
|
|
|
|
void
|
|
pic_init_mutable_string(pic_state *pic)
|
|
{
|
|
pic_deflibrary (pic, "(picrin string)") {
|
|
pic_defun(pic, "string-set!", pic_str_string_set);
|
|
pic_defun(pic, "string-copy!", pic_str_string_copy_ip);
|
|
pic_defun(pic, "string-fill!", pic_str_string_fill_ip);
|
|
}
|
|
}
|