2 * Copyright © 2014 Intel Corporation
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:
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
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
24 * Connor Abbott (cwabbott0@gmail.com)
28 #include "nir_types.h"
29 #include "compiler/glsl/ir.h"
32 glsl_get_type_name(const glsl_type
*type
)
38 glsl_array_size(const struct glsl_type
*type
)
40 return type
->array_size();
44 glsl_get_array_element(const glsl_type
* type
)
46 if (type
->is_matrix())
47 return type
->column_type();
48 else if (type
->is_vector())
49 return type
->get_scalar_type();
50 return type
->fields
.array
;
54 glsl_without_array(const glsl_type
*type
)
56 return type
->without_array();
60 glsl_without_array_or_matrix(const glsl_type
*type
)
62 type
= type
->without_array();
63 if (type
->is_matrix())
64 type
= type
->column_type();
69 glsl_get_bare_type(const glsl_type
*type
)
71 return type
->get_bare_type();
75 glsl_get_struct_field(const glsl_type
*type
, unsigned index
)
77 return type
->fields
.structure
[index
].type
;
81 glsl_get_struct_field_offset(const struct glsl_type
*type
,
84 return type
->fields
.structure
[index
].offset
;
87 const struct glsl_struct_field
*
88 glsl_get_struct_field_data(const struct glsl_type
*type
, unsigned index
)
90 assert(type
->is_struct() || type
->is_interface());
91 assert(index
< type
->length
);
92 return &type
->fields
.structure
[index
];
96 glsl_get_explicit_stride(const struct glsl_type
*type
)
98 return type
->explicit_stride
;
102 glsl_get_function_return_type(const glsl_type
*type
)
104 return type
->fields
.parameters
[0].type
;
107 const glsl_function_param
*
108 glsl_get_function_param(const glsl_type
*type
, unsigned index
)
110 return &type
->fields
.parameters
[index
+ 1];
113 const struct glsl_type
*
114 glsl_get_column_type(const struct glsl_type
*type
)
116 return type
->column_type();
120 glsl_get_gl_type(const struct glsl_type
*type
)
122 return type
->gl_type
;
126 glsl_get_base_type(const struct glsl_type
*type
)
128 return type
->base_type
;
132 glsl_get_vector_elements(const struct glsl_type
*type
)
134 return type
->vector_elements
;
138 glsl_get_components(const struct glsl_type
*type
)
140 return type
->components();
144 glsl_get_matrix_columns(const struct glsl_type
*type
)
146 return type
->matrix_columns
;
150 glsl_get_length(const struct glsl_type
*type
)
152 return type
->is_matrix() ? type
->matrix_columns
: type
->length
;
156 glsl_get_aoa_size(const struct glsl_type
*type
)
158 return type
->arrays_of_arrays_size();
162 glsl_count_vec4_slots(const struct glsl_type
*type
,
163 bool is_gl_vertex_input
, bool is_bindless
)
165 return type
->count_vec4_slots(is_gl_vertex_input
, is_bindless
);
169 glsl_count_dword_slots(const struct glsl_type
*type
, bool is_bindless
)
171 return type
->count_dword_slots(is_bindless
);
175 glsl_count_attribute_slots(const struct glsl_type
*type
,
176 bool is_gl_vertex_input
)
178 return type
->count_attribute_slots(is_gl_vertex_input
);
182 glsl_get_component_slots(const struct glsl_type
*type
)
184 return type
->component_slots();
188 glsl_varying_count(const struct glsl_type
*type
)
190 return type
->varying_count();
194 glsl_get_struct_elem_name(const struct glsl_type
*type
, unsigned index
)
196 return type
->fields
.structure
[index
].name
;
200 glsl_get_sampler_dim(const struct glsl_type
*type
)
202 assert(glsl_type_is_sampler(type
) || glsl_type_is_image(type
));
203 return (glsl_sampler_dim
)type
->sampler_dimensionality
;
207 glsl_get_sampler_result_type(const struct glsl_type
*type
)
209 assert(glsl_type_is_sampler(type
) || glsl_type_is_image(type
));
210 return (glsl_base_type
)type
->sampled_type
;
214 glsl_get_sampler_target(const struct glsl_type
*type
)
216 assert(glsl_type_is_sampler(type
));
217 return type
->sampler_index();
221 glsl_get_sampler_coordinate_components(const struct glsl_type
*type
)
223 assert(glsl_type_is_sampler(type
) || glsl_type_is_image(type
));
224 return type
->coordinate_components();
228 glsl_get_struct_location_offset(const struct glsl_type
*type
,
231 return type
->struct_location_offset(length
);
235 glsl_type_is_16bit(const glsl_type
*type
)
237 return type
->is_16bit();
241 glsl_type_is_32bit(const glsl_type
*type
)
243 return type
->is_32bit();
247 glsl_type_is_64bit(const glsl_type
*type
)
249 return type
->is_64bit();
253 glsl_type_is_void(const glsl_type
*type
)
255 return type
->is_void();
259 glsl_type_is_error(const glsl_type
*type
)
261 return type
->is_error();
265 glsl_type_is_vector(const struct glsl_type
*type
)
267 return type
->is_vector();
271 glsl_type_is_scalar(const struct glsl_type
*type
)
273 return type
->is_scalar();
277 glsl_type_is_vector_or_scalar(const struct glsl_type
*type
)
279 return type
->is_vector() || type
->is_scalar();
283 glsl_type_is_matrix(const struct glsl_type
*type
)
285 return type
->is_matrix();
289 glsl_matrix_type_is_row_major(const struct glsl_type
*type
)
291 assert(type
->is_matrix() && type
->explicit_stride
);
292 return type
->interface_row_major
;
296 glsl_type_is_array(const struct glsl_type
*type
)
298 return type
->is_array();
302 glsl_type_is_unsized_array(const struct glsl_type
*type
)
304 return type
->is_unsized_array();
308 glsl_type_is_array_of_arrays(const struct glsl_type
*type
)
310 return type
->is_array_of_arrays();
314 glsl_type_is_array_or_matrix(const struct glsl_type
*type
)
316 return type
->is_array() || type
->is_matrix();
320 glsl_type_is_struct(const struct glsl_type
*type
)
322 return type
->is_struct();
326 glsl_type_is_interface(const struct glsl_type
*type
)
328 return type
->is_interface();
332 glsl_type_is_struct_or_ifc(const struct glsl_type
*type
)
334 return type
->is_struct() || type
->is_interface();
338 glsl_type_is_sampler(const struct glsl_type
*type
)
340 return type
->is_sampler();
344 glsl_type_is_image(const struct glsl_type
*type
)
346 return type
->is_image();
350 glsl_sampler_type_is_shadow(const struct glsl_type
*type
)
352 assert(glsl_type_is_sampler(type
));
353 return type
->sampler_shadow
;
357 glsl_sampler_type_is_array(const struct glsl_type
*type
)
359 assert(glsl_type_is_sampler(type
) || glsl_type_is_image(type
));
360 return type
->sampler_array
;
364 glsl_struct_type_is_packed(const struct glsl_type
*type
)
366 assert(glsl_type_is_struct(type
));
371 glsl_type_is_dual_slot(const struct glsl_type
*type
)
373 return type
->is_dual_slot();
377 glsl_type_is_numeric(const struct glsl_type
*type
)
379 return type
->is_numeric();
383 glsl_type_is_boolean(const struct glsl_type
*type
)
385 return type
->is_boolean();
388 glsl_type_is_integer(const struct glsl_type
*type
)
390 return type
->is_integer();
394 glsl_type_contains_64bit(const struct glsl_type
*type
)
396 return type
->contains_64bit();
402 return glsl_type::void_type
;
406 glsl_float_type(void)
408 return glsl_type::float_type
;
412 glsl_double_type(void)
414 return glsl_type::double_type
;
418 glsl_float16_t_type(void)
420 return glsl_type::float16_t_type
;
424 glsl_vec_type(unsigned n
)
426 return glsl_type::vec(n
);
430 glsl_dvec_type(unsigned n
)
432 return glsl_type::dvec(n
);
438 return glsl_type::vec4_type
;
442 glsl_uvec4_type(void)
444 return glsl_type::uvec4_type
;
450 return glsl_type::int_type
;
456 return glsl_type::uint_type
;
460 glsl_int64_t_type(void)
462 return glsl_type::int64_t_type
;
466 glsl_uint64_t_type(void)
468 return glsl_type::uint64_t_type
;
472 glsl_int16_t_type(void)
474 return glsl_type::int16_t_type
;
478 glsl_uint16_t_type(void)
480 return glsl_type::uint16_t_type
;
484 glsl_int8_t_type(void)
486 return glsl_type::int8_t_type
;
490 glsl_uint8_t_type(void)
492 return glsl_type::uint8_t_type
;
498 return glsl_type::bool_type
;
502 glsl_scalar_type(enum glsl_base_type base_type
)
504 return glsl_type::get_instance(base_type
, 1, 1);
508 glsl_vector_type(enum glsl_base_type base_type
, unsigned components
)
510 const glsl_type
*t
= glsl_type::get_instance(base_type
, components
, 1);
511 assert(t
!= glsl_type::error_type
);
516 glsl_matrix_type(enum glsl_base_type base_type
, unsigned rows
, unsigned columns
)
518 const glsl_type
*t
= glsl_type::get_instance(base_type
, rows
, columns
);
519 assert(t
!= glsl_type::error_type
);
524 glsl_explicit_matrix_type(const glsl_type
*mat
,
525 unsigned stride
, bool row_major
)
528 const glsl_type
*t
= glsl_type::get_instance(mat
->base_type
,
529 mat
->vector_elements
,
532 assert(t
!= glsl_type::error_type
);
537 glsl_array_type(const glsl_type
*base
, unsigned elements
,
538 unsigned explicit_stride
)
540 return glsl_type::get_array_instance(base
, elements
, explicit_stride
);
544 glsl_replace_vector_type(const glsl_type
*t
, unsigned components
)
546 if (glsl_type_is_array(t
)) {
547 return glsl_array_type(
548 glsl_replace_vector_type(t
->fields
.array
, components
), t
->length
,
550 } else if (glsl_type_is_vector_or_scalar(t
)) {
551 return glsl_vector_type(t
->base_type
, components
);
553 unreachable("Unhandled base type glsl_replace_vector_type()");
558 glsl_struct_type(const glsl_struct_field
*fields
,
559 unsigned num_fields
, const char *name
,
562 return glsl_type::get_struct_instance(fields
, num_fields
, name
, packed
);
566 glsl_interface_type(const glsl_struct_field
*fields
,
568 enum glsl_interface_packing packing
,
570 const char *block_name
)
572 return glsl_type::get_interface_instance(fields
, num_fields
, packing
,
573 row_major
, block_name
);
576 const struct glsl_type
*
577 glsl_sampler_type(enum glsl_sampler_dim dim
, bool is_shadow
, bool is_array
,
578 enum glsl_base_type base_type
)
580 return glsl_type::get_sampler_instance(dim
, is_shadow
, is_array
, base_type
);
583 const struct glsl_type
*
584 glsl_bare_sampler_type()
586 return glsl_type::sampler_type
;
589 const struct glsl_type
*
590 glsl_image_type(enum glsl_sampler_dim dim
, bool is_array
,
591 enum glsl_base_type base_type
)
593 return glsl_type::get_image_instance(dim
, is_array
, base_type
);
597 glsl_function_type(const glsl_type
*return_type
,
598 const glsl_function_param
*params
, unsigned num_params
)
600 return glsl_type::get_function_instance(return_type
, params
, num_params
);
604 glsl_transposed_type(const struct glsl_type
*type
)
606 assert(glsl_type_is_matrix(type
));
607 return glsl_type::get_instance(type
->base_type
, type
->matrix_columns
,
608 type
->vector_elements
);
612 glsl_channel_type(const glsl_type
*t
)
614 switch (t
->base_type
) {
615 case GLSL_TYPE_ARRAY
:
616 return glsl_array_type(glsl_channel_type(t
->fields
.array
), t
->length
,
620 case GLSL_TYPE_FLOAT
:
621 case GLSL_TYPE_FLOAT16
:
622 case GLSL_TYPE_DOUBLE
:
623 case GLSL_TYPE_UINT8
:
625 case GLSL_TYPE_UINT16
:
626 case GLSL_TYPE_INT16
:
627 case GLSL_TYPE_UINT64
:
628 case GLSL_TYPE_INT64
:
630 return glsl_type::get_instance(t
->base_type
, 1, 1);
632 unreachable("Unhandled base type glsl_channel_type()");
637 glsl_float16_type(const struct glsl_type
*type
)
639 return type
->get_float16_type();
643 glsl_int16_type(const struct glsl_type
*type
)
645 return type
->get_int16_type();
649 glsl_uint16_type(const struct glsl_type
*type
)
651 return type
->get_uint16_type();
655 glsl_get_natural_size_align_bytes(const struct glsl_type
*type
,
656 unsigned *size
, unsigned *align
)
658 switch (type
->base_type
) {
660 /* We special-case Booleans to 32 bits to not cause heartburn for
661 * drivers that suddenly get an 8-bit load.
663 *size
= 4 * type
->components();
667 case GLSL_TYPE_UINT8
:
669 case GLSL_TYPE_UINT16
:
670 case GLSL_TYPE_INT16
:
671 case GLSL_TYPE_FLOAT16
:
674 case GLSL_TYPE_FLOAT
:
675 case GLSL_TYPE_DOUBLE
:
676 case GLSL_TYPE_UINT64
:
677 case GLSL_TYPE_INT64
: {
678 unsigned N
= glsl_get_bit_size(type
) / 8;
679 *size
= N
* type
->components();
684 case GLSL_TYPE_ARRAY
: {
685 unsigned elem_size
= 0, elem_align
= 0;
686 glsl_get_natural_size_align_bytes(type
->fields
.array
,
687 &elem_size
, &elem_align
);
689 *size
= type
->length
* ALIGN_POT(elem_size
, elem_align
);
693 case GLSL_TYPE_STRUCT
:
696 for (unsigned i
= 0; i
< type
->length
; i
++) {
697 unsigned elem_size
= 0, elem_align
= 0;
698 glsl_get_natural_size_align_bytes(type
->fields
.structure
[i
].type
,
699 &elem_size
, &elem_align
);
700 *align
= MAX2(*align
, elem_align
);
701 *size
= ALIGN_POT(*size
, elem_align
) + elem_size
;
705 case GLSL_TYPE_SAMPLER
:
706 case GLSL_TYPE_IMAGE
:
707 /* Bindless samplers and images. */
712 case GLSL_TYPE_ATOMIC_UINT
:
713 case GLSL_TYPE_SUBROUTINE
:
715 case GLSL_TYPE_ERROR
:
716 case GLSL_TYPE_INTERFACE
:
717 case GLSL_TYPE_FUNCTION
:
718 unreachable("type does not have a natural size");
723 glsl_atomic_uint_type(void)
725 return glsl_type::atomic_uint_type
;
729 glsl_atomic_size(const struct glsl_type
*type
)
731 return type
->atomic_size();
735 glsl_contains_atomic(const struct glsl_type
*type
)
737 return type
->contains_atomic();
741 glsl_contains_opaque(const struct glsl_type
*type
)
743 return type
->contains_opaque();
747 glsl_get_cl_size(const struct glsl_type
*type
)
749 return type
->cl_size();
753 glsl_get_cl_alignment(const struct glsl_type
*type
)
755 return type
->cl_alignment();
759 glsl_type_get_sampler_count(const struct glsl_type
*type
)
761 if (glsl_type_is_array(type
)) {
762 return (glsl_get_aoa_size(type
) *
763 glsl_type_get_sampler_count(glsl_without_array(type
)));
766 /* Ignore interface blocks - they can only contain bindless samplers,
767 * which we shouldn't count.
769 if (glsl_type_is_struct(type
)) {
771 for (unsigned i
= 0; i
< glsl_get_length(type
); i
++)
772 count
+= glsl_type_get_sampler_count(glsl_get_struct_field(type
, i
));
776 if (glsl_type_is_sampler(type
))
783 glsl_type_get_image_count(const struct glsl_type
*type
)
785 if (glsl_type_is_array(type
)) {
786 return (glsl_get_aoa_size(type
) *
787 glsl_type_get_image_count(glsl_without_array(type
)));
790 /* Ignore interface blocks - they can only contain bindless images,
791 * which we shouldn't count.
793 if (glsl_type_is_struct(type
)) {
795 for (unsigned i
= 0; i
< glsl_get_length(type
); i
++)
796 count
+= glsl_type_get_image_count(glsl_get_struct_field(type
, i
));
800 if (glsl_type_is_image(type
))
806 enum glsl_interface_packing
807 glsl_get_internal_ifc_packing(const struct glsl_type
*type
,
808 bool std430_supported
)
810 return type
->get_internal_ifc_packing(std430_supported
);
813 enum glsl_interface_packing
814 glsl_get_ifc_packing(const struct glsl_type
*type
)
816 return type
->get_interface_packing();
820 glsl_get_std140_base_alignment(const struct glsl_type
*type
, bool row_major
)
822 return type
->std140_base_alignment(row_major
);
826 glsl_get_std140_size(const struct glsl_type
*type
, bool row_major
)
828 return type
->std140_size(row_major
);
832 glsl_get_std430_base_alignment(const struct glsl_type
*type
, bool row_major
)
834 return type
->std430_base_alignment(row_major
);
838 glsl_get_std430_size(const struct glsl_type
*type
, bool row_major
)
840 return type
->std430_size(row_major
);
844 glsl_get_explicit_size(const struct glsl_type
*type
, bool align_to_stride
)
846 return type
->explicit_size(align_to_stride
);
850 glsl_type_is_leaf(const struct glsl_type
*type
)
852 if (glsl_type_is_struct_or_ifc(type
) ||
853 (glsl_type_is_array(type
) &&
854 (glsl_type_is_array(glsl_get_array_element(type
)) ||
855 glsl_type_is_struct_or_ifc(glsl_get_array_element(type
))))) {
862 const struct glsl_type
*
863 glsl_get_explicit_type_for_size_align(const struct glsl_type
*type
,
864 glsl_type_size_align_func type_info
,
865 unsigned *size
, unsigned *align
)
867 return type
->get_explicit_type_for_size_align(type_info
, size
, align
);