]> git.wincent.com - wikitext.git/blob - ext/str.c
Change emit parameter of _Wikitext_dedent to bool type
[wikitext.git] / ext / str.c
1 // Copyright 2008-2009 Wincent Colaiuta. All rights reserved.
2 //
3 // Redistribution and use in source and binary forms, with or without
4 // modification, are permitted provided that the following conditions are met:
5 //
6 // 1. Redistributions of source code must retain the above copyright notice,
7 //    this list of conditions and the following disclaimer.
8 // 2. Redistributions in binary form must reproduce the above copyright notice,
9 //    this list of conditions and the following disclaimer in the documentation
10 //    and/or other materials provided with the distribution.
11 //
12 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
13 // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
14 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
15 // ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
16 // LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
17 // CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
18 // SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
19 // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
20 // CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
21 // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
22 // POSSIBILITY OF SUCH DAMAGE.
23
24 #include "str.h"
25
26 // when allocating memory, reserve a little more than was asked for,
27 // which can help to avoid subsequent allocations
28 #define STR_OVERALLOC 256
29
30 str_t *str_new(void)
31 {
32     str_t *str      = ALLOC_N(str_t, 1);
33     str->ptr        = NULL;
34     str->len        = 0;
35     str->capacity   = 0;
36     return str;
37 }
38
39 str_t *str_new_copy(const char *src, long len)
40 {
41     str_t *str      = ALLOC_N(str_t, 1);
42     str->ptr        = ALLOC_N(char, len + STR_OVERALLOC);
43     memcpy(str->ptr, src, len);
44     str->len        = len;
45     str->capacity   = len + STR_OVERALLOC;
46     return str;
47 }
48
49 str_t *str_new_from_string(VALUE string)
50 {
51     string = StringValue(string);
52     return str_new_copy(RSTRING_PTR(string), RSTRING_LEN(string));
53 }
54
55 VALUE string_from_str(str_t *str)
56 {
57     return rb_str_new(str->ptr, str->len);
58 }
59
60 void str_grow(str_t *str, long len)
61 {
62     if (str->capacity < len)
63     {
64         if (str->ptr)
65             REALLOC_N(str->ptr, char, len + STR_OVERALLOC);
66         else
67             str->ptr = ALLOC_N(char, len + STR_OVERALLOC);
68         str->capacity = len + STR_OVERALLOC;
69     }
70 }
71
72 void str_append(str_t *str, const char *src, long len)
73 {
74     long new_len = str->len + len;
75     if (str->capacity < new_len)
76     {
77         if (str->ptr)
78             REALLOC_N(str->ptr, char, new_len + STR_OVERALLOC);
79         else
80             str->ptr = ALLOC_N(char, new_len + STR_OVERALLOC);
81         str->capacity = new_len + STR_OVERALLOC;
82     }
83     memcpy(str->ptr + str->len, src, len);
84     str->len = new_len;
85 }
86
87 void str_append_str(str_t *str, str_t *other)
88 {
89     str_append(str, other->ptr, other->len);
90 }
91
92 void str_append_string(str_t *str, VALUE other)
93 {
94     str_append(str, RSTRING_PTR(other), RSTRING_LEN(other));
95 }
96
97 void str_clear(str_t *str)
98 {
99     str->len = 0;
100 }
101
102 void str_free(str_t *str)
103 {
104     if (str->ptr)
105         free(str->ptr);
106     free(str);
107 }