suyu/src/audio_core/renderer/mix/mix_context.cpp
2022-11-22 22:22:28 -05:00

143 lines
4.1 KiB
C++

// SPDX-FileCopyrightText: Copyright 2022 yuzu Emulator Project
// SPDX-License-Identifier: GPL-2.0-or-later
#include <ranges>
#include "audio_core/renderer/mix/mix_context.h"
#include "audio_core/renderer/splitter/splitter_context.h"
#include "common/polyfill_ranges.h"
namespace AudioCore::AudioRenderer {
void MixContext::Initialize(std::span<MixInfo*> sorted_mix_infos_, std::span<MixInfo> mix_infos_,
const u32 count_, std::span<s32> effect_process_order_buffer_,
const u32 effect_count_, std::span<u8> node_states_workbuffer,
const u64 node_buffer_size, std::span<u8> edge_matrix_workbuffer,
const u64 edge_matrix_size) {
count = count_;
sorted_mix_infos = sorted_mix_infos_;
mix_infos = mix_infos_;
effect_process_order_buffer = effect_process_order_buffer_;
effect_count = effect_count_;
if (node_states_workbuffer.size() > 0 && edge_matrix_workbuffer.size() > 0) {
node_states.Initialize(node_states_workbuffer, node_buffer_size, count);
edge_matrix.Initialize(edge_matrix_workbuffer, edge_matrix_size, count);
}
for (s32 i = 0; i < count; i++) {
sorted_mix_infos[i] = &mix_infos[i];
}
}
MixInfo* MixContext::GetSortedInfo(const s32 index) {
return sorted_mix_infos[index];
}
void MixContext::SetSortedInfo(const s32 index, MixInfo& mix_info) {
sorted_mix_infos[index] = &mix_info;
}
MixInfo* MixContext::GetInfo(const s32 index) {
return &mix_infos[index];
}
MixInfo* MixContext::GetFinalMixInfo() {
return &mix_infos[0];
}
s32 MixContext::GetCount() const {
return count;
}
void MixContext::UpdateDistancesFromFinalMix() {
for (s32 i = 0; i < count; i++) {
mix_infos[i].distance_from_final_mix = InvalidDistanceFromFinalMix;
}
for (s32 i = 0; i < count; i++) {
auto& mix_info{mix_infos[i]};
sorted_mix_infos[i] = &mix_info;
if (!mix_info.in_use) {
continue;
}
auto mix_id{mix_info.mix_id};
auto distance_to_final_mix{FinalMixId};
while (distance_to_final_mix < count) {
if (mix_id == FinalMixId) {
break;
}
if (mix_id == UnusedMixId) {
distance_to_final_mix = InvalidDistanceFromFinalMix;
break;
}
auto distance_from_final_mix{mix_infos[mix_id].distance_from_final_mix};
if (distance_from_final_mix != InvalidDistanceFromFinalMix) {
distance_to_final_mix = distance_from_final_mix + 1;
break;
}
distance_to_final_mix++;
mix_id = mix_infos[mix_id].dst_mix_id;
}
if (distance_to_final_mix >= count) {
distance_to_final_mix = InvalidDistanceFromFinalMix;
}
mix_info.distance_from_final_mix = distance_to_final_mix;
}
}
void MixContext::SortInfo() {
UpdateDistancesFromFinalMix();
std::ranges::sort(sorted_mix_infos, [](const MixInfo* lhs, const MixInfo* rhs) {
return lhs->distance_from_final_mix > rhs->distance_from_final_mix;
});
CalcMixBufferOffset();
}
void MixContext::CalcMixBufferOffset() {
s16 offset{0};
for (s32 i = 0; i < count; i++) {
auto mix_info{sorted_mix_infos[i]};
if (mix_info->in_use) {
const auto buffer_count{mix_info->buffer_count};
mix_info->buffer_offset = offset;
offset += buffer_count;
}
}
}
bool MixContext::TSortInfo(const SplitterContext& splitter_context) {
if (!splitter_context.UsingSplitter()) {
CalcMixBufferOffset();
return true;
}
if (!node_states.Tsort(edge_matrix)) {
return false;
}
std::vector<s32> sorted_results{node_states.GetSortedResuls()};
const auto result_size{std::min(count, static_cast<s32>(sorted_results.size()))};
for (s32 i = 0; i < result_size; i++) {
sorted_mix_infos[i] = &mix_infos[sorted_results[i]];
}
CalcMixBufferOffset();
return true;
}
EdgeMatrix& MixContext::GetEdgeMatrix() {
return edge_matrix;
}
} // namespace AudioCore::AudioRenderer