c++ qtreewidget支持撤销恢复代码

以下是一个简单的示例代码，演示如何在QTreeWidget中使用QUndoStack和QUndoCommand来实现撤销和恢复功能：

#include <QApplication>
#include <QTreeWidget>
#include <QUndoStack>
#include <QUndoCommand>

// 自定义的添加节点的命令类
class AddNodeCommand : public QUndoCommand
{
public:
    AddNodeCommand(QTreeWidget *treeWidget, QTreeWidgetItem *parent, const QStringList &strings, int index)
        : m_treeWidget(treeWidget), m_parent(parent), m_index(index)
    {
        setText(QStringLiteral("Add Node"));
        // 在此记录添加节点前的状态
        m_beforeState = treeWidget->currentItem() ? treeWidget->currentItem()->text(0) : QString();
    }

    void undo() override
    {
        // 撤销操作：删除刚才添加的节点
        QTreeWidgetItem *child = m_parent->child(m_index);
        m_parent->removeChild(child);
        m_treeWidget->setCurrentItem(m_parent);
        // 在此记录撤销后的状态
        m_afterState = m_treeWidget->currentItem() ? m_treeWidget->currentItem()->text(0) : QString();
    }

    void redo() override
    {
        // 恢复操作：重新添加之前删除的节点
        QTreeWidgetItem *child = new QTreeWidgetItem(m_parent, m_index);
        child->setText(0, m_strings[0]);
        child->setText(1, m_strings[1]);
        child->setText(2, m_strings[2]);
        m_treeWidget->setCurrentItem(child);
        // 在此记录恢复后的状态
        m_afterState = m_treeWidget->currentItem() ? m_treeWidget->currentItem()->text(0) : QString();
    }

    bool mergeWith(const QUndoCommand *other) override
    {
        // 如果两个命令的类型相同，且父节点和索引相同，则认为它们可以合并为一个操作
        const AddNodeCommand *command = static_cast<const AddNodeCommand *>(other);
        if (command && command->m_parent == m_parent && command->m_index == m_index) {
            m_strings = command->m_strings;
            return true;
        }
        return false;
    }

private:
    QTreeWidget *m_treeWidget;
    QTreeWidgetItem *m_parent;
    QStringList m_strings;
    int m_index;
    QString m_beforeState;
    QString m_afterState;
};

int main(int argc, char *argv[])
{
    QApplication app(argc, argv);

    QTreeWidget treeWidget;
    treeWidget.setColumnCount(3);
    treeWidget.setHeaderLabels(QStringList() << "Name" << "Type" << "Value");

    QUndoStack undoStack;
    QObject::connect(&treeWidget, &QTreeWidget::itemChanged, [&undoStack](QTreeWidgetItem *item, int column) {
        // 监听节点的修改操作，每次修改都会创建一个命令
        Q_UNUSED(column)
        QStringList strings;
        strings << item->text(0) << item->text(1) << item->text(2);
        QUndoCommand *command = new QUndoCommand();
        command->setText(QStringLiteral("Edit Node"));
        command->setParent(&undoStack);
        command->addCommand(new AddNodeCommand(treeWidget, item->parent(), strings, item->parent()->indexOfChild(item)));
        undoStack.push(command);
    });

    QTreeWidgetItem *root = new QTreeWidgetItem(&treeWidget);
    root->setText(0, "Root");

    QTreeWidgetItem *child1 = new QTreeWidgetItem(root);
    child1->setText(0, "Child 1");
    child1->setText(1, "Type 1");
    child1->setText(2, "Value 1");

    QTreeWidgetItem *child2 = new QTreeWidgetItem(root);
    child2->setText(0, "Child 2");
    child2->setText(1, "Type 2");
    child2->setText(2, "Value 2");

    treeWidget.expandAll();
    treeWidget.show();

    return app.exec();
}

在上述代码中，我们使用了QUndoStack和QUndoCommand来管理QTreeWidget中节点的添加、删除和修改操作。当用户执行操作时，我们会创建一个相应的QUndoCommand，并将其推入QUndoStack中。当用户想要撤销操作时，我们可以调用QUndoStack中的undo()函数，该函数会执行当前堆栈顶部的命令的撤销操作。当用户想要恢复操作时，我们可以调用redo()函数，该函数会执行当前堆栈顶部的命令的恢复操作。