/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

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

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/

#include "hsTypes.h"
#include "plTimerCallbackManager.h"
#include "../pnMessage/plTimeMsg.h"
#include "plgDispatch.h"
#include "../pnKeyedObject/plFixedKey.h"
#include "hsTimer.h"

plTimerCallbackManager::plTimerCallbackManager() 
{
}

plTimerCallbackManager::~plTimerCallbackManager()
{
	while (fCallbacks.GetCount() > 0)
		delete fCallbacks.Pop();
}

hsBool plTimerCallbackManager::MsgReceive(plMessage* msg)
{
	plTimeMsg* pTimeMsg = plTimeMsg::ConvertNoRef(msg);
	int i = fCallbacks.Count();
	if (pTimeMsg )
	{
		if(i)
		{
			i--;
			if (pTimeMsg->GetTimeStamp() >= fCallbacks[i]->fTime)
			{
				plgDispatch::MsgSend( fCallbacks[i]->fMsg );

				// Set it nil so the TimerCallback destructor doesn't unRef it
				fCallbacks[i]->fMsg = nil; 

				delete(fCallbacks[i]);
				fCallbacks.SetCount(i);
			}
		}
		return true;
	}
	return hsKeyedObject::MsgReceive(msg);
}

plTimerCallback* plTimerCallbackManager::NewTimer(hsScalar time, plMessage* pMsg)
{
	plTimerCallback* t = TRACKED_NEW plTimerCallback( hsTimer::GetSysSeconds() + time, pMsg );
	fCallbacks.Append(t); 
	// sort them
	for (int i = 0; i < fCallbacks.Count(); i++)
	{
		for (int j = i + 1; j < fCallbacks.Count(); j++)
		{
#if 0
			hsScalar a = fCallbacks[i]->fTime;
			hsScalar b = fCallbacks[j]->fTime;
#endif
			if (fCallbacks[i]->fTime < fCallbacks[j]->fTime)
			{
				plTimerCallback* pTemp = fCallbacks[i];
				fCallbacks[i] = fCallbacks[j];
				fCallbacks[j] = pTemp;
			}
		}
	}
	return t;
}

hsBool plTimerCallbackManager::CancelCallback(plTimerCallback* pTimer)
{
	for (int i = 0; i < fCallbacks.Count(); i++)
	{
		if (fCallbacks[i] == pTimer)
		{	fCallbacks.Remove(i);
			return true;
		}
	}
	return false;
}

hsBool plTimerCallbackManager::CancelCallbacksToKey(const plKey& key)
{
	const plKey rKey;
	bool removed = false;

	for (int i = fCallbacks.Count() - 1; i >= 0 ; i--)
	{
		for (int j = 0; j < fCallbacks[i]->fMsg->GetNumReceivers(); j++)
		{
			const plKey rKey = fCallbacks[i]->fMsg->GetReceiver(j);
			
			if (rKey == key)
			{
				delete fCallbacks[i];
				fCallbacks.Remove(i);
				removed = true;
				break;
			}
		}
	}

	return removed;
}

void plTimerCallbackManager::Read(hsStream* stream, hsResMgr* mgr)
{
}
void plTimerCallbackManager::Write(hsStream* stream, hsResMgr* mgr)
{
}



plTimerCallback::plTimerCallback(double time, plMessage* pMsg) :
fTime(time),
fMsg(pMsg)
{
}

plTimerCallback::~plTimerCallback()
{
	if (fMsg)
		hsRefCnt_SafeUnRef(fMsg);
	fMsg = nil;
}

void plTimerCallback::Read(hsStream* stream, hsResMgr* mgr)
{
}
void plTimerCallback::Write(hsStream* stream, hsResMgr* mgr)
{
}


plTimerCallbackManager* plgTimerCallbackMgr::fMgr = nil;

void plgTimerCallbackMgr::Init()
{
	fMgr = TRACKED_NEW plTimerCallbackManager;
	fMgr->RegisterAs( kTimerCallbackManager_KEY );		// fixedKey from plFixedKey.h
	plgDispatch::Dispatch()->RegisterForExactType( plTimeMsg::Index(), fMgr->GetKey() );
}

hsBool plgTimerCallbackMgr::CancelCallback(plTimerCallback* pTimer)
{
	return (fMgr->CancelCallback(pTimer));
}

hsBool plgTimerCallbackMgr::CancelCallbacksToKey(const plKey& key)
{
	return (fMgr->CancelCallbacksToKey(key));
}

void plgTimerCallbackMgr::Shutdown()
{
	fMgr->UnRegisterAs(kTimerCallbackManager_KEY);
}