#ifndef MXDISKSTREAMPROVIDER_H
#define MXDISKSTREAMPROVIDER_H

#include "compat.h"
#include "decomp.h"
#include "mxcriticalsection.h"
#include "mxdsaction.h"
#include "mxstreamlist.h"
#include "mxstreamprovider.h"
#include "mxthread.h"

class MxDiskStreamProvider;

// VTABLE: LEGO1 0x100dd130
class MxDiskStreamProviderThread : public MxThread {
public:
	// Only inlined, no offset
	inline MxDiskStreamProviderThread() : MxThread() { m_target = NULL; }

	MxResult Run() override;

	MxResult StartWithTarget(MxDiskStreamProvider* p_target);
};

// VTABLE: LEGO1 0x100dd138
class MxDiskStreamProvider : public MxStreamProvider {
public:
	MxDiskStreamProvider();

	virtual ~MxDiskStreamProvider() override;

	// FUNCTION: LEGO1 0x100d1160
	inline virtual const char* ClassName() const override // vtable+0x0c
	{
		// GLOBAL: LEGO1 0x1010287c
		return "MxDiskStreamProvider";
	}

	// FUNCTION: LEGO1 0x100d1170
	inline virtual MxBool IsA(const char* name) const override // vtable+0x10
	{
		return !strcmp(name, MxDiskStreamProvider::ClassName()) || MxStreamProvider::IsA(name);
	}

	MxResult WaitForWorkToComplete();

	void PerformWork();

	virtual MxResult SetResourceToGet(MxStreamController* p_resource) override; // vtable+0x14
	virtual MxU32 GetFileSize() override;                                       // vtable+0x18
	virtual MxU32 GetStreamBuffersNum() override;                               // vtable+0x1c
	virtual void vtable0x20(undefined4 p_unknown1) override;                    // vtable+0x20
	virtual MxU32 GetLengthInDWords() override;                                 // vtable+0x24
	virtual MxU32* GetBufferForDWords() override;                               // vtable+0x28

private:
	MxDiskStreamProviderThread m_thread; // 0x10
	MxSemaphore m_busySemaphore;         // 0x2c
	undefined m_remainingWork;           // 0x34
	undefined m_unk35;                   // 0x35
	MxCriticalSection m_criticalSection; // 0x38
	MxStreamListMxDSAction m_list;       // 0x54
};

#endif // MXDISKSTREAMPROVIDER_H