src/glsl/nir/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 "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 const char *
 122 glsl_get_struct_elem_name(const struct glsl_type *type, unsigned index)
 123 {
 124    return type->fields.structure[index].name;
 125 }
 126
 127 glsl_sampler_dim
 128 glsl_get_sampler_dim(const struct glsl_type *type)
 129 {
 130    assert(glsl_type_is_sampler(type));
 131    return (glsl_sampler_dim)type->sampler_dimensionality;
 132 }
 133
 134 glsl_base_type
 135 glsl_get_sampler_result_type(const struct glsl_type *type)
 136 {
 137    assert(glsl_type_is_sampler(type));
 138    return (glsl_base_type)type->sampler_type;
 139 }
 140
 141 bool
 142 glsl_type_is_void(const glsl_type *type)
 143 {
 144    return type->is_void();
 145 }
 146
 147 bool
 148 glsl_type_is_vector(const struct glsl_type *type)
 149 {
 150    return type->is_vector();
 151 }
 152
 153 bool
 154 glsl_type_is_scalar(const struct glsl_type *type)
 155 {
 156    return type->is_scalar();
 157 }
 158
 159 bool
 160 glsl_type_is_vector_or_scalar(const struct glsl_type *type)
 161 {
 162    return type->is_vector() || type->is_scalar();
 163 }
 164
 165 bool
 166 glsl_type_is_matrix(const struct glsl_type *type)
 167 {
 168    return type->is_matrix();
 169 }
 170
 171 bool
 172 glsl_type_is_array(const struct glsl_type *type)
 173 {
 174    return type->is_array();
 175 }
 176
 177 bool
 178 glsl_type_is_struct(const struct glsl_type *type)
 179 {
 180    return type->is_record() || type->is_interface();
 181 }
 182
 183 bool
 184 glsl_type_is_sampler(const struct glsl_type *type)
 185 {
 186    return type->is_sampler();
 187 }
 188
 189 bool
 190 glsl_sampler_type_is_shadow(const struct glsl_type *type)
 191 {
 192    assert(glsl_type_is_sampler(type));
 193    return type->sampler_shadow;
 194 }
 195
 196 bool
 197 glsl_sampler_type_is_array(const struct glsl_type *type)
 198 {
 199    assert(glsl_type_is_sampler(type));
 200    return type->sampler_array;
 201 }
 202
 203 const glsl_type *
 204 glsl_void_type(void)
 205 {
 206    return glsl_type::void_type;
 207 }
 208
 209 const glsl_type *
 210 glsl_float_type(void)
 211 {
 212    return glsl_type::float_type;
 213 }
 214
 215 const glsl_type *
 216 glsl_int_type(void)
 217 {
 218    return glsl_type::int_type;
 219 }
 220
 221 const glsl_type *
 222 glsl_uint_type(void)
 223 {
 224    return glsl_type::uint_type;
 225 }
 226
 227 const glsl_type *
 228 glsl_bool_type(void)
 229 {
 230    return glsl_type::bool_type;
 231 }
 232
 233 const glsl_type *
 234 glsl_vec4_type(void)
 235 {
 236    return glsl_type::vec4_type;
 237 }
 238
 239 const glsl_type *
 240 glsl_scalar_type(enum glsl_base_type base_type)
 241 {
 242    return glsl_type::get_instance(base_type, 1, 1);
 243 }
 244
 245 const glsl_type *
 246 glsl_vector_type(enum glsl_base_type base_type, unsigned components)
 247 {
 248    assert(components > 1 && components <= 4);
 249    return glsl_type::get_instance(base_type, components, 1);
 250 }
 251
 252 const glsl_type *
 253 glsl_matrix_type(enum glsl_base_type base_type, unsigned rows, unsigned columns)
 254 {
 255    assert(rows > 1 && rows <= 4 && columns >= 1 && columns <= 4);
 256    return glsl_type::get_instance(base_type, rows, columns);
 257 }
 258
 259 const glsl_type *
 260 glsl_array_type(const glsl_type *base, unsigned elements)
 261 {
 262    return glsl_type::get_array_instance(base, elements);
 263 }
 264
 265 const glsl_type *
 266 glsl_struct_type(const glsl_struct_field *fields,
 267                  unsigned num_fields, const char *name)
 268 {
 269    return glsl_type::get_record_instance(fields, num_fields, name);
 270 }
 271
 272 const struct glsl_type *
 273 glsl_sampler_type(enum glsl_sampler_dim dim, bool is_shadow, bool is_array,
 274                   enum glsl_base_type base_type)
 275 {
 276    return glsl_type::get_sampler_instance(dim, is_shadow, is_array, base_type);
 277 }
 278
 279 const glsl_type *
 280 glsl_function_type(const glsl_type *return_type,
 281                    const glsl_function_param *params, unsigned num_params)
 282 {
 283    return glsl_type::get_function_instance(return_type, params, num_params);
 284 }
 285
 286 const glsl_type *
 287 glsl_transposed_type(const struct glsl_type *type)
 288 {
 289    return glsl_type::get_instance(type->base_type, type->matrix_columns,
 290                                   type->vector_elements);
 291 }