src/glsl/ir_clone.cpp

   1 /*
   2  * Copyright © 2010 Intel Corporation
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  21  * DEALINGS IN THE SOFTWARE.
  22  */
  23
  24 #include <string.h>
  25 #include "ir.h"
  26 #include "glsl_types.h"
  27 extern "C" {
  28 #include "program/hash_table.h"
  29 }
  30
  31 /**
  32  * Duplicate an IR variable
  33  *
  34  * \note
  35  * This will probably be made \c virtual and moved to the base class
  36  * eventually.
  37  */
  38 ir_variable *
  39 ir_variable::clone(void *mem_ctx, struct hash_table *ht) const
  40 {
  41    ir_variable *var = new(mem_ctx) ir_variable(this->type, this->name,
  42                                                (ir_variable_mode) this->mode);
  43
  44    var->max_array_access = this->max_array_access;
  45    var->read_only = this->read_only;
  46    var->centroid = this->centroid;
  47    var->invariant = this->invariant;
  48    var->interpolation = this->interpolation;
  49    var->array_lvalue = this->array_lvalue;
  50    var->location = this->location;
  51    var->warn_extension = this->warn_extension;
  52    var->origin_upper_left = this->origin_upper_left;
  53    var->pixel_center_integer = this->pixel_center_integer;
  54
  55    if (this->constant_value)
  56       var->constant_value = this->constant_value->clone(mem_ctx, ht);
  57
  58    if (ht) {
  59       hash_table_insert(ht, var, (void *)const_cast<ir_variable *>(this));
  60    }
  61
  62    return var;
  63 }
  64
  65 ir_swizzle *
  66 ir_swizzle::clone(void *mem_ctx, struct hash_table *ht) const
  67 {
  68    return new(mem_ctx) ir_swizzle(this->val->clone(mem_ctx, ht), this->mask);
  69 }
  70
  71 ir_return *
  72 ir_return::clone(void *mem_ctx, struct hash_table *ht) const
  73 {
  74    ir_rvalue *new_value = NULL;
  75
  76    if (this->value)
  77       new_value = this->value->clone(mem_ctx, ht);
  78
  79    return new(mem_ctx) ir_return(new_value);
  80 }
  81
  82 ir_discard *
  83 ir_discard::clone(void *mem_ctx, struct hash_table *ht) const
  84 {
  85    ir_rvalue *new_condition = NULL;
  86
  87    if (this->condition != NULL)
  88       new_condition = this->condition->clone(mem_ctx, ht);
  89
  90    return new(mem_ctx) ir_discard(new_condition);
  91 }
  92
  93 ir_loop_jump *
  94 ir_loop_jump::clone(void *mem_ctx, struct hash_table *ht) const
  95 {
  96    (void)ht;
  97
  98    return new(mem_ctx) ir_loop_jump(this->mode);
  99 }
 100
 101 ir_if *
 102 ir_if::clone(void *mem_ctx, struct hash_table *ht) const
 103 {
 104    ir_if *new_if = new(mem_ctx) ir_if(this->condition->clone(mem_ctx, ht));
 105
 106    foreach_iter(exec_list_iterator, iter, this->then_instructions) {
 107       ir_instruction *ir = (ir_instruction *)iter.get();
 108       new_if->then_instructions.push_tail(ir->clone(mem_ctx, ht));
 109    }
 110
 111    foreach_iter(exec_list_iterator, iter, this->else_instructions) {
 112       ir_instruction *ir = (ir_instruction *)iter.get();
 113       new_if->else_instructions.push_tail(ir->clone(mem_ctx, ht));
 114    }
 115
 116    return new_if;
 117 }
 118
 119 ir_loop *
 120 ir_loop::clone(void *mem_ctx, struct hash_table *ht) const
 121 {
 122    ir_loop *new_loop = new(mem_ctx) ir_loop();
 123
 124    if (this->from)
 125       new_loop->from = this->from->clone(mem_ctx, ht);
 126    if (this->to)
 127       new_loop->to = this->to->clone(mem_ctx, ht);
 128    if (this->increment)
 129       new_loop->increment = this->increment->clone(mem_ctx, ht);
 130    new_loop->counter = counter;
 131
 132    foreach_iter(exec_list_iterator, iter, this->body_instructions) {
 133       ir_instruction *ir = (ir_instruction *)iter.get();
 134       new_loop->body_instructions.push_tail(ir->clone(mem_ctx, ht));
 135    }
 136
 137    new_loop->cmp = this->cmp;
 138    return new_loop;
 139 }
 140
 141 ir_call *
 142 ir_call::clone(void *mem_ctx, struct hash_table *ht) const
 143 {
 144    if (this->type == glsl_type::error_type)
 145       return ir_call::get_error_instruction(mem_ctx);
 146
 147    exec_list new_parameters;
 148
 149    foreach_iter(exec_list_iterator, iter, this->actual_parameters) {
 150       ir_instruction *ir = (ir_instruction *)iter.get();
 151       new_parameters.push_tail(ir->clone(mem_ctx, ht));
 152    }
 153
 154    return new(mem_ctx) ir_call(this->callee, &new_parameters);
 155 }
 156
 157 ir_expression *
 158 ir_expression::clone(void *mem_ctx, struct hash_table *ht) const
 159 {
 160    ir_rvalue *op[2] = {NULL, NULL};
 161    unsigned int i;
 162
 163    for (i = 0; i < get_num_operands(); i++) {
 164       op[i] = this->operands[i]->clone(mem_ctx, ht);
 165    }
 166
 167    return new(mem_ctx) ir_expression(this->operation, this->type, op[0], op[1]);
 168 }
 169
 170 ir_dereference_variable *
 171 ir_dereference_variable::clone(void *mem_ctx, struct hash_table *ht) const
 172 {
 173    ir_variable *new_var;
 174
 175    if (ht) {
 176       new_var = (ir_variable *)hash_table_find(ht, this->var);
 177       if (!new_var)
 178          new_var = this->var;
 179    } else {
 180       new_var = this->var;
 181    }
 182
 183    return new(mem_ctx) ir_dereference_variable(new_var);
 184 }
 185
 186 ir_dereference_array *
 187 ir_dereference_array::clone(void *mem_ctx, struct hash_table *ht) const
 188 {
 189    return new(mem_ctx) ir_dereference_array(this->array->clone(mem_ctx, ht),
 190                                             this->array_index->clone(mem_ctx,
 191                                                                      ht));
 192 }
 193
 194 ir_dereference_record *
 195 ir_dereference_record::clone(void *mem_ctx, struct hash_table *ht) const
 196 {
 197    return new(mem_ctx) ir_dereference_record(this->record->clone(mem_ctx, ht),
 198                                              this->field);
 199 }
 200
 201 ir_texture *
 202 ir_texture::clone(void *mem_ctx, struct hash_table *ht) const
 203 {
 204    ir_texture *new_tex = new(mem_ctx) ir_texture(this->op);
 205    new_tex->type = this->type;
 206
 207    new_tex->sampler = this->sampler->clone(mem_ctx, ht);
 208    new_tex->coordinate = this->coordinate->clone(mem_ctx, ht);
 209    if (this->projector)
 210       new_tex->projector = this->projector->clone(mem_ctx, ht);
 211    if (this->shadow_comparitor) {
 212       new_tex->shadow_comparitor = this->shadow_comparitor->clone(mem_ctx, ht);
 213    }
 214
 215    for (int i = 0; i < 3; i++)
 216       new_tex->offsets[i] = this->offsets[i];
 217
 218    switch (this->op) {
 219    case ir_tex:
 220       break;
 221    case ir_txb:
 222       new_tex->lod_info.bias = this->lod_info.bias->clone(mem_ctx, ht);
 223       break;
 224    case ir_txl:
 225    case ir_txf:
 226       new_tex->lod_info.lod = this->lod_info.lod->clone(mem_ctx, ht);
 227       break;
 228    case ir_txd:
 229       new_tex->lod_info.grad.dPdx = this->lod_info.grad.dPdx->clone(mem_ctx, ht);
 230       new_tex->lod_info.grad.dPdy = this->lod_info.grad.dPdy->clone(mem_ctx, ht);
 231       break;
 232    }
 233
 234    return new_tex;
 235 }
 236
 237 ir_assignment *
 238 ir_assignment::clone(void *mem_ctx, struct hash_table *ht) const
 239 {
 240    ir_rvalue *new_condition = NULL;
 241
 242    if (this->condition)
 243       new_condition = this->condition->clone(mem_ctx, ht);
 244
 245    return new(mem_ctx) ir_assignment(this->lhs->clone(mem_ctx, ht),
 246                                      this->rhs->clone(mem_ctx, ht),
 247                                      new_condition,
 248                                      this->write_mask);
 249 }
 250
 251 ir_function *
 252 ir_function::clone(void *mem_ctx, struct hash_table *ht) const
 253 {
 254    ir_function *copy = new(mem_ctx) ir_function(this->name);
 255
 256    foreach_list_const(node, &this->signatures) {
 257       const ir_function_signature *const sig =
 258          (const ir_function_signature *const) node;
 259
 260       ir_function_signature *sig_copy = sig->clone(mem_ctx, ht);
 261       copy->add_signature(sig_copy);
 262
 263       if (ht != NULL)
 264          hash_table_insert(ht, sig_copy,
 265                            (void *)const_cast<ir_function_signature *>(sig));
 266    }
 267
 268    return copy;
 269 }
 270
 271 ir_function_signature *
 272 ir_function_signature::clone(void *mem_ctx, struct hash_table *ht) const
 273 {
 274    ir_function_signature *copy =
 275       new(mem_ctx) ir_function_signature(this->return_type);
 276
 277    copy->is_defined = this->is_defined;
 278    copy->is_builtin = this->is_builtin;
 279
 280    /* Clone the parameter list.
 281     */
 282    foreach_list_const(node, &this->parameters) {
 283       const ir_variable *const param = (const ir_variable *) node;
 284
 285       assert(const_cast<ir_variable *>(param)->as_variable() != NULL);
 286
 287       ir_variable *const param_copy = param->clone(mem_ctx, ht);
 288       copy->parameters.push_tail(param_copy);
 289    }
 290
 291    /* Clone the instruction list.
 292     */
 293    foreach_list_const(node, &this->body) {
 294       const ir_instruction *const inst = (const ir_instruction *) node;
 295
 296       ir_instruction *const inst_copy = inst->clone(mem_ctx, ht);
 297       copy->body.push_tail(inst_copy);
 298    }
 299
 300    return copy;
 301 }
 302
 303 ir_constant *
 304 ir_constant::clone(void *mem_ctx, struct hash_table *ht) const
 305 {
 306    (void)ht;
 307
 308    switch (this->type->base_type) {
 309    case GLSL_TYPE_UINT:
 310    case GLSL_TYPE_INT:
 311    case GLSL_TYPE_FLOAT:
 312    case GLSL_TYPE_BOOL:
 313       return new(mem_ctx) ir_constant(this->type, &this->value);
 314
 315    case GLSL_TYPE_STRUCT: {
 316       ir_constant *c = new(mem_ctx) ir_constant;
 317
 318       c->type = this->type;
 319       for (exec_node *node = this->components.head
 320               ; !node->is_tail_sentinel()
 321               ; node = node->next) {
 322          ir_constant *const orig = (ir_constant *) node;
 323
 324          c->components.push_tail(orig->clone(mem_ctx, NULL));
 325       }
 326
 327       return c;
 328    }
 329
 330    case GLSL_TYPE_ARRAY: {
 331       ir_constant *c = new(mem_ctx) ir_constant;
 332
 333       c->type = this->type;
 334       c->array_elements = talloc_array(c, ir_constant *, this->type->length);
 335       for (unsigned i = 0; i < this->type->length; i++) {
 336          c->array_elements[i] = this->array_elements[i]->clone(mem_ctx, NULL);
 337       }
 338       return c;
 339    }
 340
 341    default:
 342       assert(!"Should not get here.");
 343       return NULL;
 344    }
 345 }
 346
 347
 348 class fixup_ir_call_visitor : public ir_hierarchical_visitor {
 349 public:
 350    fixup_ir_call_visitor(struct hash_table *ht)
 351    {
 352       this->ht = ht;
 353    }
 354
 355    virtual ir_visitor_status visit_enter(ir_call *ir)
 356    {
 357       /* Try to find the function signature referenced by the ir_call in the
 358        * table.  If it is found, replace it with the value from the table.
 359        */
 360       ir_function_signature *sig =
 361          (ir_function_signature *) hash_table_find(this->ht, ir->get_callee());
 362       if (sig != NULL)
 363          ir->set_callee(sig);
 364
 365       /* Since this may be used before function call parameters are flattened,
 366        * the children also need to be processed.
 367        */
 368       return visit_continue;
 369    }
 370
 371 private:
 372    struct hash_table *ht;
 373 };
 374
 375
 376 static void
 377 fixup_function_calls(struct hash_table *ht, exec_list *instructions)
 378 {
 379    fixup_ir_call_visitor v(ht);
 380    v.run(instructions);
 381 }
 382
 383
 384 void
 385 clone_ir_list(void *mem_ctx, exec_list *out, const exec_list *in)
 386 {
 387    struct hash_table *ht =
 388       hash_table_ctor(0, hash_table_pointer_hash, hash_table_pointer_compare);
 389
 390    foreach_list_const(node, in) {
 391       const ir_instruction *const original = (ir_instruction *) node;
 392       ir_instruction *copy = original->clone(mem_ctx, ht);
 393
 394       out->push_tail(copy);
 395    }
 396
 397    /* Make a pass over the cloned tree to fix up ir_call nodes to point to the
 398     * cloned ir_function_signature nodes.  This cannot be done automatically
 399     * during cloning because the ir_call might be a forward reference (i.e.,
 400     * the function signature that it references may not have been cloned yet).
 401     */
 402    fixup_function_calls(ht, out);
 403
 404    hash_table_dtor(ht);
 405 }