DANodeGraphicsScene.h

#ifndef DANODEGRAPHICSSCENE_H
#define DANODEGRAPHICSSCENE_H
#include <QtCore/qglobal.h>
#include <QImage>
#include "DANodeMetaData.h"
#include "DAWorkFlowAPI.h"
#include "DAGraphicsScene.h"
#include "DAAbstractNodeGraphicsItem.h"
#include "DAAbstractNodeLinkGraphicsItem.h"
#include "DAWorkFlow.h"
 
class QGraphicsSceneMouseEvent;
namespace DA
{
class DAGraphicsPixmapItem;
class DAGraphicsRectItem;
class DAGraphicsStandardTextItem;
class DAGraphicsTextItem;
class DAWORKFLOW_API DANodeGraphicsScene : public DAGraphicsScene
{
    Q_OBJECT
    DA_DECLARE_PRIVATE(DANodeGraphicsScene)
    friend class DAAbstractNodeGraphicsItem;
    friend class DAAbstractNodeLinkGraphicsItem;
 
public:
    DANodeGraphicsScene(QObject* p = nullptr);
    DANodeGraphicsScene(const QRectF& sceneRect, QObject* p = nullptr);
    DANodeGraphicsScene(qreal x, qreal y, qreal width, qreal height, QObject* p = nullptr);
    ~DANodeGraphicsScene();
 
    // 取消链接
    virtual void cancelLink() override;
 
    // 设置工作流
    void setWorkFlow(DAWorkFlow* wf);
    DAWorkFlow* getWorkflow();
 
    // 通过node找到item,时间复杂度<O(n)
    DAAbstractNodeGraphicsItem* findItemByNode(DAAbstractNode* n);
 
    // 获取选中的NodeGraphicsItem,如果没有返回nullptr，如果选中多个，返回第一个
    DAAbstractNodeGraphicsItem* getSelectedNodeGraphicsItem() const;
 
    // 获取所有的graphicsitem
    QList< DAAbstractNodeGraphicsItem* > getNodeGraphicsItems() const;
 
    // 获取所有的DAAbstractNodeGraphicsItem
    QList< DAGraphicsStandardTextItem* > getTextGraphicsItems() const;
 
    // 获取选中的NodeGraphicsItem,如果没有返回一个空list
    QList< DAAbstractNodeGraphicsItem* > getSelectedNodeGraphicsItems() const;
 
    // 获取选中的NodeLinkGraphicsItem,如果没有返回nullptr，如果选中多个，返回第一个
    DAAbstractNodeLinkGraphicsItem* getSelectedNodeLinkGraphicsItem() const;
 
    // 获取选中的NodeLinkGraphicsItem,如果没有返回一个空list
    QList< DAAbstractNodeLinkGraphicsItem* > getSelectedNodeLinkGraphicsItems() const;
 
    // 获取除了连接线以外的item
    QList< QGraphicsItem* > getGraphicsItemsWithoutLink() const;
 
    // 删除选中的item，此函数支持redo/undo,返回删除的数量
    int removeSelectedItems_();
    // 带redo/undo的节点item删除，此操作会顺带删除节点对应的连接线，避免悬空连接线
    void removeNodeItem_(DAAbstractNodeGraphicsItem* i);
    // 带redo/undo的节点item的添加，此函数适用于自己通过factory创建node，然后用node创建item，再加入场景中
    void addNodeItem_(DAAbstractNodeGraphicsItem* i);
    // 添加带redo/undo的链接线
    void addNodeLink_(DAAbstractNodeLinkGraphicsItem* link);
    // 通过node元对象创建工作流节点，（item没有设置到场景中）
    DAAbstractNodeGraphicsItem* createNode(const DANodeMetaData& md, const QPointF& pos, bool addItemToScene = true);
    // 创建节点同时把item设置到场景中
    DAAbstractNodeGraphicsItem* createNode_(const DANodeMetaData& md, const QPointF& pos);
 
    // 添加一个图片item
    DAGraphicsPixmapItem* addPixmapItem_(const QImage& img);
    // 通过位置获取DAAbstractNodeGraphicsItem，此函数是加强版的itemAt
    DAAbstractNodeGraphicsItem* nodeItemAt(const QPointF& scenePos) const;
signals:
 
    void nodeItemLinkPointSelected(DA::DAAbstractNodeGraphicsItem* item,
                                   const DA::DANodeLinkPoint& lp,
                                   QGraphicsSceneMouseEvent* event);
 
    void nodeLinkItemIsEmpty(DA::DAAbstractNodeLinkGraphicsItem* link);
 
    void selectNodeItemChanged(DA::DAAbstractNodeGraphicsItem* item);
 
    void selectNodeLinkChanged(DA::DAAbstractNodeLinkGraphicsItem* link);
 
    void nodeItemsRemoved(const QList< DA::DAAbstractNodeGraphicsItem* >& items);
 
    void nodeLinksRemoved(const QList< DA::DAAbstractNodeLinkGraphicsItem* >& items);
 
protected slots:
    void onSelectItemChanged(DAGraphicsItem* item);
    void onSelectLinkChanged(DAGraphicsLinkItem* item);
    // item选择改变
    //    void onItemSelectionChanged();
 
    // node的名字改变
    void onNodeNameChanged(DAAbstractNode::SharedPointer node, const QString& oldName, const QString& newName);
 
protected:
    // 鼠标点击事件
    void mousePressEvent(QGraphicsSceneMouseEvent* mouseEvent) override;
 
    // itemlink都没用节点连接时会调用这个函数，发出
    void callNodeItemLinkIsEmpty(DAAbstractNodeLinkGraphicsItem* link);
 
    // 对item进行分类
    static void classifyItems(const QList< QGraphicsItem* >& sourceItems,
                              QList< DAAbstractNodeGraphicsItem* >& nodeItems,
                              QList< DAAbstractNodeLinkGraphicsItem* >& linkItems,
                              QList< QGraphicsItem* >& normalItem);
    // 获取node item的所有连接线
    static QList< DAAbstractNodeLinkGraphicsItem* > getNodesAllLinkItems(const QList< DAAbstractNodeGraphicsItem* >& nodeItems);
    // 所有节点item对应的node
    static QList< DAAbstractNode::SharedPointer > nodeItemsToNodes(const QList< DAAbstractNodeGraphicsItem* >& nodeItems);
 
private:
    void initConnect();
};
}
#endif  // FCNODEGRAPHICSSCENE_H