void
nir_lower_alpha_test(nir_shader *shader, enum compare_func func,
- bool alpha_to_one)
+ bool alpha_to_one,
+ const gl_state_index16 *alpha_ref_state_tokens)
{
assert(shader->info.stage == MESA_SHADER_FRAGMENT);
- nir_assert_lowered_derefs(shader, nir_lower_load_store_derefs);
-
nir_foreach_function(function, shader) {
nir_function_impl *impl = function->impl;
nir_builder b;
nir_variable *out = NULL;
switch (intr->intrinsic) {
- case nir_intrinsic_store_var:
- out = intr->variables[0]->var;
+ case nir_intrinsic_store_deref:
+ out = nir_deref_instr_get_variable(nir_src_as_deref(intr->src[0]));
break;
case nir_intrinsic_store_output:
/* already had i/o lowered.. lookup the matching output var: */
- nir_foreach_variable(var, &shader->outputs) {
+ nir_foreach_shader_out_variable(var, shader) {
int drvloc = var->data.driver_location;
if (nir_intrinsic_base(intr) == drvloc) {
out = var;
nir_ssa_def *alpha;
if (alpha_to_one) {
alpha = nir_imm_float(&b, 1.0);
+ } else if (intr->intrinsic == nir_intrinsic_store_deref) {
+ alpha = nir_channel(&b, nir_ssa_for_src(&b, intr->src[1], 4),
+ 3);
} else {
alpha = nir_channel(&b, nir_ssa_for_src(&b, intr->src[0], 4),
3);
}
+ nir_ssa_def *alpha_ref;
+ if (alpha_ref_state_tokens) {
+ nir_variable *var = nir_variable_create(shader,
+ nir_var_uniform,
+ glsl_float_type(),
+ "gl_AlphaRefMESA");
+ var->num_state_slots = 1;
+ var->state_slots = ralloc_array(var, nir_state_slot, 1);
+ memcpy(var->state_slots[0].tokens,
+ alpha_ref_state_tokens,
+ sizeof(var->state_slots[0].tokens));
+ alpha_ref = nir_load_var(&b, var);
+ } else
+ alpha_ref = nir_load_alpha_ref_float(&b);
+
nir_ssa_def *condition =
- nir_compare_func(&b, func,
- alpha, nir_load_alpha_ref_float(&b));
+ nir_compare_func(&b, func, alpha, alpha_ref);
nir_intrinsic_instr *discard =
nir_intrinsic_instr_create(b.shader,