#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();
}