]> git.wincent.com - wikitext.git/blob - ext/str.c
Change 4 VALUE (String) members of the parser_t struct to str_t 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 str_t *str_new(void)
27 {
28     str_t *str      = ALLOC_N(str_t, 1);
29     str->ptr        = NULL;
30     str->len        = 0;
31     str->capacity   = 0;
32     return str;
33 }
34
35 str_t *str_new_size(long len)
36 {
37     str_t *str      = ALLOC_N(str_t, 1);
38     str->ptr        = ALLOC_N(char, len);
39     str->len        = 0;
40     str->capacity   = len;
41     return str;
42 }
43
44 str_t *str_new_copy(const char *src, long len)
45 {
46     str_t *str      = ALLOC_N(str_t, 1);
47     str->ptr        = ALLOC_N(char, len);
48     memcpy(str->ptr, src, len);
49     str->len        = len;
50     str->capacity   = len;
51     return str;
52 }
53
54 str_t *str_new_no_copy(char *src, long len)
55 {
56     str_t *str      = ALLOC_N(str_t, 1);
57     str->ptr        = src;
58     str->len        = len;
59     str->capacity   = len;
60     return str;
61 }
62
63 str_t *str_new_from_string(VALUE string)
64 {
65     string = StringValue(string);
66     return str_new_copy(RSTRING_PTR(string), RSTRING_LEN(string));
67 }
68
69 VALUE string_from_str(str_t *str)
70 {
71     return rb_str_new(str->ptr, str->len);
72 }
73
74 void str_grow(str_t *str, long len)
75 {
76     if (str->capacity < len)
77     {
78         if (str->ptr)
79             REALLOC_N(str->ptr, char, len);
80         else
81             str->ptr = ALLOC_N(char, len);
82         str->capacity = len;
83     }
84 }
85
86 // try overallocating once we've switched more things over to str_t
87 // ie. allocate 32 bytes too much, 64 bytes etc
88 // and run benchmarks to see if it runs faster
89 // and other tests to count the number of allocations
90 void str_append(str_t *str, const char *src, long len)
91 {
92     long new_len = str->len + len;
93     if (str->capacity < new_len)
94     {
95         if (str->ptr)
96             REALLOC_N(str->ptr, char, new_len);
97         else
98             str->ptr = ALLOC_N(char, new_len);
99         str->capacity = new_len;
100     }
101     memcpy(str->ptr + str->len, src, len);
102     str->len = new_len;
103 }
104
105 void str_append_str(str_t *str, str_t *other)
106 {
107     str_append(str, other->ptr, other->len);
108 }
109
110 void str_append_string(str_t *str, VALUE other)
111 {
112     str_append(str, RSTRING_PTR(other), RSTRING_LEN(other));
113 }
114
115 void str_swap(str_t **a, str_t **b)
116 {
117     str_t *c;
118     c = *a;
119     *a = *b;
120     *b = c;
121 }
122
123 void str_clear(str_t *str)
124 {
125     str->len = 0;
126 }
127
128 void str_free(str_t *str)
129 {
130     if (str->ptr)
131         free(str->ptr);
132     free(str);
133 }