#include <algorithm>
#include "Common/Serialize/Serializer.h"
#include "Common/Serialize/SerializeFuncs.h"
#include "Core/HLE/HLE.h"
#include "Core/HLE/ErrorCodes.h"
#include "Core/HLE/FunctionWrappers.h"
#include "Core/MIPS/MIPS.h"
#include "Core/CoreTiming.h"
#include "Core/MemMapHelpers.h"
#include "Core/Reporting.h"
#include "Core/Config.h"
#include "Core/Debugger/MemBlockInfo.h"
#include "Core/HW/BufferQueue.h"
#include "Core/HLE/sceKernel.h"
#include "Core/HLE/sceUtility.h"
#include "Core/HLE/sceKernelMemory.h"
#include "Core/HLE/sceAtrac.h"
#include "Core/HLE/AtracCtx.h"
#include "Core/HLE/AtracCtx2.h"
#include "Core/System.h"
static const int atracDecodeDelay = 2300;
static bool atracInited = true;
static AtracBase *atracContexts[PSP_MAX_ATRAC_IDS];
static u32 atracContextTypes[PSP_MAX_ATRAC_IDS];
static int atracLibVersion = 0;
static u32 atracLibCrc = 0;
static int g_atracMaxContexts = 6;
static int g_atracBSS = 0;
static bool g_muteFlag[PSP_MAX_ATRAC_IDS]{};
bool *__AtracMuteFlag(int atracID) {
if (atracID < 0 || atracID >= PSP_MAX_ATRAC_IDS) {
return nullptr;
}
return &g_muteFlag[atracID];
}
const AtracBase *__AtracGetCtx(int i, u32 *type) {
if (type) {
*type = atracContextTypes[i];
}
return atracContexts[i];
}
void __AtracInit() {
_assert_(sizeof(SceAtracContext) == 256);
atracLibVersion = 0;
atracLibCrc = 0;
g_atracMaxContexts = 6;
g_atracBSS = 0;
atracInited = true;
memset(atracContexts, 0, sizeof(atracContexts));
memset(g_muteFlag, 0, sizeof(g_muteFlag));
atracContextTypes[0] = PSP_CODEC_AT3PLUS;
atracContextTypes[1] = PSP_CODEC_AT3PLUS;
atracContextTypes[2] = PSP_CODEC_AT3;
atracContextTypes[3] = PSP_CODEC_AT3;
atracContextTypes[4] = 0;
atracContextTypes[5] = 0;
}
void __AtracShutdown() {
for (size_t i = 0; i < ARRAY_SIZE(atracContexts); ++i) {
delete atracContexts[i];
atracContexts[i] = nullptr;
}
}
int __AtracMaxContexts() {
return g_atracMaxContexts;
}
void __AtracNotifyLoadModule(int version, u32 crc, u32 bssAddr, int bssSize) {
atracLibVersion = version;
atracLibCrc = crc;
INFO_LOG(Log::Atrac, "Atrac module loaded: atracLibVersion 0x%0x, atracLibcrc %x, bss: %x (%x bytes)", atracLibVersion, atracLibCrc, bssAddr, bssSize);
g_atracBSS = bssAddr;
g_atracMaxContexts = atracLibVersion <= 0x101 ? 4 : 6;
_dbg_assert_(bssSize >= g_atracMaxContexts * sizeof(SceAtracContext));
Memory::Memset(g_atracBSS, 0, g_atracMaxContexts * sizeof(SceAtracContext));
NotifyMemInfo(MemBlockFlags::ALLOC, g_atracBSS, g_atracMaxContexts * sizeof(SceAtracContext), "AtracContext");
}
void __AtracNotifyUnloadModule() {
atracLibVersion = 0;
atracLibCrc = 0;
INFO_LOG(Log::Atrac, "Atrac module unloaded.");
g_atracBSS = 0;
g_atracMaxContexts = 6;
NotifyMemInfo(MemBlockFlags::FREE, g_atracBSS, g_atracMaxContexts * sizeof(SceAtracContext), "AtracContext");
}
static u32 GetAtracContextAddress(int atracID) {
return g_atracBSS + atracID * sizeof(SceAtracContext);
}
void __AtracDoState(PointerWrap &p) {
auto s = p.Section("sceAtrac", 1, 4);
if (!s)
return;
Do(p, atracInited);
if (s >= 4) {
Do(p, g_atracBSS);
Do(p, g_atracMaxContexts);
if (g_atracMaxContexts > PSP_MAX_ATRAC_IDS)
g_atracMaxContexts = PSP_MAX_ATRAC_IDS;
} else {
g_atracBSS = 0;
g_atracMaxContexts = 6;
}
for (int i = 0; i < PSP_MAX_ATRAC_IDS; ++i) {
bool valid = atracContexts[i] != nullptr;
Do(p, valid);
if (valid) {
int version = atracContexts[i] ? atracContexts[i]->GetContextVersion() : 0;
if (s >= 4) {
Do(p, version);
_dbg_assert_(version != 0);
} else {
version = 1;
}
switch (version) {
case 1:
DoSubClass<AtracBase, Atrac>(p, atracContexts[i], i);
break;
case 2:
DoSubClass<AtracBase, Atrac2>(p, atracContexts[i]);
break;
}
} else {
delete atracContexts[i];
atracContexts[i] = nullptr;
}
}
DoArray(p, atracContextTypes, PSP_MAX_ATRAC_IDS);
if (s >= 2) {
Do(p, atracLibVersion);
Do(p, atracLibCrc);
}
else {
atracLibVersion = 0;
atracLibCrc = 0;
}
}
static AtracBase *getAtrac(int atracID) {
if (atracID < 0 || atracID >= PSP_MAX_ATRAC_IDS) {
return nullptr;
}
AtracBase *atrac = atracContexts[atracID];
if (atrac) {
atrac->UpdateContextFromPSPMem();
}
return atrac;
}
static int AllocAndRegisterAtrac(int codecType) {
for (int i = 0; i < g_atracMaxContexts; ++i) {
if (atracContextTypes[i] == codecType && atracContexts[i] == 0) {
if (!g_Config.bUseOldAtrac && g_atracBSS != 0) {
atracContexts[i] = new Atrac2(GetAtracContextAddress(i), codecType);
} else {
atracContexts[i] = new Atrac(i, codecType);
}
return i;
}
}
return SCE_ERROR_ATRAC_NO_ATRACID;
}
static int UnregisterAndDeleteAtrac(int atracID) {
if (atracID >= 0 && atracID < PSP_MAX_ATRAC_IDS) {
if (atracContexts[atracID] != nullptr) {
delete atracContexts[atracID];
atracContexts[atracID] = nullptr;
return 0;
}
}
return SCE_ERROR_ATRAC_BAD_ATRACID;
}
static u32 sceAtracGetAtracID(int codecType) {
if (codecType != PSP_CODEC_AT3 && codecType != PSP_CODEC_AT3PLUS) {
return hleReportError(Log::Atrac, SCE_ERROR_ATRAC_INVALID_CODECTYPE, "invalid codecType");
}
int atracID = AllocAndRegisterAtrac(codecType);
if (atracID < 0) {
return hleLogError(Log::Atrac, atracID, "no free ID");
}
return hleLogInfo(Log::Atrac, atracID);
}
static int AtracValidateData(const AtracBase *atrac) {
if (!atrac) {
return SCE_ERROR_ATRAC_BAD_ATRACID;
} else if (atrac->BufferState() == ATRAC_STATUS_NO_DATA) {
return SCE_ERROR_ATRAC_NO_DATA;
} else {
return 0;
}
}
static int AtracValidateManaged(const AtracBase *atrac) {
if (!atrac) {
return SCE_ERROR_ATRAC_BAD_ATRACID;
} else if (atrac->BufferState() == ATRAC_STATUS_NO_DATA) {
return SCE_ERROR_ATRAC_NO_DATA;
} else if (atrac->BufferState() == ATRAC_STATUS_LOW_LEVEL) {
return SCE_ERROR_ATRAC_IS_LOW_LEVEL;
} else if (atrac->BufferState() == ATRAC_STATUS_FOR_SCESAS) {
return SCE_ERROR_ATRAC_IS_FOR_SCESAS;
} else {
return 0;
}
}
static u32 sceAtracAddStreamData(int atracID, u32 bytesToAdd) {
AtracBase *atrac = getAtrac(atracID);
int err = AtracValidateManaged(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
if (atrac->BufferState() == ATRAC_STATUS_ALL_DATA_LOADED) {
if (bytesToAdd == 0)
return hleLogDebug(Log::Atrac, SCE_ERROR_ATRAC_ALL_DATA_LOADED, "stream entirely loaded");
return hleLogWarning(Log::Atrac, SCE_ERROR_ATRAC_ALL_DATA_LOADED, "stream entirely loaded");
}
int ret = atrac->AddStreamData(bytesToAdd);
return hleLogDebugOrError(Log::Atrac, ret);
}
static u32 sceAtracDecodeData(int atracID, u32 outAddr, u32 numSamplesAddr, u32 finishFlagAddr, u32 remainAddr) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateData(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
if (outAddr & 1) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_BAD_ALIGNMENT);
}
int numSamplesWritten = 0;
int finish = 0;
int remains = 0;
if (outAddr != 0 && !Memory::IsValidAddress(outAddr)) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_SIZE_TOO_SMALL);
}
u8 *outPtr = outAddr ? Memory::GetPointerWrite(outAddr) : nullptr;
int ret = atrac->DecodeData(outPtr, outAddr, &numSamplesWritten, &finish, &remains);
if (ret != (int)SCE_ERROR_ATRAC_BAD_ATRACID && ret != (int)SCE_ERROR_ATRAC_NO_DATA) {
if (Memory::IsValidAddress(numSamplesAddr))
Memory::WriteUnchecked_U32(numSamplesWritten, numSamplesAddr);
if (Memory::IsValidAddress(finishFlagAddr))
Memory::WriteUnchecked_U32(finish, finishFlagAddr);
if (ret == 0 && Memory::IsValidAddress(remainAddr))
Memory::WriteUnchecked_U32(remains, remainAddr);
}
DEBUG_LOG(Log::Atrac, "%08x=sceAtracDecodeData(%i, %08x, %08x[%08x], %08x[%08x], %08x[%d])", ret, atracID, outAddr,
numSamplesAddr, numSamplesWritten,
finishFlagAddr, finish,
remainAddr, remains);
if (outPtr && g_muteFlag[atracID]) {
memset(outPtr, 0, atrac->GetOutputChannels() * 2 * numSamplesWritten);
}
if (ret == 0 || ret == SCE_ERROR_ATRAC_API_FAIL) {
return hleDelayResult(hleNoLog(ret), "atrac decode data", atracDecodeDelay);
}
return hleNoLog(ret);
}
static u32 sceAtracReleaseResources() {
return hleLogDebug(Log::Atrac, 0);
}
static u32 sceAtracGetBufferInfoForResetting(int atracID, int sample, u32 bufferInfoAddr) {
auto bufferInfo = PSPPointer<AtracResetBufferInfo>::Create(bufferInfoAddr);
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateManaged(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
if (!bufferInfo.IsValid()) {
return hleLogError(Log::Atrac, SCE_KERNEL_ERROR_ILLEGAL_ADDR, "invalid buffer, should crash");
}
bool delay = false;
int ret = atrac->GetBufferInfoForResetting(bufferInfo, sample, &delay);
if (delay) {
return hleDelayResult(hleLogDebugOrError(Log::Atrac, ret), "getreset_frameskip", 300);
} else {
return hleLogDebugOrError(Log::Atrac, ret);
}
}
static u32 sceAtracGetBitrate(int atracID, u32 outBitrateAddr) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateData(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
int bitrate = atrac->Bitrate();
if (Memory::IsValidAddress(outBitrateAddr)) {
Memory::WriteUnchecked_U32(atrac->Bitrate(), outBitrateAddr);
return hleLogDebug(Log::Atrac, 0);
} else {
return hleLogError(Log::Atrac, 0, "invalid address");
}
}
static u32 sceAtracGetChannel(int atracID, u32 channelAddr) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateData(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
if (Memory::IsValidAddress(channelAddr)){
Memory::WriteUnchecked_U32(atrac->Channels(), channelAddr);
return hleLogDebug(Log::Atrac, 0);
} else {
return hleLogError(Log::Atrac, 0, "invalid address");
}
}
static u32 sceAtracGetLoopStatus(int atracID, u32 loopNumAddr, u32 statusAddr) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateData(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
if (Memory::IsValidAddress(loopNumAddr)) {
Memory::WriteUnchecked_U32(atrac->LoopNum(), loopNumAddr);
}
if (Memory::IsValidAddress(statusAddr)) {
const int loopStatus = atrac->LoopStatus();
Memory::WriteUnchecked_U32(loopStatus, statusAddr);
return hleLogDebug(Log::Atrac, 0);
} else {
return hleLogError(Log::Atrac, 0, "invalid address");
}
}
static u32 sceAtracGetInternalErrorInfo(int atracID, u32 errorAddr) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateData(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
const u32 errorCode = atrac->GetInternalCodecError();
if (Memory::IsValidAddress(errorAddr)) {
Memory::WriteUnchecked_U32(errorCode, errorAddr);
}
if (errorCode) {
return hleLogWarning(Log::Atrac, 0, "code: %08x", errorCode);
} else {
return hleLogDebug(Log::Atrac, 0);
}
}
static u32 sceAtracGetMaxSample(int atracID, u32 maxSamplesAddr) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateData(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
if (Memory::IsValidAddress(maxSamplesAddr)) {
Memory::WriteUnchecked_U32(atrac->SamplesPerFrame(), maxSamplesAddr);
return hleLogDebug(Log::Atrac, 0);
} else {
return hleLogError(Log::Atrac, 0, "invalid address");
}
}
static u32 sceAtracGetNextDecodePosition(int atracID, u32 outposAddr) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateData(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
if (!Memory::IsValidAddress(outposAddr)) {
return hleLogError(Log::Atrac, 0, "invalid address");
}
int pos = 0;
int ret = atrac->GetNextDecodePosition(&pos);
if (ret < 0) {
if (ret == SCE_ERROR_ATRAC_ALL_DATA_DECODED) {
return hleLogDebug(Log::Atrac, ret);
} else {
return hleLogError(Log::Atrac, ret);
}
}
Memory::WriteUnchecked_U32(pos, outposAddr);
return hleLogDebug(Log::Atrac, 0);
}
static u32 sceAtracGetNextSample(int atracID, u32 outNAddr) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateData(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
int numSamples = atrac->GetNextSamples();
if (Memory::IsValidAddress(outNAddr)) {
Memory::WriteUnchecked_U32(numSamples, outNAddr);
}
return hleLogDebug(Log::Atrac, 0, "%d samples left", numSamples);
}
static u32 sceAtracGetRemainFrame(int atracID, u32 remainAddr) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateManaged(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
if (!Memory::IsValidAddress(remainAddr)) {
return hleReportError(Log::Atrac, SCE_KERNEL_ERROR_ILLEGAL_ADDR, "invalid remainingFrames pointer");
}
u32 remaining = atrac->RemainingFrames();
Memory::WriteUnchecked_U32(remaining, remainAddr);
return hleLogDebug(Log::Atrac, 0);
}
static u32 sceAtracGetSecondBufferInfo(int atracID, u32 fileOffsetAddr, u32 desiredSizeAddr) {
auto fileOffset = PSPPointer<u32_le>::Create(fileOffsetAddr);
auto desiredSize = PSPPointer<u32_le>::Create(desiredSizeAddr);
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateManaged(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
if (!fileOffset.IsValid() || !desiredSize.IsValid()) {
return hleReportError(Log::Atrac, SCE_KERNEL_ERROR_ILLEGAL_ADDR, "invalid addresses");
}
int result = atrac->GetSecondBufferInfo(fileOffset, desiredSize);
switch (result) {
case (int)SCE_ERROR_ATRAC_SECOND_BUFFER_NOT_NEEDED:
return hleLogDebug(Log::Atrac, result);
default:
return hleLogDebugOrError(Log::Atrac, result);
}
}
static u32 sceAtracGetSoundSample(int atracID, u32 outEndSampleAddr, u32 outLoopStartSampleAddr, u32 outLoopEndSampleAddr) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateManaged(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
int endSample = -1;
int loopStart = -1;
int loopEnd = -1;
int retval = atrac->GetSoundSample(&endSample, &loopStart, &loopEnd);
if (retval < 0) {
return hleLogError(Log::Atrac, retval);
}
if (Memory::IsValidAddress(outEndSampleAddr)) {
Memory::WriteUnchecked_U32(endSample, outEndSampleAddr);
}
if (Memory::IsValidAddress(outLoopStartSampleAddr)) {
Memory::WriteUnchecked_U32(loopStart, outLoopStartSampleAddr);
}
if (Memory::IsValidAddress(outLoopEndSampleAddr)) {
Memory::WriteUnchecked_U32(loopEnd, outLoopEndSampleAddr);
}
return hleLogDebug(Log::Atrac, retval);
}
static u32 sceAtracGetStreamDataInfo(int atracID, u32 writePtrAddr, u32 writableBytesAddr, u32 readOffsetAddr) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateManaged(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
u32 writePtr;
u32 writableBytes;
u32 readOffset;
atrac->GetStreamDataInfo(&writePtr, &writableBytes, &readOffset);
if (Memory::IsValidAddress(writePtrAddr))
Memory::WriteUnchecked_U32(writePtr, writePtrAddr);
if (Memory::IsValidAddress(writableBytesAddr))
Memory::WriteUnchecked_U32(writableBytes, writableBytesAddr);
if (Memory::IsValidAddress(readOffsetAddr))
Memory::WriteUnchecked_U32(readOffset, readOffsetAddr);
return hleLogDebug(Log::Atrac, 0);
}
static u32 sceAtracReleaseAtracID(int atracID) {
int result = UnregisterAndDeleteAtrac(atracID);
if (result < 0) {
if (atracID >= 0) {
return hleLogError(Log::Atrac, result, "did not exist");
} else {
return hleLogWarning(Log::Atrac, result, "did not exist");
}
}
return hleLogDebug(Log::Atrac, result);
}
static u32 sceAtracResetPlayPosition(int atracID, int sample, int bytesWrittenFirstBuf, int bytesWrittenSecondBuf) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateManaged(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
bool delay = false;
int res = atrac->ResetPlayPosition(sample, bytesWrittenFirstBuf, bytesWrittenSecondBuf, &delay);
if (res < 0) {
if (delay) {
return hleDelayResult(hleLogError(Log::Atrac, res), "reset play pos", 200);
} else {
return hleLogError(Log::Atrac, res);
}
}
return hleDelayResult(res, "reset play pos", 3000);
}
static u32 sceAtracSetHalfwayBuffer(int atracID, u32 buffer, u32 readSize, u32 bufferSize) {
AtracBase *atrac = getAtrac(atracID);
if (!atrac) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_BAD_ATRACID, "invalid atrac ID");
}
if (readSize > bufferSize) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_INCORRECT_READ_SIZE, "read size too large");
}
Track track;
std::string error;
int ret = AnalyzeAtracTrack(Memory::GetPointerOrNull(buffer), readSize, &track, &error);
if (ret < 0) {
return hleLogError(Log::Atrac, ret, "%s", error.c_str());
}
if (track.codecType != atracContextTypes[atracID]) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_WRONG_CODECTYPE, "atracID uses different codec type than data");
}
ret = atrac->SetData(track, buffer, readSize, bufferSize, 0, 2, false);
if (ret < 0) {
return hleLogError(Log::Atrac, ret);
}
return hleDelayResult(hleLogDebug(Log::Atrac, ret), "atrac set data", 100);
}
static u32 sceAtracSetSecondBuffer(int atracID, u32 secondBuffer, u32 secondBufferSize) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateManaged(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
return hleLogDebugOrError(Log::Atrac, atrac->SetSecondBuffer(secondBuffer, secondBufferSize));
}
static u32 sceAtracSetData(int atracID, u32 buffer, u32 bufferSize) {
AtracBase *atrac = getAtrac(atracID);
if (!atrac) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_BAD_ATRACID, "bad atrac ID");
}
Track track;
std::string error;
int ret = AnalyzeAtracTrack(Memory::GetPointerOrNull(buffer), bufferSize, &track, &error);
if (ret < 0) {
return hleLogError(Log::Atrac, ret, "%s", error.c_str());
}
if (track.codecType != atracContextTypes[atracID]) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_WRONG_CODECTYPE, "atracID uses different codec type than data");
}
ret = atrac->SetData(track, buffer, bufferSize, bufferSize, 0, 2, false);
if (ret < 0) {
return hleLogError(Log::Atrac, ret);
}
return hleDelayResult(hleLogDebug(Log::Atrac, ret), "atrac set data", 100);
}
static int sceAtracSetDataAndGetID(u32 buffer, int bufferSize) {
if (bufferSize < 0) {
WARN_LOG(Log::Atrac, "sceAtracSetDataAndGetID(%08x, %08x): negative bufferSize", buffer, bufferSize);
bufferSize = 0x10000000;
}
Track track;
std::string error;
int ret = AnalyzeAtracTrack(Memory::GetPointerOrNull(buffer), bufferSize, &track, &error);
if (ret < 0) {
return hleLogError(Log::Atrac, ret, "%s", error.c_str());
}
int atracID = AllocAndRegisterAtrac(track.codecType);
if (atracID < 0) {
return hleLogError(Log::Atrac, atracID, "no free ID");
}
ret = atracContexts[atracID]->SetData(track, buffer, bufferSize, bufferSize, 0, 2, false);
if (ret < 0) {
UnregisterAndDeleteAtrac(atracID);
return hleLogError(Log::Atrac, ret);
}
return hleDelayResult(hleLogDebug(Log::Atrac, atracID), "atrac set data", 100);
}
static int sceAtracSetHalfwayBufferAndGetID(u32 buffer, u32 readSize, u32 bufferSize) {
if (readSize > bufferSize) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_INCORRECT_READ_SIZE, "read size too large");
}
Track track;
std::string error;
int ret = AnalyzeAtracTrack(Memory::GetPointerOrNull(buffer), readSize, &track, &error);
if (ret < 0) {
return hleLogError(Log::Atrac, ret, "%s", error.c_str());
}
int atracID = AllocAndRegisterAtrac(track.codecType);
if (atracID < 0) {
return hleLogError(Log::Atrac, atracID, "no free ID");
}
ret = atracContexts[atracID]->SetData(track, buffer, readSize, bufferSize, 0, 2, false);
if (ret < 0) {
UnregisterAndDeleteAtrac(atracID);
return hleLogError(Log::Atrac, ret);
}
return hleDelayResult(hleLogDebug(Log::Atrac, atracID), "atrac set data", 100);
}
static u32 sceAtracStartEntry() {
ERROR_LOG_REPORT(Log::Atrac, "UNIMPL sceAtracStartEntry()");
return 0;
}
static u32 sceAtracSetLoopNum(int atracID, int loopNum) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateData(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
int ret = atrac->SetLoopNum(loopNum);
if (ret == SCE_ERROR_ATRAC_NO_LOOP_INFORMATION && (loopNum == -1 || loopNum == 0)) {
return hleLogDebug(Log::Atrac, ret);
}
return hleLogDebugOrError(Log::Atrac, ret);
}
static int sceAtracReinit(int at3Count, int at3plusCount) {
for (int i = 0; i < PSP_MAX_ATRAC_IDS; ++i) {
if (atracContexts[i] != nullptr) {
return hleReportError(Log::Atrac, SCE_KERNEL_ERROR_BUSY, "cannot reinit while IDs in use");
}
}
memset(atracContextTypes, 0, sizeof(atracContextTypes));
int next = 0;
int space = g_atracMaxContexts;
if (at3Count == 0 && at3plusCount == 0) {
atracInited = false;
return hleDelayResult(hleLogInfo(Log::Atrac, 0, "deinit"), "atrac reinit", 200);
}
for (int i = 0; i < at3plusCount; ++i) {
space -= 2;
if (space >= 0) {
atracContextTypes[next++] = PSP_CODEC_AT3PLUS;
}
}
for (int i = 0; i < at3Count; ++i) {
space -= 1;
if (space >= 0) {
atracContextTypes[next++] = PSP_CODEC_AT3;
}
}
int result = space >= 0 ? 0 : (int)SCE_KERNEL_ERROR_OUT_OF_MEMORY;
if (atracInited || next == 0) {
atracInited = true;
return hleLogInfo(Log::Atrac, result);
} else {
atracInited = true;
return hleDelayResult(hleLogInfo(Log::Atrac, result), "atrac reinit", 400);
}
}
static int sceAtracGetOutputChannel(int atracID, u32 outputChanPtr) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateData(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
if (Memory::IsValidAddress(outputChanPtr)) {
Memory::WriteUnchecked_U32(atrac->GetOutputChannels(), outputChanPtr);
return hleLogDebug(Log::Atrac, 0);
} else {
return hleLogError(Log::Atrac, 0, "invalid address");
}
}
static int sceAtracIsSecondBufferNeeded(int atracID) {
AtracBase *atrac = getAtrac(atracID);
u32 err = AtracValidateManaged(atrac);
if (err != 0) {
return hleLogError(Log::Atrac, err);
}
int needed = atrac->BufferState() == ATRAC_STATUS_STREAMED_LOOP_WITH_TRAILER ? 1 : 0;
return hleLogDebug(Log::Atrac, needed);
}
static int sceAtracSetMOutHalfwayBuffer(int atracID, u32 buffer, u32 readSize, u32 bufferSize) {
AtracBase *atrac = getAtrac(atracID);
if (!atrac) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_BAD_ATRACID, "bad atrac ID");
}
if (readSize > bufferSize) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_INCORRECT_READ_SIZE, "read size too large");
}
Track track;
std::string error;
int ret = AnalyzeAtracTrack(Memory::GetPointerOrNull(buffer), readSize, &track, &error);
if (ret < 0) {
return hleLogError(Log::Atrac, ret, "%s", error.c_str());
}
ret = atrac->SetData(track, buffer, readSize, bufferSize, 1, 0, false);
if (ret < 0 && ret != SCE_ERROR_ATRAC_NOT_MONO) {
return hleLogError(Log::Atrac, ret);
}
return hleDelayResult(hleLogDebugOrError(Log::Atrac, ret), "atrac set data mono", 100);
}
static u32 sceAtracSetMOutData(int atracID, u32 buffer, u32 bufferSize) {
AtracBase *atrac = getAtrac(atracID);
if (!atrac) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_BAD_ATRACID, "bad atrac ID");
}
Track track;
std::string error;
int ret = AnalyzeAtracTrack(Memory::GetPointerOrNull(buffer), bufferSize, &track, &error);
if (ret < 0) {
return hleLogError(Log::Atrac, ret, "%s", error.c_str());
}
ret = atrac->SetData(track, buffer, bufferSize, bufferSize, 1, 0, false);
if (ret < 0 && ret != SCE_ERROR_ATRAC_NOT_MONO) {
return hleLogError(Log::Atrac, ret);
}
return hleDelayResult(hleLogDebugOrError(Log::Atrac, ret), "atrac set data mono", 100);
}
static int sceAtracSetMOutDataAndGetID(u32 buffer, u32 bufferSize) {
Track track;
std::string error;
int ret = AnalyzeAtracTrack(Memory::GetPointerOrNull(buffer), bufferSize, &track, &error);
if (ret < 0) {
return hleLogError(Log::Atrac, ret, "%s", error.c_str());
}
if (track.channels != 1) {
return hleReportError(Log::Atrac, SCE_ERROR_ATRAC_NOT_MONO, "not mono data");
}
int atracID = AllocAndRegisterAtrac(track.codecType);
if (atracID < 0) {
return hleLogError(Log::Atrac, atracID, "no free ID");
}
ret = atracContexts[atracID]->SetData(track, buffer, bufferSize, bufferSize, 0, 1, false);
if (ret < 0 && ret != SCE_ERROR_ATRAC_NOT_MONO) {
UnregisterAndDeleteAtrac(atracID);
return hleLogError(Log::Atrac, ret);
}
return hleDelayResult(hleLogDebugOrError(Log::Atrac, atracID), "atrac set data", 100);
}
static int sceAtracSetMOutHalfwayBufferAndGetID(u32 buffer, u32 readSize, u32 bufferSize) {
if (readSize > bufferSize) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_INCORRECT_READ_SIZE, "read size too large");
}
Track track;
std::string error;
int ret = AnalyzeAtracTrack(Memory::GetPointerOrNull(buffer), readSize, &track, &error);
if (ret < 0) {
return hleLogError(Log::Atrac, ret, "%s", error.c_str());
}
if (track.channels != 1) {
return hleReportError(Log::Atrac, SCE_ERROR_ATRAC_NOT_MONO, "not mono data");
}
int atracID = AllocAndRegisterAtrac(track.codecType);
if (atracID < 0) {
return hleLogError(Log::Atrac, atracID, "no free ID");
}
ret = atracContexts[atracID]->SetData(track, buffer, readSize, bufferSize, 0, 1, false);
if (ret < 0 && ret != SCE_ERROR_ATRAC_NOT_MONO) {
UnregisterAndDeleteAtrac(atracID);
return hleLogError(Log::Atrac, ret);
}
return hleDelayResult(hleLogDebug(Log::Atrac, atracID), "atrac set data", 100);
}
static int sceAtracSetAA3DataAndGetID(u32 buffer, u32 bufferSize, u32 fileSize, u32 metadataSizeAddr) {
Track track;
std::string error;
int ret = AnalyzeAA3Track(Memory::GetPointerOrNull(buffer), bufferSize, fileSize, &track, &error);
if (ret < 0) {
return hleLogError(Log::Atrac, ret, "%s", error.c_str());
}
int atracID = AllocAndRegisterAtrac(track.codecType);
if (atracID < 0) {
return hleLogError(Log::Atrac, atracID, "no free ID");
}
ret = atracContexts[atracID]->SetData(track, buffer, bufferSize, bufferSize, fileSize, 2, true);
if (ret < 0) {
UnregisterAndDeleteAtrac(atracID);
return hleLogError(Log::Atrac, ret);
}
return hleDelayResult(hleLogDebug(Log::Atrac, atracID), "atrac set aa3 data", 100);
}
static int sceAtracSetAA3HalfwayBufferAndGetID(u32 buffer, u32 readSize, u32 bufferSize, u32 fileSize) {
if (readSize > bufferSize) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_INCORRECT_READ_SIZE, "read size too large");
}
Track track;
std::string error;
int ret = AnalyzeAA3Track(Memory::GetPointerOrNull(buffer), readSize, fileSize, &track, &error);
if (ret < 0) {
return hleLogError(Log::Atrac, ret, "%s", error.c_str());
}
int atracID = AllocAndRegisterAtrac(track.codecType);
if (atracID < 0) {
return hleLogError(Log::Atrac, atracID, "no free ID");
}
ret = atracContexts[atracID]->SetData(track, buffer, readSize, bufferSize, fileSize, 2, true);
if (ret < 0) {
UnregisterAndDeleteAtrac(atracID);
return hleLogError(Log::Atrac, ret);
}
return hleDelayResult(hleLogDebug(Log::Atrac, atracID), "atrac set data", 100);
}
static u32 _sceAtracGetContextAddress(int atracID) {
AtracBase *atrac = getAtrac(atracID);
if (!atrac) {
return hleLogError(Log::Atrac, 0, "bad atrac id");
}
atrac->NotifyGetContextAddress();
return hleLogDebug(Log::Atrac, atrac->context_.ptr);
}
static int sceAtracLowLevelInitDecoder(int atracID, u32 paramsAddr) {
AtracBase *atrac = getAtrac(atracID);
if (!atrac) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_BAD_ATRACID, "bad atrac ID");
}
if (!Memory::IsValidRange(paramsAddr, 12)) {
return hleReportError(Log::Atrac, 0, "invalid pointers");
}
auto params = PSPPointer<Atrac3LowLevelParams>::Create(paramsAddr);
const int codecType = atracContextTypes[atracID];
atrac->InitLowLevel(*params, codecType);
const char *codecName = codecType == PSP_CODEC_AT3 ? "atrac3" : "atrac3+";
const char *encodedChannelName = params->encodedChannels == 1 ? "mono" : "stereo";
const char *outputChannelName = params->outputChannels == 1 ? "mono" : "stereo";
return hleLogInfo(Log::Atrac, 0, "%s %s->%s audio", codecName, encodedChannelName, outputChannelName);
}
static int sceAtracLowLevelDecode(int atracID, u32 sourceAddr, u32 sourceBytesConsumedAddr, u32 samplesAddr, u32 sampleBytesAddr) {
auto srcp = PSPPointer<u8>::Create(sourceAddr);
auto srcConsumed = PSPPointer<u32_le>::Create(sourceBytesConsumedAddr);
auto outp = PSPPointer<s16>::Create(samplesAddr);
auto outWritten = PSPPointer<u32_le>::Create(sampleBytesAddr);
AtracBase *atrac = getAtrac(atracID);
if (!atrac) {
return hleLogError(Log::Atrac, SCE_ERROR_ATRAC_BAD_ATRACID, "bad atrac ID");
}
if (!srcp.IsValid() || !srcConsumed.IsValid() || !outp.IsValid() || !outWritten.IsValid()) {
return hleReportError(Log::Atrac, 0, "invalid pointers");
}
int bytesConsumed = 0;
int bytesWritten = 0;
int retval = atrac->DecodeLowLevel(srcp, &bytesConsumed, outp, &bytesWritten);
*srcConsumed = bytesConsumed;
*outWritten = bytesWritten;
if (g_muteFlag[atracID]) {
memset(outp, 0, bytesWritten);
}
NotifyMemInfo(MemBlockFlags::WRITE, samplesAddr, bytesWritten, "AtracLowLevelDecode");
return hleDelayResult(hleLogDebug(Log::Atrac, retval), "low level atrac decode data", atracDecodeDelay);
}
u32 AtracSasAddStreamData(int atracID, u32 bufPtr, u32 bytesToAdd) {
AtracBase *atrac = getAtrac(atracID);
if (!atrac) {
WARN_LOG(Log::Atrac, "bad atrac ID");
}
return atrac->EnqueueForSas(bufPtr, bytesToAdd);
}
void AtracSasDecodeData(int atracID, u8* outbuf, int *SamplesNum, int *finish) {
AtracBase *atrac = getAtrac(atracID);
if (!atrac) {
WARN_LOG(Log::Atrac, "bad atrac ID");
}
atrac->DecodeForSas((s16 *)outbuf, SamplesNum, finish);
}
int AtracSasBindContextAndGetID(u32 contextAddr) {
int atracID = (int)Memory::Read_U32(contextAddr + 0xfc);
if (atracID < PSP_MAX_ATRAC_IDS && atracContexts[atracID] && atracContexts[atracID]->GetContextVersion() == 1) {
} else {
atracID = -1;
for (int i = 0; i < PSP_MAX_ATRAC_IDS; i++) {
if (!atracContexts[i]) {
continue;
}
if (atracContexts[i]->GetContextVersion() == 2 && atracContexts[i]->context_.Equals(contextAddr)) {
atracID = i;
break;
}
}
_dbg_assert_(atracID != -1);
}
AtracBase *atrac = getAtrac(atracID);
atrac->CheckForSas();
return atracID;
}
const char *AtracStatusToString(AtracStatus status) {
switch (status) {
case ATRAC_STATUS_NO_DATA: return "NO_DATA";
case ATRAC_STATUS_ALL_DATA_LOADED: return "ALL_DATA_LOADED";
case ATRAC_STATUS_HALFWAY_BUFFER: return "HALFWAY_BUFFER";
case ATRAC_STATUS_STREAMED_WITHOUT_LOOP: return "STREAMED_WITHOUT_LOOP";
case ATRAC_STATUS_STREAMED_LOOP_FROM_END: return "STREAMED_LOOP_FROM_END";
case ATRAC_STATUS_STREAMED_LOOP_WITH_TRAILER: return "STREAMED_LOOP_WITH_TRAILER";
case ATRAC_STATUS_LOW_LEVEL: return "LOW_LEVEL";
case ATRAC_STATUS_FOR_SCESAS: return "FOR_SCESAS";
default: return "(unknown!)";
}
}
const HLEFunction sceAtrac3plus[] = {
{0X7DB31251, &WrapU_IU<sceAtracAddStreamData>, "sceAtracAddStreamData", 'x', "ix" },
{0X6A8C3CD5, &WrapU_IUUUU<sceAtracDecodeData>, "sceAtracDecodeData", 'x', "ixppp"},
{0XD5C28CC0, &WrapU_V<sceAtracReleaseResources>, "sceAtracReleaseResources", 'x', "" },
{0X780F88D1, &WrapU_I<sceAtracGetAtracID>, "sceAtracGetAtracID", 'i', "x" },
{0XCA3CA3D2, &WrapU_IIU<sceAtracGetBufferInfoForResetting>, "sceAtracGetBufferInfoForReseting", 'x', "iix" },
{0XA554A158, &WrapU_IU<sceAtracGetBitrate>, "sceAtracGetBitrate", 'x', "ip" },
{0X31668BAA, &WrapU_IU<sceAtracGetChannel>, "sceAtracGetChannel", 'x', "ip" },
{0XFAA4F89B, &WrapU_IUU<sceAtracGetLoopStatus>, "sceAtracGetLoopStatus", 'x', "ipp" },
{0XE88F759B, &WrapU_IU<sceAtracGetInternalErrorInfo>, "sceAtracGetInternalErrorInfo", 'x', "ip" },
{0XD6A5F2F7, &WrapU_IU<sceAtracGetMaxSample>, "sceAtracGetMaxSample", 'x', "ip" },
{0XE23E3A35, &WrapU_IU<sceAtracGetNextDecodePosition>, "sceAtracGetNextDecodePosition", 'x', "ip" },
{0X36FAABFB, &WrapU_IU<sceAtracGetNextSample>, "sceAtracGetNextSample", 'x', "ip" },
{0X9AE849A7, &WrapU_IU<sceAtracGetRemainFrame>, "sceAtracGetRemainFrame", 'x', "ip" },
{0X83E85EA0, &WrapU_IUU<sceAtracGetSecondBufferInfo>, "sceAtracGetSecondBufferInfo", 'x', "ipp" },
{0XA2BBA8BE, &WrapU_IUUU<sceAtracGetSoundSample>, "sceAtracGetSoundSample", 'x', "ippp" },
{0X5D268707, &WrapU_IUUU<sceAtracGetStreamDataInfo>, "sceAtracGetStreamDataInfo", 'x', "ippp" },
{0X61EB33F5, &WrapU_I<sceAtracReleaseAtracID>, "sceAtracReleaseAtracID", 'x', "i" },
{0X644E5607, &WrapU_IIII<sceAtracResetPlayPosition>, "sceAtracResetPlayPosition", 'x', "iiii" },
{0X3F6E26B5, &WrapU_IUUU<sceAtracSetHalfwayBuffer>, "sceAtracSetHalfwayBuffer", 'x', "ixxx" },
{0X83BF7AFD, &WrapU_IUU<sceAtracSetSecondBuffer>, "sceAtracSetSecondBuffer", 'x', "ixx" },
{0X0E2A73AB, &WrapU_IUU<sceAtracSetData>, "sceAtracSetData", 'x', "ixx" },
{0X7A20E7AF, &WrapI_UI<sceAtracSetDataAndGetID>, "sceAtracSetDataAndGetID", 'i', "xx" },
{0XD1F59FDB, &WrapU_V<sceAtracStartEntry>, "sceAtracStartEntry", 'x', "" },
{0X868120B5, &WrapU_II<sceAtracSetLoopNum>, "sceAtracSetLoopNum", 'x', "ii" },
{0X132F1ECA, &WrapI_II<sceAtracReinit>, "sceAtracReinit", 'x', "ii" },
{0XECA32A99, &WrapI_I<sceAtracIsSecondBufferNeeded>, "sceAtracIsSecondBufferNeeded", 'i', "i" },
{0X0FAE370E, &WrapI_UUU<sceAtracSetHalfwayBufferAndGetID>, "sceAtracSetHalfwayBufferAndGetID", 'i', "xxx" },
{0X2DD3E298, &WrapU_IIU<sceAtracGetBufferInfoForResetting>, "sceAtracGetBufferInfoForResetting", 'x', "iix" },
{0X5CF9D852, &WrapI_IUUU<sceAtracSetMOutHalfwayBuffer>, "sceAtracSetMOutHalfwayBuffer", 'x', "ixxx" },
{0XF6837A1A, &WrapU_IUU<sceAtracSetMOutData>, "sceAtracSetMOutData", 'x', "ixx" },
{0X472E3825, &WrapI_UU<sceAtracSetMOutDataAndGetID>, "sceAtracSetMOutDataAndGetID", 'i', "xx" },
{0X9CD7DE03, &WrapI_UUU<sceAtracSetMOutHalfwayBufferAndGetID>, "sceAtracSetMOutHalfwayBufferAndGetID", 'i', "xxx" },
{0XB3B5D042, &WrapI_IU<sceAtracGetOutputChannel>, "sceAtracGetOutputChannel", 'x', "ip" },
{0X5622B7C1, &WrapI_UUUU<sceAtracSetAA3DataAndGetID>, "sceAtracSetAA3DataAndGetID", 'i', "xxxp" },
{0X5DD66588, &WrapI_UUUU<sceAtracSetAA3HalfwayBufferAndGetID>, "sceAtracSetAA3HalfwayBufferAndGetID", 'i', "xxxx" },
{0X231FC6B7, &WrapU_I<_sceAtracGetContextAddress>, "_sceAtracGetContextAddress", 'x', "i" },
{0X1575D64B, &WrapI_IU<sceAtracLowLevelInitDecoder>, "sceAtracLowLevelInitDecoder", 'x', "ix" },
{0X0C116E1B, &WrapI_IUUUU<sceAtracLowLevelDecode>, "sceAtracLowLevelDecode", 'x', "ixpxp"},
};
void Register_sceAtrac3plus() {
RegisterHLEModule("sceATRAC3plus_Library", ARRAY_SIZE(sceAtrac3plus), sceAtrac3plus);
RegisterHLEModule("sceAtrac3plus", ARRAY_SIZE(sceAtrac3plus), sceAtrac3plus);
}