src/mesa/shader/slang/slang_compile_operation.c

   1 /*
   2  * Mesa 3-D graphics library
   3  * Version:  6.5.2
   4  *
   5  * Copyright (C) 2005-2006  Brian Paul   All Rights Reserved.
   6  *
   7  * Permission is hereby granted, free of charge, to any person obtaining a
   8  * copy of this software and associated documentation files (the "Software"),
   9  * to deal in the Software without restriction, including without limitation
  10  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  11  * and/or sell copies of the Software, and to permit persons to whom the
  12  * Software is furnished to do so, subject to the following conditions:
  13  *
  14  * The above copyright notice and this permission notice shall be included
  15  * in all copies or substantial portions of the Software.
  16  *
  17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  18  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  19  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  20  * BRIAN PAUL BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN
  21  * AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  22  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  23  */
  24
  25 /**
  26  * \file slang_compile_operation.c
  27  * slang front-end compiler
  28  * \author Michal Krol
  29  */
  30
  31 #include "main/imports.h"
  32 #include "slang_compile.h"
  33 #include "slang_mem.h"
  34
  35
  36 /**
  37  * Init a slang_operation object
  38  */
  39 GLboolean
  40 slang_operation_construct(slang_operation * oper)
  41 {
  42    oper->type = SLANG_OPER_NONE;
  43    oper->children = NULL;
  44    oper->num_children = 0;
  45    oper->literal[0] = 0.0;
  46    oper->literal_size = 1;
  47    oper->array_constructor = GL_FALSE;
  48    oper->a_id = SLANG_ATOM_NULL;
  49    oper->locals = _slang_variable_scope_new(NULL);
  50    if (oper->locals == NULL)
  51       return GL_FALSE;
  52    _slang_variable_scope_ctr(oper->locals);
  53    oper->fun = NULL;
  54    oper->var = NULL;
  55    return GL_TRUE;
  56 }
  57
  58 void
  59 slang_operation_destruct(slang_operation * oper)
  60 {
  61    GLuint i;
  62
  63    for (i = 0; i < oper->num_children; i++)
  64       slang_operation_destruct(oper->children + i);
  65    _slang_free(oper->children);
  66    slang_variable_scope_destruct(oper->locals);
  67    _slang_free(oper->locals);
  68    oper->children = NULL;
  69    oper->num_children = 0;
  70    oper->locals = NULL;
  71 }
  72
  73
  74 /**
  75  * Recursively traverse 'oper', replacing occurances of 'oldScope' with
  76  * 'newScope' in the oper->locals->outer_scope field.
  77  */
  78 void
  79 slang_replace_scope(slang_operation *oper,
  80                     slang_variable_scope *oldScope,
  81                     slang_variable_scope *newScope)
  82 {
  83    GLuint i;
  84
  85    if (oper->locals != newScope &&
  86        oper->locals->outer_scope == oldScope) {
  87       /* found.  replace old w/ new */
  88       oper->locals->outer_scope = newScope;
  89    }
  90
  91    if (oper->type == SLANG_OPER_VARIABLE_DECL) {
  92       /* search/replace in the initializer */
  93       slang_variable *var;
  94       var = _slang_variable_locate(oper->locals, oper->a_id, GL_TRUE);
  95       if (var && var->initializer) {
  96          slang_replace_scope(var->initializer, oldScope, newScope);
  97       }
  98    }
  99
 100    /* search/replace in children */
 101    for (i = 0; i < oper->num_children; i++) {
 102       slang_replace_scope(&oper->children[i], oldScope, newScope);
 103    }
 104 }
 105
 106
 107 /**
 108  * Recursively copy a slang_operation node.
 109  * \param x  copy target
 110  * \param y  copy source
 111  * \return GL_TRUE for success, GL_FALSE if failure
 112  */
 113 GLboolean
 114 slang_operation_copy(slang_operation * x, const slang_operation * y)
 115 {
 116    slang_operation z;
 117    GLuint i;
 118
 119    if (!slang_operation_construct(&z))
 120       return GL_FALSE;
 121    z.type = y->type;
 122    z.children = (slang_operation *)
 123       _slang_alloc(y->num_children * sizeof(slang_operation));
 124    if (z.children == NULL) {
 125       slang_operation_destruct(&z);
 126       return GL_FALSE;
 127    }
 128    for (z.num_children = 0; z.num_children < y->num_children;
 129         z.num_children++) {
 130       if (!slang_operation_construct(&z.children[z.num_children])) {
 131          slang_operation_destruct(&z);
 132          return GL_FALSE;
 133       }
 134    }
 135    for (i = 0; i < z.num_children; i++) {
 136       if (!slang_operation_copy(&z.children[i], &y->children[i])) {
 137          slang_operation_destruct(&z);
 138          return GL_FALSE;
 139       }
 140    }
 141    z.literal[0] = y->literal[0];
 142    z.literal[1] = y->literal[1];
 143    z.literal[2] = y->literal[2];
 144    z.literal[3] = y->literal[3];
 145    z.literal_size = y->literal_size;
 146    assert(y->literal_size >= 1);
 147    assert(y->literal_size <= 4);
 148    z.a_id = y->a_id;
 149    if (y->locals) {
 150       if (!slang_variable_scope_copy(z.locals, y->locals)) {
 151          slang_operation_destruct(&z);
 152          return GL_FALSE;
 153       }
 154    }
 155 #if 0
 156    z.var = y->var;
 157    z.fun = y->fun;
 158 #endif
 159    slang_operation_destruct(x);
 160    *x = z;
 161
 162    /* If this operation declares a new scope, we need to make sure
 163     * all children point to it, not the original operation's scope!
 164     */
 165    if (x->type == SLANG_OPER_BLOCK_NEW_SCOPE ||
 166        x->type == SLANG_OPER_WHILE ||
 167        x->type == SLANG_OPER_FOR) {
 168       slang_replace_scope(x, y->locals, x->locals);
 169    }
 170
 171    return GL_TRUE;
 172 }
 173
 174
 175 slang_operation *
 176 slang_operation_new(GLuint count)
 177 {
 178    slang_operation *ops
 179        = (slang_operation *) _slang_alloc(count * sizeof(slang_operation));
 180    assert(count > 0);
 181    if (ops) {
 182       GLuint i;
 183       for (i = 0; i < count; i++)
 184          slang_operation_construct(ops + i);
 185    }
 186    return ops;
 187 }
 188
 189
 190 /**
 191  * Delete operation and all children
 192  */
 193 void
 194 slang_operation_delete(slang_operation *oper)
 195 {
 196    slang_operation_destruct(oper);
 197    _slang_free(oper);
 198 }
 199
 200
 201 slang_operation *
 202 slang_operation_grow(GLuint *numChildren, slang_operation **children)
 203 {
 204    slang_operation *ops;
 205
 206    ops = (slang_operation *)
 207       _slang_realloc(*children,
 208                      *numChildren * sizeof(slang_operation),
 209                      (*numChildren + 1) * sizeof(slang_operation));
 210    if (ops) {
 211       slang_operation *newOp = ops + *numChildren;
 212       if (!slang_operation_construct(newOp)) {
 213          _slang_free(ops);
 214          *children = NULL;
 215          return NULL;
 216       }
 217       *children = ops;
 218       (*numChildren)++;
 219       return newOp;
 220    }
 221    return NULL;
 222 }
 223
 224 /**
 225  * Insert a new slang_operation into an array.
 226  * \param numElements  pointer to current array size (in/out)
 227  * \param array  address of the array (in/out)
 228  * \param pos  position to insert new element
 229  * \return  pointer to the new operation/element
 230  */
 231 slang_operation *
 232 slang_operation_insert(GLuint *numElements, slang_operation **array,
 233                        GLuint pos)
 234 {
 235    slang_operation *ops;
 236
 237    assert(pos <= *numElements);
 238
 239    ops = (slang_operation *)
 240       _slang_alloc((*numElements + 1) * sizeof(slang_operation));
 241    if (ops) {
 242       slang_operation *newOp;
 243       newOp = ops + pos;
 244       if (pos > 0)
 245          _mesa_memcpy(ops, *array, pos * sizeof(slang_operation));
 246       if (pos < *numElements)
 247          _mesa_memcpy(newOp + 1, (*array) + pos,
 248                       (*numElements - pos) * sizeof(slang_operation));
 249
 250       if (!slang_operation_construct(newOp)) {
 251          _slang_free(ops);
 252          *numElements = 0;
 253          *array = NULL;
 254          return NULL;
 255       }
 256       if (*array)
 257          _slang_free(*array);
 258       *array = ops;
 259       (*numElements)++;
 260       return newOp;
 261    }
 262    return NULL;
 263 }
 264
 265
 266 void
 267 _slang_operation_swap(slang_operation *oper0, slang_operation *oper1)
 268 {
 269    slang_operation tmp = *oper0;
 270    *oper0 = *oper1;
 271    *oper1 = tmp;
 272 }
 273
 274