1 /* -*- c++ -*- */
2 /*
3 * Copyright © 2010 Intel Corporation
4 *
5 * Permission is hereby granted, free of charge, to any person obtaining a
6 * copy of this software and associated documentation files (the "Software"),
7 * to deal in the Software without restriction, including without limitation
8 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9 * and/or sell copies of the Software, and to permit persons to whom the
10 * Software is furnished to do so, subject to the following conditions:
11 *
12 * The above copyright notice and this permission notice (including the next
13 * paragraph) shall be included in all copies or substantial portions of the
14 * Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22 * DEALINGS IN THE SOFTWARE.
23 */
24
25 #include <assert.h>
26 #include <limits>
27 #include "s_expression.h"
28
s_symbol(const char * str,size_t n)29 s_symbol::s_symbol(const char *str, size_t n)
30 {
31 /* Assume the given string is already nul-terminated and in memory that
32 * will live as long as this node.
33 */
34 assert(str[n] == '\0');
35 this->str = str;
36 }
37
s_list()38 s_list::s_list()
39 {
40 }
41
42 static void
skip_whitespace(const char * & src,char * & symbol_buffer)43 skip_whitespace(const char *&src, char *&symbol_buffer)
44 {
45 size_t n = strspn(src, " \v\t\r\n");
46 src += n;
47 symbol_buffer += n;
48 /* Also skip Scheme-style comments: semi-colon 'til end of line */
49 if (src[0] == ';') {
50 n = strcspn(src, "\n");
51 src += n;
52 symbol_buffer += n;
53 skip_whitespace(src, symbol_buffer);
54 }
55 }
56
57 static s_expression *
read_atom(void * ctx,const char * & src,char * & symbol_buffer)58 read_atom(void *ctx, const char *&src, char *&symbol_buffer)
59 {
60 s_expression *expr = NULL;
61
62 skip_whitespace(src, symbol_buffer);
63
64 size_t n = strcspn(src, "( \v\t\r\n);");
65 if (n == 0)
66 return NULL; // no atom
67
68 // Check for the special symbol '+INF', which means +Infinity. Note: C99
69 // requires strtod to parse '+INF' as +Infinity, but we still support some
70 // non-C99-compliant compilers (e.g. MSVC).
71 if (n == 4 && strncmp(src, "+INF", 4) == 0) {
72 expr = new(ctx) s_float(std::numeric_limits<float>::infinity());
73 } else {
74 // Check if the atom is a number.
75 char *float_end = NULL;
76 double f = glsl_strtod(src, &float_end);
77 if (float_end != src) {
78 char *int_end = NULL;
79 int i = strtol(src, &int_end, 10);
80 // If strtod matched more characters, it must have a decimal part
81 if (float_end > int_end)
82 expr = new(ctx) s_float(f);
83 else
84 expr = new(ctx) s_int(i);
85 } else {
86 // Not a number; return a symbol.
87 symbol_buffer[n] = '\0';
88 expr = new(ctx) s_symbol(symbol_buffer, n);
89 }
90 }
91
92 src += n;
93 symbol_buffer += n;
94
95 return expr;
96 }
97
98 static s_expression *
__read_expression(void * ctx,const char * & src,char * & symbol_buffer)99 __read_expression(void *ctx, const char *&src, char *&symbol_buffer)
100 {
101 s_expression *atom = read_atom(ctx, src, symbol_buffer);
102 if (atom != NULL)
103 return atom;
104
105 skip_whitespace(src, symbol_buffer);
106 if (src[0] == '(') {
107 ++src;
108 ++symbol_buffer;
109
110 s_list *list = new(ctx) s_list;
111 s_expression *expr;
112
113 while ((expr = __read_expression(ctx, src, symbol_buffer)) != NULL) {
114 list->subexpressions.push_tail(expr);
115 }
116 skip_whitespace(src, symbol_buffer);
117 if (src[0] != ')') {
118 printf("Unclosed expression (check your parenthesis).\n");
119 return NULL;
120 }
121 ++src;
122 ++symbol_buffer;
123 return list;
124 }
125 return NULL;
126 }
127
128 s_expression *
read_expression(void * ctx,const char * & src)129 s_expression::read_expression(void *ctx, const char *&src)
130 {
131 assert(src != NULL);
132
133 /* When we encounter a Symbol, we need to save a nul-terminated copy of
134 * the string. However, ralloc_strndup'ing every individual Symbol is
135 * extremely expensive. We could avoid this by simply overwriting the
136 * next character (guaranteed to be whitespace, parens, or semicolon) with
137 * a nul-byte. But overwriting non-whitespace would mess up parsing.
138 *
139 * So, just copy the whole buffer ahead of time. Walk both, leaving the
140 * original source string unmodified, and altering the copy to contain the
141 * necessary nul-bytes whenever we encounter a symbol.
142 */
143 char *symbol_buffer = ralloc_strdup(ctx, src);
144 return __read_expression(ctx, src, symbol_buffer);
145 }
146
print()147 void s_int::print()
148 {
149 printf("%d", this->val);
150 }
151
print()152 void s_float::print()
153 {
154 printf("%f", this->val);
155 }
156
print()157 void s_symbol::print()
158 {
159 printf("%s", this->str);
160 }
161
print()162 void s_list::print()
163 {
164 printf("(");
165 foreach_iter(exec_list_iterator, it, this->subexpressions) {
166 s_expression *expr = (s_expression*) it.get();
167 expr->print();
168 if (!expr->next->is_tail_sentinel())
169 printf(" ");
170 }
171 printf(")");
172 }
173
174 // --------------------------------------------------
175
176 bool
match(s_expression * expr)177 s_pattern::match(s_expression *expr)
178 {
179 switch (type)
180 {
181 case EXPR: *p_expr = expr; break;
182 case LIST: if (expr->is_list()) *p_list = (s_list *) expr; break;
183 case SYMBOL: if (expr->is_symbol()) *p_symbol = (s_symbol *) expr; break;
184 case NUMBER: if (expr->is_number()) *p_number = (s_number *) expr; break;
185 case INT: if (expr->is_int()) *p_int = (s_int *) expr; break;
186 case STRING:
187 s_symbol *sym = SX_AS_SYMBOL(expr);
188 if (sym != NULL && strcmp(sym->value(), literal) == 0)
189 return true;
190 return false;
191 };
192
193 return *p_expr == expr;
194 }
195
196 bool
s_match(s_expression * top,unsigned n,s_pattern * pattern,bool partial)197 s_match(s_expression *top, unsigned n, s_pattern *pattern, bool partial)
198 {
199 s_list *list = SX_AS_LIST(top);
200 if (list == NULL)
201 return false;
202
203 unsigned i = 0;
204 foreach_iter(exec_list_iterator, it, list->subexpressions) {
205 if (i >= n)
206 return partial; /* More actual items than the pattern expected */
207
208 s_expression *expr = (s_expression *) it.get();
209 if (expr == NULL || !pattern[i].match(expr))
210 return false;
211
212 i++;
213 }
214
215 if (i < n)
216 return false; /* Less actual items than the pattern expected */
217
218 return true;
219 }
220