dust3d/src/skeletoneditgraphicsview.cpp

#include <QGraphicsPixmapItem>
#include <QXmlStreamWriter>
#include <QFile>
#include <cmath>
#include <map>
#include <vector>
#include "skeletoneditgraphicsview.h"
#include "skeletoneditnodeitem.h"
#include "skeletoneditedgeitem.h"

qreal SkeletonEditGraphicsView::m_initialNodeSize = 128;
qreal SkeletonEditGraphicsView::m_minimalNodeSize = 8;

SkeletonEditGraphicsView::SkeletonEditGraphicsView(QWidget *parent) :
    QGraphicsView(parent),
    m_pendingNodeItem(NULL),
    m_pendingEdgeItem(NULL),
    m_inAddNodeMode(true),
    m_nextStartNodeItem(NULL),
    m_lastHoverNodeItem(NULL),
    m_lastMousePos(0, 0),
    m_isMovingNodeItem(false),
    m_backgroundLoaded(false)
{
    setScene(new QGraphicsScene());

    m_backgroundItem = new QGraphicsPixmapItem();
    m_backgroundItem->setOpacity(0.25);
    scene()->addItem(m_backgroundItem);
    
    m_pendingNodeItem = new QGraphicsEllipseItem(0, 0, m_initialNodeSize, m_initialNodeSize);
    m_pendingNodeItem->setVisible(false);
    scene()->addItem(m_pendingNodeItem);
    
    m_pendingEdgeItem = new QGraphicsLineItem(0, 0, 0, 0);
    m_pendingEdgeItem->setVisible(false);
    scene()->addItem(m_pendingEdgeItem);
}

void SkeletonEditGraphicsView::toggleAddNodeMode()
{
    if (!m_backgroundLoaded)
        return;
    m_inAddNodeMode = !m_inAddNodeMode;
    applyAddNodeMode();
}

void SkeletonEditGraphicsView::applyAddNodeMode()
{
    m_pendingNodeItem->setVisible(m_inAddNodeMode);
    m_pendingEdgeItem->setVisible(m_inAddNodeMode);
    setMouseTracking(true);
}

void SkeletonEditGraphicsView::turnOffAddNodeMode()
{
    if (!m_backgroundLoaded)
        return;
    m_inAddNodeMode = false;
    applyAddNodeMode();
}

void SkeletonEditGraphicsView::turnOnAddNodeMode()
{
    if (!m_backgroundLoaded)
        return;
    m_inAddNodeMode = true;
    applyAddNodeMode();
}

SkeletonEditNodeItem *SkeletonEditGraphicsView::findNodeItemByPos(QPointF pos)
{
    QList<QGraphicsItem *>::iterator it;
    QList<QGraphicsItem *> list = scene()->items();
    for (it = list.begin(); it != list.end(); ++it) {
        if ((*it)->data(0).toString() == "node") {
            SkeletonEditNodeItem *nodeItem = static_cast<SkeletonEditNodeItem *>(*it);
            if (nodeItem->rect().contains(pos)) {
                return nodeItem;
            }
        }
    }
    return NULL;
}

void SkeletonEditGraphicsView::mousePressEvent(QMouseEvent *event)
{
    QWidget::mousePressEvent(event);
    if (!m_backgroundLoaded)
        return;
    if (event->button() == Qt::LeftButton) {
        if (!m_inAddNodeMode) {
            if (m_lastHoverNodeItem) {
                setNextStartNodeItem(m_lastHoverNodeItem->master());
                m_lastHoverNodeItem = NULL;
            }
        }
    }
    m_lastMousePos = mapToScene(event->pos());
}

void SkeletonEditGraphicsView::mouseDoubleClickEvent(QMouseEvent *event)
{
    QWidget::mouseDoubleClickEvent(event);
    emit changeTurnaroundTriggered();
}

float SkeletonEditGraphicsView::findXForSlave(float x)
{
    return x - m_backgroundItem->boundingRect().width() / 4;
}

void SkeletonEditGraphicsView::mouseReleaseEvent(QMouseEvent *event)
{
    QWidget::mouseReleaseEvent(event);
    if (!m_backgroundLoaded)
        return;
    if (event->button() == Qt::LeftButton) {
        if (m_inAddNodeMode) {
            SkeletonEditNodeItem *masterNode = new SkeletonEditNodeItem(m_pendingNodeItem->rect());
            scene()->addItem(masterNode);
            QRectF slaveRect = m_pendingNodeItem->rect();
            float x = m_pendingNodeItem->x() + m_pendingNodeItem->rect().width() / 2;
            slaveRect.translate(findXForSlave(x) - x, 0);
            SkeletonEditNodeItem *slaveNode = new SkeletonEditNodeItem(slaveRect);
            scene()->addItem(slaveNode);
            masterNode->setSlave(slaveNode);
            slaveNode->setMaster(masterNode);
            if (m_nextStartNodeItem) {
                SkeletonEditEdgeItem *newEdge = new SkeletonEditEdgeItem();
                newEdge->setNodes(masterNode, m_nextStartNodeItem);
                scene()->addItem(newEdge);
            }
            setNextStartNodeItem(masterNode);
            emit nodesChanged();
        }
        m_isMovingNodeItem = false;
    } else if (event->button() == Qt::RightButton) {
        toggleAddNodeMode();
    }
}

bool SkeletonEditGraphicsView::canNodeItemMoveTo(SkeletonEditNodeItem *item, QPointF moveTo)
{
    if (moveTo.x() < 0)
        return false;
    if (moveTo.y() < 0)
        return false;
    if (moveTo.x() + item->rect().width() >= m_backgroundItem->boundingRect().width())
        return false;
    if (moveTo.y() + item->rect().height() >= m_backgroundItem->boundingRect().height())
        return false;
    return true;
}

void SkeletonEditGraphicsView::mouseMoveEvent(QMouseEvent *event)
{
    QWidget::mouseMoveEvent(event);
    if (!m_backgroundLoaded)
        return;
    QPointF pos = mapToScene(event->pos());
    QPointF moveTo = QPointF(pos.x() - m_pendingNodeItem->rect().width() / 2, pos.y() - m_pendingNodeItem->rect().height() / 2);
    if (moveTo.x() < 0)
        moveTo.setX(0);
    if (moveTo.y() < 0)
        moveTo.setY(0);
    if (moveTo.x() + m_pendingNodeItem->rect().width() >= m_backgroundItem->boundingRect().width())
        moveTo.setX(m_backgroundItem->boundingRect().width() - m_pendingNodeItem->rect().width());
    if (moveTo.y() + m_pendingNodeItem->rect().height() >= m_backgroundItem->boundingRect().height())
        moveTo.setY(m_backgroundItem->boundingRect().height() - m_pendingNodeItem->rect().height());
    QSizeF oldSize = m_pendingNodeItem->rect().size();
    m_pendingNodeItem->setRect(moveTo.x(), moveTo.y(),
        oldSize.width(),
        oldSize.height());
    if (m_nextStartNodeItem) {
        m_pendingEdgeItem->setLine(QLineF(m_nextStartNodeItem->origin(), QPointF(moveTo.x() + m_pendingNodeItem->rect().width() / 2,
            moveTo.y() + m_pendingNodeItem->rect().height() / 2)));
    }
    if (!m_isMovingNodeItem) {
        if (!m_inAddNodeMode) {
            SkeletonEditNodeItem *hoverNodeItem = findNodeItemByPos(pos);
            if (hoverNodeItem) {
                hoverNodeItem->setHighlighted(true);
            }
            if (hoverNodeItem != m_lastHoverNodeItem) {
                if (m_lastHoverNodeItem)
                    m_lastHoverNodeItem->setHighlighted(false);
                m_lastHoverNodeItem = hoverNodeItem;
            }
        } else {
            if (m_lastHoverNodeItem) {
                m_lastHoverNodeItem->setHighlighted(false);
                m_lastHoverNodeItem = NULL;
            }
        }
    }
    QPointF curMousePos = pos;
    if (m_lastHoverNodeItem) {
        if ((event->buttons() & Qt::LeftButton) &&
                (curMousePos != m_lastMousePos || m_isMovingNodeItem)) {
            m_isMovingNodeItem = true;
            QRectF rect = m_lastHoverNodeItem->rect();
            QRectF slaveRect;
            if (m_lastHoverNodeItem->isMaster()) {
                rect.translate(curMousePos.x() - m_lastMousePos.x(), curMousePos.y() - m_lastMousePos.y());
                slaveRect = m_lastHoverNodeItem->slave()->rect();
                slaveRect.translate(0, curMousePos.y() - m_lastMousePos.y());
            } else {
                rect.translate(curMousePos.x() - m_lastMousePos.x(), 0);
            }
            if (canNodeItemMoveTo(m_lastHoverNodeItem, QPointF(rect.x(), rect.y()))) {
                if (m_lastHoverNodeItem->isMaster()) {
                    m_lastHoverNodeItem->slave()->setRect(slaveRect);
                }
                m_lastHoverNodeItem->setRect(rect);
                QList<QGraphicsItem *>::iterator it;
                QList<QGraphicsItem *> list = scene()->items();
                for (it = list.begin(); it != list.end(); ++it) {
                    if ((*it)->data(0).toString() == "edge") {
                        SkeletonEditEdgeItem *edgeItem = static_cast<SkeletonEditEdgeItem *>(*it);
                        if (edgeItem->connects(m_lastHoverNodeItem))
                            edgeItem->updatePosition();
                    }
                }
                emit nodesChanged();
            }
        }
    }
    m_lastMousePos = curMousePos;
}

void SkeletonEditGraphicsView::AddItemRadius(QGraphicsEllipseItem *item, float delta)
{
    QSizeF oldSize = item->rect().size();
    QPointF originPt = QPointF(item->rect().left() + oldSize.width() / 2,
        item->rect().top() + oldSize.height() / 2);
    QSizeF newSize = QSizeF(oldSize.width() + delta, oldSize.height() + delta);
    if (newSize.width() < m_minimalNodeSize || newSize.height() < m_minimalNodeSize) {
        newSize.setWidth(m_minimalNodeSize);
        newSize.setHeight(m_minimalNodeSize);
    }
    QPointF newLeftTop = QPointF(originPt.x() - newSize.width() / 2,
        originPt.y() - newSize.height() / 2);
    if (newLeftTop.x() < 0 || newLeftTop.x() + newSize.width() >= m_backgroundItem->boundingRect().width())
        return;
    if (newLeftTop.y() < 0 || newLeftTop.y() + newSize.height() >= m_backgroundItem->boundingRect().height())
        return;
    item->setRect(newLeftTop.x(),
        newLeftTop.y(),
        newSize.width(),
        newSize.height());
}

bool SkeletonEditGraphicsView::canAddItemRadius(QGraphicsEllipseItem *item, float delta)
{
    QSizeF oldSize = item->rect().size();
    QPointF originPt = QPointF(item->rect().left() + oldSize.width() / 2,
        item->rect().top() + oldSize.height() / 2);
    QSizeF newSize = QSizeF(oldSize.width() + delta, oldSize.height() + delta);
    if (newSize.width() < m_minimalNodeSize || newSize.height() < m_minimalNodeSize) {
        newSize.setWidth(m_minimalNodeSize);
        newSize.setHeight(m_minimalNodeSize);
    }
    QPointF newLeftTop = QPointF(originPt.x() - newSize.width() / 2,
        originPt.y() - newSize.height() / 2);
    if (newLeftTop.x() < 0 || newLeftTop.x() + newSize.width() >= m_backgroundItem->boundingRect().width())
        return false;
    if (newLeftTop.y() < 0 || newLeftTop.y() + newSize.height() >= m_backgroundItem->boundingRect().height())
        return false;
    return true;
}

void SkeletonEditGraphicsView::wheelEvent(QWheelEvent *event)
{
    QWidget::wheelEvent(event);
    if (!m_backgroundLoaded)
        return;
    qreal delta = event->delta() / 10;
    if (fabs(delta) < 1)
        delta = delta < 0 ? -1.0 : 1.0;
    AddItemRadius(m_pendingNodeItem, delta);
    if (!m_inAddNodeMode && m_lastHoverNodeItem) {
        if (canAddItemRadius(m_lastHoverNodeItem, delta) &&
                canAddItemRadius(m_lastHoverNodeItem->pair(), delta)) {
            AddItemRadius(m_lastHoverNodeItem, delta);
            AddItemRadius(m_lastHoverNodeItem->pair(), delta);
        }
        emit nodesChanged();
    }
}

void SkeletonEditGraphicsView::setNextStartNodeItem(SkeletonEditNodeItem *item)
{
    if (m_nextStartNodeItem != item) {
        if (m_nextStartNodeItem)
            m_nextStartNodeItem->setIsNextStartNode(false);
    }
    m_nextStartNodeItem = item;
    if (m_nextStartNodeItem)
        m_nextStartNodeItem->setIsNextStartNode(true);
}

void SkeletonEditGraphicsView::updateBackgroundImage(const QImage &image)
{
    QSizeF oldSceneSize = scene()->sceneRect().size();
    QPixmap pixmap = QPixmap::fromImage(image);
    m_backgroundItem->setPixmap(pixmap);
    scene()->setSceneRect(pixmap.rect());
    adjustItems(oldSceneSize, scene()->sceneRect().size());
    if (!m_backgroundLoaded) {
        m_backgroundLoaded = true;
        applyAddNodeMode();
    }
}

void SkeletonEditGraphicsView::adjustItems(QSizeF oldSceneSize, QSizeF newSceneSize)
{
    if (oldSceneSize == newSceneSize)
        return;
    float radiusMul = (float)newSceneSize.height() / oldSceneSize.height();
    float xMul = (float)newSceneSize.width() / oldSceneSize.width();
    float yMul = radiusMul;
    QList<QGraphicsItem *>::iterator it;
    QList<QGraphicsItem *> list = scene()->items();
    for (it = list.begin(); it != list.end(); ++it) {
        if ((*it)->data(0).toString() == "node") {
            SkeletonEditNodeItem *nodeItem = static_cast<SkeletonEditNodeItem *>(*it);
            nodeItem->setRadius(nodeItem->radius() * radiusMul);
            QPointF oldOrigin = nodeItem->origin();
            nodeItem->setOrigin(QPointF(oldOrigin.x() * xMul, oldOrigin.y() * yMul));
        }
    }
    for (it = list.begin(); it != list.end(); ++it) {
        if ((*it)->data(0).toString() == "edge") {
            SkeletonEditEdgeItem *edgeItem = static_cast<SkeletonEditEdgeItem *>(*it);
            edgeItem->updatePosition();
        }
    }
}

void SkeletonEditGraphicsView::loadFromXml(const QString &filename)
{
    QFile file(filename);
    if (!file.open(QFile::ReadOnly | QFile::Text)) {
        return;
    }
    
    float radiusMul = 1.0;
    float xMul = 1.0;
    float yMul = radiusMul;
    
    std::vector<std::map<QString, QString>> pendingNodes;
    std::vector<std::map<QString, QString>> pendingEdges;
    std::map<QString, SkeletonEditNodeItem *> addedNodeMapById;
    
    QXmlStreamReader xml;
    xml.setDevice(&file);
    while (!xml.atEnd()) {
        xml.readNext();
        printf("tokenString:%s\n", xml.tokenString().toUtf8().constData());
        if (xml.isStartElement()) {
            printf("name:%s\n", xml.name().toUtf8().constData());
            if (xml.name() == "canvas") {
                QString canvasWidth = xml.attributes().value("width").toString();
                QString canvasHeight = xml.attributes().value("height").toString();
                float canvasHeightWidth = canvasWidth.toFloat();
                float canvasHeightVal = canvasHeight.toFloat();
                if (canvasHeightVal > 0)
                    radiusMul = (float)scene()->sceneRect().height() / canvasHeightVal;
                if (canvasHeightWidth > 0)
                    xMul = (float)scene()->sceneRect().width() / canvasHeightWidth;
                yMul = radiusMul;
            } else if (xml.name() == "origin") {
                if (pendingNodes.size() > 0) {
                    pendingNodes[pendingNodes.size() - 1]["x"] = QString("%1").arg(xml.attributes().value("x").toString().toFloat() * xMul);
                    pendingNodes[pendingNodes.size() - 1]["y"] = QString("%1").arg(xml.attributes().value("y").toString().toFloat() * yMul);
                }
            } else if (xml.name() == "node") {
                QString nodeId = xml.attributes().value("id").toString();
                QString nodeType = xml.attributes().value("type").toString();
                QString nodePairId = xml.attributes().value("pair").toString();
                QString nodeRadius = xml.attributes().value("radius").toString();
                std::map<QString, QString> pendingNode;
                pendingNode["id"] = nodeId;
                pendingNode["type"] = nodeType;
                pendingNode["pair"] = nodePairId;
                pendingNode["radius"] = QString("%1").arg(nodeRadius.toFloat() * radiusMul);
                pendingNode["x"] = "0";
                pendingNode["y"] = "0";
                pendingNodes.push_back(pendingNode);
            } else if (xml.name() == "edge") {
                QString edgeId = xml.attributes().value("id").toString();
                QString edgeFromNodeId = xml.attributes().value("from").toString();
                QString edgeToNodeId = xml.attributes().value("to").toString();
                if (!edgeFromNodeId.isEmpty() && !edgeToNodeId.isEmpty()) {
                    std::map<QString, QString> pendingEdge;
                    pendingEdge["id"] = edgeId;
                    pendingEdge["from"] = edgeFromNodeId;
                    pendingEdge["to"] = edgeToNodeId;
                    pendingEdges.push_back(pendingEdge);
                }
            }
        }
    }
    
    for (size_t i = 0; i < pendingNodes.size(); i++) {
        std::map<QString, QString> *pendingNode = &pendingNodes[i];
        float radius = (*pendingNode)["radius"].toFloat();
        QRectF nodeRect((*pendingNode)["x"].toFloat() - radius, (*pendingNode)["y"].toFloat() - radius,
            radius * 2, radius * 2);
        addedNodeMapById[(*pendingNode)["id"]] = new SkeletonEditNodeItem(nodeRect);
    }
    
    for (size_t i = 0; i < pendingNodes.size(); i++) {
        std::map<QString, QString> *pendingNode = &pendingNodes[i];
        if ((*pendingNode)["type"] == "master") {
            addedNodeMapById[(*pendingNode)["id"]]->setSlave(addedNodeMapById[(*pendingNode)["pair"]]);
        } else if ((*pendingNode)["type"] == "slave") {
            addedNodeMapById[(*pendingNode)["id"]]->setMaster(addedNodeMapById[(*pendingNode)["pair"]]);
        }
        scene()->addItem(addedNodeMapById[(*pendingNode)["id"]]);
    }
    
    for (size_t i = 0; i < pendingEdges.size(); i++) {
        std::map<QString, QString> *pendingEdge = &pendingEdges[i];
        SkeletonEditEdgeItem *newEdge = new SkeletonEditEdgeItem();
        newEdge->setNodes(addedNodeMapById[(*pendingEdge)["from"]], addedNodeMapById[(*pendingEdge)["to"]]);
        scene()->addItem(newEdge);
    }
    
    emit nodesChanged();
}

void SkeletonEditGraphicsView::saveToXml(const QString &filename)
{
    QFile file(filename);
    file.open(QIODevice::WriteOnly);
    
    QXmlStreamWriter stream(&file);
    stream.setAutoFormatting(true);
    stream.writeStartDocument();
    
    stream.writeStartElement("canvas");
        stream.writeAttribute("width", QString("%1").arg(scene()->sceneRect().width()));
        stream.writeAttribute("height", QString("%1").arg(scene()->sceneRect().height()));
    
        QList<QGraphicsItem *>::iterator it;
        QList<QGraphicsItem *> list = scene()->items();
        std::map<SkeletonEditNodeItem *, int> nodeIdMap;
        int nextNodeId = 1;
        for (it = list.begin(); it != list.end(); ++it) {
            if ((*it)->data(0).toString() == "node") {
                SkeletonEditNodeItem *nodeItem = static_cast<SkeletonEditNodeItem *>(*it);
                nodeIdMap[nodeItem] = nextNodeId;
                nextNodeId++;
            }
        }
        stream.writeStartElement("nodes");
        for (it = list.begin(); it != list.end(); ++it) {
            if ((*it)->data(0).toString() == "node") {
                SkeletonEditNodeItem *nodeItem = static_cast<SkeletonEditNodeItem *>(*it);
                stream.writeStartElement("node");
                    stream.writeAttribute("id", QString("node%1").arg(nodeIdMap[nodeItem]));
                    stream.writeAttribute("type", nodeItem->isMaster() ? "master" : "slave");
                    stream.writeAttribute("pair", QString("node%1").arg(nodeIdMap[nodeItem->pair()]));
                    stream.writeAttribute("radius", QString("%1").arg(nodeItem->radius()));
                    stream.writeStartElement("origin");
                        QPointF origin = nodeItem->origin();
                        stream.writeAttribute("x", QString("%1").arg(origin.x()));
                        stream.writeAttribute("y", QString("%1").arg(origin.y()));
                    stream.writeEndElement();
                stream.writeEndElement();
            }
        }
        stream.writeEndElement();
        stream.writeStartElement("edges");
        int nextEdgeId = 1;
        for (it = list.begin(); it != list.end(); ++it) {
            if ((*it)->data(0).toString() == "edge") {
                SkeletonEditEdgeItem *edgeItem = static_cast<SkeletonEditEdgeItem *>(*it);
                stream.writeStartElement("edge");
                    stream.writeAttribute("id", QString("edge%1").arg(nextEdgeId));
                    stream.writeAttribute("from", QString("node%1").arg(nodeIdMap[edgeItem->firstNode()]));
                    stream.writeAttribute("to", QString("node%1").arg(nodeIdMap[edgeItem->secondNode()]));
                stream.writeEndElement();
                nextEdgeId++;
            }
        }
        stream.writeEndElement();
    
    stream.writeEndElement();
    stream.writeEndDocument();
}