// SPDX-FileCopyrightText: Copyright 2022 yuzu Emulator Project
// SPDX-License-Identifier: GPL-2.0-or-later
#include <array>
#include "audio_core/renderer/adsp/command_list_processor.h"
#include "audio_core/renderer/command/resample/upsample.h"
#include "audio_core/renderer/upsampler/upsampler_info.h"
namespace AudioCore::AudioRenderer {
/**
* Upsampling impl. Input must be 8K, 16K or 32K, output is 48K.
*
* @param output - Output buffer.
* @param input - Input buffer.
* @param target_sample_count - Number of samples for output.
* @param state - Upsampler state, updated each call.
*/
static void SrcProcessFrame(std::span<s32> output, std::span<const s32> input,
const u32 target_sample_count, const u32 source_sample_count,
UpsamplerState* state) {
constexpr u32 WindowSize = 10;
constexpr std::array<Common::FixedPoint<24, 8>, WindowSize> SincWindow1{
51.93359375f, -18.80078125f, 9.73046875f, -5.33203125f, 2.84375f,
-1.41015625f, 0.62109375f, -0.2265625f, 0.0625f, -0.00390625f,
};
constexpr std::array<Common::FixedPoint<24, 8>, WindowSize> SincWindow2{
105.35546875f, -24.52734375f, 11.9609375f, -6.515625f, 3.52734375f,
-1.796875f, 0.828125f, -0.32421875f, 0.1015625f, -0.015625f,
};
constexpr std::array<Common::FixedPoint<24, 8>, WindowSize> SincWindow3{
122.08203125f, -16.47656250f, 7.68359375f, -4.15625000f, 2.26171875f,
-1.16796875f, 0.54687500f, -0.22265625f, 0.07421875f, -0.01171875f,
};
constexpr std::array<Common::FixedPoint<24, 8>, WindowSize> SincWindow4{
23.73437500f, -9.62109375f, 5.07812500f, -2.78125000f, 1.46875000f,
-0.71484375f, 0.30859375f, -0.10546875f, 0.02734375f, 0.00000000f,
};
constexpr std::array<Common::FixedPoint<24, 8>, WindowSize> SincWindow5{
80.62500000f, -24.67187500f, 12.44921875f, -6.80859375f, 3.66406250f,
-1.83984375f, 0.83203125f, -0.31640625f, 0.09375000f, -0.01171875f,
};
if (!state->initialized) {
switch (source_sample_count) {
case 40:
state->window_size = WindowSize;
state->ratio = 6.0f;
state->history.fill(0);
break;
case 80:
state->window_size = WindowSize;
state->ratio = 3.0f;
state->history.fill(0);
break;
case 160:
state->window_size = WindowSize;
state->ratio = 1.5f;
state->history.fill(0);
break;
default:
LOG_ERROR(Service_Audio, "Invalid upsampling source count {}!", source_sample_count);
// This continues anyway, but let's assume 160 for sanity
state->window_size = WindowSize;
state->ratio = 1.5f;
state->history.fill(0);
break;
}
state->history_input_index = 0;
state->history_output_index = 9;
state->history_start_index = 0;
state->history_end_index = UpsamplerState::HistorySize - 1;
state->initialized = true;
}
if (target_sample_count == 0) {
return;
}
u32 read_index{0};
auto increment = [&]() -> void {
state->history[state->history_input_index] = input[read_index++];
state->history_input_index =
static_cast<u16>((state->history_input_index + 1) % UpsamplerState::HistorySize);
state->history_output_index =
static_cast<u16>((state->history_output_index + 1) % UpsamplerState::HistorySize);
};
auto calculate_sample = [&state](std::span<const Common::FixedPoint<24, 8>> coeffs1,
std::span<const Common::FixedPoint<24, 8>> coeffs2) -> s32 {
auto output_index{state->history_output_index};
auto start_pos{output_index - state->history_start_index + 1U};
auto end_pos{10U};
if (start_pos < 10) {
end_pos = start_pos;
}
u64 prev_contrib{0};
u32 coeff_index{0};
for (; coeff_index < end_pos; coeff_index++, output_index--) {
prev_contrib += static_cast<u64>(state->history[output_index].to_raw()) *
coeffs1[coeff_index].to_raw();
}
auto end_index{state->history_end_index};
for (; start_pos < 9; start_pos++, coeff_index++, end_index--) {
prev_contrib += static_cast<u64>(state->history[end_index].to_raw()) *
coeffs1[coeff_index].to_raw();
}
output_index =
static_cast<u16>((state->history_output_index + 1) % UpsamplerState::HistorySize);
start_pos = state->history_end_index - output_index + 1U;
end_pos = 10U;
if (start_pos < 10) {
end_pos = start_pos;
}
u64 next_contrib{0};
coeff_index = 0;
for (; coeff_index < end_pos; coeff_index++, output_index++) {
next_contrib += static_cast<u64>(state->history[output_index].to_raw()) *
coeffs2[coeff_index].to_raw();
}
auto start_index{state->history_start_index};
for (; start_pos < 9; start_pos++, start_index++, coeff_index++) {
next_contrib += static_cast<u64>(state->history[start_index].to_raw()) *
coeffs2[coeff_index].to_raw();
}
return static_cast<s32>(((prev_contrib >> 15) + (next_contrib >> 15)) >> 8);
};
switch (state->ratio.to_int_floor()) {
// 40 -> 240
case 6:
for (u32 write_index = 0; write_index < target_sample_count; write_index++) {
switch (state->sample_index) {
case 0:
increment();
output[write_index] = state->history[state->history_output_index].to_int_floor();
break;
case 1:
output[write_index] = calculate_sample(SincWindow3, SincWindow4);
break;
case 2:
output[write_index] = calculate_sample(SincWindow2, SincWindow1);
break;
case 3:
output[write_index] = calculate_sample(SincWindow5, SincWindow5);
break;
case 4:
output[write_index] = calculate_sample(SincWindow1, SincWindow2);
break;
case 5:
output[write_index] = calculate_sample(SincWindow4, SincWindow3);
break;
}
state->sample_index = static_cast<u8>((state->sample_index + 1) % 6);
}
break;
// 80 -> 240
case 3:
for (u32 write_index = 0; write_index < target_sample_count; write_index++) {
switch (state->sample_index) {
case 0:
increment();
output[write_index] = state->history[state->history_output_index].to_int_floor();
break;
case 1:
output[write_index] = calculate_sample(SincWindow2, SincWindow1);
break;
case 2:
output[write_index] = calculate_sample(SincWindow1, SincWindow2);
break;
}
state->sample_index = static_cast<u8>((state->sample_index + 1) % 3);
}
break;
// 160 -> 240
default:
for (u32 write_index = 0; write_index < target_sample_count; write_index++) {
switch (state->sample_index) {
case 0:
increment();
output[write_index] = state->history[state->history_output_index].to_int_floor();
break;
case 1:
output[write_index] = calculate_sample(SincWindow1, SincWindow2);
break;
case 2:
increment();
output[write_index] = calculate_sample(SincWindow2, SincWindow1);
break;
}
state->sample_index = static_cast<u8>((state->sample_index + 1) % 3);
}
break;
}
}
auto UpsampleCommand::Dump([[maybe_unused]] const ADSP::CommandListProcessor& processor,
std::string& string) -> void {
string += fmt::format("UpsampleCommand\n\tsource_sample_count {} source_sample_rate {}",
source_sample_count, source_sample_rate);
const auto upsampler{reinterpret_cast<UpsamplerInfo*>(upsampler_info)};
if (upsampler != nullptr) {
string += fmt::format("\n\tUpsampler\n\t\tenabled {} sample count {}\n\tinputs: ",
upsampler->enabled, upsampler->sample_count);
for (u32 i = 0; i < upsampler->input_count; i++) {
string += fmt::format("{:02X}, ", upsampler->inputs[i]);
}
}
string += "\n";
}
void UpsampleCommand::Process(const ADSP::CommandListProcessor& processor) {
const auto info{reinterpret_cast<UpsamplerInfo*>(upsampler_info)};
const auto input_count{std::min(info->input_count, buffer_count)};
const std::span<const s16> inputs_{reinterpret_cast<const s16*>(inputs), input_count};
for (u32 i = 0; i < input_count; i++) {
const auto channel{inputs_[i]};
if (channel >= 0 && channel < static_cast<s16>(processor.buffer_count)) {
auto state{&info->states[i]};
std::span<s32> output{
reinterpret_cast<s32*>(samples_buffer + info->sample_count * channel * sizeof(s32)),
info->sample_count};
auto input{processor.mix_buffers.subspan(channel * processor.sample_count,
processor.sample_count)};
SrcProcessFrame(output, input, info->sample_count, source_sample_count, state);
}
}
}
bool UpsampleCommand::Verify(const ADSP::CommandListProcessor& processor) {
return true;
}
} // namespace AudioCore::AudioRenderer
