CWE-ou-minkata/Sources/Tools/MaxSceneViewer/plPluginClient.cpp

/*==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 "plPluginClient.h"

#include "plPipeline/hsG3DDeviceSelector.h"

#include "hsThread.h"
#include "hsSTLStream.h"

#include "pnKeyedObject/plUoid.h"
#include "plResMgr/plUpdatableResManager.h"
#include "plStatusLog/plStatusLog.h"

// Needed for IEnableProxies
#include "plPipeline.h"
#include "plDrawable/plDrawableSpans.h"
#include "pnMessage/plProxyDrawMsg.h"
#include "plgDispatch.h"

#define LOG_SCENEVIWER

static plUpdatableResManager *GetResMgr()
{
    return (plUpdatableResManager*)hsgResMgr::ResMgr();
}

plUpdatableClient::plUpdatableClient() : fPipeName(nil), fUpdateSignal(nil), fDataPath(nil), fActive(true), fDirty(false)
{
#ifdef LOG_SCENEVIWER
    fLog = plStatusLogMgr::GetInstance().CreateStatusLog(25, "SceneViewer",
        plStatusLog::kDontWriteFile | plStatusLog::kFilledBackground | plStatusLog::kAlignToTop);
#endif // LOG_SCENEVIWER
}

plUpdatableClient::~plUpdatableClient()
{
    if (fUpdateSignal)
    {
        delete fUpdateSignal;
        fUpdateSignal = nil;
    }

    if (fLog)
    {
        delete fLog;
        fLog = nil;
    }
}

void plUpdatableClient::InitUpdate(const char *semaphoreName, const char *pipeName, const char *dir)
{
    fUpdateSignal = TRACKED_NEW hsSemaphore(0, semaphoreName);
    fPipeName = pipeName;
    fDataPath = dir;
}

hsG3DDeviceModeRecord plUpdatableClient::ILoadDevMode(const char* devModeFile)
{
    // Get the usual devmode
    hsG3DDeviceModeRecord dmr = plClient::ILoadDevMode(devModeFile);

    // Override the mode with a windowed one
    hsG3DDeviceMode *mode = (hsG3DDeviceMode*)dmr.GetMode();
    mode->SetColorDepth(0);

    return hsG3DDeviceModeRecord(*dmr.GetDevice(), *mode);
}

#include "../../../Plasma/Apps/plClient/plClientUpdateFormat.h"
#include "../pnKeyedObject/plKey.h"

void plUpdatableClient::IGetUpdate()
{
    // If the semaphore is signaled an update is ready
    if (fUpdateSignal && fUpdateSignal->Wait(0))
    {
        hsNamedPipeStream s;
        s.Open(fPipeName, "r");

#ifdef LOG_SCENEVIWER
        fLog->Clear();
        static int numUpdates = 0;
        numUpdates++;
        fLog->AddLineF(plStatusLog::kBlue, "SceneViewer Update #%d", numUpdates);
#endif // LOG_SCENEVIWER

        UInt8 type = s.ReadByte();

        if (type == ClientUpdate::kShutdown)
        {
            #ifdef LOG_SCENEVIWER
            fLog->AddLine("Client shutdown");
            #endif // LOG_SCENEVIWER

            PostMessage(GetWindowHandle(), WM_SYSCOMMAND, SC_CLOSE, 0);
        }
        else if (type == ClientUpdate::kUpdate)
        {
            fDirty = true;
            
            IEnableProxies(false);

            int i;

            //
            // Delete the deleted keys
            //
            int numDeleted = s.ReadSwap32();
            std::vector<plKey*> delKeys;
            delKeys.reserve(numDeleted);

            for (i = 0; i < numDeleted; i++)
            {
                plUoid uoid;
                uoid.Read(&s);
                plKey *key = hsgResMgr::ResMgr()->FindKey(uoid);
                hsAssert(key, "Key to delete not found");
                if (key)
                {
                    #ifdef LOG_SCENEVIWER
                    fLog->AddLineF("Remove: %s", key->GetName());
                    #endif // LOG_SCENEVIWER

                    GetResMgr()->RemoveObject(key, false);
                    delKeys.push_back(key);
                }
            }

            GetResMgr()->DelayLoad(true);

            //
            // Read in the changed spans
            //
            hsStatusMessage("ReadChangedSpans\n");
            GetResMgr()->ReadChangedSpans(&s);

            //
            // Read in the new keys and objects
            //
            int numNew = s.ReadSwap32();
            for (i = 0; i < numNew; i++)
            {
                plCreatable *cre = GetResMgr()->ReadCreatable(&s);

                hsKeyedObject *ko = hsKeyedObject::ConvertNoRef(cre);

                #ifdef LOG_SCENEVIWER
                if (ko)
                    fLog->AddLineF("Read: %s", ko->GetKey()->GetName());
                else
                    fLog->AddLine("Read: (null)");
                #endif // LOG_SCENEVIWER
            }

            GetResMgr()->DelayLoad(false);

            // Clear out any objects that were never reloaded (really deleted)
            for (i = 0; i < delKeys.size(); i++)
            {
                plKey *key = delKeys[i];
                if (!key->ObjectIsLoaded())
                {
                    #ifdef LOG_SCENEVIWER
                    fLog->AddLineF("Key deleted: %s", key->GetName());
                    #endif // LOG_SCENEVIWER

                    GetResMgr()->RemoveObject(key);
                }
            }

            IEnableProxies(true);
        }

        s.Close();
    }
}

hsBool plUpdatableClient::Init()
{
    if (plClient::Init())
    {
        GetResMgr()->ForceLoadDirectory(fDataPath, true);
        // Page in the SceneViewer now that our key is ready
        GetResMgr()->PageInSceneViewer();
        return true;
    }

    return false;
}

hsBool plUpdatableClient::MainLoop()
{
    IGetUpdate();

    if (fActive)
        return plClient::MainLoop();
    else
    {
        Sleep(100);
        return true;
    }
}

#include <direct.h>

hsBool plUpdatableClient::Shutdown()
{
    if (fDirty && fDataPath)
    {
        char oldCwd[MAX_PATH];
        getcwd(oldCwd, sizeof(oldCwd));

        // Even bigger hack
        char tempCrap[MAX_PATH];
        strcpy(tempCrap, fDataPath);
        tempCrap[strlen(tempCrap)-strlen("dat\\")] = '\0';
        chdir(tempCrap);

        GetResMgr()->WriteSceneViewer();

        chdir(oldCwd);
    }

    return plClient::Shutdown();
}

void plUpdatableClient::IEnableProxies(bool enable)
{
    if (enable)
    {
        // switch back on any drawable proxies
        if (fPipeline->GetDrawableTypeMask() & plDrawableSpans::kAudibleProxy)
        {   
            plProxyDrawMsg* msg = TRACKED_NEW plProxyDrawMsg(plProxyDrawMsg::kAudible | plProxyDrawMsg::kCreate);
            plgDispatch::MsgSend(msg);
        }
        if (fPipeline->GetDrawableTypeMask() & plDrawableSpans::kOccluderProxy)
        {   
            plProxyDrawMsg* msg = TRACKED_NEW plProxyDrawMsg(plProxyDrawMsg::kOccluder | plProxyDrawMsg::kCreate);
            plgDispatch::MsgSend(msg);
        }
        if (fPipeline->GetDrawableTypeMask() & plDrawableSpans::kPhysicalProxy)
        {   
            plProxyDrawMsg* msg = TRACKED_NEW plProxyDrawMsg(plProxyDrawMsg::kPhysical | plProxyDrawMsg::kCreate);
            plgDispatch::MsgSend(msg);
        }
        if (fPipeline->GetDrawableTypeMask() & plDrawableSpans::kLightProxy)
        {   
            plProxyDrawMsg* msg = TRACKED_NEW plProxyDrawMsg(plProxyDrawMsg::kLight | plProxyDrawMsg::kCreate);
            plgDispatch::MsgSend(msg);
        }
    }
    else
    {
        // notify any and all drawable proxies to stop drawing...
        plProxyDrawMsg* nuke = TRACKED_NEW plProxyDrawMsg(plProxyDrawMsg::kAudible
                                                | plProxyDrawMsg::kOccluder
                                                | plProxyDrawMsg::kPhysical
                                                | plProxyDrawMsg::kLight
                                                | plProxyDrawMsg::kDestroy);
        plgDispatch::MsgSend(nuke);
    }
}

void plUpdatableClient::WindowActivate(bool active)
{
    fActive = active;
}