CWE-ou-minkata/Sources/Plasma/PubUtilLib/plDrawable/plInstanceDrawInterface.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 "plInstanceDrawInterface.h"
#include "plSharedMesh.h"
#include "plMorphSequence.h"
#include "plMessage/plReplaceGeometryMsg.h"
#include "plDrawableSpans.h"
#include "plGeometrySpan.h"
#include "plScene/plSceneNode.h"
#include "pnMessage/plDISpansMsg.h"
#include "hsResMgr.h"

plInstanceDrawInterface::plInstanceDrawInterface() : plDrawInterface(), fTargetID(-1) {}

plInstanceDrawInterface::~plInstanceDrawInterface() {}

void plInstanceDrawInterface::Read(hsStream* stream, hsResMgr* mgr)
{
    plDrawInterface::Read(stream, mgr);

    fTargetID = stream->ReadSwap32();
    plSwapSpansRefMsg *sMsg = TRACKED_NEW plSwapSpansRefMsg(GetKey(), plRefMsg::kOnCreate, -1, -1);
    mgr->ReadKeyNotifyMe(stream, sMsg, plRefFlags::kActiveRef);
}

void plInstanceDrawInterface::Write(hsStream* stream, hsResMgr* mgr)
{
    plDrawInterface::Write(stream, mgr);
    
    stream->WriteSwap32(fTargetID);
    mgr->WriteKey(stream, fDrawable->GetKey());
}

hsBool plInstanceDrawInterface::MsgReceive(plMessage* msg)
{
#if 0 // UNUSED
    // This currently isn't being used, so I'm commenting it out at
    // Bob's preference. If it does come back into play, the plReplaceGeometryMsg should also
    // contain an LOD field, saying which avatar LOD this is targetted for.
    // Just always specifying 0 won't break anything, it just circumvents some
    // optimizations the pipeline can make, so it'll look right, just slower.
    plReplaceGeometryMsg *rMsg = plReplaceGeometryMsg::ConvertNoRef(msg);
    if (rMsg)
    {
        if (rMsg->fFlags & rMsg->kAddingGeom)
            AddSharedMesh(rMsg->fMesh, rMsg->fMaterial, rMsg->fFlags & rMsg->kAddToFront);
        else
            RemoveSharedMesh(rMsg->fMesh);
        
        return true;
    }
#endif // UNUSED

    plSwapSpansRefMsg* refMsg = plSwapSpansRefMsg::ConvertNoRef(msg);
    if (refMsg)
    {
        if( refMsg->GetContext() & (plRefMsg::kOnDestroy|plRefMsg::kOnRemove) )
            fDrawable = nil;
        else
            fDrawable = plDrawableSpans::ConvertNoRef(refMsg->GetRef());
        return true;
    }
    
    return plDrawInterface::MsgReceive(msg);
}

void plInstanceDrawInterface::AddSharedMesh(plSharedMesh *mesh, hsGMaterial *mat, hsBool addToFront, int lod, hsBool partialSort)
{
    if (fDrawable == nil)
    {
        hsAssert(false, "Missing drawable when instancing a shared mesh. Ignoring instance.");
        return;
    }

#ifdef MF_NOSHADOW_ACC
    // TESTHACKERY FOLLOWS - GlassesNoShadow
    UInt32 noShadHack = 0;
    if( mesh->GetKey() && (strstr(mesh->GetKey()->GetName(), "lasses") || strstr(mesh->GetKey()->GetName(), "oggles")) )
        noShadHack = plGeometrySpan::kPropNoShadowCast;
#endif // MF_NOSHADOW_ACC

    int i;
    for (i = 0; i < mesh->fSpans.GetCount(); i++)
    {
        mesh->fSpans[i]->fMaterial = mat;

        if( partialSort )
        {
            mesh->fSpans[i]->fProps |= plGeometrySpan::kPartialSort;
        }
        else
            mesh->fSpans[i]->fProps &= ~plGeometrySpan::kPartialSort;

#ifdef MF_NOSHADOW_ACC
        mesh->fSpans[i]->fProps |= noShadHack;
#endif // MF_NOSHADOW_ACC
    }
            
    // Add the spans to the drawable
    UInt32 index = (UInt32)-1;
    index = fDrawable->AppendDISpans(mesh->fSpans, index, false, true, addToFront, lod);
            
    // Tell the drawInterface what drawable and index it wants.
    UInt8 iDraw = (UInt8)GetNumDrawables();
    ISetDrawable(iDraw, fDrawable);
    SetDrawableMeshIndex(iDraw, index);
    SetSharedMesh(iDraw, mesh);
    if (GetProperty(kDisable))
        fDrawables[iDraw]->SetProperty(fDrawableIndices[iDraw], kDisable, true);

#ifdef HS_DEBUGGING
    plDISpansMsg* diMsg = TRACKED_NEW plDISpansMsg(fDrawable->GetKey(), plDISpansMsg::kAddingSpan, index, plDISpansMsg::kLeaveEmptyDrawable);
    diMsg->SetSender(GetKey());
    diMsg->Send();
#endif          

    plSharedMeshBCMsg *smMsg = TRACKED_NEW plSharedMeshBCMsg;
    smMsg->SetSender(GetKey());
    smMsg->fDraw = fDrawable;
    smMsg->fMesh = mesh;
    smMsg->fIsAdding = true;
    smMsg->Send();
            
    if (mesh->fMorphSet)
    {
        plMorphSequence *morph = const_cast<plMorphSequence*>(plMorphSequence::ConvertNoRef(fOwner->GetModifierByType(plMorphSequence::Index())));
        if (morph)
        {
            //hsgResMgr::ResMgr()->AddViaNotify(mesh->GetKey(), TRACKED_NEW plGenRefMsg(morph->GetKey(), plRefMsg::kOnCreate, -1, -1), plRefFlags::kPassiveRef);
            morph->AddSharedMesh(mesh);
        }
    }
}

void plInstanceDrawInterface::RemoveSharedMesh(plSharedMesh *mesh)
{
    UInt32 geoIndex = fMeshes.Find(mesh);
    if (geoIndex != fMeshes.kMissingIndex)
    {
        IClearIndex((UInt8)geoIndex);
                
        plSharedMeshBCMsg *smMsg = TRACKED_NEW plSharedMeshBCMsg;
        smMsg->SetSender(GetKey());
        smMsg->fDraw = fDrawable;
        smMsg->fMesh = mesh;
        smMsg->fIsAdding = false;
        smMsg->Send();
                
        if (mesh->fMorphSet)
        {
            plMorphSequence *morph = const_cast<plMorphSequence*>(plMorphSequence::ConvertNoRef(fOwner->GetModifierByType(plMorphSequence::Index())));
            if (morph)
            {
                //morph->GetKey()->Release(mesh->GetKey());
                morph->RemoveSharedMesh(mesh);
            }
        }
    }
}

void plInstanceDrawInterface::ICheckDrawableIndex(UInt8 which)
{
    if( which >= fMeshes.GetCount() )
    {
        fMeshes.ExpandAndZero(which+1);
    }
    
    plDrawInterface::ICheckDrawableIndex(which);
}

void plInstanceDrawInterface::ReleaseData()
{
    fMeshes.Reset();
    
    plDrawInterface::ReleaseData();
}

void plInstanceDrawInterface::SetSharedMesh(UInt8 which, plSharedMesh *mesh)
{
    ICheckDrawableIndex(which);
    fMeshes[which] = mesh;
}

void plInstanceDrawInterface::IClearIndex(UInt8 which)
{
    plDrawableSpans *drawable = plDrawableSpans::ConvertNoRef(fDrawables[which]);
    if (drawable != nil)
    {
        plDISpansMsg* diMsg = TRACKED_NEW plDISpansMsg(fDrawable->GetKey(), plDISpansMsg::kRemovingSpan, fDrawableIndices[which], plDISpansMsg::kLeaveEmptyDrawable);
        diMsg->SetSender(GetKey());
        diMsg->Send();
    }
    
    fDrawables.Remove(which);
    fDrawableIndices.Remove(which);
    fMeshes.Remove(which);
}

// Temp testing - not really ideal. Check with Bob for real soln.
Int32 plInstanceDrawInterface::GetSharedMeshIndex(const plSharedMesh *mesh) const
{
    int i;
    for( i = 0; i < fMeshes.GetCount(); i++ )
    {
        if( fMeshes[i] == mesh )
            return i;
    }
    return -1;
}