2 * © Copyright 2018 Alyssa Rosenzweig
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 FROM,
20 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
28 #include "pan_context.h"
30 #include "compiler/nir/nir.h"
31 #include "nir/tgsi_to_nir.h"
32 #include "midgard/midgard_compile.h"
33 #include "util/u_dynarray.h"
35 #include "tgsi/tgsi_dump.h"
38 panfrost_shader_compile(struct panfrost_context
*ctx
, struct mali_shader_meta
*meta
, const char *src
, int type
, struct panfrost_shader_state
*state
)
44 struct pipe_shader_state
*cso
= state
->base
;
46 if (cso
->type
== PIPE_SHADER_IR_NIR
) {
47 s
= nir_shader_clone(NULL
, cso
->ir
.nir
);
49 assert (cso
->type
== PIPE_SHADER_IR_TGSI
);
50 //tgsi_dump(cso->tokens, 0);
51 s
= tgsi_to_nir(cso
->tokens
, ctx
->base
.screen
);
54 s
->info
.stage
= type
== JOB_TYPE_VERTEX
? MESA_SHADER_VERTEX
: MESA_SHADER_FRAGMENT
;
56 if (s
->info
.stage
== MESA_SHADER_FRAGMENT
) {
57 /* Inject the alpha test now if we need to */
59 if (state
->alpha_state
.enabled
) {
60 NIR_PASS_V(s
, nir_lower_alpha_test
, state
->alpha_state
.func
, false);
64 /* Call out to Midgard compiler given the above NIR */
66 midgard_program program
= {
67 .alpha_ref
= state
->alpha_state
.ref_value
70 midgard_compile_shader_nir(&ctx
->compiler
, s
, &program
, false);
72 /* Prepare the compiled binary for upload */
73 int size
= program
.compiled
.size
;
74 dst
= program
.compiled
.data
;
76 /* Upload the shader. The lookahead tag is ORed on as a tagged pointer.
77 * I bet someone just thought that would be a cute pun. At least,
78 * that's how I'd do it. */
80 meta
->shader
= panfrost_upload(&ctx
->shaders
, dst
, size
) | program
.first_tag
;
82 util_dynarray_fini(&program
.compiled
);
84 /* Sysvals are prepended */
85 program
.uniform_count
+= program
.sysval_count
;
86 state
->sysval_count
= program
.sysval_count
;
87 memcpy(state
->sysval
, program
.sysvals
, sizeof(state
->sysval
[0]) * state
->sysval_count
);
89 meta
->midgard1
.uniform_count
= MIN2(program
.uniform_count
, program
.uniform_cutoff
);
90 meta
->midgard1
.work_count
= program
.work_register_count
;
92 switch (s
->info
.stage
) {
93 case MESA_SHADER_VERTEX
:
94 meta
->attribute_count
= util_bitcount64(s
->info
.inputs_read
);
95 meta
->varying_count
= util_bitcount64(s
->info
.outputs_written
);
97 case MESA_SHADER_FRAGMENT
:
98 meta
->attribute_count
= 0;
99 meta
->varying_count
= util_bitcount64(s
->info
.inputs_read
);
102 unreachable("Unknown shader state");
105 state
->can_discard
= s
->info
.fs
.uses_discard
;
106 state
->writes_point_size
= s
->info
.outputs_written
& VARYING_SLOT_PSIZ
;
107 state
->reads_point_coord
= s
->info
.inputs_read
& VARYING_SLOT_PNTC
;
108 state
->helper_invocations
= s
->info
.fs
.needs_helper_invocations
;
110 /* Separate as primary uniform count is truncated */
111 state
->uniform_count
= program
.uniform_count
;
113 meta
->midgard1
.unknown2
= 8; /* XXX */
115 unsigned default_vec1_swizzle
= panfrost_get_default_swizzle(1);
116 unsigned default_vec2_swizzle
= panfrost_get_default_swizzle(2);
117 unsigned default_vec4_swizzle
= panfrost_get_default_swizzle(4);
119 /* Iterate the varyings and emit the corresponding descriptor */
120 for (unsigned i
= 0; i
< meta
->varying_count
; ++i
) {
121 unsigned location
= program
.varyings
[i
];
123 /* Default to a vec4 varying */
124 struct mali_attr_meta v
= {
125 .format
= MALI_RGBA32F
,
126 .swizzle
= default_vec4_swizzle
,
130 /* Check for special cases, otherwise assume general varying */
132 if (location
== VARYING_SLOT_POS
) {
134 v
.format
= MALI_VARYING_POS
;
135 } else if (location
== VARYING_SLOT_PSIZ
) {
137 v
.format
= MALI_R16F
;
138 v
.swizzle
= default_vec1_swizzle
;
139 } else if (location
== VARYING_SLOT_PNTC
) {
141 v
.format
= MALI_RG16F
;
142 v
.swizzle
= default_vec2_swizzle
;
147 state
->varyings
[i
] = v
;
148 state
->varyings_loc
[i
] = location
;