forked from suyu/suyu
glsl: Implement geometry shaders
This commit is contained in:
parent
df53046d68
commit
19247ba4fa
4 changed files with 62 additions and 9 deletions
|
@ -21,6 +21,15 @@ std::string_view InterpDecorator(Interpolation interp) {
|
||||||
throw InvalidArgument("Invalid interpolation {}", interp);
|
throw InvalidArgument("Invalid interpolation {}", interp);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::string_view ArrayDecorator(Stage stage) {
|
||||||
|
switch (stage) {
|
||||||
|
case Stage::Geometry:
|
||||||
|
return "[1]";
|
||||||
|
default:
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
std::string_view SamplerType(TextureType type, bool is_depth) {
|
std::string_view SamplerType(TextureType type, bool is_depth) {
|
||||||
if (is_depth) {
|
if (is_depth) {
|
||||||
switch (type) {
|
switch (type) {
|
||||||
|
@ -64,6 +73,33 @@ std::string_view SamplerType(TextureType type, bool is_depth) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::string_view InputPrimitive(InputTopology topology) {
|
||||||
|
switch (topology) {
|
||||||
|
case InputTopology::Points:
|
||||||
|
return "points";
|
||||||
|
case InputTopology::Lines:
|
||||||
|
return "lines";
|
||||||
|
case InputTopology::LinesAdjacency:
|
||||||
|
return "lines_adjacency";
|
||||||
|
case InputTopology::Triangles:
|
||||||
|
return "triangles";
|
||||||
|
case InputTopology::TrianglesAdjacency:
|
||||||
|
return "triangles_adjacency";
|
||||||
|
}
|
||||||
|
throw InvalidArgument("Invalid input topology {}", topology);
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string_view OutputPrimitive(OutputTopology topology) {
|
||||||
|
switch (topology) {
|
||||||
|
case OutputTopology::PointList:
|
||||||
|
return "points";
|
||||||
|
case OutputTopology::LineStrip:
|
||||||
|
return "line_strip";
|
||||||
|
case OutputTopology::TriangleStrip:
|
||||||
|
return "triangle_strip";
|
||||||
|
}
|
||||||
|
throw InvalidArgument("Invalid output topology {}", topology);
|
||||||
|
}
|
||||||
} // namespace
|
} // namespace
|
||||||
|
|
||||||
EmitContext::EmitContext(IR::Program& program, Bindings& bindings, const Profile& profile_,
|
EmitContext::EmitContext(IR::Program& program, Bindings& bindings, const Profile& profile_,
|
||||||
|
@ -85,6 +121,9 @@ EmitContext::EmitContext(IR::Program& program, Bindings& bindings, const Profile
|
||||||
break;
|
break;
|
||||||
case Stage::Geometry:
|
case Stage::Geometry:
|
||||||
stage_name = "gs";
|
stage_name = "gs";
|
||||||
|
header += fmt::format("layout({})in;layout({}, max_vertices={})out;\n",
|
||||||
|
InputPrimitive(runtime_info.input_topology),
|
||||||
|
OutputPrimitive(program.output_topology), program.output_vertices);
|
||||||
break;
|
break;
|
||||||
case Stage::Fragment:
|
case Stage::Fragment:
|
||||||
stage_name = "fs";
|
stage_name = "fs";
|
||||||
|
@ -99,8 +138,9 @@ EmitContext::EmitContext(IR::Program& program, Bindings& bindings, const Profile
|
||||||
for (size_t index = 0; index < info.input_generics.size(); ++index) {
|
for (size_t index = 0; index < info.input_generics.size(); ++index) {
|
||||||
const auto& generic{info.input_generics[index]};
|
const auto& generic{info.input_generics[index]};
|
||||||
if (generic.used) {
|
if (generic.used) {
|
||||||
header += fmt::format("layout(location={}) {} in vec4 in_attr{};", index,
|
header +=
|
||||||
InterpDecorator(generic.interpolation), index);
|
fmt::format("layout(location={}){} in vec4 in_attr{}{};", index,
|
||||||
|
InterpDecorator(generic.interpolation), index, ArrayDecorator(stage));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
for (size_t index = 0; index < info.stores_frag_color.size(); ++index) {
|
for (size_t index = 0; index < info.stores_frag_color.size(); ++index) {
|
||||||
|
@ -126,8 +166,6 @@ EmitContext::EmitContext(IR::Program& program, Bindings& bindings, const Profile
|
||||||
void EmitContext::SetupExtensions(std::string&) {
|
void EmitContext::SetupExtensions(std::string&) {
|
||||||
// TODO: track this usage
|
// TODO: track this usage
|
||||||
header += "#extension GL_ARB_sparse_texture2 : enable\n";
|
header += "#extension GL_ARB_sparse_texture2 : enable\n";
|
||||||
header += "#extension GL_ARB_shader_viewport_layer_array : enable\n";
|
|
||||||
header += "#extension GL_NV_viewport_array2 : enable\n";
|
|
||||||
header += "#extension GL_EXT_texture_shadow_lod : enable\n";
|
header += "#extension GL_EXT_texture_shadow_lod : enable\n";
|
||||||
if (info.uses_int64) {
|
if (info.uses_int64) {
|
||||||
header += "#extension GL_ARB_gpu_shader_int64 : enable\n";
|
header += "#extension GL_ARB_gpu_shader_int64 : enable\n";
|
||||||
|
@ -157,6 +195,10 @@ void EmitContext::SetupExtensions(std::string&) {
|
||||||
header += "#extension GL_ARB_gpu_shader_int64 : enable\n";
|
header += "#extension GL_ARB_gpu_shader_int64 : enable\n";
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if (info.stores_viewport_index) {
|
||||||
|
header += "#extension GL_ARB_shader_viewport_layer_array : enable\n";
|
||||||
|
header += "#extension GL_NV_viewport_array2 : enable\n";
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void EmitContext::DefineConstantBuffers(Bindings& bindings) {
|
void EmitContext::DefineConstantBuffers(Bindings& bindings) {
|
||||||
|
|
|
@ -162,7 +162,7 @@ void EmitCode(EmitContext& ctx, const IR::Program& program) {
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
fmt::print("{}", node.type);
|
fmt::print("{}", node.type);
|
||||||
throw NotImplementedException("{}", node.type);
|
throw NotImplementedException("AbstractSyntaxNode::Type {}", node.type);
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -19,6 +19,15 @@ u32 CbufIndex(u32 offset) {
|
||||||
char OffsetSwizzle(u32 offset) {
|
char OffsetSwizzle(u32 offset) {
|
||||||
return SWIZZLE[CbufIndex(offset)];
|
return SWIZZLE[CbufIndex(offset)];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool IsInputArray(Stage stage) {
|
||||||
|
return stage == Stage::Geometry || stage == Stage::TessellationControl ||
|
||||||
|
stage == Stage::TessellationEval;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string VertexIndex(EmitContext& ctx, std::string_view vertex) {
|
||||||
|
return IsInputArray(ctx.stage) ? fmt::format("[{}]", vertex) : "";
|
||||||
|
}
|
||||||
} // namespace
|
} // namespace
|
||||||
|
|
||||||
void EmitGetCbufU8([[maybe_unused]] EmitContext& ctx, [[maybe_unused]] IR::Inst& inst,
|
void EmitGetCbufU8([[maybe_unused]] EmitContext& ctx, [[maybe_unused]] IR::Inst& inst,
|
||||||
|
@ -128,7 +137,7 @@ void EmitGetAttribute(EmitContext& ctx, IR::Inst& inst, IR::Attribute attr,
|
||||||
const char swizzle{"xyzw"[element]};
|
const char swizzle{"xyzw"[element]};
|
||||||
if (IR::IsGeneric(attr)) {
|
if (IR::IsGeneric(attr)) {
|
||||||
const u32 index{IR::GenericAttributeIndex(attr)};
|
const u32 index{IR::GenericAttributeIndex(attr)};
|
||||||
ctx.AddF32("{}=in_attr{}.{};", inst, index, swizzle);
|
ctx.AddF32("{}=in_attr{}{}.{};", inst, index, VertexIndex(ctx, vertex), swizzle);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
switch (attr) {
|
switch (attr) {
|
||||||
|
@ -139,9 +148,11 @@ void EmitGetAttribute(EmitContext& ctx, IR::Inst& inst, IR::Attribute attr,
|
||||||
switch (ctx.stage) {
|
switch (ctx.stage) {
|
||||||
case Stage::VertexA:
|
case Stage::VertexA:
|
||||||
case Stage::VertexB:
|
case Stage::VertexB:
|
||||||
case Stage::Geometry:
|
|
||||||
ctx.AddF32("{}=gl_Position.{};", inst, swizzle);
|
ctx.AddF32("{}=gl_Position.{};", inst, swizzle);
|
||||||
break;
|
break;
|
||||||
|
case Stage::Geometry:
|
||||||
|
ctx.AddF32("{}=gl_in[{}].gl_Position.{};", inst, vertex, swizzle);
|
||||||
|
break;
|
||||||
case Stage::Fragment:
|
case Stage::Fragment:
|
||||||
ctx.AddF32("{}=gl_FragCoord.{};", inst, swizzle);
|
ctx.AddF32("{}=gl_FragCoord.{};", inst, swizzle);
|
||||||
break;
|
break;
|
||||||
|
|
|
@ -103,11 +103,11 @@ void EmitEpilogue(EmitContext& ctx) {
|
||||||
}
|
}
|
||||||
|
|
||||||
void EmitEmitVertex(EmitContext& ctx, const IR::Value& stream) {
|
void EmitEmitVertex(EmitContext& ctx, const IR::Value& stream) {
|
||||||
NotImplemented();
|
ctx.Add("EmitStreamVertex(int({}));", ctx.var_alloc.Consume(stream));
|
||||||
}
|
}
|
||||||
|
|
||||||
void EmitEndPrimitive(EmitContext& ctx, const IR::Value& stream) {
|
void EmitEndPrimitive(EmitContext& ctx, const IR::Value& stream) {
|
||||||
NotImplemented();
|
ctx.Add("EndStreamPrimitive(int({}));", ctx.var_alloc.Consume(stream));
|
||||||
}
|
}
|
||||||
|
|
||||||
void EmitGetRegister(EmitContext& ctx) {
|
void EmitGetRegister(EmitContext& ctx) {
|
||||||
|
|
Loading…
Reference in a new issue