mirror of
https://github.com/isledecomp/isle.git
synced 2025-10-23 08:24:16 +00:00
402 lines
9.1 KiB
C++
402 lines
9.1 KiB
C++
#include "mxdiskstreamprovider.h"
|
|
|
|
#include "mxautolock.h"
|
|
#include "mxdiskstreamcontroller.h"
|
|
#include "mxdsbuffer.h"
|
|
#include "mxdsfile.h"
|
|
#include "mxdsstreamingaction.h"
|
|
#include "mxomni.h"
|
|
#include "mxramstreamprovider.h"
|
|
#include "mxstreamcontroller.h"
|
|
#include "mxstring.h"
|
|
#include "mxthread.h"
|
|
|
|
DECOMP_SIZE_ASSERT(MxDiskStreamProviderThread, 0x1c)
|
|
DECOMP_SIZE_ASSERT(MxDiskStreamProvider, 0x60);
|
|
|
|
// GLOBAL: LEGO1 0x10102878
|
|
MxU32 g_unk0x10102878 = 0;
|
|
|
|
// FUNCTION: LEGO1 0x100d0f30
|
|
MxResult MxDiskStreamProviderThread::Run()
|
|
{
|
|
if (m_target) {
|
|
((MxDiskStreamProvider*) m_target)->WaitForWorkToComplete();
|
|
}
|
|
MxThread::Run();
|
|
// They should probably have writen "return MxThread::Run()" but they didn't.
|
|
return SUCCESS;
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d0f50
|
|
MxResult MxDiskStreamProviderThread::StartWithTarget(MxDiskStreamProvider* p_target)
|
|
{
|
|
m_target = p_target;
|
|
return Start(0x1000, 0);
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d0f70
|
|
MxDiskStreamProvider::MxDiskStreamProvider()
|
|
{
|
|
m_pFile = NULL;
|
|
m_remainingWork = FALSE;
|
|
m_unk0x35 = FALSE;
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d1240
|
|
MxDiskStreamProvider::~MxDiskStreamProvider()
|
|
{
|
|
MxDSObject* action;
|
|
m_unk0x35 = FALSE;
|
|
|
|
do {
|
|
action = NULL;
|
|
|
|
{
|
|
AUTOLOCK(m_criticalSection);
|
|
m_list.PopFront(action);
|
|
}
|
|
|
|
if (!action) {
|
|
break;
|
|
}
|
|
|
|
if (((MxDSStreamingAction*) action)->GetUnknowna0()->GetWriteOffset() < 0x20000) {
|
|
g_unk0x10102878--;
|
|
}
|
|
|
|
((MxDiskStreamController*) m_pLookup)->FUN_100c8670((MxDSStreamingAction*) action);
|
|
} while (action);
|
|
|
|
if (m_remainingWork) {
|
|
m_remainingWork = FALSE;
|
|
m_busySemaphore.Release(1);
|
|
m_thread.Terminate();
|
|
}
|
|
|
|
if (m_pFile) {
|
|
delete m_pFile;
|
|
}
|
|
|
|
m_pFile = NULL;
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d13d0
|
|
MxResult MxDiskStreamProvider::SetResourceToGet(MxStreamController* p_resource)
|
|
{
|
|
MxResult result = FAILURE;
|
|
MxString path;
|
|
m_pLookup = p_resource;
|
|
|
|
path = (MxString(MxOmni::GetHD()) + p_resource->GetAtom().GetInternal() + ".si");
|
|
|
|
m_pFile = new MxDSFile(path.GetData(), 0);
|
|
if (m_pFile != NULL) {
|
|
if (m_pFile->Open(OF_READ) != 0) {
|
|
path = MxString(MxOmni::GetCD()) + p_resource->GetAtom().GetInternal() + ".si";
|
|
m_pFile->SetFileName(path.GetData());
|
|
|
|
if (m_pFile->Open(OF_READ) != 0) {
|
|
goto done;
|
|
}
|
|
}
|
|
|
|
m_remainingWork = TRUE;
|
|
m_busySemaphore.Init(0, 100);
|
|
|
|
if (m_thread.StartWithTarget(this) == SUCCESS && p_resource != NULL) {
|
|
result = SUCCESS;
|
|
}
|
|
}
|
|
|
|
done:
|
|
return result;
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d15e0
|
|
void MxDiskStreamProvider::VTable0x20(MxDSAction* p_action)
|
|
{
|
|
MxDSObject* action;
|
|
|
|
if (p_action->GetObjectId() == -1) {
|
|
m_unk0x35 = FALSE;
|
|
|
|
do {
|
|
action = NULL;
|
|
|
|
{
|
|
AUTOLOCK(m_criticalSection);
|
|
m_list.PopFront(action);
|
|
}
|
|
|
|
if (!action) {
|
|
return;
|
|
}
|
|
|
|
if (((MxDSStreamingAction*) action)->GetUnknowna0()->GetWriteOffset() < 0x20000) {
|
|
g_unk0x10102878--;
|
|
}
|
|
|
|
((MxDiskStreamController*) m_pLookup)->FUN_100c8670((MxDSStreamingAction*) action);
|
|
} while (action);
|
|
}
|
|
else {
|
|
do {
|
|
{
|
|
AUTOLOCK(m_criticalSection);
|
|
action = (MxDSStreamingAction*) m_list.FindAndErase(p_action);
|
|
}
|
|
|
|
if (!action) {
|
|
return;
|
|
}
|
|
|
|
if (((MxDSStreamingAction*) action)->GetUnknowna0()->GetWriteOffset() < 0x20000) {
|
|
g_unk0x10102878--;
|
|
}
|
|
|
|
((MxDiskStreamController*) m_pLookup)->FUN_100c8670((MxDSStreamingAction*) action);
|
|
} while (action);
|
|
}
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d1750
|
|
MxResult MxDiskStreamProvider::WaitForWorkToComplete()
|
|
{
|
|
while (m_remainingWork) {
|
|
m_busySemaphore.Wait(INFINITE);
|
|
if (m_unk0x35) {
|
|
PerformWork();
|
|
}
|
|
}
|
|
|
|
return SUCCESS;
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d1780
|
|
MxResult MxDiskStreamProvider::FUN_100d1780(MxDSStreamingAction* p_action)
|
|
{
|
|
if (!m_remainingWork) {
|
|
return FAILURE;
|
|
}
|
|
|
|
if (p_action->GetUnknown9c() > 0 && !p_action->GetUnknowna0()) {
|
|
MxDSBuffer* buffer = new MxDSBuffer();
|
|
|
|
if (!buffer) {
|
|
return FAILURE;
|
|
}
|
|
|
|
if (buffer->AllocateBuffer(GetFileSize(), MxDSBuffer::e_allocate) != SUCCESS) {
|
|
delete buffer;
|
|
return FAILURE;
|
|
}
|
|
|
|
p_action->SetUnknowna0(buffer);
|
|
}
|
|
|
|
if (p_action->GetUnknowna0()->GetWriteOffset() < 0x20000) {
|
|
g_unk0x10102878++;
|
|
}
|
|
|
|
{
|
|
AUTOLOCK(m_criticalSection);
|
|
m_list.PushBack(p_action);
|
|
}
|
|
|
|
m_unk0x35 = TRUE;
|
|
m_busySemaphore.Release(1);
|
|
return SUCCESS;
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d18f0
|
|
void MxDiskStreamProvider::PerformWork()
|
|
{
|
|
MxDiskStreamController* controller = (MxDiskStreamController*) m_pLookup;
|
|
MxDSObject* streamingAction = NULL;
|
|
|
|
{
|
|
AUTOLOCK(m_criticalSection);
|
|
if (!m_list.empty()) {
|
|
streamingAction = m_list.front();
|
|
|
|
if (streamingAction && !FUN_100d1af0((MxDSStreamingAction*) streamingAction)) {
|
|
m_thread.Sleep(500);
|
|
m_busySemaphore.Release(1);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
MxDSBuffer* buffer;
|
|
|
|
{
|
|
AUTOLOCK(m_criticalSection);
|
|
|
|
if (!m_list.PopFront(streamingAction)) {
|
|
goto done;
|
|
}
|
|
}
|
|
|
|
if (((MxDSStreamingAction*) streamingAction)->GetUnknowna0()->GetWriteOffset() < 0x20000) {
|
|
g_unk0x10102878--;
|
|
}
|
|
|
|
buffer = ((MxDSStreamingAction*) streamingAction)->GetUnknowna0();
|
|
|
|
if (m_pFile->GetPosition() == ((MxDSStreamingAction*) streamingAction)->GetBufferOffset() ||
|
|
m_pFile->Seek(((MxDSStreamingAction*) streamingAction)->GetBufferOffset(), SEEK_SET) == 0) {
|
|
buffer->SetUnknown14(m_pFile->GetPosition());
|
|
|
|
if (m_pFile->ReadToBuffer(buffer) == SUCCESS) {
|
|
buffer->SetUnknown1c(m_pFile->GetPosition());
|
|
|
|
if (((MxDSStreamingAction*) streamingAction)->GetUnknown9c() > 0) {
|
|
FUN_100d1b20(((MxDSStreamingAction*) streamingAction));
|
|
}
|
|
else {
|
|
if (m_pLookup == NULL || !((MxDiskStreamController*) m_pLookup)->GetUnk0xc4()) {
|
|
controller->FUN_100c8670(((MxDSStreamingAction*) streamingAction));
|
|
}
|
|
else {
|
|
controller->FUN_100c7f40(((MxDSStreamingAction*) streamingAction));
|
|
}
|
|
}
|
|
|
|
streamingAction = NULL;
|
|
}
|
|
}
|
|
|
|
done:
|
|
if (streamingAction) {
|
|
controller->FUN_100c8670(((MxDSStreamingAction*) streamingAction));
|
|
}
|
|
|
|
m_thread.Sleep(0);
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d1af0
|
|
MxBool MxDiskStreamProvider::FUN_100d1af0(MxDSStreamingAction* p_action)
|
|
{
|
|
if (p_action->GetUnknowna0()->GetWriteOffset() == 0x20000) {
|
|
return g_unk0x10102878 == 0;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d1b20
|
|
// FUNCTION: BETA10 0x10163712
|
|
MxResult MxDiskStreamProvider::FUN_100d1b20(MxDSStreamingAction* p_action)
|
|
{
|
|
MxDSBuffer* buffer = new MxDSBuffer();
|
|
|
|
if (!buffer) {
|
|
return FAILURE;
|
|
}
|
|
|
|
MxU32 size = (p_action->GetUnknowna4() ? p_action->GetUnknowna4()->GetWriteOffset() : 0) +
|
|
p_action->GetUnknowna0()->GetWriteOffset() - (p_action->GetUnknown94() - p_action->GetBufferOffset());
|
|
|
|
if (buffer->AllocateBuffer(size, MxDSBuffer::e_allocate) != SUCCESS) {
|
|
delete buffer;
|
|
return FAILURE;
|
|
}
|
|
|
|
MxU8* data;
|
|
|
|
if (p_action->GetUnknowna4()) {
|
|
buffer->FUN_100c7090(p_action->GetUnknowna4());
|
|
data = buffer->GetBuffer() + p_action->GetUnknowna4()->GetWriteOffset();
|
|
|
|
memcpy(data, p_action->GetUnknowna0()->GetBuffer(), p_action->GetUnknowna0()->GetWriteOffset());
|
|
|
|
delete p_action->GetUnknowna4();
|
|
}
|
|
else {
|
|
data = buffer->GetBuffer();
|
|
|
|
memcpy(
|
|
data,
|
|
p_action->GetUnknowna0()->GetBuffer() + (p_action->GetUnknown94() - p_action->GetBufferOffset()),
|
|
size
|
|
);
|
|
}
|
|
|
|
p_action->SetUnknowna4(buffer);
|
|
|
|
#define IntoType(p) ((MxU32*) (p))
|
|
|
|
while (data) {
|
|
if (*IntoType(data) != FOURCC('M', 'x', 'O', 'b') &&
|
|
*MxStreamChunk::IntoTime(data) > p_action->GetUnknown9c()) {
|
|
*IntoType(data) = FOURCC('p', 'a', 'd', ' ');
|
|
|
|
// DECOMP: prefer order that matches retail versus beta
|
|
*(MxU32*) (data + 4) = buffer->GetBuffer() + buffer->GetWriteOffset() - data - 8;
|
|
memset(data + 8, 0, *(MxU32*) (data + 4));
|
|
size = ReadData(buffer->GetBuffer(), buffer->GetWriteOffset());
|
|
|
|
buffer = new MxDSBuffer();
|
|
if (buffer == NULL || buffer->AllocateBuffer(size, MxDSBuffer::e_allocate) != SUCCESS) {
|
|
delete buffer;
|
|
return FAILURE;
|
|
}
|
|
|
|
memcpy(buffer->GetBuffer(), p_action->GetUnknowna4()->GetBuffer(), size);
|
|
p_action->GetUnknowna4()->SetMode(MxDSBuffer::e_allocate);
|
|
delete p_action->GetUnknowna4();
|
|
|
|
buffer->SetMode(MxDSBuffer::e_unknown);
|
|
p_action->SetUnknowna4(buffer);
|
|
MxU32 unk0x14 = p_action->GetUnknowna0()->GetUnknown14();
|
|
|
|
for (data = p_action->GetUnknowna0()->GetBuffer();
|
|
*MxStreamChunk::IntoTime(data) <= p_action->GetUnknown9c();
|
|
data = MxDSChunk::End(data)) {
|
|
unk0x14 += MxDSChunk::Size(data);
|
|
}
|
|
|
|
p_action->SetUnknown94(unk0x14);
|
|
p_action->SetBufferOffset(p_action->GetUnknowna0()->GetUnknown14());
|
|
delete p_action->GetUnknowna0();
|
|
p_action->ClearUnknowna0();
|
|
((MxDiskStreamController*) m_pLookup)->FUN_100c7890(p_action);
|
|
return SUCCESS;
|
|
}
|
|
|
|
data = buffer->FUN_100c6fa0(data);
|
|
}
|
|
|
|
p_action->SetUnknown94(GetFileSize() + p_action->GetBufferOffset());
|
|
p_action->SetBufferOffset(GetFileSize() + p_action->GetBufferOffset());
|
|
FUN_100d1780(p_action);
|
|
return SUCCESS;
|
|
|
|
#undef IntoType
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d1e90
|
|
MxU32 MxDiskStreamProvider::GetFileSize()
|
|
{
|
|
return m_pFile->GetBufferSize();
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d1ea0
|
|
MxS32 MxDiskStreamProvider::GetStreamBuffersNum()
|
|
{
|
|
return m_pFile->GetStreamBuffersNum();
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d1eb0
|
|
MxU32 MxDiskStreamProvider::GetLengthInDWords()
|
|
{
|
|
return m_pFile->GetLengthInDWords();
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x100d1ec0
|
|
MxU32* MxDiskStreamProvider::GetBufferForDWords()
|
|
{
|
|
return m_pFile->GetBuffer();
|
|
}
|