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 "main/compiler.h"
  26 #include "ir.h"
  27 #include "glsl_types.h"
  28 #include "program/hash_table.h"
  29
  30 ir_rvalue *
  31 ir_rvalue::clone(void *mem_ctx, struct hash_table *ht) const
  32 {
  33    /* The only possible instantiation is the generic error value. */
  34    return error_value(mem_ctx);
  35 }
  36
  37 /**
  38  * Duplicate an IR variable
  39  */
  40 ir_variable *
  41 ir_variable::clone(void *mem_ctx, struct hash_table *ht) const
  42 {
  43    ir_variable *var = new(mem_ctx) ir_variable(this->type, this->name,
  44                                                (ir_variable_mode) this->mode);
  45
  46    var->max_array_access = this->max_array_access;
  47    if (this->is_interface_instance()) {
  48       var->max_ifc_array_access =
  49          rzalloc_array(var, unsigned, this->interface_type->length);
  50       memcpy(var->max_ifc_array_access, this->max_ifc_array_access,
  51              this->interface_type->length * sizeof(unsigned));
  52    }
  53    var->data.read_only = this->data.read_only;
  54    var->data.centroid = this->data.centroid;
  55    var->data.sample = this->data.sample;
  56    var->data.invariant = this->data.invariant;
  57    var->interpolation = this->interpolation;
  58    var->location = this->location;
  59    var->index = this->index;
  60    var->binding = this->binding;
  61    var->atomic.buffer_index = this->atomic.buffer_index;
  62    var->atomic.offset = this->atomic.offset;
  63    var->warn_extension = this->warn_extension;
  64    var->origin_upper_left = this->origin_upper_left;
  65    var->pixel_center_integer = this->pixel_center_integer;
  66    var->explicit_location = this->explicit_location;
  67    var->explicit_index = this->explicit_index;
  68    var->explicit_binding = this->explicit_binding;
  69    var->has_initializer = this->has_initializer;
  70    var->depth_layout = this->depth_layout;
  71    var->assigned = this->assigned;
  72    var->how_declared = this->how_declared;
  73    var->used = this->used;
  74
  75    var->num_state_slots = this->num_state_slots;
  76    if (this->state_slots) {
  77       /* FINISHME: This really wants to use something like talloc_reference, but
  78        * FINISHME: ralloc doesn't have any similar function.
  79        */
  80       var->state_slots = ralloc_array(var, ir_state_slot,
  81                                       this->num_state_slots);
  82       memcpy(var->state_slots, this->state_slots,
  83              sizeof(this->state_slots[0]) * var->num_state_slots);
  84    }
  85
  86    if (this->constant_value)
  87       var->constant_value = this->constant_value->clone(mem_ctx, ht);
  88
  89    if (this->constant_initializer)
  90       var->constant_initializer =
  91          this->constant_initializer->clone(mem_ctx, ht);
  92
  93    var->interface_type = this->interface_type;
  94
  95    if (ht) {
  96       hash_table_insert(ht, var, (void *)const_cast<ir_variable *>(this));
  97    }
  98
  99    return var;
 100 }
 101
 102 ir_swizzle *
 103 ir_swizzle::clone(void *mem_ctx, struct hash_table *ht) const
 104 {
 105    return new(mem_ctx) ir_swizzle(this->val->clone(mem_ctx, ht), this->mask);
 106 }
 107
 108 ir_return *
 109 ir_return::clone(void *mem_ctx, struct hash_table *ht) const
 110 {
 111    ir_rvalue *new_value = NULL;
 112
 113    if (this->value)
 114       new_value = this->value->clone(mem_ctx, ht);
 115
 116    return new(mem_ctx) ir_return(new_value);
 117 }
 118
 119 ir_discard *
 120 ir_discard::clone(void *mem_ctx, struct hash_table *ht) const
 121 {
 122    ir_rvalue *new_condition = NULL;
 123
 124    if (this->condition != NULL)
 125       new_condition = this->condition->clone(mem_ctx, ht);
 126
 127    return new(mem_ctx) ir_discard(new_condition);
 128 }
 129
 130 ir_loop_jump *
 131 ir_loop_jump::clone(void *mem_ctx, struct hash_table *ht) const
 132 {
 133    (void)ht;
 134
 135    return new(mem_ctx) ir_loop_jump(this->mode);
 136 }
 137
 138 ir_if *
 139 ir_if::clone(void *mem_ctx, struct hash_table *ht) const
 140 {
 141    ir_if *new_if = new(mem_ctx) ir_if(this->condition->clone(mem_ctx, ht));
 142
 143    foreach_iter(exec_list_iterator, iter, this->then_instructions) {
 144       ir_instruction *ir = (ir_instruction *)iter.get();
 145       new_if->then_instructions.push_tail(ir->clone(mem_ctx, ht));
 146    }
 147
 148    foreach_iter(exec_list_iterator, iter, this->else_instructions) {
 149       ir_instruction *ir = (ir_instruction *)iter.get();
 150       new_if->else_instructions.push_tail(ir->clone(mem_ctx, ht));
 151    }
 152
 153    return new_if;
 154 }
 155
 156 ir_loop *
 157 ir_loop::clone(void *mem_ctx, struct hash_table *ht) const
 158 {
 159    ir_loop *new_loop = new(mem_ctx) ir_loop();
 160
 161    foreach_iter(exec_list_iterator, iter, this->body_instructions) {
 162       ir_instruction *ir = (ir_instruction *)iter.get();
 163       new_loop->body_instructions.push_tail(ir->clone(mem_ctx, ht));
 164    }
 165
 166    return new_loop;
 167 }
 168
 169 ir_call *
 170 ir_call::clone(void *mem_ctx, struct hash_table *ht) const
 171 {
 172    ir_dereference_variable *new_return_ref = NULL;
 173    if (this->return_deref != NULL)
 174       new_return_ref = this->return_deref->clone(mem_ctx, ht);
 175
 176    exec_list new_parameters;
 177
 178    foreach_iter(exec_list_iterator, iter, this->actual_parameters) {
 179       ir_instruction *ir = (ir_instruction *)iter.get();
 180       new_parameters.push_tail(ir->clone(mem_ctx, ht));
 181    }
 182
 183    return new(mem_ctx) ir_call(this->callee, new_return_ref, &new_parameters);
 184 }
 185
 186 ir_expression *
 187 ir_expression::clone(void *mem_ctx, struct hash_table *ht) const
 188 {
 189    ir_rvalue *op[Elements(this->operands)] = { NULL, };
 190    unsigned int i;
 191
 192    for (i = 0; i < get_num_operands(); i++) {
 193       op[i] = this->operands[i]->clone(mem_ctx, ht);
 194    }
 195
 196    return new(mem_ctx) ir_expression(this->operation, this->type,
 197                                      op[0], op[1], op[2], op[3]);
 198 }
 199
 200 ir_dereference_variable *
 201 ir_dereference_variable::clone(void *mem_ctx, struct hash_table *ht) const
 202 {
 203    ir_variable *new_var;
 204
 205    if (ht) {
 206       new_var = (ir_variable *)hash_table_find(ht, this->var);
 207       if (!new_var)
 208          new_var = this->var;
 209    } else {
 210       new_var = this->var;
 211    }
 212
 213    return new(mem_ctx) ir_dereference_variable(new_var);
 214 }
 215
 216 ir_dereference_array *
 217 ir_dereference_array::clone(void *mem_ctx, struct hash_table *ht) const
 218 {
 219    return new(mem_ctx) ir_dereference_array(this->array->clone(mem_ctx, ht),
 220                                             this->array_index->clone(mem_ctx,
 221                                                                      ht));
 222 }
 223
 224 ir_dereference_record *
 225 ir_dereference_record::clone(void *mem_ctx, struct hash_table *ht) const
 226 {
 227    return new(mem_ctx) ir_dereference_record(this->record->clone(mem_ctx, ht),
 228                                              this->field);
 229 }
 230
 231 ir_texture *
 232 ir_texture::clone(void *mem_ctx, struct hash_table *ht) const
 233 {
 234    ir_texture *new_tex = new(mem_ctx) ir_texture(this->op);
 235    new_tex->type = this->type;
 236
 237    new_tex->sampler = this->sampler->clone(mem_ctx, ht);
 238    if (this->coordinate)
 239       new_tex->coordinate = this->coordinate->clone(mem_ctx, ht);
 240    if (this->projector)
 241       new_tex->projector = this->projector->clone(mem_ctx, ht);
 242    if (this->shadow_comparitor) {
 243       new_tex->shadow_comparitor = this->shadow_comparitor->clone(mem_ctx, ht);
 244    }
 245
 246    if (this->offset != NULL)
 247       new_tex->offset = this->offset->clone(mem_ctx, ht);
 248
 249    switch (this->op) {
 250    case ir_tex:
 251    case ir_lod:
 252    case ir_query_levels:
 253       break;
 254    case ir_txb:
 255       new_tex->lod_info.bias = this->lod_info.bias->clone(mem_ctx, ht);
 256       break;
 257    case ir_txl:
 258    case ir_txf:
 259    case ir_txs:
 260       new_tex->lod_info.lod = this->lod_info.lod->clone(mem_ctx, ht);
 261       break;
 262    case ir_txf_ms:
 263       new_tex->lod_info.sample_index = this->lod_info.sample_index->clone(mem_ctx, ht);
 264       break;
 265    case ir_txd:
 266       new_tex->lod_info.grad.dPdx = this->lod_info.grad.dPdx->clone(mem_ctx, ht);
 267       new_tex->lod_info.grad.dPdy = this->lod_info.grad.dPdy->clone(mem_ctx, ht);
 268       break;
 269    case ir_tg4:
 270       new_tex->lod_info.component = this->lod_info.component->clone(mem_ctx, ht);
 271       break;
 272    }
 273
 274    return new_tex;
 275 }
 276
 277 ir_assignment *
 278 ir_assignment::clone(void *mem_ctx, struct hash_table *ht) const
 279 {
 280    ir_rvalue *new_condition = NULL;
 281
 282    if (this->condition)
 283       new_condition = this->condition->clone(mem_ctx, ht);
 284
 285    return new(mem_ctx) ir_assignment(this->lhs->clone(mem_ctx, ht),
 286                                      this->rhs->clone(mem_ctx, ht),
 287                                      new_condition,
 288                                      this->write_mask);
 289 }
 290
 291 ir_function *
 292 ir_function::clone(void *mem_ctx, struct hash_table *ht) const
 293 {
 294    ir_function *copy = new(mem_ctx) ir_function(this->name);
 295
 296    foreach_list_const(node, &this->signatures) {
 297       const ir_function_signature *const sig =
 298          (const ir_function_signature *const) node;
 299
 300       ir_function_signature *sig_copy = sig->clone(mem_ctx, ht);
 301       copy->add_signature(sig_copy);
 302
 303       if (ht != NULL)
 304          hash_table_insert(ht, sig_copy,
 305                            (void *)const_cast<ir_function_signature *>(sig));
 306    }
 307
 308    return copy;
 309 }
 310
 311 ir_function_signature *
 312 ir_function_signature::clone(void *mem_ctx, struct hash_table *ht) const
 313 {
 314    ir_function_signature *copy = this->clone_prototype(mem_ctx, ht);
 315
 316    copy->is_defined = this->is_defined;
 317
 318    /* Clone the instruction list.
 319     */
 320    foreach_list_const(node, &this->body) {
 321       const ir_instruction *const inst = (const ir_instruction *) node;
 322
 323       ir_instruction *const inst_copy = inst->clone(mem_ctx, ht);
 324       copy->body.push_tail(inst_copy);
 325    }
 326
 327    return copy;
 328 }
 329
 330 ir_function_signature *
 331 ir_function_signature::clone_prototype(void *mem_ctx, struct hash_table *ht) const
 332 {
 333    ir_function_signature *copy =
 334       new(mem_ctx) ir_function_signature(this->return_type);
 335
 336    copy->is_defined = false;
 337    copy->builtin_avail = this->builtin_avail;
 338    copy->origin = this;
 339
 340    /* Clone the parameter list, but NOT the body.
 341     */
 342    foreach_list_const(node, &this->parameters) {
 343       const ir_variable *const param = (const ir_variable *) node;
 344
 345       assert(const_cast<ir_variable *>(param)->as_variable() != NULL);
 346
 347       ir_variable *const param_copy = param->clone(mem_ctx, ht);
 348       copy->parameters.push_tail(param_copy);
 349    }
 350
 351    return copy;
 352 }
 353
 354 ir_constant *
 355 ir_constant::clone(void *mem_ctx, struct hash_table *ht) const
 356 {
 357    (void)ht;
 358
 359    switch (this->type->base_type) {
 360    case GLSL_TYPE_UINT:
 361    case GLSL_TYPE_INT:
 362    case GLSL_TYPE_FLOAT:
 363    case GLSL_TYPE_BOOL:
 364       return new(mem_ctx) ir_constant(this->type, &this->value);
 365
 366    case GLSL_TYPE_STRUCT: {
 367       ir_constant *c = new(mem_ctx) ir_constant;
 368
 369       c->type = this->type;
 370       for (exec_node *node = this->components.head
 371               ; !node->is_tail_sentinel()
 372               ; node = node->next) {
 373          ir_constant *const orig = (ir_constant *) node;
 374
 375          c->components.push_tail(orig->clone(mem_ctx, NULL));
 376       }
 377
 378       return c;
 379    }
 380
 381    case GLSL_TYPE_ARRAY: {
 382       ir_constant *c = new(mem_ctx) ir_constant;
 383
 384       c->type = this->type;
 385       c->array_elements = ralloc_array(c, ir_constant *, this->type->length);
 386       for (unsigned i = 0; i < this->type->length; i++) {
 387          c->array_elements[i] = this->array_elements[i]->clone(mem_ctx, NULL);
 388       }
 389       return c;
 390    }
 391
 392    case GLSL_TYPE_SAMPLER:
 393    case GLSL_TYPE_ATOMIC_UINT:
 394    case GLSL_TYPE_VOID:
 395    case GLSL_TYPE_ERROR:
 396    case GLSL_TYPE_INTERFACE:
 397       assert(!"Should not get here.");
 398       break;
 399    }
 400
 401    return NULL;
 402 }
 403
 404
 405 class fixup_ir_call_visitor : public ir_hierarchical_visitor {
 406 public:
 407    fixup_ir_call_visitor(struct hash_table *ht)
 408    {
 409       this->ht = ht;
 410    }
 411
 412    virtual ir_visitor_status visit_enter(ir_call *ir)
 413    {
 414       /* Try to find the function signature referenced by the ir_call in the
 415        * table.  If it is found, replace it with the value from the table.
 416        */
 417       ir_function_signature *sig =
 418          (ir_function_signature *) hash_table_find(this->ht, ir->callee);
 419       if (sig != NULL)
 420          ir->callee = sig;
 421
 422       /* Since this may be used before function call parameters are flattened,
 423        * the children also need to be processed.
 424        */
 425       return visit_continue;
 426    }
 427
 428 private:
 429    struct hash_table *ht;
 430 };
 431
 432
 433 static void
 434 fixup_function_calls(struct hash_table *ht, exec_list *instructions)
 435 {
 436    fixup_ir_call_visitor v(ht);
 437    v.run(instructions);
 438 }
 439
 440
 441 void
 442 clone_ir_list(void *mem_ctx, exec_list *out, const exec_list *in)
 443 {
 444    struct hash_table *ht =
 445       hash_table_ctor(0, hash_table_pointer_hash, hash_table_pointer_compare);
 446
 447    foreach_list_const(node, in) {
 448       const ir_instruction *const original = (ir_instruction *) node;
 449       ir_instruction *copy = original->clone(mem_ctx, ht);
 450
 451       out->push_tail(copy);
 452    }
 453
 454    /* Make a pass over the cloned tree to fix up ir_call nodes to point to the
 455     * cloned ir_function_signature nodes.  This cannot be done automatically
 456     * during cloning because the ir_call might be a forward reference (i.e.,
 457     * the function signature that it references may not have been cloned yet).
 458     */
 459    fixup_function_calls(ht, out);
 460
 461    hash_table_dtor(ht);
 462 }