1 files changed, 136 insertions, 0 deletions
diff --git a/Ryujinx.Audio.Renderer/Dsp/State/Reverb3dState.cs b/Ryujinx.Audio.Renderer/Dsp/State/Reverb3dState.cs
new file mode 100644
index 00000000..3b119b92
--- /dev/null
+++ b/Ryujinx.Audio.Renderer/Dsp/State/Reverb3dState.cs
@@ -0,0 +1,136 @@
+//
+// 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.Dsp.Effect;
+using Ryujinx.Audio.Renderer.Parameter.Effect;
+using System;
+
+namespace Ryujinx.Audio.Renderer.Dsp.State
+{
+    public class Reverb3dState
+    {
+        private readonly float[] FdnDelayMinTimes = new float[4] { 5.0f, 6.0f, 13.0f, 14.0f };
+        private readonly float[] FdnDelayMaxTimes = new float[4] { 45.704f, 82.782f, 149.94f, 271.58f };
+        private readonly float[] DecayDelayMaxTimes1 = new float[4] { 17.0f, 13.0f, 9.0f, 7.0f };
+        private readonly float[] DecayDelayMaxTimes2 = new float[4] { 19.0f, 11.0f, 10.0f, 6.0f };
+        private readonly float[] EarlyDelayTimes = new float[20] { 0.017136f, 0.059154f, 0.16173f, 0.39019f, 0.42526f, 0.45541f, 0.68974f, 0.74591f, 0.83384f, 0.8595f, 0.0f, 0.075024f, 0.16879f, 0.2999f, 0.33744f, 0.3719f, 0.59901f, 0.71674f, 0.81786f, 0.85166f };
+        public readonly float[] EarlyGain = new float[20] { 0.67096f, 0.61027f, 1.0f, 0.35680f, 0.68361f, 0.65978f, 0.51939f, 0.24712f, 0.45945f, 0.45021f, 0.64196f, 0.54879f, 0.92925f, 0.38270f, 0.72867f, 0.69794f, 0.5464f, 0.24563f, 0.45214f, 0.44042f };
+
+        public IDelayLine[] FdnDelayLines { get; }
+        public DecayDelay[] DecayDelays1 { get; }
+        public DecayDelay[] DecayDelays2 { get; }
+        public IDelayLine PreDelayLine { get; }
+        public IDelayLine BackLeftDelayLine { get; }
+        public float DryGain { get; private set; }
+        public uint[] EarlyDelayTime { get; private set; }
+        public float PreviousPreDelayValue { get; set; }
+        public float PreviousPreDelayGain { get; private set; }
+        public float TargetPreDelayGain { get; private set; }
+        public float EarlyReflectionsGain { get; private set; }
+        public float LateReverbGain { get; private set; }
+        public uint ReflectionDelayTime { get; private set; }
+        public float EchoLateReverbDecay { get; private set; }
+        public float[] DecayDirectFdnGain { get; private set; }
+        public float[] DecayCurrentFdnGain { get; private set; }
+        public float[] DecayCurrentOutputGain { get; private set; }
+        public float[] PreviousFeedbackOutputDecayed { get; private set; }
+
+        public Reverb3dState(ref Reverb3dParameter parameter, ulong workBuffer)
+        {
+            FdnDelayLines = new IDelayLine[4];
+            DecayDelays1 = new DecayDelay[4];
+            DecayDelays2 = new DecayDelay[4];
+            DecayDirectFdnGain = new float[4];
+            DecayCurrentFdnGain = new float[4];
+            DecayCurrentOutputGain = new float[4];
+            PreviousFeedbackOutputDecayed = new float[4];
+
+            uint sampleRate = parameter.SampleRate / 1000;
+
+            for (int i = 0; i < 4; i++)
+            {
+                FdnDelayLines[i] = new DelayLine3d(sampleRate, FdnDelayMaxTimes[i]);
+                DecayDelays1[i] = new DecayDelay(new DelayLine3d(sampleRate, DecayDelayMaxTimes1[i]));
+                DecayDelays2[i] = new DecayDelay(new DelayLine3d(sampleRate, DecayDelayMaxTimes2[i]));
+            }
+
+            PreDelayLine = new DelayLine3d(sampleRate, 400);
+            BackLeftDelayLine = new DelayLine3d(sampleRate, 5);
+
+            UpdateParameter(ref parameter);
+        }
+
+        public void UpdateParameter(ref Reverb3dParameter parameter)
+        {
+            uint sampleRate = parameter.SampleRate / 1000;
+
+            EarlyDelayTime = new uint[20];
+            DryGain = parameter.DryGain;
+            PreviousFeedbackOutputDecayed.AsSpan().Fill(0);
+            PreviousPreDelayValue = 0;
+
+            EarlyReflectionsGain = FloatingPointHelper.Pow10(Math.Min(parameter.RoomGain + parameter.ReflectionsGain, 5000.0f) / 2000.0f);
+            LateReverbGain = FloatingPointHelper.Pow10(Math.Min(parameter.RoomGain + parameter.ReverbGain, 5000.0f) / 2000.0f);
+
+            float highFrequencyRoomGain = FloatingPointHelper.Pow10(parameter.RoomHf / 2000.0f);
+
+            if (highFrequencyRoomGain < 1.0f)
+            {
+                float tempA = 1.0f - highFrequencyRoomGain;
+                float tempB = 2.0f - ((2.0f * highFrequencyRoomGain) * FloatingPointHelper.Cos(256.0f * parameter.HfReference / parameter.SampleRate));
+                float tempC = MathF.Sqrt(MathF.Pow(tempB, 2) - (4.0f * (1.0f - highFrequencyRoomGain) * (1.0f - highFrequencyRoomGain)));
+
+                PreviousPreDelayGain = (tempB - tempC) / (2.0f * tempA);
+                TargetPreDelayGain = 1.0f - PreviousPreDelayGain;
+            }
+            else
+            {
+                PreviousPreDelayGain = 0.0f;
+                TargetPreDelayGain = 1.0f;
+            }
+
+            ReflectionDelayTime = IDelayLine.GetSampleCount(sampleRate, 1000.0f * (parameter.ReflectionDelay + parameter.ReverbDelayTime));
+            EchoLateReverbDecay = 0.6f * parameter.Diffusion * 0.01f;
+
+            for (int i = 0; i < FdnDelayLines.Length; i++)
+            {
+                FdnDelayLines[i].SetDelay(FdnDelayMinTimes[i] + (parameter.Density / 100 * (FdnDelayMaxTimes[i] - FdnDelayMinTimes[i])));
+
+                uint tempSampleCount = FdnDelayLines[i].CurrentSampleCount + DecayDelays1[i].CurrentSampleCount + DecayDelays2[i].CurrentSampleCount;
+
+                float tempA = (-60.0f * tempSampleCount) / (parameter.DecayTime * parameter.SampleRate);
+                float tempB = tempA / parameter.HfDecayRatio;
+                float tempC = FloatingPointHelper.Cos(128.0f * 0.5f * parameter.HfReference / parameter.SampleRate) / FloatingPointHelper.Sin(128.0f * 0.5f * parameter.HfReference / parameter.SampleRate);
+                float tempD = FloatingPointHelper.Pow10((tempB - tempA) / 40.0f);
+                float tempE = FloatingPointHelper.Pow10((tempB + tempA) / 40.0f) * 0.7071f;
+
+                DecayDirectFdnGain[i] = tempE * ((tempD * tempC) + 1.0f) / (tempC + tempD);
+                DecayCurrentFdnGain[i] = tempE * (1.0f - (tempD * tempC)) / (tempC + tempD);
+                DecayCurrentOutputGain[i] = (tempC - tempD) / (tempC + tempD);
+
+                DecayDelays1[i].SetDecayRate(EchoLateReverbDecay);
+                DecayDelays2[i].SetDecayRate(EchoLateReverbDecay * -0.9f);
+            }
+
+            for (int i = 0; i < EarlyDelayTime.Length; i++)
+            {
+                uint sampleCount = Math.Min(IDelayLine.GetSampleCount(sampleRate, (parameter.ReflectionDelay * 1000.0f) + (EarlyDelayTimes[i] * 1000.0f * ((parameter.ReverbDelayTime * 0.9998f) + 0.02f))), PreDelayLine.SampleCountMax);
+                EarlyDelayTime[i] = sampleCount;
+            }
+        }
+    }
+}