/* ScummVM - Graphic Adventure Engine
 *
 * ScummVM is the legal property of its developers, whose names
 * are too numerous to list here. Please refer to the COPYRIGHT
 * file distributed with this source distribution.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#ifndef MTROPOLIS_VTHREAD_H
#define MTROPOLIS_VTHREAD_H

#include "mtropolis/coroutine_protos.h"
#include "mtropolis/coroutine_return_value.h"
#include "mtropolis/debug.h"

namespace MTropolis {

struct ICoroutineManager;
class VThread;

// Virtual thread, really a task stack
enum VThreadState {
	kVThreadReturn,
	kVThreadSuspended,
	kVThreadError,
};

struct VThreadFaultIdentifier {
};

template<typename T>
struct VThreadFaultIdentifierSingleton {
	static VThreadFaultIdentifier _identifier;
};

template<typename T>
VThreadFaultIdentifier VThreadFaultIdentifierSingleton<T>::_identifier;

class VThreadTaskData : public Debuggable {
public:
	VThreadTaskData();
	virtual ~VThreadTaskData();

	virtual VThreadState execute(VThread *thread) = 0;

#ifdef MTROPOLIS_DEBUG_ENABLE
public:
	void debugInit(const char *name);

protected:
	SupportStatus debugGetSupportStatus() const override { return kSupportStatusDone; }
	const char *debugGetTypeName() const override { return "Task"; }
	const Common::String &debugGetName() const override { return _debugName; }
	void debugInspect(IDebugInspectionReport *report) const override;

	Common::String _debugName;
#endif
};

struct VThreadStackFrame;

class VThreadStackChunk {
public:
	explicit VThreadStackChunk(size_t capacity);
	VThreadStackChunk(VThreadStackChunk &&other);
	~VThreadStackChunk();

	VThreadStackFrame *_topFrame;
	byte *_memory;
	size_t _size;

private:
	VThreadStackChunk() = delete;
	VThreadStackChunk(const VThreadStackChunk &) = delete;
};

struct VThreadStackFrame {
	VThreadTaskData *data;
	VThreadStackFrame *prevFrame;
	bool isLastInChunk;
};

template<typename TClass, typename TData>
class VThreadMethodData : public VThreadTaskData {
public:
	VThreadMethodData(const VThreadFaultIdentifier *faultID, TClass *target, VThreadState (TClass::*method)(const TData &data));
	VThreadMethodData(const VThreadMethodData &other);
	VThreadMethodData(VThreadMethodData &&other);

	VThreadState execute(VThread *thread) override;

	TData &getData();

private:
	const VThreadFaultIdentifier *_faultID;
	TClass *_target;
	VThreadState (TClass::*_method)(const TData &data);
	TData _data;
};

template<typename TData>
class VThreadFunctionData : public VThreadTaskData {

public:
	explicit VThreadFunctionData(const VThreadFaultIdentifier *faultID, VThreadState (*func)(const TData &data));
	VThreadFunctionData(const VThreadFunctionData &other);

	VThreadFunctionData(VThreadFunctionData &&other);

	VThreadState execute(VThread *thread) override;

	TData &getData();

private:
	const VThreadFaultIdentifier *_faultID;
	VThreadState (*_func)(const TData &data);
	TData _data;
};

class VThread {
public:
	explicit VThread(ICoroutineManager *coroManager);
	~VThread();

	template<typename TClass, typename TData>
	TData *pushTask(const char *name, TClass *obj, VThreadState (TClass::*method)(const TData &data));

	template<typename TData>
	TData *pushTask(const char *name, VThreadState (*func)(const TData &data));

	VThreadState step();

	bool hasTasks() const;

	bool popFrame();

	VThreadTaskData *pushCoroutineFrame(const CompiledCoroutine *compiledCoro, const CoroutineParamsBase &params, const CoroutineReturnValueRefBase &returnValueRef);

	template<typename TCoroutine, typename TReturnValue, typename ...TParams>
	void pushCoroutineWithReturn(TReturnValue *returnValuePtr, TParams &&...args);

	template<typename TCoroutine, typename TReturnValue>
	void pushCoroutineWithReturn(TReturnValue *returnValuePtr);

	template<typename TCoroutine, typename... TParams>
	void pushCoroutine(TParams &&...args);

	template<typename TCoroutine>
	void pushCoroutine();


private:
	void reserveFrame(size_t frameAlignment, size_t frameSize, VThreadStackFrame *&outFramePtr, size_t dataAlignment, size_t dataSize, void *&outDataPtr, bool &outIsNewChunk);
	static bool reserveFrameInChunk(VThreadStackChunk *chunk, size_t frameAlignment, size_t frameSize, VThreadStackFrame *&outFramePtr, size_t dataAlignment, size_t dataSize, void *&outDataPtr);

	void pushCoroutineInternal(CompiledCoroutine **compiledCoroPtr, CoroutineCompileFunction_t compileFunc, bool isVoidReturn, const CoroutineParamsBase &params, const CoroutineReturnValueRefBase &returnValueRef);

	template<typename TClass, typename TData>
	TData *pushTaskWithFaultHandler(const VThreadFaultIdentifier *faultID, const char *name, TClass *obj, VThreadState (TClass::*method)(const TData &data));

	template<typename TData>
	TData *pushTaskWithFaultHandler(const VThreadFaultIdentifier *faultID, const char *name, VThreadState (*func)(const TData &data));

	Common::Array<VThreadStackChunk> _stackChunks;
	ICoroutineManager *_coroManager;
	uint _numActiveStackChunks;
};

template<typename TClass, typename TData>
VThreadMethodData<TClass, TData>::VThreadMethodData(const VThreadFaultIdentifier *faultID, TClass *target, VThreadState (TClass::*method)(const TData &data))
	: _faultID(faultID), _target(target), _method(method) {
}

template<typename TClass, typename TData>
VThreadMethodData<TClass, TData>::VThreadMethodData(const VThreadMethodData& other)
	: _faultID(other._faultID), _target(other._target), _method(other._method), _data(other._data) {
}

template<typename TClass, typename TData>
VThreadMethodData<TClass, TData>::VThreadMethodData(VThreadMethodData &&other)
	: _faultID(other._faultID), _target(other._target), _method(other._method), _data(static_cast<TData &&>(other._data)) {
}

template<typename TClass, typename TData>
VThreadState VThreadMethodData<TClass, TData>::execute(VThread *thread) {
	TData data(static_cast<TData &&>(_data));

	TClass *target = _target;
	VThreadState (TClass::*method)(const TData &) = _method;

	thread->popFrame();

	return (target->*method)(data);
}

template<typename TClass, typename TData>
TData &VThreadMethodData<TClass, TData>::getData() {
	return _data;
}

template<typename TData>
VThreadFunctionData<TData>::VThreadFunctionData(const VThreadFaultIdentifier *faultID, VThreadState (*func)(const TData &data))
	: _faultID(faultID), _func(func) {
}

template<typename TData>
VThreadFunctionData<TData>::VThreadFunctionData(const VThreadFunctionData &other)
	: _faultID(other._faultID), _func(other._func), _data(other._data) {
}

template<typename TData>
VThreadFunctionData<TData>::VThreadFunctionData(VThreadFunctionData &&other)
	: _faultID(other._faultID), _func(other._func), _data(static_cast<TData &&>(other._data)) {
}

template<typename TData>
VThreadState VThreadFunctionData<TData>::execute(VThread *thread) {
	TData data(static_cast<TData &&>(_data));

	VThreadState (*func)(const TData &) = _func;

	thread->popFrame();

	return func(data);
}

template<typename TData>
TData &VThreadFunctionData<TData>::getData() {
	return _data;
}

template<typename TCoroutine, typename TReturnValue, typename... TParams>
void VThread::pushCoroutineWithReturn(TReturnValue *returnValuePtr, TParams &&...args) {
	assert(returnValuePtr != nullptr);
	this->pushCoroutineInternal(&TCoroutine::ms_compiledCoro, TCoroutine::compileCoroutine, CoroutineReturnValueRef<typename TCoroutine::ReturnValue_t>::isVoid(), typename TCoroutine::Params(Common::forward<TParams>(args)...), CoroutineReturnValueRef<typename TCoroutine::ReturnValue_t>(returnValuePtr));
}

template<typename TCoroutine, typename TReturnValue>
void VThread::pushCoroutineWithReturn(TReturnValue *returnValuePtr) {
	assert(returnValuePtr != nullptr);
	this->pushCoroutineInternal(&TCoroutine::ms_compiledCoro, TCoroutine::compileCoroutine, CoroutineReturnValueRef<typename TCoroutine::ReturnValue_t>::isVoid(), typename TCoroutine::Params(), CoroutineReturnValueRef<typename TCoroutine::ReturnValue_t>(returnValuePtr));
}

template<typename TCoroutine, typename... TParams>
void VThread::pushCoroutine(TParams &&...args) {
	this->pushCoroutineInternal(&TCoroutine::ms_compiledCoro, TCoroutine::compileCoroutine, CoroutineReturnValueRef<typename TCoroutine::ReturnValue_t>::isVoid(), typename TCoroutine::Params(Common::forward<TParams>(args)...), CoroutineReturnValueRef<typename TCoroutine::ReturnValue_t>());
}

template<typename TCoroutine>
void VThread::pushCoroutine() {
	this->pushCoroutineInternal(&TCoroutine::ms_compiledCoro, TCoroutine::compileCoroutine, CoroutineReturnValueRef<typename TCoroutine::ReturnValue_t>::isVoid(), typename TCoroutine::Params(), CoroutineReturnValueRef<typename TCoroutine::ReturnValue_t>());
}

template<typename TClass, typename TData>
TData *VThread::pushTask(const char *name, TClass *obj, VThreadState (TClass::*method)(const TData &data)) {
	return this->pushTaskWithFaultHandler(nullptr, name, obj, method);
}

template<typename TData>
TData *VThread::pushTask(const char *name, VThreadState (*func)(const TData &data)) {
	return this->pushTaskWithFaultHandler(nullptr, name, func);
}

template<typename TClass, typename TData>
TData *VThread::pushTaskWithFaultHandler(const VThreadFaultIdentifier *faultID, const char *name, TClass *obj, VThreadState (TClass::*method)(const TData &data)) {
	typedef VThreadMethodData<TClass, TData> FrameData_t;

	const size_t frameAlignment = alignof(VThreadStackFrame);
	const size_t dataAlignment = alignof(FrameData_t);

	VThreadStackFrame *prevFrame = nullptr;
	if (_numActiveStackChunks > 0)
		prevFrame = _stackChunks[_numActiveStackChunks - 1]._topFrame;

	VThreadStackFrame *framePtr = nullptr;
	void *dataPtr = nullptr;
	bool isNewChunk = false;
	reserveFrame(frameAlignment, sizeof(VThreadStackFrame), framePtr, dataAlignment, sizeof(FrameData_t), dataPtr, isNewChunk);

	VThreadStackFrame *frame = new (framePtr) VThreadStackFrame();
	FrameData_t *frameData = new (dataPtr) FrameData_t(faultID, obj, method);

	frame->data = frameData;
	frame->prevFrame = prevFrame;
	frame->isLastInChunk = isNewChunk;

#ifdef MTROPOLIS_DEBUG_ENABLE
	frameData->debugInit(name);
#endif

	return &frameData->getData();
}

template<typename TData>
TData *VThread::pushTaskWithFaultHandler(const VThreadFaultIdentifier *faultID, const char *name, VThreadState (*func)(const TData &data)) {
	typedef VThreadFunctionData<TData> FrameData_t;

	const size_t frameAlignment = alignof(VThreadStackFrame);
	const size_t dataAlignment = alignof(FrameData_t);

	VThreadStackFrame *prevFrame = nullptr;
	if (_numActiveStackChunks > 0)
		prevFrame = _stackChunks[_numActiveStackChunks - 1]._topFrame;

	VThreadStackFrame *framePtr = nullptr;
	void *dataPtr = nullptr;
	bool isNewChunk = false;
	reserveFrame(frameAlignment, sizeof(VThreadStackFrame), framePtr, dataAlignment, sizeof(FrameData_t), dataPtr, isNewChunk);

	VThreadStackFrame *frame = new (framePtr) VThreadStackFrame();
	FrameData_t *frameData = new (dataPtr) FrameData_t(faultID, func);

	frame->data = frameData;
	frame->prevFrame = prevFrame;
	frame->isLastInChunk = isNewChunk;

#ifdef MTROPOLIS_DEBUG_ENABLE
	frameData->debugInit(name);
#endif

	return &frameData->getData();
}

} // End of namespace MTropolis

#endif