[+] SoundRouting & HeadphoneVolumeMultiply (#75)

* [+] SoundRouting * [+] HeadphoneVolumeMultiply
2026-08-20 16:24:38 -05:00 · 2025-10-20 03:37:11 +08:00
parent f7a95c0af4
commit 0edc51ef65
3 changed files with 209 additions and 0 deletions
--- a/AquaMai.Config/Types/SoundChannel.cs
+++ b/AquaMai.Config/Types/SoundChannel.cs
@@ -0,0 +1,16 @@
+namespace AquaMai.Config.Types;
+
+// The enum values are the column indices in a 8x8 float BE matrix in Mai2.acf file
+// It's NOT consistent with WASAPI's channel indices (bits in dwChannelMask).
+public enum SoundChannel
+{
+    P1SpeakerLeft = 0, // Routes to WASAPI's SPEAKER_FRONT_LEFT (0x1 = 1 << 0)
+    P1SpeakerRight = 1, // Routes to WASAPI's SPEAKER_FRONT_RIGHT (0x2 = 1 << 1)
+    P1HeadphoneLeft = 2, // Routes to WASAPI's SPEAKER_FRONT_CENTER (0x4 = 1 << 2)
+    P1HeadphoneRight = 3, // Routes to WASAPI's SPEAKER_LOW_FREQUENCY (0x8 = 1 << 3)
+    P2SpeakerLeft = 6, // Routes to WASAPI's SPEAKER_BACK_LEFT (0x10) = 1 << 4)
+    P2SpeakerRight = 7, // Routes to WASAPI's SPEAKER_BACK_RIGHT (0x20) = 1 << 5)
+    P2HeadphoneLeft = 4, // Routes to WASAPI's SPEAKER_SIDE_LEFT (0x200 = 1 << 9)
+    P2HeadphoneRight = 5, // Routes to WASAPI's SPEAKER_SIDE_RIGHT (0x400 = 1 << 10)
+    None = 8,
+}
--- a/AquaMai.Mods/GameSystem/HeadphoneVolumeMultiply.cs
+++ b/AquaMai.Mods/GameSystem/HeadphoneVolumeMultiply.cs
@@ -0,0 +1,35 @@
+using HarmonyLib;
+
+using AquaMai.Config.Attributes;
+using UnityEngine;
+
+namespace AquaMai.Mods.GameSystem;
+
+[ConfigSection(
+    name: "耳机音量倍率",
+    en: "Multiply 1P/2P headphone volume by a factor. Can decrease or increase, but not higher than original 20.",
+    zh: "将 1P/2P 耳机音量乘以一个倍率，可以调低或调高，但不能高于原本的 20。")]
+public static class HeadphoneVolumeMultiply
+{
+    [ConfigEntry]
+    private static readonly float p1 = 1.0f;
+
+    [ConfigEntry]
+    private static readonly float p2 = 1.0f;
+
+    [HarmonyPrefix]
+    [HarmonyPatch(typeof(CriAtomExPlayer), "SetAisacControl", [typeof(uint), typeof(float)])]
+    public static void PreSetAisacControl(uint controlId, ref float value)
+    {
+        if (controlId == 2)
+        {
+            value *= p1;
+            value = Mathf.Min(value, 1.0f);
+        }
+        else if (controlId == 3)
+        {
+            value *= p2;
+            value = Mathf.Min(value, 1.0f);
+        }
+    }
+}
--- a/AquaMai.Mods/GameSystem/SoundRouting.cs
+++ b/AquaMai.Mods/GameSystem/SoundRouting.cs
@@ -0,0 +1,158 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using AquaMai.Config.Attributes;
+using AquaMai.Config.Types;
+using HarmonyLib;
+using MelonLoader;
+
+namespace AquaMai.Mods.GameSystem;
+
+[ConfigSection(
+    name: "8Ch 音频路由重映射",
+    en: """
+        Redirect each channel in 8Ch to specific channel.
+        Mixing is allowed. Use "None" to mute a channel. Volumes are applied on the source channels (> 1.0 won't work, to increase volume, use HeadphoneVolumeMultiply instead).
+        (NOTE: For 2Ch setups, please route the channels you want to P1SpeakerLeft/Right and other channels to None.)
+        """,
+    zh: """
+        将 8Ch 的各个声道重新映射到指定的声道
+        可进行混合，可使用 "None" 以禁用一个声道，音量在源声道应用（> 1.0 不会生效，若要提高音量，请使用 HeadphoneVolumeMultiply）
+        （注：对于 2Ch 用户，请将需要的声道路由到 P1SpeakerLeft/Right，其他声道路由到 None）
+        """)]
+public static class SoundRouting
+{
+    [ConfigEntry]
+    private static readonly SoundChannel routeP1SpeakerLeftTo = SoundChannel.P1SpeakerLeft;
+
+    [ConfigEntry]
+    private static readonly SoundChannel routeP1SpeakerRightTo = SoundChannel.P1SpeakerRight;
+
+    [ConfigEntry]
+    private static readonly SoundChannel routeP1HeadphoneLeftTo = SoundChannel.P1HeadphoneLeft;
+
+    [ConfigEntry]
+    private static readonly SoundChannel routeP1HeadphoneRightTo = SoundChannel.P1HeadphoneRight;
+
+    [ConfigEntry]
+    private static readonly SoundChannel routeP2SpeakerLeftTo = SoundChannel.P2SpeakerLeft;
+
+    [ConfigEntry]
+    private static readonly SoundChannel routeP2SpeakerRightTo = SoundChannel.P2SpeakerRight;
+
+    [ConfigEntry]
+    private static readonly SoundChannel routeP2HeadphoneLeftTo = SoundChannel.P2HeadphoneLeft;
+
+    [ConfigEntry]
+    private static readonly SoundChannel routeP2HeadphoneRightTo = SoundChannel.P2HeadphoneRight;
+
+    [ConfigEntry]
+    private static readonly float volumeP1Speaker = 1.0f;
+    
+    [ConfigEntry]
+    private static readonly float volumeP1Headphone = 1.0f;
+    
+    [ConfigEntry]
+    private static readonly float volumeP2Speaker = 1.0f;
+    
+    [ConfigEntry]
+    private static readonly float volumeP2Headphone = 1.0f;
+
+    private const int PARAMETER_MATRIX_SIZE = 8 * 8 * 4; // 8 input * 8 output * 4 sizeof(float BE)
+
+    private static void WriteFloatBigEndian(byte[] buffer, int index, float value)
+    {
+        var bytes = BitConverter.GetBytes(value);
+        if (BitConverter.IsLittleEndian) Array.Reverse(bytes);
+        Array.Copy(bytes, 0, buffer, index, 4);
+    }
+
+    private static byte[] GenerateParameterMatrix(SoundChannel routeLeftTo, SoundChannel routeRightTo, float volume)
+    {
+        var matrix8x8 = new byte[PARAMETER_MATRIX_SIZE];
+        int index = 0;
+        for (int i = 0; i < 2; i++) // 2ch (stereo) input
+        {
+            for (int j = 0; j < 8; j++) // 8ch output
+            {
+                if (i == 0 && j == (int)routeLeftTo || i == 1 && j == (int)routeRightTo)
+                {
+                    WriteFloatBigEndian(matrix8x8, index, volume);
+                }
+                else
+                {
+                    WriteFloatBigEndian(matrix8x8, index, 0.0f);
+                }
+                index += 4;
+            }
+        }
+        return matrix8x8;
+    }
+
+    private static byte[] GetOriginalParameters()
+    {
+        var memoryStream = new MemoryStream();
+        memoryStream.Write(GenerateParameterMatrix(SoundChannel.P1SpeakerLeft, SoundChannel.P1SpeakerRight, 1.0f), 0, PARAMETER_MATRIX_SIZE);
+        memoryStream.Write(GenerateParameterMatrix(SoundChannel.P2SpeakerLeft, SoundChannel.P2SpeakerRight, 1.0f), 0, PARAMETER_MATRIX_SIZE);
+        memoryStream.Write(GenerateParameterMatrix(SoundChannel.P1HeadphoneLeft, SoundChannel.P1HeadphoneRight, 1.0f), 0, PARAMETER_MATRIX_SIZE);
+        memoryStream.Write(GenerateParameterMatrix(SoundChannel.P2HeadphoneLeft, SoundChannel.P2HeadphoneRight, 1.0f), 0, PARAMETER_MATRIX_SIZE);
+        return memoryStream.ToArray();
+    }
+
+    private static byte[] GenerateParameters()
+    {
+        var memoryStream = new MemoryStream();
+        memoryStream.Write(GenerateParameterMatrix(routeP1SpeakerLeftTo, routeP1SpeakerRightTo, volumeP1Speaker), 0, PARAMETER_MATRIX_SIZE);
+        memoryStream.Write(GenerateParameterMatrix(routeP2SpeakerLeftTo, routeP2SpeakerRightTo, volumeP2Speaker), 0, PARAMETER_MATRIX_SIZE);
+        memoryStream.Write(GenerateParameterMatrix(routeP1HeadphoneLeftTo, routeP1HeadphoneRightTo, volumeP1Headphone), 0, PARAMETER_MATRIX_SIZE);
+        memoryStream.Write(GenerateParameterMatrix(routeP2HeadphoneLeftTo, routeP2HeadphoneRightTo, volumeP2Headphone), 0, PARAMETER_MATRIX_SIZE);
+        return memoryStream.ToArray();
+    }
+
+    // Naive O(n*m) search but enough
+    private static int FindBytesInBytes(byte[] haystack, byte[] needle)
+    {
+        for (int i = 0; i < haystack.Length - needle.Length + 1; i++) if (haystack.Skip(i).Take(needle.Length).SequenceEqual(needle)) return i;
+        return -1;
+    }
+
+    private static byte[] RewriteAcf(byte[] acfData)
+    {
+        var originalParameters = GetOriginalParameters();
+        int i = FindBytesInBytes(acfData, originalParameters);
+        if (i == -1)
+        {
+            MelonLogger.Error("[SoundRouting] Failed to find original parameters in ACF data");
+            return acfData;
+        }
+        var newParameters = GenerateParameters();
+        var newAcfData = new byte[acfData.Length];
+        Array.Copy(acfData, 0, newAcfData, 0, i);
+        Array.Copy(newParameters, 0, newAcfData, i, newParameters.Length);
+        Array.Copy(acfData, i + originalParameters.Length, newAcfData, i + newParameters.Length, acfData.Length - i - originalParameters.Length);
+        return newAcfData;
+    }
+
+    private static readonly Dictionary<string, byte[]> _patchedAcfDataCache = [];
+
+    [HarmonyPrefix]
+    [HarmonyPatch(typeof(CriAtomEx), "RegisterAcf", [typeof(CriFsBinder), typeof(string)])]
+    public static bool PreRegisterAcf(string acfPath)
+    {
+        try
+        {
+            if (!_patchedAcfDataCache.ContainsKey(acfPath))
+            {
+                var acfData = File.ReadAllBytes(acfPath);
+                _patchedAcfDataCache[acfPath] = RewriteAcf(acfData);
+            }
+            CriAtomEx.RegisterAcf(_patchedAcfDataCache[acfPath]);
+        }
+        catch (Exception e)
+        {
+            MelonLogger.Error("[SoundRouting] Failed to rewrite ACF data", e);
+        }
+        return false;
+    }
+}