1 files changed, 320 insertions, 0 deletions
diff --git a/Ryujinx.Audio.Renderer/Server/Splitter/SplitterContext.cs b/Ryujinx.Audio.Renderer/Server/Splitter/SplitterContext.cs
new file mode 100644
index 00000000..f5599acc
--- /dev/null
+++ b/Ryujinx.Audio.Renderer/Server/Splitter/SplitterContext.cs
@@ -0,0 +1,320 @@
+//
+// Copyright (c) 2019-2020 Ryujinx
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Lesser General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+//
+
+using Ryujinx.Audio.Renderer.Common;
+using Ryujinx.Audio.Renderer.Parameter;
+using Ryujinx.Audio.Renderer.Utils;
+using Ryujinx.Common;
+using System;
+using System.Diagnostics;
+using System.Runtime.CompilerServices;
+using System.Runtime.InteropServices;
+
+namespace Ryujinx.Audio.Renderer.Server.Splitter
+{
+    /// <summary>
+    /// Splitter context.
+    /// </summary>
+    public class SplitterContext
+    {
+        /// <summary>
+        /// Storage for <see cref="SplitterState"/>.
+        /// </summary>
+        private Memory<SplitterState> _splitters;
+
+        /// <summary>
+        /// Storage for <see cref="SplitterDestination"/>.
+        /// </summary>
+        private Memory<SplitterDestination> _splitterDestinations;
+
+        /// <summary>
+        /// If set to true, trust the user destination count in <see cref="SplitterState.Update(SplitterContext, ref SplitterInParameter, ReadOnlySpan{byte})"/>.
+        /// </summary>
+        public bool IsBugFixed { get; private set; }
+
+        /// <summary>
+        /// Initialize <see cref="SplitterContext"/>.
+        /// </summary>
+        /// <param name="behaviourContext">The behaviour context.</param>
+        /// <param name="parameter">The audio renderer configuration.</param>
+        /// <param name="workBufferAllocator">The <see cref="WorkBufferAllocator"/>.</param>
+        /// <returns>Return true if the initialization was successful.</returns>
+        public bool Initialize(ref BehaviourContext behaviourContext, ref AudioRendererConfiguration parameter, WorkBufferAllocator workBufferAllocator)
+        {
+            if (!behaviourContext.IsSplitterSupported() || parameter.SplitterCount <= 0 || parameter.SplitterDestinationCount <= 0)
+            {
+                Setup(Memory<SplitterState>.Empty, Memory<SplitterDestination>.Empty, false);
+
+                return true;
+            }
+
+            Memory<SplitterState> splitters = workBufferAllocator.Allocate<SplitterState>(parameter.SplitterCount, SplitterState.Alignment);
+
+            if (splitters.IsEmpty)
+            {
+                return false;
+            }
+
+            int splitterId = 0;
+
+            foreach (ref SplitterState splitter in splitters.Span)
+            {
+                splitter = new SplitterState(splitterId++);
+            }
+
+            Memory<SplitterDestination> splitterDestinations = workBufferAllocator.Allocate<SplitterDestination>(parameter.SplitterDestinationCount,
+                SplitterDestination.Alignment);
+
+            if (splitterDestinations.IsEmpty)
+            {
+                return false;
+            }
+
+            int splitterDestinationId = 0;
+            foreach (ref SplitterDestination data in splitterDestinations.Span)
+            {
+                data = new SplitterDestination(splitterDestinationId++);
+            }
+
+            SplitterState.InitializeSplitters(splitters.Span);
+
+            Setup(splitters, splitterDestinations, behaviourContext.IsSplitterBugFixed());
+
+            return true;
+        }
+
+        /// <summary>
+        /// Get the work buffer size while adding the size needed for splitter to operate.
+        /// </summary>
+        /// <param name="size">The current size.</param>
+        /// <param name="behaviourContext">The behaviour context.</param>
+        /// <param name="parameter">The renderer configuration.</param>
+        /// <returns>Return the new size taking splitter into account.</returns>
+        public static ulong GetWorkBufferSize(ulong size, ref BehaviourContext behaviourContext, ref AudioRendererConfiguration parameter)
+        {
+            if (behaviourContext.IsSplitterSupported())
+            {
+                size = WorkBufferAllocator.GetTargetSize<SplitterState>(size, parameter.SplitterCount, SplitterState.Alignment);
+                size = WorkBufferAllocator.GetTargetSize<SplitterDestination>(size, parameter.SplitterDestinationCount, SplitterDestination.Alignment);
+
+                if (behaviourContext.IsSplitterBugFixed())
+                {
+                    size = WorkBufferAllocator.GetTargetSize<int>(size, parameter.SplitterDestinationCount, 0x10);
+                }
+
+                return size;
+            }
+            else
+            {
+                return size;
+            }
+        }
+
+        /// <summary>
+        /// Setup the <see cref="SplitterContext"/> instance.
+        /// </summary>
+        /// <param name="splitters">The <see cref="SplitterState"/> storage.</param>
+        /// <param name="splitterDestinations">The <see cref="SplitterDestination"/> storage.</param>
+        /// <param name="isBugFixed">If set to true, trust the user destination count in <see cref="SplitterState.Update(SplitterContext, ref SplitterInParameter, ReadOnlySpan{byte})"/>.</param>
+        private void Setup(Memory<SplitterState> splitters, Memory<SplitterDestination> splitterDestinations, bool isBugFixed)
+        {
+            _splitters = splitters;
+            _splitterDestinations = splitterDestinations;
+            IsBugFixed = isBugFixed;
+        }
+
+        /// <summary>
+        /// Clear the new connection flag.
+        /// </summary>
+        private void ClearAllNewConnectionFlag()
+        {
+            foreach (ref SplitterState splitter in _splitters.Span)
+            {
+                splitter.ClearNewConnectionFlag();
+            }
+        }
+
+        /// <summary>
+        /// Get the destination count using the count of splitter.
+        /// </summary>
+        /// <returns>The destination count using the count of splitter.</returns>
+        public int GetDestinationCountPerStateForCompatibility()
+        {
+            if (_splitters.IsEmpty)
+            {
+                return 0;
+            }
+
+            return _splitterDestinations.Length / _splitters.Length;
+        }
+
+        /// <summary>
+        /// Update one or multiple <see cref="SplitterState"/> from user parameters.
+        /// </summary>
+        /// <param name="inputHeader">The splitter header.</param>
+        /// <param name="input">The raw data after the splitter header.</param>
+        private void UpdateState(ref SplitterInParameterHeader inputHeader, ref ReadOnlySpan<byte> input)
+        {
+            for (int i = 0; i < inputHeader.SplitterCount; i++)
+            {
+                SplitterInParameter parameter = MemoryMarshal.Read<SplitterInParameter>(input);
+
+                Debug.Assert(parameter.IsMagicValid());
+
+                if (parameter.IsMagicValid())
+                {
+                    if (parameter.Id >= 0 && parameter.Id < _splitters.Length)
+                    {
+                        ref SplitterState splitter = ref GetState(parameter.Id);
+
+                        splitter.Update(this, ref parameter, input.Slice(Unsafe.SizeOf<SplitterInParameter>()));
+                    }
+
+                    input = input.Slice(0x1C + (int)parameter.DestinationCount * 4);
+                }
+            }
+        }
+
+        /// <summary>
+        /// Update one or multiple <see cref="SplitterDestination"/> from user parameters.
+        /// </summary>
+        /// <param name="inputHeader">The splitter header.</param>
+        /// <param name="input">The raw data after the splitter header.</param>
+        private void UpdateData(ref SplitterInParameterHeader inputHeader, ref ReadOnlySpan<byte> input)
+        {
+            for (int i = 0; i < inputHeader.SplitterDestinationCount; i++)
+            {
+                SplitterDestinationInParameter parameter = MemoryMarshal.Read<SplitterDestinationInParameter>(input);
+
+                Debug.Assert(parameter.IsMagicValid());
+
+                if (parameter.IsMagicValid())
+                {
+                    if (parameter.Id >= 0 && parameter.Id < _splitterDestinations.Length)
+                    {
+                        ref SplitterDestination destination = ref GetDestination(parameter.Id);
+
+                        destination.Update(parameter);
+                    }
+
+                    input = input.Slice(Unsafe.SizeOf<SplitterDestinationInParameter>());
+                }
+            }
+        }
+
+        /// <summary>
+        /// Update splitter from user parameters.
+        /// </summary>
+        /// <param name="input">The input raw user data.</param>
+        /// <param name="consumedSize">The total consumed size.</param>
+        /// <returns>Return true if the update was successful.</returns>
+        public bool Update(ReadOnlySpan<byte> input, out int consumedSize)
+        {
+            if (_splitterDestinations.IsEmpty || _splitters.IsEmpty)
+            {
+                consumedSize = 0;
+
+                return true;
+            }
+
+            int originalSize = input.Length;
+
+            SplitterInParameterHeader header = SpanIOHelper.Read<SplitterInParameterHeader>(ref input);
+
+            if (header.IsMagicValid())
+            {
+                ClearAllNewConnectionFlag();
+
+                UpdateState(ref header, ref input);
+                UpdateData(ref header, ref input);
+
+                consumedSize = BitUtils.AlignUp(originalSize - input.Length, 0x10);
+
+                return true;
+            }
+            else
+            {
+                consumedSize = 0;
+
+                return false;
+            }
+        }
+
+        /// <summary>
+        /// Get a reference to a <see cref="SplitterState"/> at the given <paramref name="id"/>.
+        /// </summary>
+        /// <param name="id">The index to use.</param>
+        /// <returns>A reference to a <see cref="SplitterState"/> at the given <paramref name="id"/>.</returns>
+        public ref SplitterState GetState(int id)
+        {
+            return ref SpanIOHelper.GetFromMemory(_splitters, id, (uint)_splitters.Length);
+        }
+
+        /// <summary>
+        /// Get a reference to a <see cref="SplitterDestination"/> at the given <paramref name="id"/>.
+        /// </summary>
+        /// <param name="id">The index to use.</param>
+        /// <returns>A reference to a <see cref="SplitterDestination"/> at the given <paramref name="id"/>.</returns>
+        public ref SplitterDestination GetDestination(int id)
+        {
+            return ref SpanIOHelper.GetFromMemory(_splitterDestinations, id, (uint)_splitterDestinations.Length);
+        }
+
+        /// <summary>
+        /// Get a <see cref="Memory{SplitterDestination}"/> at the given <paramref name="id"/>.
+        /// </summary>
+        /// <param name="id">The index to use.</param>
+        /// <returns>A <see cref="Memory{SplitterDestination}"/> at the given <paramref name="id"/>.</returns>
+        public Memory<SplitterDestination> GetDestinationMemory(int id)
+        {
+            return SpanIOHelper.GetMemory(_splitterDestinations, id, (uint)_splitterDestinations.Length);
+        }
+
+        /// <summary>
+        /// Get a <see cref="Span{SplitterDestination}"/> in the <see cref="SplitterState"/> at <paramref name="id"/> and pass <paramref name="destinationId"/> to <see cref="SplitterState.GetData(int)"/>.
+        /// </summary>
+        /// <param name="id">The index to use to get the <see cref="SplitterState"/>.</param>
+        /// <param name="destinationId">The index of the <see cref="SplitterDestination"/>.</param>
+        /// <returns>A <see cref="Span{SplitterDestination}"/>.</returns>
+        public Span<SplitterDestination> GetDestination(int id, int destinationId)
+        {
+            ref SplitterState splitter = ref GetState(id);
+
+            return splitter.GetData(destinationId);
+        }
+
+        /// <summary>
+        /// Return true if the audio renderer has any splitters.
+        /// </summary>
+        /// <returns>True if the audio renderer has any splitters.</returns>
+        public bool UsingSplitter()
+        {
+            return !_splitters.IsEmpty && !_splitterDestinations.IsEmpty;
+        }
+
+        /// <summary>
+        /// Update the internal state of all splitters.
+        /// </summary>
+        public void UpdateInternalState()
+        {
+            foreach (ref SplitterState splitter in _splitters.Span)
+            {
+                splitter.UpdateInternalState();
+            }
+        }
+    }
+}