/*==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);
}