From 26bc383ef8fd3b0a7b210bcfe0bd3bbccdee212b Mon Sep 17 00:00:00 2001 From: nobu Date: Tue, 12 Aug 2008 10:07:08 +0000 Subject: * string.c (rb_str_drop): new function to drop first bytes. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@18540 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- string.c | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) (limited to 'string.c') diff --git a/string.c b/string.c index 9262dceea1..fbe6ea88bb 100644 --- a/string.c +++ b/string.c @@ -2829,9 +2829,43 @@ rb_str_aref_m(int argc, VALUE *argv, VALUE str) return rb_str_aref(str, argv[0]); } +VALUE +rb_str_drop(VALUE str, long len) +{ + char *ptr = RSTRING_PTR(str); + long olen = RSTRING_LEN(str), nlen; + + str_modifiable(str); + if (len > olen) len = olen; + nlen = olen - len; + if (nlen <= RSTRING_EMBED_LEN_MAX) { + char *oldptr = ptr; + int fl = (RBASIC(str)->flags & (STR_NOEMBED|ELTS_SHARED)); + STR_SET_EMBED(str); + STR_SET_EMBED_LEN(str, nlen); + ptr = RSTRING(str)->as.ary; + memcpy(ptr, oldptr + len, nlen); + if (fl == STR_NOEMBED) xfree(oldptr); + } + else { + if (!STR_SHARED_P(str)) rb_str_new4(str); + ptr = RSTRING(str)->as.heap.ptr += len; + RSTRING(str)->as.heap.len = nlen; + } + ptr[nlen] = 0; + ENC_CODERANGE_CLEAR(str); + return str; +} + static void rb_str_splice_0(VALUE str, long beg, long len, VALUE val) { + if (beg == 0 && RSTRING_LEN(val) == 0) { + rb_str_drop(str, len); + OBJ_INFECT(str, val); + return; + } + rb_str_modify(str); if (len < RSTRING_LEN(val)) { /* expand string */ -- cgit v1.2.3