PrimeWorldEditor/src/Editor/Undo/CChangeLayerCommand.cpp

#include "CChangeLayerCommand.h"

CChangeLayerCommand::CChangeLayerCommand(CWorldEditor *pEditor, const QList<CScriptNode*>& rkNodeList, CScriptLayer *pNewLayer)
    : IUndoCommand("Change Layer")
    , mpNewLayer(pNewLayer)
    , mpEditor(pEditor)
{
    foreach (CScriptNode *pNode, rkNodeList)
    {
        CScriptLayer *pLayer = pNode->Instance()->Layer();

        if (pLayer != pNewLayer && !mNodes.contains(pNode))
        {
            mNodes << pNode;
            mOldLayers[pNode->ID()] = pLayer;
        }
    }
}

void CChangeLayerCommand::undo()
{
    mpEditor->InstancesLayerAboutToChange();
    QList<CSceneNode*> Nodes = mNodes.DereferenceList();

    QList<CScriptNode*> ScriptNodes;
    foreach (CSceneNode *pNode, Nodes) ScriptNodes << static_cast<CScriptNode*>(pNode);

    foreach (CScriptNode *pNode, ScriptNodes)
        pNode->Instance()->SetLayer(mOldLayers[pNode->ID()]);

    mpEditor->InstancesLayerChanged(ScriptNodes);
}

void CChangeLayerCommand::redo()
{
    mpEditor->InstancesLayerAboutToChange();
    QList<CSceneNode*> Nodes = mNodes.DereferenceList();

    QList<CScriptNode*> ScriptNodes;
    foreach (CSceneNode *pNode, Nodes) ScriptNodes << static_cast<CScriptNode*>(pNode);

    foreach (CScriptNode *pNode, ScriptNodes)
        pNode->Instance()->SetLayer(mpNewLayer);

    mpEditor->InstancesLayerChanged(ScriptNodes);
}