ADOFAIRenderer/Patch/Engine/AudioPatches.cs

using System;
using ADOFAIRenderer.Core;
using HarmonyLib;
using UnityEngine;
using UnityEngine.Audio;

namespace ADOFAIRenderer.Patch.Engine;

[HarmonyPatch(typeof(AudioManager), "_Play")]
public static class AudioManagerPlayPatch
{
    [HarmonyPrefix]
    public static bool Prefix(string snd, double time, AudioMixerGroup group, float volume, int priority)
    {
        if (RenderManager.IsCapturingActive && RenderManager.Instance.Configuration.AudioConfig.RenderAudio)
        {
            // Cache the clip on the main thread
            RenderManager.Instance.AudioMixer?.CacheClip(snd);
        }
        return true; // Let the original method execute to prevent NullReferenceException
    }

    [HarmonyPostfix]
    // ReSharper disable once InconsistentNaming
    public static void Postfix(AudioSource __result, string snd, double time, AudioMixerGroup group, float volume, int priority)
    {
        if (RenderManager.IsCapturingActive && RenderManager.Instance.Configuration.AudioConfig.RenderAudio && __result != null)
        {
            // Convert absolute DSP time to relative time in seconds from render start
            double relativeStartTime = time - RenderManager.Instance.dspTimeBase;

            // Register event to the offline mixer (default pitch = 1.0f) and map it to the AudioSource
            RenderManager.Instance.AudioMixer?.RegisterSoundEvent(__result, snd, relativeStartTime, -1.0, volume, 1.0f);
        }
    }
}

[HarmonyPatch(typeof(scrConductor), "PlayWithEndTime")]
public static class PlayWithEndTimePatch
{
    [HarmonyPrefix]
    public static bool Prefix(string snd, double time, double endTime, float volume, int priority)
    {
        if (RenderManager.IsCapturingActive && RenderManager.Instance.Configuration.AudioConfig.RenderAudio)
        {
            var mixer = RenderManager.Instance.AudioMixer;
            if (mixer != null)
            {
                // Cache the clip on the main thread
                mixer.CacheClip(snd);

                // Convert absolute DSP times to relative times in seconds from render start
                double relativeStartTime = time - RenderManager.Instance.dspTimeBase;
                double relativeEndTime = endTime - RenderManager.Instance.dspTimeBase;

                // Register looping sound event to the offline mixer (default pitch = 1.0f)
                mixer.RegisterSoundEvent(snd, relativeStartTime, relativeEndTime, volume, 1.0f);
            }
        }
        return true; // Let the original method execute
    }
}

[HarmonyPatch(typeof(AudioSource), "pitch", MethodType.Setter)]
public static class AudioSourcePitchPatch
{
    [HarmonyPostfix]
    // ReSharper disable once InconsistentNaming
    public static void Postfix(AudioSource __instance, float value)
    {
        if (RenderManager.IsCapturingActive && RenderManager.Instance.Configuration.AudioConfig.RenderAudio)
        {
            var mixer = RenderManager.Instance.AudioMixer;
            if (mixer != null && mixer.TryGetSoundEvent(__instance, out var soundEvent))
            {
                soundEvent.Pitch = value;
            }
        }
    }
}

[HarmonyPatch(typeof(AudioSource), "SetScheduledEndTime", typeof(double))]
public static class AudioSourceSetScheduledEndTimePatch
{
    [HarmonyPostfix]
    // ReSharper disable once InconsistentNaming
    public static void Postfix(AudioSource __instance, double time)
    {
        if (RenderManager.IsCapturingActive && RenderManager.Instance.Configuration.AudioConfig.RenderAudio)
        {
            var mixer = RenderManager.Instance.AudioMixer;
            if (mixer != null && mixer.TryGetSoundEvent(__instance, out var soundEvent))
            {
                soundEvent.EndTime = time - RenderManager.Instance.dspTimeBase;
            }
        }
    }
}

[HarmonyPatch(typeof(AudioSource), "Stop", new Type[0])]
public static class AudioSourceStopPatch
{
    [HarmonyPostfix]
    // ReSharper disable once InconsistentNaming
    public static void Postfix(AudioSource __instance)
    {
        if (RenderManager.IsCapturingActive && RenderManager.Instance.Configuration.AudioConfig.RenderAudio)
        {
            RenderManager.Instance.AudioMixer?.RemoveSourceMapping(__instance);
        }
    }
}