mirror of
https://gitlab.freedesktop.org/mesa/mesa.git
synced 2026-05-18 09:28:06 +02:00
349219df6a
It has all the poly NIR things so it should have a more generic name. Acked-by: Alyssa Rosenzweig <alyssa.rosenzweig@intel.com> Reviewed-by: Mary Guillemard <mary@mary.zone> Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/38404>
59 lines
1.8 KiB
C
59 lines
1.8 KiB
C
/*
|
|
* Copyright 2023 Valve Corporation
|
|
* SPDX-License-Identifier: MIT
|
|
*/
|
|
|
|
#include "compiler/nir/nir_builder.h"
|
|
#include "poly/cl/libpoly.h"
|
|
#include "poly/nir/poly_nir.h"
|
|
#include "poly/geometry.h"
|
|
#include "nir.h"
|
|
|
|
/*
|
|
* This file implements basic input assembly in software. It runs on software
|
|
* vertex shaders, as part of geometry/tessellation lowering. It does not apply
|
|
* the topology, which happens in the geometry shader.
|
|
*/
|
|
nir_def *
|
|
poly_nir_load_vertex_id(nir_builder *b, nir_def *id, unsigned index_size_B)
|
|
{
|
|
/* If drawing with an index buffer, pull the vertex ID. Otherwise, the
|
|
* vertex ID is just the index as-is.
|
|
*/
|
|
if (index_size_B) {
|
|
nir_def *p = nir_load_vertex_param_buffer_poly(b);
|
|
id = poly_load_index_buffer(b, p, id, nir_imm_int(b, index_size_B));
|
|
}
|
|
|
|
/* Add the "start", either an index bias or a base vertex. This must happen
|
|
* after indexing for proper index bias behaviour.
|
|
*/
|
|
return nir_iadd(b, id, nir_load_first_vertex(b));
|
|
}
|
|
|
|
static bool
|
|
lower(nir_builder *b, nir_intrinsic_instr *intr, void *data)
|
|
{
|
|
unsigned *index_size_B = data;
|
|
b->cursor = nir_before_instr(&intr->instr);
|
|
|
|
if (intr->intrinsic == nir_intrinsic_load_vertex_id) {
|
|
nir_def *id = nir_channel(b, nir_load_global_invocation_id(b, 32), 0);
|
|
nir_def_replace(&intr->def,
|
|
poly_nir_load_vertex_id(b, id, *index_size_B));
|
|
return true;
|
|
} else if (intr->intrinsic == nir_intrinsic_load_instance_id) {
|
|
nir_def_replace(&intr->def,
|
|
nir_channel(b, nir_load_global_invocation_id(b, 32), 1));
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
poly_nir_lower_sw_vs(nir_shader *s, unsigned index_size_B)
|
|
{
|
|
return nir_shader_intrinsics_pass(s, lower, nir_metadata_control_flow,
|
|
&index_size_B);
|
|
}
|