PrimeWorldEditor/src/Editor/Undo/CSelectNodeCommand.cpp

#include "CSelectNodeCommand.h"
#include "Editor/INodeEditor.h"

CSelectNodeCommand::CSelectNodeCommand(INodeEditor *pEditor, CSceneNode *pNode, QList<CSceneNode*>& selection)
    : IUndoCommand("Select"),
      mpEditor(pEditor),
      mpNode(pNode),
      mpSelection(&selection)
{
}

void CSelectNodeCommand::undo()
{
    if (mpNode->IsSelected())
    {
        mpNode->SetSelected(false);

        for (auto it = mpSelection->begin(); it != mpSelection->end(); it++)
        {
            if (*it == mpNode)
            {
                mpSelection->erase(it);
                break;
            }
        }
    }

    mpEditor->RecalculateSelectionBounds();
    mpEditor->NotifySelectionModified();
}

void CSelectNodeCommand::redo()
{
    if (!mpNode->IsSelected())
    {
        mpNode->SetSelected(true);
        mpSelection->push_back(mpNode);
    }

    mpEditor->ExpandSelectionBounds(mpNode);
    mpEditor->NotifySelectionModified();
}