From b7589fe1154d9e810e83f6c609dad1d646ec0359 Mon Sep 17 00:00:00 2001
From: ameerj <52414509+ameerj@users.noreply.github.com>
Date: Fri, 26 Mar 2021 18:52:06 -0400
Subject: shader: Add PointSize attribute

---
 src/shader_recompiler/backend/spirv/emit_context.cpp               | 6 ++++++
 src/shader_recompiler/backend/spirv/emit_context.h                 | 1 +
 src/shader_recompiler/backend/spirv/emit_spirv_context_get_set.cpp | 2 ++
 3 files changed, 9 insertions(+)

(limited to 'src/shader_recompiler/backend')

diff --git a/src/shader_recompiler/backend/spirv/emit_context.cpp b/src/shader_recompiler/backend/spirv/emit_context.cpp
index 7531f8b21..ecee1220e 100644
--- a/src/shader_recompiler/backend/spirv/emit_context.cpp
+++ b/src/shader_recompiler/backend/spirv/emit_context.cpp
@@ -492,6 +492,12 @@ void EmitContext::DefineOutputs(const Info& info) {
     if (info.stores_position || stage == Stage::VertexB) {
         output_position = DefineOutput(*this, F32[4], spv::BuiltIn::Position);
     }
+    if (info.stores_point_size) {
+        if (stage == Stage::Fragment) {
+            throw NotImplementedException("Storing PointSize in Fragment stage");
+        }
+        output_point_size = DefineOutput(*this, F32[1], spv::BuiltIn::PointSize);
+    }
     for (size_t i = 0; i < info.stores_generics.size(); ++i) {
         if (info.stores_generics[i]) {
             output_generics[i] = DefineOutput(*this, F32[4]);
diff --git a/src/shader_recompiler/backend/spirv/emit_context.h b/src/shader_recompiler/backend/spirv/emit_context.h
index ffac39c4f..97e055db4 100644
--- a/src/shader_recompiler/backend/spirv/emit_context.h
+++ b/src/shader_recompiler/backend/spirv/emit_context.h
@@ -120,6 +120,7 @@ public:
     Id input_position{};
     std::array<Id, 32> input_generics{};
 
+    Id output_point_size{};
     Id output_position{};
     std::array<Id, 32> output_generics{};
 
diff --git a/src/shader_recompiler/backend/spirv/emit_spirv_context_get_set.cpp b/src/shader_recompiler/backend/spirv/emit_spirv_context_get_set.cpp
index 4a267b16c..c870fac47 100644
--- a/src/shader_recompiler/backend/spirv/emit_spirv_context_get_set.cpp
+++ b/src/shader_recompiler/backend/spirv/emit_spirv_context_get_set.cpp
@@ -37,6 +37,8 @@ Id OutputAttrPointer(EmitContext& ctx, IR::Attribute attr) {
         return ctx.OpAccessChain(ctx.output_f32, ctx.output_generics.at(index), element_id());
     }
     switch (attr) {
+    case IR::Attribute::PointSize:
+        return ctx.output_point_size;
     case IR::Attribute::PositionX:
     case IR::Attribute::PositionY:
     case IR::Attribute::PositionZ:
-- 
cgit v1.2.3