src/compiler/nir_types.cpp

   1 /*
   2  * Copyright © 2014 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 DEALINGS
  21  * IN THE SOFTWARE.
  22  *
  23  * Authors:
  24  *    Connor Abbott (cwabbott0@gmail.com)
  25  *
  26  */
  27
  28 #include "nir_types.h"
  29 #include "compiler/glsl/ir.h"
  30
  31 void
  32 glsl_print_type(const glsl_type *type, FILE *fp)
  33 {
  34    if (type->base_type == GLSL_TYPE_ARRAY) {
  35       glsl_print_type(type->fields.array, fp);
  36       fprintf(fp, "[%u]", type->length);
  37    } else if ((type->base_type == GLSL_TYPE_STRUCT)
  38               && !is_gl_identifier(type->name)) {
  39       fprintf(fp, "%s@%p", type->name, (void *) type);
  40    } else {
  41       fprintf(fp, "%s", type->name);
  42    }
  43 }
  44
  45 void
  46 glsl_print_struct(const glsl_type *type, FILE *fp)
  47 {
  48    assert(type->base_type == GLSL_TYPE_STRUCT);
  49
  50    fprintf(fp, "struct {\n");
  51    for (unsigned i = 0; i < type->length; i++) {
  52       fprintf(fp, "\t");
  53       glsl_print_type(type->fields.structure[i].type, fp);
  54       fprintf(fp, " %s;\n", type->fields.structure[i].name);
  55    }
  56    fprintf(fp, "}\n");
  57 }
  58
  59 const glsl_type *
  60 glsl_get_array_element(const glsl_type* type)
  61 {
  62    if (type->is_matrix())
  63       return type->column_type();
  64    return type->fields.array;
  65 }
  66
  67 const glsl_type *
  68 glsl_get_struct_field(const glsl_type *type, unsigned index)
  69 {
  70    return type->fields.structure[index].type;
  71 }
  72
  73 const glsl_type *
  74 glsl_get_function_return_type(const glsl_type *type)
  75 {
  76    return type->fields.parameters[0].type;
  77 }
  78
  79 const glsl_function_param *
  80 glsl_get_function_param(const glsl_type *type, unsigned index)
  81 {
  82    return &type->fields.parameters[index + 1];
  83 }
  84
  85 const struct glsl_type *
  86 glsl_get_column_type(const struct glsl_type *type)
  87 {
  88    return type->column_type();
  89 }
  90
  91 enum glsl_base_type
  92 glsl_get_base_type(const struct glsl_type *type)
  93 {
  94    return type->base_type;
  95 }
  96
  97 unsigned
  98 glsl_get_vector_elements(const struct glsl_type *type)
  99 {
 100    return type->vector_elements;
 101 }
 102
 103 unsigned
 104 glsl_get_components(const struct glsl_type *type)
 105 {
 106    return type->components();
 107 }
 108
 109 unsigned
 110 glsl_get_matrix_columns(const struct glsl_type *type)
 111 {
 112    return type->matrix_columns;
 113 }
 114
 115 unsigned
 116 glsl_get_length(const struct glsl_type *type)
 117 {
 118    return type->is_matrix() ? type->matrix_columns : type->length;
 119 }
 120
 121 unsigned
 122 glsl_get_aoa_size(const struct glsl_type *type)
 123 {
 124    return type->arrays_of_arrays_size();
 125 }
 126
 127 const char *
 128 glsl_get_struct_elem_name(const struct glsl_type *type, unsigned index)
 129 {
 130    return type->fields.structure[index].name;
 131 }
 132
 133 unsigned
 134 glsl_get_record_location_offset(const struct glsl_type *type,
 135                                 unsigned length)
 136 {
 137    return type->record_location_offset(length);
 138 }
 139
 140 bool
 141 glsl_type_is_void(const glsl_type *type)
 142 {
 143    return type->is_void();
 144 }
 145
 146 bool
 147 glsl_type_is_vector(const struct glsl_type *type)
 148 {
 149    return type->is_vector();
 150 }
 151
 152 bool
 153 glsl_type_is_scalar(const struct glsl_type *type)
 154 {
 155    return type->is_scalar();
 156 }
 157
 158 bool
 159 glsl_type_is_vector_or_scalar(const struct glsl_type *type)
 160 {
 161    return type->is_vector() || type->is_scalar();
 162 }
 163
 164 bool
 165 glsl_type_is_matrix(const struct glsl_type *type)
 166 {
 167    return type->is_matrix();
 168 }
 169
 170 const glsl_type *
 171 glsl_void_type(void)
 172 {
 173    return glsl_type::void_type;
 174 }
 175
 176 const glsl_type *
 177 glsl_float_type(void)
 178 {
 179    return glsl_type::float_type;
 180 }
 181
 182 const glsl_type *
 183 glsl_vec_type(unsigned n)
 184 {
 185    return glsl_type::vec(n);
 186 }
 187
 188 const glsl_type *
 189 glsl_vec4_type(void)
 190 {
 191    return glsl_type::vec4_type;
 192 }
 193
 194 const glsl_type *
 195 glsl_uint_type(void)
 196 {
 197    return glsl_type::uint_type;
 198 }
 199
 200 const glsl_type *
 201 glsl_array_type(const glsl_type *base, unsigned elements)
 202 {
 203    return glsl_type::get_array_instance(base, elements);
 204 }
 205
 206 const glsl_type *
 207 glsl_function_type(const glsl_type *return_type,
 208                    const glsl_function_param *params, unsigned num_params)
 209 {
 210    return glsl_type::get_function_instance(return_type, params, num_params);
 211 }