src/glsl/ir_print_visitor.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 "ir_print_visitor.h"
  25 #include "glsl_types.h"
  26 #include "glsl_parser_extras.h"
  27 #include "main/macros.h"
  28 #include "program/hash_table.h"
  29
  30 static void print_type(const glsl_type *t);
  31
  32 void
  33 ir_instruction::print(void) const
  34 {
  35    ir_instruction *deconsted = const_cast<ir_instruction *>(this);
  36
  37    ir_print_visitor v;
  38    deconsted->accept(&v);
  39 }
  40
  41 extern "C" {
  42 void
  43 _mesa_print_ir(exec_list *instructions,
  44                struct _mesa_glsl_parse_state *state)
  45 {
  46    if (state) {
  47       for (unsigned i = 0; i < state->num_user_structures; i++) {
  48          const glsl_type *const s = state->user_structures[i];
  49
  50          printf("(structure (%s) (%s@%p) (%u) (\n",
  51                 s->name, s->name, (void *) s, s->length);
  52
  53          for (unsigned j = 0; j < s->length; j++) {
  54             printf("\t((");
  55             print_type(s->fields.structure[j].type);
  56             printf(")(%s))\n", s->fields.structure[j].name);
  57          }
  58
  59          printf(")\n");
  60       }
  61    }
  62
  63    printf("(\n");
  64    foreach_iter(exec_list_iterator, iter, *instructions) {
  65       ir_instruction *ir = (ir_instruction *)iter.get();
  66       ir->print();
  67       if (ir->ir_type != ir_type_function)
  68          printf("\n");
  69    }
  70    printf("\n)");
  71 }
  72
  73 } /* extern "C" */
  74
  75 ir_print_visitor::ir_print_visitor()
  76 {
  77    indentation = 0;
  78    printable_names =
  79       hash_table_ctor(32, hash_table_pointer_hash, hash_table_pointer_compare);
  80    symbols = _mesa_symbol_table_ctor();
  81    mem_ctx = ralloc_context(NULL);
  82 }
  83
  84 ir_print_visitor::~ir_print_visitor()
  85 {
  86    hash_table_dtor(printable_names);
  87    _mesa_symbol_table_dtor(symbols);
  88    ralloc_free(mem_ctx);
  89 }
  90
  91 void ir_print_visitor::indent(void)
  92 {
  93    for (int i = 0; i < indentation; i++)
  94       printf("  ");
  95 }
  96
  97 const char *
  98 ir_print_visitor::unique_name(ir_variable *var)
  99 {
 100    /* var->name can be NULL in function prototypes when a type is given for a
 101     * parameter but no name is given.  In that case, just return an empty
 102     * string.  Don't worry about tracking the generated name in the printable
 103     * names hash because this is the only scope where it can ever appear.
 104     */
 105    if (var->name == NULL) {
 106       static unsigned arg = 1;
 107       return ralloc_asprintf(this->mem_ctx, "parameter@%u", arg++);
 108    }
 109
 110    /* Do we already have a name for this variable? */
 111    const char *name = (const char *) hash_table_find(this->printable_names, var);
 112    if (name != NULL)
 113       return name;
 114
 115    /* If there's no conflict, just use the original name */
 116    if (_mesa_symbol_table_find_symbol(this->symbols, -1, var->name) == NULL) {
 117       name = var->name;
 118    } else {
 119       static unsigned i = 1;
 120       name = ralloc_asprintf(this->mem_ctx, "%s@%u", var->name, ++i);
 121    }
 122    hash_table_insert(this->printable_names, (void *) name, var);
 123    _mesa_symbol_table_add_symbol(this->symbols, -1, name, var);
 124    return name;
 125 }
 126
 127 static void
 128 print_type(const glsl_type *t)
 129 {
 130    if (t->base_type == GLSL_TYPE_ARRAY) {
 131       printf("(array ");
 132       print_type(t->fields.array);
 133       printf(" %u)", t->length);
 134    } else if ((t->base_type == GLSL_TYPE_STRUCT)
 135               && (strncmp("gl_", t->name, 3) != 0)) {
 136       printf("%s@%p", t->name, (void *) t);
 137    } else {
 138       printf("%s", t->name);
 139    }
 140 }
 141
 142 void ir_print_visitor::visit(ir_rvalue *ir)
 143 {
 144    printf("error");
 145 }
 146
 147 void ir_print_visitor::visit(ir_variable *ir)
 148 {
 149    printf("(declare ");
 150
 151    const char *const cent = (ir->centroid) ? "centroid " : "";
 152    const char *const inv = (ir->invariant) ? "invariant " : "";
 153    const char *const mode[] = { "", "uniform ", "shader_in ", "shader_out ",
 154                                 "in ", "out ", "inout ",
 155                                 "const_in ", "sys ", "temporary " };
 156    STATIC_ASSERT(ARRAY_SIZE(mode) == ir_var_mode_count);
 157    const char *const interp[] = { "", "smooth", "flat", "noperspective" };
 158    STATIC_ASSERT(ARRAY_SIZE(interp) == INTERP_QUALIFIER_COUNT);
 159
 160    printf("(%s%s%s%s) ",
 161           cent, inv, mode[ir->mode], interp[ir->interpolation]);
 162
 163    print_type(ir->type);
 164    printf(" %s)", unique_name(ir));
 165 }
 166
 167
 168 void ir_print_visitor::visit(ir_function_signature *ir)
 169 {
 170    _mesa_symbol_table_push_scope(symbols);
 171    printf("(signature ");
 172    indentation++;
 173
 174    print_type(ir->return_type);
 175    printf("\n");
 176    indent();
 177
 178    printf("(parameters\n");
 179    indentation++;
 180
 181    foreach_iter(exec_list_iterator, iter, ir->parameters) {
 182       ir_variable *const inst = (ir_variable *) iter.get();
 183
 184       indent();
 185       inst->accept(this);
 186       printf("\n");
 187    }
 188    indentation--;
 189
 190    indent();
 191    printf(")\n");
 192
 193    indent();
 194
 195    printf("(\n");
 196    indentation++;
 197
 198    foreach_iter(exec_list_iterator, iter, ir->body) {
 199       ir_instruction *const inst = (ir_instruction *) iter.get();
 200
 201       indent();
 202       inst->accept(this);
 203       printf("\n");
 204    }
 205    indentation--;
 206    indent();
 207    printf("))\n");
 208    indentation--;
 209    _mesa_symbol_table_pop_scope(symbols);
 210 }
 211
 212
 213 void ir_print_visitor::visit(ir_function *ir)
 214 {
 215    printf("(function %s\n", ir->name);
 216    indentation++;
 217    foreach_iter(exec_list_iterator, iter, *ir) {
 218       ir_function_signature *const sig = (ir_function_signature *) iter.get();
 219       indent();
 220       sig->accept(this);
 221       printf("\n");
 222    }
 223    indentation--;
 224    indent();
 225    printf(")\n\n");
 226 }
 227
 228
 229 void ir_print_visitor::visit(ir_expression *ir)
 230 {
 231    printf("(expression ");
 232
 233    print_type(ir->type);
 234
 235    printf(" %s ", ir->operator_string());
 236
 237    for (unsigned i = 0; i < ir->get_num_operands(); i++) {
 238       ir->operands[i]->accept(this);
 239    }
 240
 241    printf(") ");
 242 }
 243
 244
 245 void ir_print_visitor::visit(ir_texture *ir)
 246 {
 247    printf("(%s ", ir->opcode_string());
 248
 249    print_type(ir->type);
 250    printf(" ");
 251
 252    ir->sampler->accept(this);
 253    printf(" ");
 254
 255    if (ir->op != ir_txs && ir->op != ir_query_levels) {
 256       ir->coordinate->accept(this);
 257
 258       printf(" ");
 259
 260       if (ir->offset != NULL) {
 261          ir->offset->accept(this);
 262       } else {
 263          printf("0");
 264       }
 265
 266       printf(" ");
 267    }
 268
 269    if (ir->op != ir_txf && ir->op != ir_txf_ms &&
 270        ir->op != ir_txs && ir->op != ir_tg4 &&
 271        ir->op != ir_query_levels) {
 272       if (ir->projector)
 273          ir->projector->accept(this);
 274       else
 275          printf("1");
 276
 277       if (ir->shadow_comparitor) {
 278          printf(" ");
 279          ir->shadow_comparitor->accept(this);
 280       } else {
 281          printf(" ()");
 282       }
 283    }
 284
 285    printf(" ");
 286    switch (ir->op)
 287    {
 288    case ir_tex:
 289    case ir_lod:
 290    case ir_query_levels:
 291       break;
 292    case ir_txb:
 293       ir->lod_info.bias->accept(this);
 294       break;
 295    case ir_txl:
 296    case ir_txf:
 297    case ir_txs:
 298       ir->lod_info.lod->accept(this);
 299       break;
 300    case ir_txf_ms:
 301       ir->lod_info.sample_index->accept(this);
 302       break;
 303    case ir_txd:
 304       printf("(");
 305       ir->lod_info.grad.dPdx->accept(this);
 306       printf(" ");
 307       ir->lod_info.grad.dPdy->accept(this);
 308       printf(")");
 309       break;
 310    case ir_tg4:
 311       ir->lod_info.component->accept(this);
 312       break;
 313    };
 314    printf(")");
 315 }
 316
 317
 318 void ir_print_visitor::visit(ir_swizzle *ir)
 319 {
 320    const unsigned swiz[4] = {
 321       ir->mask.x,
 322       ir->mask.y,
 323       ir->mask.z,
 324       ir->mask.w,
 325    };
 326
 327    printf("(swiz ");
 328    for (unsigned i = 0; i < ir->mask.num_components; i++) {
 329       printf("%c", "xyzw"[swiz[i]]);
 330    }
 331    printf(" ");
 332    ir->val->accept(this);
 333    printf(")");
 334 }
 335
 336
 337 void ir_print_visitor::visit(ir_dereference_variable *ir)
 338 {
 339    ir_variable *var = ir->variable_referenced();
 340    printf("(var_ref %s) ", unique_name(var));
 341 }
 342
 343
 344 void ir_print_visitor::visit(ir_dereference_array *ir)
 345 {
 346    printf("(array_ref ");
 347    ir->array->accept(this);
 348    ir->array_index->accept(this);
 349    printf(") ");
 350 }
 351
 352
 353 void ir_print_visitor::visit(ir_dereference_record *ir)
 354 {
 355    printf("(record_ref ");
 356    ir->record->accept(this);
 357    printf(" %s) ", ir->field);
 358 }
 359
 360
 361 void ir_print_visitor::visit(ir_assignment *ir)
 362 {
 363    printf("(assign ");
 364
 365    if (ir->condition)
 366       ir->condition->accept(this);
 367
 368    char mask[5];
 369    unsigned j = 0;
 370
 371    for (unsigned i = 0; i < 4; i++) {
 372       if ((ir->write_mask & (1 << i)) != 0) {
 373          mask[j] = "xyzw"[i];
 374          j++;
 375       }
 376    }
 377    mask[j] = '\0';
 378
 379    printf(" (%s) ", mask);
 380
 381    ir->lhs->accept(this);
 382
 383    printf(" ");
 384
 385    ir->rhs->accept(this);
 386    printf(") ");
 387 }
 388
 389
 390 void ir_print_visitor::visit(ir_constant *ir)
 391 {
 392    printf("(constant ");
 393    print_type(ir->type);
 394    printf(" (");
 395
 396    if (ir->type->is_array()) {
 397       for (unsigned i = 0; i < ir->type->length; i++)
 398          ir->get_array_element(i)->accept(this);
 399    } else if (ir->type->is_record()) {
 400       ir_constant *value = (ir_constant *) ir->components.get_head();
 401       for (unsigned i = 0; i < ir->type->length; i++) {
 402          printf("(%s ", ir->type->fields.structure[i].name);
 403          value->accept(this);
 404          printf(")");
 405
 406          value = (ir_constant *) value->next;
 407       }
 408    } else {
 409       for (unsigned i = 0; i < ir->type->components(); i++) {
 410          if (i != 0)
 411             printf(" ");
 412          switch (ir->type->base_type) {
 413          case GLSL_TYPE_UINT:  printf("%u", ir->value.u[i]); break;
 414          case GLSL_TYPE_INT:   printf("%d", ir->value.i[i]); break;
 415          case GLSL_TYPE_FLOAT:
 416             if (ir->value.f[i] == 0.0f)
 417                /* 0.0 == -0.0, so print with %f to get the proper sign. */
 418                printf("%.1f", ir->value.f[i]);
 419             else if (abs(ir->value.f[i]) < 0.000001f)
 420                printf("%a", ir->value.f[i]);
 421             else if (abs(ir->value.f[i]) > 1000000.0f)
 422                printf("%e", ir->value.f[i]);
 423             else
 424                printf("%f", ir->value.f[i]);
 425             break;
 426          case GLSL_TYPE_BOOL:  printf("%d", ir->value.b[i]); break;
 427          default: assert(0);
 428          }
 429       }
 430    }
 431    printf(")) ");
 432 }
 433
 434
 435 void
 436 ir_print_visitor::visit(ir_call *ir)
 437 {
 438    printf("(call %s ", ir->callee_name());
 439    if (ir->return_deref)
 440       ir->return_deref->accept(this);
 441    printf(" (");
 442    foreach_iter(exec_list_iterator, iter, *ir) {
 443       ir_instruction *const inst = (ir_instruction *) iter.get();
 444
 445       inst->accept(this);
 446    }
 447    printf("))\n");
 448 }
 449
 450
 451 void
 452 ir_print_visitor::visit(ir_return *ir)
 453 {
 454    printf("(return");
 455
 456    ir_rvalue *const value = ir->get_value();
 457    if (value) {
 458       printf(" ");
 459       value->accept(this);
 460    }
 461
 462    printf(")");
 463 }
 464
 465
 466 void
 467 ir_print_visitor::visit(ir_discard *ir)
 468 {
 469    printf("(discard ");
 470
 471    if (ir->condition != NULL) {
 472       printf(" ");
 473       ir->condition->accept(this);
 474    }
 475
 476    printf(")");
 477 }
 478
 479
 480 void
 481 ir_print_visitor::visit(ir_if *ir)
 482 {
 483    printf("(if ");
 484    ir->condition->accept(this);
 485
 486    printf("(\n");
 487    indentation++;
 488
 489    foreach_iter(exec_list_iterator, iter, ir->then_instructions) {
 490       ir_instruction *const inst = (ir_instruction *) iter.get();
 491
 492       indent();
 493       inst->accept(this);
 494       printf("\n");
 495    }
 496
 497    indentation--;
 498    indent();
 499    printf(")\n");
 500
 501    indent();
 502    if (!ir->else_instructions.is_empty()) {
 503       printf("(\n");
 504       indentation++;
 505
 506       foreach_iter(exec_list_iterator, iter, ir->else_instructions) {
 507          ir_instruction *const inst = (ir_instruction *) iter.get();
 508
 509          indent();
 510          inst->accept(this);
 511          printf("\n");
 512       }
 513       indentation--;
 514       indent();
 515       printf("))\n");
 516    } else {
 517       printf("())\n");
 518    }
 519 }
 520
 521
 522 void
 523 ir_print_visitor::visit(ir_loop *ir)
 524 {
 525    printf("(loop (");
 526    if (ir->counter != NULL)
 527       ir->counter->accept(this);
 528    printf(") (");
 529    if (ir->from != NULL)
 530       ir->from->accept(this);
 531    printf(") (");
 532    if (ir->to != NULL)
 533       ir->to->accept(this);
 534    printf(") (");
 535    if (ir->increment != NULL)
 536       ir->increment->accept(this);
 537    printf(") (\n");
 538    indentation++;
 539
 540    foreach_iter(exec_list_iterator, iter, ir->body_instructions) {
 541       ir_instruction *const inst = (ir_instruction *) iter.get();
 542
 543       indent();
 544       inst->accept(this);
 545       printf("\n");
 546    }
 547    indentation--;
 548    indent();
 549    printf("))\n");
 550 }
 551
 552
 553 void
 554 ir_print_visitor::visit(ir_loop_jump *ir)
 555 {
 556    printf("%s", ir->is_break() ? "break" : "continue");
 557 }
 558
 559 void
 560 ir_print_visitor::visit(ir_emit_vertex *ir)
 561 {
 562    printf("(emit-vertex)");
 563 }
 564
 565 void
 566 ir_print_visitor::visit(ir_end_primitive *ir)
 567 {
 568    printf("(end-primitive)");
 569 }