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 unsigned
 128 glsl_count_attribute_slots(const struct glsl_type *type,
 129                            bool vertex_input_slots)
 130 {
 131    return type->count_attribute_slots(vertex_input_slots);
 132 }
 133
 134 const char *
 135 glsl_get_struct_elem_name(const struct glsl_type *type, unsigned index)
 136 {
 137    return type->fields.structure[index].name;
 138 }
 139
 140 glsl_sampler_dim
 141 glsl_get_sampler_dim(const struct glsl_type *type)
 142 {
 143    assert(glsl_type_is_sampler(type) || glsl_type_is_image(type));
 144    return (glsl_sampler_dim)type->sampler_dimensionality;
 145 }
 146
 147 glsl_base_type
 148 glsl_get_sampler_result_type(const struct glsl_type *type)
 149 {
 150    assert(glsl_type_is_sampler(type) || glsl_type_is_image(type));
 151    return (glsl_base_type)type->sampler_type;
 152 }
 153
 154 unsigned
 155 glsl_get_record_location_offset(const struct glsl_type *type,
 156                                 unsigned length)
 157 {
 158    return type->record_location_offset(length);
 159 }
 160
 161 bool
 162 glsl_type_is_void(const glsl_type *type)
 163 {
 164    return type->is_void();
 165 }
 166
 167 bool
 168 glsl_type_is_error(const glsl_type *type)
 169 {
 170    return type->is_error();
 171 }
 172
 173 bool
 174 glsl_type_is_vector(const struct glsl_type *type)
 175 {
 176    return type->is_vector();
 177 }
 178
 179 bool
 180 glsl_type_is_scalar(const struct glsl_type *type)
 181 {
 182    return type->is_scalar();
 183 }
 184
 185 bool
 186 glsl_type_is_vector_or_scalar(const struct glsl_type *type)
 187 {
 188    return type->is_vector() || type->is_scalar();
 189 }
 190
 191 bool
 192 glsl_type_is_matrix(const struct glsl_type *type)
 193 {
 194    return type->is_matrix();
 195 }
 196
 197 bool
 198 glsl_type_is_array(const struct glsl_type *type)
 199 {
 200    return type->is_array();
 201 }
 202
 203 bool
 204 glsl_type_is_struct(const struct glsl_type *type)
 205 {
 206    return type->is_record() || type->is_interface();
 207 }
 208
 209 bool
 210 glsl_type_is_sampler(const struct glsl_type *type)
 211 {
 212    return type->is_sampler();
 213 }
 214
 215 bool
 216 glsl_type_is_image(const struct glsl_type *type)
 217 {
 218    return type->is_image();
 219 }
 220
 221 bool
 222 glsl_sampler_type_is_shadow(const struct glsl_type *type)
 223 {
 224    assert(glsl_type_is_sampler(type));
 225    return type->sampler_shadow;
 226 }
 227
 228 bool
 229 glsl_sampler_type_is_array(const struct glsl_type *type)
 230 {
 231    assert(glsl_type_is_sampler(type) || glsl_type_is_image(type));
 232    return type->sampler_array;
 233 }
 234
 235 const glsl_type *
 236 glsl_void_type(void)
 237 {
 238    return glsl_type::void_type;
 239 }
 240
 241 const glsl_type *
 242 glsl_float_type(void)
 243 {
 244    return glsl_type::float_type;
 245 }
 246
 247 const glsl_type *
 248 glsl_vec_type(unsigned n)
 249 {
 250    return glsl_type::vec(n);
 251 }
 252
 253 const glsl_type *
 254 glsl_vec4_type(void)
 255 {
 256    return glsl_type::vec4_type;
 257 }
 258
 259 const glsl_type *
 260 glsl_int_type(void)
 261 {
 262    return glsl_type::int_type;
 263 }
 264
 265 const glsl_type *
 266 glsl_uint_type(void)
 267 {
 268    return glsl_type::uint_type;
 269 }
 270
 271 const glsl_type *
 272 glsl_bool_type(void)
 273 {
 274    return glsl_type::bool_type;
 275 }
 276
 277 const glsl_type *
 278 glsl_scalar_type(enum glsl_base_type base_type)
 279 {
 280    return glsl_type::get_instance(base_type, 1, 1);
 281 }
 282
 283 const glsl_type *
 284 glsl_vector_type(enum glsl_base_type base_type, unsigned components)
 285 {
 286    assert(components > 1 && components <= 4);
 287    return glsl_type::get_instance(base_type, components, 1);
 288 }
 289
 290 const glsl_type *
 291 glsl_matrix_type(enum glsl_base_type base_type, unsigned rows, unsigned columns)
 292 {
 293    assert(rows > 1 && rows <= 4 && columns >= 1 && columns <= 4);
 294    return glsl_type::get_instance(base_type, rows, columns);
 295 }
 296
 297 const glsl_type *
 298 glsl_array_type(const glsl_type *base, unsigned elements)
 299 {
 300    return glsl_type::get_array_instance(base, elements);
 301 }
 302
 303 const glsl_type *
 304 glsl_struct_type(const glsl_struct_field *fields,
 305                  unsigned num_fields, const char *name)
 306 {
 307    return glsl_type::get_record_instance(fields, num_fields, name);
 308 }
 309
 310 const struct glsl_type *
 311 glsl_sampler_type(enum glsl_sampler_dim dim, bool is_shadow, bool is_array,
 312                   enum glsl_base_type base_type)
 313 {
 314    return glsl_type::get_sampler_instance(dim, is_shadow, is_array, base_type);
 315 }
 316
 317 const struct glsl_type *
 318 glsl_image_type(enum glsl_sampler_dim dim, bool is_array,
 319                 enum glsl_base_type base_type)
 320 {
 321    return glsl_type::get_image_instance(dim, is_array, base_type);
 322 }
 323
 324 const glsl_type *
 325 glsl_function_type(const glsl_type *return_type,
 326                    const glsl_function_param *params, unsigned num_params)
 327 {
 328    return glsl_type::get_function_instance(return_type, params, num_params);
 329 }
 330
 331 const glsl_type *
 332 glsl_transposed_type(const struct glsl_type *type)
 333 {
 334    return glsl_type::get_instance(type->base_type, type->matrix_columns,
 335                                   type->vector_elements);
 336 }