#include "mxdsfile.h"

#include "decomp.h"

#include <stdio.h>

#define SI_MAJOR_VERSION 2
#define SI_MINOR_VERSION 2

DECOMP_SIZE_ASSERT(MxDSFile::ChunkHeader, 0x0c)
DECOMP_SIZE_ASSERT(MxDSFile, 0x7c)

// FUNCTION: LEGO1 0x100cc4b0
MxDSFile::MxDSFile(const char* p_filename, MxULong p_skipReadingChunks)
{
	m_filename = p_filename;
	m_skipReadingChunks = p_skipReadingChunks;
}

// FUNCTION: LEGO1 0x100cc590
MxLong MxDSFile::Open(MxULong p_uStyle)
{
	MXIOINFO& io = m_io;
	MxLong longResult = 1;
	memset(&io, 0, sizeof(MXIOINFO));

	if (io.Open(m_filename.GetData(), p_uStyle) != 0) {
		return -1;
	}

	io.SetBuffer(NULL, 0, 0);
	m_position = 0;

	if (m_skipReadingChunks == 0) {
		longResult = ReadChunks();
	}

	if (longResult != 0) {
		Close();
	}
	else {
		Seek(0, 0);
	}

	return longResult;
}

// FUNCTION: LEGO1 0x100cc620
MxLong MxDSFile::ReadChunks()
{
	_MMCKINFO topChunk;
	_MMCKINFO childChunk;
	char tempBuffer[80];

	topChunk.fccType = FOURCC('O', 'M', 'N', 'I');
	if (m_io.Descend(&topChunk, NULL, MMIO_FINDRIFF) != 0) {
		return -1;
	}
	childChunk.ckid = FOURCC('M', 'x', 'H', 'd');
	if (m_io.Descend(&childChunk, &topChunk, 0) != 0) {
		return -1;
	}

	m_io.Read(&m_header, 0x0c);
	if ((m_header.m_majorVersion == SI_MAJOR_VERSION) && (m_header.m_minorVersion == SI_MINOR_VERSION)) {
		childChunk.ckid = FOURCC('M', 'x', 'O', 'f');
		if (m_io.Descend(&childChunk, &topChunk, 0) != 0) {
			return -1;
		}
		MxULong* pLengthInDWords = &m_lengthInDWords;
		m_io.Read(pLengthInDWords, 4);
		m_pBuffer = new MxU32[*pLengthInDWords];
		m_io.Read(m_pBuffer, *pLengthInDWords * 4);
		return 0;
	}
	else {
		sprintf(tempBuffer, "Wrong SI file version. %d.%d expected.", SI_MAJOR_VERSION, SI_MINOR_VERSION);
		MessageBoxA(NULL, tempBuffer, NULL, MB_ICONERROR);
		return -1;
	}
}

// FUNCTION: LEGO1 0x100cc740
MxLong MxDSFile::Close()
{
	m_io.Close(0);
	m_position = -1;
	memset(&m_header, 0, sizeof(m_header));
	if (m_lengthInDWords != 0) {
		m_lengthInDWords = 0;
		delete[] m_pBuffer;
		m_pBuffer = NULL;
	}
	return 0;
}

// FUNCTION: LEGO1 0x100cc780
MxResult MxDSFile::Read(unsigned char* p_buf, MxULong p_nbytes)
{
	if (m_io.Read(p_buf, p_nbytes) != p_nbytes) {
		return FAILURE;
	}

	m_position += p_nbytes;
	return SUCCESS;
}

// FUNCTION: LEGO1 0x100cc7b0
MxLong MxDSFile::Seek(MxLong p_lOffset, MxS32 p_iOrigin)
{
	return (m_position = m_io.Seek(p_lOffset, p_iOrigin)) == -1 ? -1 : 0;
}

// FUNCTION: LEGO1 0x100cc7e0
MxULong MxDSFile::GetBufferSize()
{
	return m_header.m_bufferSize;
}

// FUNCTION: LEGO1 0x100cc7f0
MxULong MxDSFile::GetStreamBuffersNum()
{
	return m_header.m_streamBuffersNum;
}
