glsl/pp: Add support for user-defined macros.
[mesa.git] / src / glsl / pp / sl_pp_context.c
1 /**************************************************************************
2 *
3 * Copyright 2009 VMware, Inc.
4 * All Rights Reserved.
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the
8 * "Software"), to deal in the Software without restriction, including
9 * without limitation the rights to use, copy, modify, merge, publish,
10 * distribute, sub license, and/or sell copies of the Software, and to
11 * permit persons to whom the Software is furnished to do so, subject to
12 * the following conditions:
13 *
14 * The above copyright notice and this permission notice (including the
15 * next paragraph) shall be included in all copies or substantial portions
16 * of the Software.
17 *
18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21 * IN NO EVENT SHALL TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR
22 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25 *
26 **************************************************************************/
27
28 #include <stdlib.h>
29 #include <string.h>
30 #include "sl_pp_public.h"
31 #include "sl_pp_context.h"
32
33
34 struct sl_pp_context *
35 sl_pp_context_create(void)
36 {
37 struct sl_pp_context *context;
38
39 context = calloc(1, sizeof(struct sl_pp_context));
40 if (!context) {
41 return NULL;
42 }
43
44 if (sl_pp_dict_init(context)) {
45 sl_pp_context_destroy(context);
46 return NULL;
47 }
48
49 context->getc_buf = malloc(64 * sizeof(char));
50 if (!context->getc_buf) {
51 sl_pp_context_destroy(context);
52 return NULL;
53 }
54
55 context->macro_tail = &context->macro;
56 context->if_ptr = SL_PP_MAX_IF_NESTING;
57 context->if_value = 1;
58 memset(context->error_msg, 0, sizeof(context->error_msg));
59 context->line = 1;
60 context->file = 0;
61
62 return context;
63 }
64
65 void
66 sl_pp_context_destroy(struct sl_pp_context *context)
67 {
68 if (context) {
69 free(context->cstr_pool);
70 sl_pp_macro_free(context->macro);
71 free(context->getc_buf);
72 free(context);
73 }
74 }
75
76 const char *
77 sl_pp_context_error_message(const struct sl_pp_context *context)
78 {
79 return context->error_msg;
80 }
81
82 int
83 sl_pp_context_add_predefined(struct sl_pp_context *context,
84 const char *name,
85 const char *value)
86 {
87 struct sl_pp_predefined pre;
88
89 if (context->num_predefined == SL_PP_MAX_PREDEFINED) {
90 return -1;
91 }
92
93 pre.name = sl_pp_context_add_unique_str(context, name);
94 if (pre.name == -1) {
95 return -1;
96 }
97
98 pre.value = sl_pp_context_add_unique_str(context, value);
99 if (pre.value == -1) {
100 return -1;
101 }
102
103 context->predefined[context->num_predefined++] = pre;
104 return 0;
105 }
106
107 int
108 sl_pp_context_add_unique_str(struct sl_pp_context *context,
109 const char *str)
110 {
111 unsigned int size;
112 unsigned int offset = 0;
113
114 size = strlen(str) + 1;
115
116 /* Find out if this is a unique string. */
117 while (offset < context->cstr_pool_len) {
118 const char *str2;
119 unsigned int size2;
120
121 str2 = &context->cstr_pool[offset];
122 size2 = strlen(str2) + 1;
123 if (size == size2 && !memcmp(str, str2, size - 1)) {
124 return offset;
125 }
126
127 offset += size2;
128 }
129
130 if (context->cstr_pool_len + size > context->cstr_pool_max) {
131 context->cstr_pool_max = (context->cstr_pool_len + size + 0xffff) & ~0xffff;
132 context->cstr_pool = realloc(context->cstr_pool, context->cstr_pool_max);
133 }
134
135 if (!context->cstr_pool) {
136 strcpy(context->error_msg, "out of memory");
137 return -1;
138 }
139
140 offset = context->cstr_pool_len;
141 memcpy(&context->cstr_pool[offset], str, size);
142 context->cstr_pool_len += size;
143
144 return offset;
145 }
146
147 const char *
148 sl_pp_context_cstr(const struct sl_pp_context *context,
149 int offset)
150 {
151 if (offset == -1) {
152 return NULL;
153 }
154 return &context->cstr_pool[offset];
155 }