KisSnapshotModel.cpp

Go to the documentation of this file.

/*
 *  SPDX-FileCopyrightText: 2019 Tusooa Zhu <tusooa@vista.aero>
 *
 *  SPDX-License-Identifier: GPL-2.0-or-later
 */
 
#include "KisSnapshotModel.h"
 
#include <QMap>
#include <QList>
#include <QPointer>
#include <QPair>
#include <QString>
 
#include <KisDocument.h>
#include <KisView.h>
#include <KisViewManager.h>
#include <kis_node_manager.h>
#include <kis_name_server.h>
#include <kis_image_signal_router.h>
 
struct KisSnapshotModel::Private
{
    Private();
    virtual ~Private();
 
    QPointer<KisDocument> curDocument();
    QSharedPointer<KisNameServer> curNameServer;
    bool switchToDocument(QPointer<KisDocument> doc);
 
    using DocPList = QList<QPair<QString, QPointer<KisDocument> > >;
 
    DocPList curDocList;
 
    QMap<KisDocument *, DocPList> documentGroups;
    QMap<KisDocument *, QSharedPointer<KisNameServer>> nameServers;
    QPointer<KisCanvas2> curCanvas;
};
 
KisSnapshotModel::Private::Private()
{
}
 
KisSnapshotModel::Private::~Private()
{
}
 
QPointer<KisDocument> KisSnapshotModel::Private::curDocument()
{
    if (curCanvas && curCanvas->imageView()) {
        return curCanvas->imageView()->document();
    }
    return 0;
}
 
bool KisSnapshotModel::Private::switchToDocument(QPointer<KisDocument> doc)
{
    if (curCanvas && curCanvas->imageView()) {
        KisDocument *curDoc = curDocument();
        if (curDoc && doc) {
            curDoc->copyFromDocument(*doc);
            curDoc->image()->signalRouter()->emitNotification(ComplexNodeReselectionSignal(curDoc->preActivatedNode(), {}));
        }
        // FIXME: more things need to be done
        return true;
    }
    return false;
}
 
KisSnapshotModel::KisSnapshotModel()
    : QAbstractListModel()
    , m_d(new Private)
{
}
 
KisSnapshotModel::~KisSnapshotModel()
{
}
 
int KisSnapshotModel::rowCount(const QModelIndex &parent) const
{
    if (parent.isValid()) {
        return 0;
    } else {
        return m_d->curDocList.size();
    }
}
 
QVariant KisSnapshotModel::data(const QModelIndex &index, int role) const
{
    if (!index.isValid() || index.row() >= rowCount(QModelIndex())) {
        return QVariant();
    }
    int i = index.row();
    switch (role) {
    case Qt::DisplayRole:
    case Qt::EditRole:
        return m_d->curDocList[i].first;
        break;
    }
    return QVariant();
}
 
bool KisSnapshotModel::setData(const QModelIndex &index, const QVariant &value, int role)
{
    if (!index.isValid() || index.row() >= rowCount(QModelIndex())) {
        return false;
    }
    int i = index.row();
    switch (role) {
    case Qt::DisplayRole:
    case Qt::EditRole:
        m_d->curDocList[i].first = value.toString();
        Q_EMIT dataChanged(index, index);
        return true;
        break;
    }
    return false;
}
 
Qt::ItemFlags KisSnapshotModel::flags(const QModelIndex &index) const
{
    if (!index.isValid()) {
        return Qt::ItemIsEnabled;
    }
 
    return QAbstractListModel::flags(index) | Qt::ItemIsEditable;
}
 
void KisSnapshotModel::setCanvas(QPointer<KisCanvas2> canvas)
{
    if (m_d->curCanvas == canvas) {
        return;
    }
 
    m_d->curNameServer.reset();
 
    if (m_d->curDocument()) {
        m_d->documentGroups.insert(m_d->curDocument(), m_d->curDocList);
    } else {
        Q_FOREACH (auto const &i, m_d->curDocList) {
            delete i.second.data();
        }
    }
 
    if (!m_d->curDocList.isEmpty()) {
        beginRemoveRows(QModelIndex(), 0, m_d->curDocList.size() - 1);
        m_d->curDocList.clear();
        endRemoveRows();
    }
    m_d->curCanvas = canvas;
 
    QPointer<KisDocument> curDoc = m_d->curDocument();
    if (curDoc) {
        Private::DocPList docList = m_d->documentGroups.take(curDoc);
        beginInsertRows(QModelIndex(), docList.size(), docList.size());
        m_d->curDocList = docList;
        endInsertRows();
 
        QSharedPointer<KisNameServer> nameServer = m_d->nameServers[curDoc];
        if (!nameServer) {
            nameServer.reset(new KisNameServer);
            m_d->nameServers.insert(curDoc, nameServer);
        }
        m_d->curNameServer = nameServer;
    }
 
}
 
bool KisSnapshotModel::slotCreateSnapshot()
{
    if (!m_d->curDocument()) {
        return false;
    }
    QPointer<KisDocument> clonedDoc(m_d->curDocument()->lockAndCreateSnapshot());
    if (clonedDoc) {
        beginInsertRows(QModelIndex(), m_d->curDocList.size(), m_d->curDocList.size());
        m_d->curDocList << qMakePair(i18nc("snapshot names, e.g. \"Snapshot 1\"", "Snapshot %1", m_d->curNameServer->number()), clonedDoc);
        endInsertRows();
        return true;
    }
    return false;
}
 
bool KisSnapshotModel::slotRemoveSnapshot(const QModelIndex &index)
{
    if (!index.isValid() || index.row() >= m_d->curDocList.size()) {
        return false;
    }
    int i = index.row();
    beginRemoveRows(QModelIndex(), i, i);
    QPair<QString, QPointer<KisDocument> > pair = m_d->curDocList.takeAt(i);
    endRemoveRows();
    delete pair.second.data();
    return true;
}
 
bool KisSnapshotModel::slotSwitchToSnapshot(const QModelIndex &index)
{
    if (!index.isValid() || index.row() >= m_d->curDocList.size()) {
        return false;
    }
 
    return m_d->switchToDocument(m_d->curDocList[index.row()].second);
}