1 files changed, 71 insertions, 0 deletions
diff --git a/Ryujinx.Graphics.Nvdec.Vp9/Types/Segmentation.cs b/Ryujinx.Graphics.Nvdec.Vp9/Types/Segmentation.cs
new file mode 100644
index 00000000..53d1f2cc
--- /dev/null
+++ b/Ryujinx.Graphics.Nvdec.Vp9/Types/Segmentation.cs
@@ -0,0 +1,71 @@
+using Ryujinx.Common.Memory;
+using System.Diagnostics;
+using System.Runtime.InteropServices;
+
+namespace Ryujinx.Graphics.Nvdec.Vp9.Types
+{
+    internal struct Segmentation
+    {
+        private static readonly int[] SegFeatureDataSigned = new int[] { 1, 1, 0, 0 };
+        private static readonly int[] SegFeatureDataMax = new int[] { QuantCommon.MaxQ, Vp9.LoopFilter.MaxLoopFilter, 3, 0 };
+
+        public bool Enabled;
+        public bool UpdateMap;
+        public byte UpdateData;
+        public byte AbsDelta;
+        public bool TemporalUpdate;
+
+        public Array8<Array4<short>> FeatureData;
+        public Array8<uint> FeatureMask;
+        public int AqAvOffset;
+
+        public static byte GetPredProbSegId(ref Array3<byte> segPredProbs, ref MacroBlockD xd)
+        {
+            return segPredProbs[xd.GetPredContextSegId()];
+        }
+
+        public void ClearAllSegFeatures()
+        {
+            MemoryMarshal.CreateSpan(ref FeatureData[0][0], 8 * 4).Fill(0);
+            MemoryMarshal.CreateSpan(ref FeatureMask[0], 8).Fill(0);
+            AqAvOffset = 0;
+        }
+
+        internal void EnableSegFeature(int segmentId, SegLvlFeatures featureId)
+        {
+            FeatureMask[segmentId] |= 1u << (int)featureId;
+        }
+
+        internal static int FeatureDataMax(SegLvlFeatures featureId)
+        {
+            return SegFeatureDataMax[(int)featureId];
+        }
+
+        internal static int IsSegFeatureSigned(SegLvlFeatures featureId)
+        {
+            return SegFeatureDataSigned[(int)featureId];
+        }
+
+        internal void SetSegData(int segmentId, SegLvlFeatures featureId, int segData)
+        {
+            Debug.Assert(segData <= SegFeatureDataMax[(int)featureId]);
+            if (segData < 0)
+            {
+                Debug.Assert(SegFeatureDataSigned[(int)featureId] != 0);
+                Debug.Assert(-segData <= SegFeatureDataMax[(int)featureId]);
+            }
+
+            FeatureData[segmentId][(int)featureId] = (short)segData;
+        }
+
+        internal int IsSegFeatureActive(int segmentId, SegLvlFeatures featureId)
+        {
+            return Enabled && (FeatureMask[segmentId] & (1 << (int)featureId)) != 0 ? 1 : 0;
+        }
+
+        internal short GetSegData(int segmentId, SegLvlFeatures featureId)
+        {
+            return FeatureData[segmentId][(int)featureId];
+        }
+    }
+}