|  | Home · Overviews · Examples

# Undo Framework Example

This example shows how to implement undo/redo functionality with the Qt undo framework.

In the Qt undo framework, all actions that the user performs are implemented in classes that inherit QUndoCommand. An undo command class knows how to both redo() - or just do the first time - and undo() an action. For each action the user performs, a command is placed on a QUndoStack. Since the stack contains all commands executed (stacked in chronological order) on the document, it can roll the state of the document backwards and forwards by undoing and redoing its commands. See the overview document for a high-level introduction to the undo framework.

The undo example implements a simple diagram application. It is possible to add and delete items, which are either box or rectangular shaped, and move the items by dragging them with the mouse. The undo stack is shown in a QUndoView, which is a list in which the commands are shown as list items. Undo and redo are available through the edit menu. The user can also select a command from the undo view.

We use the graphics view framework to implement the diagram. We only treat the related code briefly as the framework has examples of its own (e.g., the Diagram Scene Example).

The example consists of the following classes:

- `MainWindow` is the main window and arranges the example's widgets. It creates the commands based on user input and keeps them on the command stack.
- `AddCommand` adds an item to the scene.
- `DeleteCommand` deletes an item from the scene.
- `MoveCommand` when an item is moved the MoveCommand keeps record of the start and stop positions of the move, and it moves the item according to these when `redo()` and `undo()` is called.
- `DiagramScene` inherits QGraphicsScene and emits signals for the `MoveComands` when an item is moved.
- `DiagramItem` inherits QGraphicsPolygonItem and represents an item in the diagram.

## MainWindow Class Definition

    class MainWindow : public QMainWindow
    {
        Q_OBJECT

    public:
        MainWindow();

    public slots:
        void itemMoved(DiagramItem *movedDiagram, const QPointF &moveStartPosition);

    private slots:
        void deleteItem();
        void addBox();
        void addTriangle();
        void about();
        void itemMenuAboutToShow();
        void itemMenuAboutToHide();

    private:
        void createActions();
        void createMenus();
        void createUndoView();

        QAction *deleteAction;
        QAction *addBoxAction;
        QAction *addTriangleAction;
        QAction *undoAction;
        QAction *redoAction;
        QAction *exitAction;
        QAction *aboutAction;

        QMenu *fileMenu;
        QMenu *editMenu;
        QMenu *itemMenu;
        QMenu *helpMenu;

        DiagramScene *diagramScene;
        QUndoStack *undoStack;
        QUndoView *undoView;
    };

The `MainWindow` class maintains the undo stack, i.e., it creates QUndoCommands and pushes and pops them from the stack when it receives the `triggered()` signal from `undoAction` and `redoAction`.

## MainWindow Class Implementation

We will start with a look at the constructor:
The following code example is written in c++.

    MainWindow::MainWindow()
    {
        undoStack = new QUndoStack();

        createActions();
        createMenus();

        connect(undoStack, SIGNAL(canRedoChanged(bool)),
                redoAction, SLOT(setEnabled(bool)));
        connect(undoStack, SIGNAL(canUndoChanged(bool)),
                undoAction, SLOT(setEnabled(bool)));
        createUndoView();

        diagramScene = new DiagramScene();
        QBrush pixmapBrush(QPixmap(":/images/cross.png").scaled(30, 30));
        diagramScene->setBackgroundBrush(pixmapBrush);
        diagramScene->setSceneRect(QRect(0, 0, 500, 500));

        connect(diagramScene, SIGNAL(itemMoved(DiagramItem *, const QPointF &)),
                this, SLOT(itemMoved(DiagramItem *, const QPointF &)));

        setWindowTitle("Undo Framework");
        QGraphicsView *view = new QGraphicsView(diagramScene);
        setCentralWidget(view);
        resize(700, 500);
    }

By connecting the undo stack's `canRedoChanged()` and canUndoChanged() signals to our undo and redo action's setEnabled() slot we make the actions disabled when the stack cannot undo and redo commands. The rest of the constructor sets up the DiagramScene and QGraphicsView.

Here is the `createUndoView()` function:
The following code example is written in c++.

    void MainWindow::createUndoView()
    {
        undoView = new QUndoView(undoStack);
        undoView->setWindowTitle(tr("Command List"));
        undoView->show();
        undoView->setAttribute(Qt::WA_QuitOnClose, false);
    }

The QUndoView is a widget that display the text, which is set with the setText() function, for each QUndoCommand in the undo stack in a list.

Here is the `createActions()` function:
The following code example is written in c++.

    void MainWindow::createActions()
    {
        deleteAction = new QAction(tr("&Delete Item"), this);
        deleteAction->setShortcut(tr("Del"));
        connect(deleteAction, SIGNAL(triggered()), this, SLOT(deleteItem()));

    ...
        undoAction = new QAction(tr("&Undo"), this);
        undoAction->setShortcut(tr("Ctrl+Z"));
        undoAction->setEnabled(false);
        connect(undoAction, SIGNAL(triggered()), undoStack, SLOT(undo()));

        redoAction = new QAction(tr("&Redo"), this);
        QList<QKeySequence> redoShortcuts;
        redoShortcuts << tr("Ctrl+Y") << tr("Shift+Ctrl+Z");
        redoAction->setShortcuts(redoShortcuts);
        redoAction->setEnabled(false);
        connect(redoAction, SIGNAL(triggered()), undoStack, SLOT(redo()));

The `createActions()` function sets up all the examples actions in the manner shown above. We can connect our `undoAction` and `redoAction` directly to the stack's `undo()` and `redo()` slots as we disable the actions when the stack cannot do undo and redo. For the other actions we have implemented slots in the `MainWindow` class.

Here is the `createMenues()` function:
The following code example is written in c++.

    void MainWindow::createMenus()
    {
    ...
        editMenu = menuBar()->addMenu(tr("&Edit"));
        editMenu->addAction(undoAction);
        editMenu->addAction(redoAction);
        editMenu->addSeparator();
        editMenu->addAction(deleteAction);
        connect(editMenu, SIGNAL(aboutToShow()),
                this, SLOT(itemMenuAboutToShow()));
        connect(editMenu, SIGNAL(aboutToHide()),
                this, SLOT(itemMenuAboutToHide()));

    ...
    }

We have to use the QMenu `aboutToShow()` and `aboutToHide()` signals since we only want `deleteAction` to be enabled when we have selected an item. We also want the text() to be shown in the `undoAction` and `redoAction` menu items.

Here is the `itemMoved()` slot:
The following code example is written in c++.

    void MainWindow::itemMoved(DiagramItem *movedItem,
                               const QPointF &oldPosition)
    {
        undoStack->push(new MoveCommand(movedItem, oldPosition));
    }

We simply push a MoveCommand on the stack, which calls `redo()` on it.

Here is the `deleteItem()` slot:
The following code example is written in c++.

    void MainWindow::deleteItem()
    {
        if (diagramScene->selectedItems().isEmpty())
            return;

        QUndoCommand *deleteCommand = new DeleteCommand(diagramScene);
        undoStack->push(deleteCommand);
    }

An item must be selected to be deleted. We need to check if it is selected as the `deleteAction` may be enabled even if an item is not selected. This can happen as we do not catch a signal or event when an item is selected.

Here is the `itemMenuAboutToShow()` and itemMenuAboutToHide() slots:
The following code example is written in c++.

    void MainWindow::itemMenuAboutToHide()
    {
        deleteAction->setEnabled(true);
    }

    void MainWindow::itemMenuAboutToShow()
    {
        undoAction->setText(tr("Undo ") + undoStack->undoText());
        redoAction->setText(tr("Redo ") + undoStack->redoText());
        deleteAction->setEnabled(!diagramScene->selectedItems().isEmpty());
    }

We implement `itemMenuAboutToShow()` and `itemMenuAboutToHide()` to get a dynamic item menu. These slots are connected to the aboutToShow() and aboutToHide() signals. We need this to disable or enable the `deleteAction` and fill the `redoAction` and `undoAction` menu item with the text from the next QUndoCommand that will be redone or undone.

Here is the `addBox()` slot:
The following code example is written in c++.

    void MainWindow::addBox()
    {
        QUndoCommand *addCommand = new AddCommand(DiagramItem::Box, diagramScene);
        undoStack->push(addCommand);
    }

The `addBox()` function creates an AddCommand and pushes it on the undo stack.

Here is the `addTriangle()` sot:
The following code example is written in c++.

    void MainWindow::addTriangle()
    {
        QUndoCommand *addCommand = new AddCommand(DiagramItem::Triangle,
                                                  diagramScene);
        undoStack->push(addCommand);
    }

The `addTriangle()` function creates an AddCommand and pushes it on the undo stack.

Here is the implementation of `about()`:
The following code example is written in c++.

    void MainWindow::about()
    {
        QMessageBox::about(this, tr("About Undo"),
                           tr("The <b>Undo</b> example demonstrates how to "
                              "use Qt's undo framework."));
    }

The about slot is triggered by the `aboutAction` and displays an about box for the example.

## AddCommand Class Definition

    class AddCommand : public QUndoCommand
    {
    public:
        AddCommand(DiagramItem::DiagramType addType, QGraphicsScene *graphicsScene,
                   QUndoCommand *parent = 0);

        void undo();
        void redo();

    private:
        DiagramItem *myDiagramItem;
        QGraphicsScene *myGraphicsScene;
        QPointF initialPosition;
    };

The `AddCommand` class adds DiagramItem graphics items to the DiagramScene.

## AddCommand Class Implementation

We start with the constructor:
The following code example is written in c++.

    AddCommand::AddCommand(DiagramItem::DiagramType addType,
                           QGraphicsScene *scene, QUndoCommand *parent)
        : QUndoCommand(parent)
    {
        static int itemCount = 0;

        myGraphicsScene = scene;
        myDiagramItem = new DiagramItem(addType);
        initialPosition = QPointF((itemCount * 15) % int(scene->width()),
                                  (itemCount * 15) % int(scene->height()));
        scene->update();
        ++itemCount;
        setText(QObject::tr("Add %1")
            .arg(createCommandString(myDiagramItem, initialPosition)));
    }

We first create the DiagramItem to add to the DiagramScene. The setText() function let us set a QString that describes the command. We use this to get custom messages in the QUndoView and in the menu of the main window.
The following code example is written in c++.

    void AddCommand::undo()
    {
        myGraphicsScene->removeItem(myDiagramItem);
        myGraphicsScene->update();
    }

`undo()` removes the item from the scene. We need to update the scene as ...(ask Andreas)
The following code example is written in c++.

    void AddCommand::redo()
    {
        myGraphicsScene->addItem(myDiagramItem);
        myDiagramItem->setPos(initialPosition);
        myGraphicsScene->clearSelection();
        myGraphicsScene->update();
    }

We set the position of the item as we do not do this in the constructor.

## DeleteCommand Class Definition

    class DeleteCommand : public QUndoCommand
    {
    public:
        DeleteCommand(QGraphicsScene *graphicsScene, QUndoCommand *parent = 0);

        void undo();
        void redo();

    private:
        DiagramItem *myDiagramItem;
        QGraphicsScene *myGraphicsScene;
    };

The DeleteCommand class implements the functionality to remove an item from the scene.

## DeleteCommand Class Implementation

The following code example is written in c++.

    DeleteCommand::DeleteCommand(QGraphicsScene *scene, QUndoCommand *parent)
        : QUndoCommand(parent)
    {
        myGraphicsScene = scene;
        QList<QGraphicsItem *> list = myGraphicsScene->selectedItems();
        list.first()->setSelected(false);
        myDiagramItem = static_cast<DiagramItem *>(list.first());
        setText(QObject::tr("Delete %1")
            .arg(createCommandString(myDiagramItem, myDiagramItem->pos())));
    }

We know that there must be one selected item as it is not possible to create a DeleteCommand unless the item to be deleted is selected and that only one item can be selected at any time. The item must be unselected if it is inserted back into the scene.
The following code example is written in c++.

    void DeleteCommand::undo()
    {
        myGraphicsScene->addItem(myDiagramItem);
        myGraphicsScene->update();
    }

The item is simply reinserted into the scene.
The following code example is written in c++.

    void DeleteCommand::redo()
    {
        myGraphicsScene->removeItem(myDiagramItem);
    }

The item is removed from the scene.

## MoveCommand Class Definition

    class MoveCommand : public QUndoCommand
    {
    public:
        enum { Id = 1234 };

        MoveCommand(DiagramItem *diagramItem, const QPointF &oldPos,
                    QUndoCommand *parent = 0);

        void undo();
        void redo();
        bool mergeWith(const QUndoCommand *command);
        int id() const { return Id; }

    private:
        DiagramItem *myDiagramItem;
        QPointF myOldPos;
        QPointF newPos;
    };

The mergeWith() is reimplemented to make consecutive moves of an item one MoveCommand, i.e, the item will be moved back to the start position of the first move.

## MoveCommand Class Implementation

The constructor of MoveCommand looks like this:
The following code example is written in c++.

    MoveCommand::MoveCommand(DiagramItem *diagramItem, const QPointF &oldPos,
                     QUndoCommand *parent)
        : QUndoCommand(parent)
    {
        myDiagramItem = diagramItem;
        newPos = diagramItem->pos();
        myOldPos = oldPos;
    }

We save both the old and new positions for undo and redo respectively.
The following code example is written in c++.

    void MoveCommand::undo()
    {
        myDiagramItem->setPos(myOldPos);
        myDiagramItem->scene()->update();
        setText(QObject::tr("Move %1")
            .arg(createCommandString(myDiagramItem, newPos)));
    }

We simply set the items old position and update the scene.
The following code example is written in c++.

    void MoveCommand::redo()
    {
        myDiagramItem->setPos(newPos);
        setText(QObject::tr("Move %1")
            .arg(createCommandString(myDiagramItem, newPos)));
    }

We set the item to its new position.
The following code example is written in c++.

    bool MoveCommand::mergeWith(const QUndoCommand *command)
    {
        const MoveCommand *moveCommand = static_cast<const MoveCommand *>(command);
        DiagramItem *item = moveCommand->myDiagramItem;

        if (myDiagramItem != item)
        return false;

        newPos = item->pos();
        setText(QObject::tr("Move %1")
            .arg(createCommandString(myDiagramItem, newPos)));

        return true;
    }

Whenever a MoveCommand is created, this function is called to check if it should be merged with the previous command. It is the previous command object that is kept on the stack. The function returns true if the command is merged; otherwise false.

We first check whether it is the same item that has been moved twice, in which case we merge the commands. We update the position of the item so that it will take the last position in the move sequence when undone.

## DiagramScene Class Definition

    class DiagramScene : public QGraphicsScene
    {
        Q_OBJECT

    public:
        DiagramScene(QObject *parent = 0);

    signals:
        void itemMoved(DiagramItem *movedItem, const QPointF &movedFromPosition);

    protected:
        void mousePressEvent(QGraphicsSceneMouseEvent *event);
        void mouseReleaseEvent(QGraphicsSceneMouseEvent *event);

    private:
        QGraphicsItem *movingItem;
        QPointF oldPos;
    };

The DiagramScene implements the functionality to move a DiagramItem with the mouse. It emits a signal when a move is completed. This is caught by the `MainWindow`, which makes MoveCommands. We do not examine the implementation of DiagramScene as it only deals with graphics framework issues.

## The main() Function

`main()` The `main()` function of the program looks like this:
The following code example is written in c++.

    int main(int argv, char *args[])
    {
        Q_INIT_RESOURCE(undoframework);

        QApplication app(argv, args);

        MainWindow mainWindow;
        mainWindow.show();

        return app.exec();
    }

We draw a grid in the background of the DiagramScene, so we use a resource file. The rest of the function creates the `MainWindow` and shows it as a top level window.