#include "mxdiskstreamprovider.h"

#include "mxautolocker.h"
#include "mxdiskstreamcontroller.h"
#include "mxdsbuffer.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()
{
	this->m_pFile = NULL;
	this->m_remainingWork = FALSE;
	this->m_unk0x35 = FALSE;
}

// FUNCTION: LEGO1 0x100d1240
MxDiskStreamProvider::~MxDiskStreamProvider()
{
	MxDSStreamingAction* action;
	m_unk0x35 = FALSE;

	do {
		action = NULL;

		{
			MxAutoLocker lock(&m_criticalSection);
			m_list.PopFrontStreamingAction(action);
		}

		if (!action)
			break;

		if (action->GetUnknowna0()->GetWriteOffset() < 0x20000)
			g_unk0x10102878--;

		((MxDiskStreamController*) m_pLookup)->FUN_100c8670(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(0) != 0) {
			path = MxString(MxOmni::GetCD()) + p_resource->GetAtom().GetInternal() + ".si";
			m_pFile->SetFileName(path.GetData());

			if (m_pFile->Open(0) != 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)
{
	MxDSStreamingAction* action;

	if (p_action->GetObjectId() == -1) {
		m_unk0x35 = FALSE;

		do {
			action = NULL;

			{
				MxAutoLocker lock(&m_criticalSection);
				m_list.PopFrontStreamingAction(action);
			}

			if (!action)
				return;

			if (action->GetUnknowna0()->GetWriteOffset() < 0x20000)
				g_unk0x10102878--;

			((MxDiskStreamController*) m_pLookup)->FUN_100c8670(action);
		} while (action);
	}
	else {
		do {
			{
				MxAutoLocker lock(&m_criticalSection);
				action = (MxDSStreamingAction*) m_list.Find(p_action, TRUE);
			}

			if (!action)
				return;

			if (action->GetUnknowna0()->GetWriteOffset() < 0x20000)
				g_unk0x10102878--;

			((MxDiskStreamController*) m_pLookup)->FUN_100c8670(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(), MxDSBufferType_Allocate) != SUCCESS) {
			delete buffer;
			return FAILURE;
		}

		p_action->SetUnknowna0(buffer);
	}

	if (p_action->GetUnknowna0()->GetWriteOffset() < 0x20000) {
		g_unk0x10102878++;
	}

	{
		MxAutoLocker lock(&m_criticalSection);
		m_list.push_back(p_action);
	}

	m_unk0x35 = TRUE;
	m_busySemaphore.Release(1);
	return SUCCESS;
}

// FUNCTION: LEGO1 0x100d18f0
void MxDiskStreamProvider::PerformWork()
{
	MxDiskStreamController* controller = (MxDiskStreamController*) m_pLookup;
	MxDSStreamingAction* streamingAction = NULL;

	{
		MxAutoLocker lock(&m_criticalSection);
		if (!m_list.empty()) {
			streamingAction = (MxDSStreamingAction*) m_list.front();

			if (streamingAction && !FUN_100d1af0(streamingAction)) {
				m_thread.Sleep(500);
				m_busySemaphore.Release(1);
				return;
			}
		}
	}

	MxDSBuffer* buffer;

	{
		MxAutoLocker lock(&m_criticalSection);

		if (!m_list.PopFrontStreamingAction(streamingAction))
			goto done;
	}

	if (streamingAction->GetUnknowna0()->GetWriteOffset() < 0x20000) {
		g_unk0x10102878--;
	}

	buffer = streamingAction->GetUnknowna0();

	if (m_pFile->GetPosition() == streamingAction->GetBufferOffset() ||
		m_pFile->Seek(streamingAction->GetBufferOffset(), 0) == 0) {
		buffer->SetUnknown14(m_pFile->GetPosition());

		if (m_pFile->ReadToBuffer(buffer) == SUCCESS) {
			buffer->SetUnknown1c(m_pFile->GetPosition());

			if (streamingAction->GetUnknown9c() > 0) {
				FUN_100d1b20(streamingAction);
			}
			else {
				if (m_pLookup == NULL || !((MxDiskStreamController*) m_pLookup)->GetUnk0xc4()) {
					controller->FUN_100c8670(streamingAction);
				}
				else {
					controller->FUN_100c7f40(streamingAction);
				}
			}

			streamingAction = NULL;
		}
	}

done:
	if (streamingAction) {
		controller->FUN_100c8670(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
MxResult MxDiskStreamProvider::FUN_100d1b20(MxDSStreamingAction* p_action)
{
	MxDSBuffer* buffer = new MxDSBuffer();

	if (!buffer)
		return FAILURE;

	MxU32 size = p_action->GetUnknowna0()->GetWriteOffset() - p_action->GetUnknown94() + p_action->GetBufferOffset() +
				 (p_action->GetUnknowna4() ? p_action->GetUnknowna4()->GetWriteOffset() : 0);

	if (buffer->AllocateBuffer(size, MxDSBufferType_Allocate) != SUCCESS) {
		if (!buffer)
			return FAILURE;

		delete buffer;
		return FAILURE;
	}

	MxDSBuffer* buffer2 = p_action->GetUnknowna4();
	MxU8** pdata;
	MxU8* data;

	if (buffer2 == NULL) {
		pdata = buffer->GetBufferRef();

		memcpy(
			data = *pdata,
			p_action->GetUnknowna0()->GetBuffer() - p_action->GetBufferOffset() + p_action->GetUnknown94(),
			size
		);
	}
	else {
		buffer->FUN_100c7090(buffer2);
		pdata = buffer->GetBufferRef();

		memcpy(
			data = (p_action->GetUnknowna4()->GetWriteOffset() + *pdata),
			p_action->GetUnknowna0()->GetBuffer(),
			p_action->GetUnknowna0()->GetWriteOffset()
		);

		delete p_action->GetUnknowna4();
	}

	p_action->SetUnknowna4(buffer);

	while (data) {
		if (*MxDSChunk::IntoType(data) != FOURCC('M', 'x', 'O', 'b')) {
			if (*MxStreamChunk::IntoTime(data) > p_action->GetUnknown9c()) {
				*MxDSChunk::IntoType(data) = FOURCC('p', 'a', 'd', ' ');

				memcpy(data + 8, *pdata, buffer->GetWriteOffset() + *pdata - data - 8);
				size = ReadData(*pdata, buffer->GetWriteOffset());

				MxDSBuffer* buffer3 = new MxDSBuffer();
				if (!buffer3)
					return FAILURE;

				if (buffer3->AllocateBuffer(size, MxDSBufferType_Allocate) == SUCCESS) {
					memcpy(buffer3->GetBuffer(), p_action->GetUnknowna4()->GetBuffer(), size);
					p_action->GetUnknowna4()->SetMode(MxDSBufferType_Allocate);
					delete p_action->GetUnknowna4();

					buffer3->SetMode(MxDSBufferType_Unknown);
					p_action->SetUnknowna4(buffer3);
					MxDSBuffer* buffer4 = p_action->GetUnknowna0();
					MxU32 unk0x14 = buffer4->GetUnknown14();
					MxU8* data2 = buffer4->GetBuffer();

					while (TRUE) {
						if (*MxStreamChunk::IntoTime(data2) > p_action->GetUnknown9c())
							break;

						data += MxDSChunk::Size(*MxDSChunk::IntoLength(data));
						unk0x14 += MxDSChunk::Size(*MxDSChunk::IntoLength(data));
					}

					p_action->SetUnknown94(unk0x14);
					p_action->SetBufferOffset(p_action->GetUnknowna0()->GetUnknown14());
					delete p_action->GetUnknowna0();
					p_action->SetUnknowna0(NULL);
					((MxDiskStreamController*) m_pLookup)->FUN_100c7890(p_action);
					return SUCCESS;
				}
				else {
					delete buffer3;
					return FAILURE;
				}
			}
		}

		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;
}

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