/**

 * vim: set ts=4 :

 * =============================================================================

 * SourceMod

 * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.

 * =============================================================================

 *

 * This program is free software; you can redistribute it and/or modify it under

 * the terms of the GNU General Public License, version 3.0, as published by the

 * Free Software Foundation.

 * 

 * 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/>.

 *

 * As a special exception, AlliedModders LLC gives you permission to link the

 * code of this program (as well as its derivative works) to "Half-Life 2," the

 * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software

 * by the Valve Corporation.  You must obey the GNU General Public License in

 * all respects for all other code used.  Additionally, AlliedModders LLC grants

 * this exception to all derivative works.  AlliedModders LLC defines further

 * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),

 * or <http://www.sourcemod.net/license.php>.

 *

 * Version: $Id$

 */

#include "BaseWorker.h"

BaseWorker::BaseWorker(IThreadWorkerCallbacks *hooks) : 

	m_perFrame(SM_DEFAULT_THREADS_PER_FRAME),

	m_state(Worker_Stopped),

	m_pHooks(hooks)

{

}

BaseWorker::~BaseWorker()

{

	if (m_state != Worker_Stopped || m_state != Worker_Invalid)

		Stop(true);

	if (m_ThreadQueue.size())

		Flush(true);

}

void BaseWorker::MakeThread(IThread *pThread)

{

	ThreadParams pt;

	pt.flags = Thread_AutoRelease;

	pt.prio = ThreadPrio_Normal;

	MakeThread(pThread, &pt);

}

IThreadHandle *BaseWorker::MakeThread(IThread *pThread, ThreadFlags flags)

{

	ThreadParams pt;

	pt.flags = flags;

	pt.prio = ThreadPrio_Normal;

	return MakeThread(pThread, &pt);

}

IThreadHandle *BaseWorker::MakeThread(IThread *pThread, const ThreadParams *params)

{

	if (m_state != Worker_Running)

		return NULL;

	SWThreadHandle *swt = new SWThreadHandle(this, params, pThread);

	AddThreadToQueue(swt);

	return swt;

}

void BaseWorker::GetPriorityBounds(ThreadPriority &max, ThreadPriority &min)

{

	max = ThreadPrio_Normal;

	min = ThreadPrio_Normal;

}

unsigned int BaseWorker::Flush(bool flush_cancel)

{

	SWThreadHandle *swt;

	unsigned int num = 0;

	while ((swt=PopThreadFromQueue()) != NULL)

	{

		swt->m_state = Thread_Done;

		if (!flush_cancel)

			swt->pThread->RunThread(swt);

		swt->pThread->OnTerminate(swt, flush_cancel);

		if (swt->m_params.flags & Thread_AutoRelease)

			delete swt;

		num++;

	}

	return num;

}

SWThreadHandle *BaseWorker::PopThreadFromQueue()

{

	if (!m_ThreadQueue.size())

		return NULL;

	SourceHook::List<SWThreadHandle *>::iterator begin;

	SWThreadHandle *swt;

	begin = m_ThreadQueue.begin();

	swt = (*begin);

	m_ThreadQueue.erase(begin);

	return swt;

}

void BaseWorker::AddThreadToQueue(SWThreadHandle *pHandle)

{

	m_ThreadQueue.push_back(pHandle);

}

WorkerState BaseWorker::GetStatus(unsigned int *threads)

{

	if (threads)

		*threads = m_perFrame;

	return m_state;

}

unsigned int BaseWorker::RunFrame()

{

	unsigned int done = 0;

	unsigned int max = m_perFrame;

	SWThreadHandle *swt = NULL;

	IThread *pThread = NULL;

	while (done < max)

	{

		if ((swt=PopThreadFromQueue()) == NULL)

		{

			break;

		}

		pThread = swt->pThread;

		swt->m_state = Thread_Running;

		pThread->RunThread(swt);

		swt->m_state = Thread_Done;

		pThread->OnTerminate(swt, false);

		if (swt->m_params.flags & Thread_AutoRelease)

		{

			delete swt;

		}

		done++;

	}

	return done;

}

void BaseWorker::SetMaxThreadsPerFrame(unsigned int threads)

{

	m_perFrame = threads;

}

bool BaseWorker::Start()

{

	if (m_state != Worker_Invalid && m_state != Worker_Stopped)

	{

		return false;

	}

	m_state = Worker_Running;

	if (m_pHooks)

	{

		m_pHooks->OnWorkerStart(this);

	}

    return true;

}

bool BaseWorker::Stop(bool flush_cancel)

{

	if (m_state == Worker_Invalid || m_state == Worker_Stopped)

	{

		return false;

	}

	if (m_state == Worker_Paused)

	{

		if (!Unpause())

		{

			return false;

		}

	}

	m_state = Worker_Stopped;

	Flush(flush_cancel);

	if (m_pHooks)

	{

		m_pHooks->OnWorkerStop(this);

	}

	return true;

}

bool BaseWorker::Pause()

{

	if (m_state != Worker_Running)

	{

		return false;

	}

	m_state = Worker_Paused;

	return true;

}

bool BaseWorker::Unpause()

{

	if (m_state != Worker_Paused)

	{

		return false;

	}

	m_state = Worker_Running;

	return true;

}

/***********************

 * THREAD HANDLE STUFF *

 ***********************/

void SWThreadHandle::DestroyThis()

{

	delete this;

}

void SWThreadHandle::GetParams(ThreadParams *p)

{

	*p = m_params;

}

ThreadPriority SWThreadHandle::GetPriority()

{

	return m_params.prio;

}

ThreadState SWThreadHandle::GetState()

{

	return m_state;

}

IThreadCreator *SWThreadHandle::Parent()

{

	return m_parent;

}

bool SWThreadHandle::SetPriority(ThreadPriority prio)

{

	if (m_params.prio != ThreadPrio_Normal)

		return false;

	m_params.prio = prio;

	return true;

}

bool SWThreadHandle::Unpause()

{

	if (m_state != Thread_Paused)

		return false;

	m_state = Thread_Running;

	return true;

}

bool SWThreadHandle::WaitForThread()

{

	return false;

}

SWThreadHandle::SWThreadHandle(IThreadCreator *parent, const ThreadParams *p, IThread *thread) : 

	m_state(Thread_Paused), m_params(*p), m_parent(parent), pThread(thread)

{

}

IThread *SWThreadHandle::GetThread()

{

	return pThread;

}