DiagramDesigner/include/graphicsItem/graphicsBaseItem.h

#ifndef GRAPHICSBASEITEM_H
#define GRAPHICSBASEITEM_H

#include "itemControlHandle.h"
#include "global.h"

#include <QObject>
#include <QGraphicsItem>
#include <QGraphicsSceneMouseEvent>
#include <QPen>
#include <QGraphicsSvgItem>
//#include "graphicsItem/itemPort.h"

enum ShapeType
{
    T_undefined,
    T_item,
    T_group
};

enum ItemState
{
    S_normal = 0,
    S_prepareConnect,
};


//基类采用模板形式，QGraphicsItem是默认值,也可以是别的类型，比如QGraphicsItemGroup，这样不同的基类继承可以共用一些高层的行为定义
template <typename BaseType = QGraphicsItem>
class AbstractShapeType : public BaseType
{
public:
    explicit AbstractShapeType(QGraphicsItem *parent = 0)
        : BaseType(parent)
    {
        m_type = T_undefined;
        m_pen = QPen(Qt::NoPen);
        m_brush = QBrush(QColor(rand() % 32 * 8, rand() % 32 * 8, rand() % 32 * 8));
        m_dWidth = m_dHeight = 0;
        m_pOperationCopy = nullptr;
        m_dSyncRotationByParent = 0.0;
        m_touched = false;
    }

    virtual ~AbstractShapeType()
    {
        /*for (size_t i = 0; i < m_vecHanle.size(); i++)
        {
            ItemControlHandle* pHandle = m_vecHanle[i];
            if (pHandle)
            {
                delete pHandle;
                pHandle = nullptr;
            }
        }*/
        foreach (int key, m_vecHanle.keys())
        {
            ItemControlHandle* pHandle = m_vecHanle.value(key);
            if (pHandle)
            {
                delete pHandle;
                pHandle = nullptr;
            }
        }
    }

public:
    virtual ShapeType getType() {return m_type;}

    QPen pen() { return m_pen; }
    void setPen(const QPen &pen) { m_pen = pen; }
    QColor penColor() { return m_pen.color(); }
    void setPenColor(const QColor &color) { m_pen.setColor(color); }

    QBrush brush() { return m_brush; }
    void setBrush(const QBrush &brush) { m_brush = brush; }
    QColor brushColor() { return m_brush.color(); }
    void setBrushColor(const QColor &color) { m_brush.setColor(color); }

    double width() { return m_dWidth; }
    void setWidth(double width)
    {
        m_dWidth = width;
        updateCoordinate();
    }

    double height() { return m_dHeight; }
    void setHeight(double height)
    {
        m_dHeight = height;
        updateCoordinate();
    }

    int collidesWithHandle(const QPointF& point)
    {
        /*for(auto it = m_vecHanle.begin(); it != m_vecHanle.end(); it++)
        {
            QPointF pt = (*it)->mapFromScene(point);
            if((*it)->contains(pt))
                return (*it)->getTag();
        }*/

        foreach (int key, m_vecHanle.keys())
        {
            ItemControlHandle* pHandle = m_vecHanle.value(key);
            if (pHandle)
            {
                QPointF pt = pHandle->mapFromScene(point);
                if(pHandle->contains(pt))
                {
                    if(pHandle->enable())
                        return pHandle->getTag();
                }
            }
        }
        return HandleTag::H_none;
    }

    //操作副本相关
    virtual void createOperationCopy() {}
    virtual void removeOperationCopy() {}
    virtual void moveOperationCopy(const QPointF&) {}
    virtual void rotateOperationCopy(const double&) {}

    virtual void resize(int,double, double, const QPointF&) {}
    virtual void move(const QPointF&) {}
    virtual void editShape(int, const QPointF&) {}

    virtual void updateCoordinate() {}

    //handle相关
    virtual int handleCount() { return m_vecHanle.count(); }
    virtual ItemControlHandle* getHandle(int nHandle)
    {
        ItemControlHandle* handle = nullptr;
        /*for(auto it = m_vecHanle.begin(); it != m_vecHanle.end(); it++)
        {
            if((*it)->getTag() == nHandle)
            {
                handle = (*it);
                return handle;
            }
        }*/

        foreach (int key, m_vecHanle.keys())
        {
            ItemControlHandle* pHandle = m_vecHanle.value(key);
            if (pHandle)
            {
                if(pHandle->getTag() == nHandle)
                {
                    handle = pHandle;
                    return handle;
                }
            }
        }
        return handle;
    }
    virtual void setHandleVisible(bool bVisible)
    {
        /*for(auto it = m_vecHanle.begin(); it != m_vecHanle.end(); it++)
        {
            if(bVisible)
                (*it)->show();
            else
                (*it)->hide();
        }*/

        foreach (int key, m_vecHanle.keys())
        {
            ItemControlHandle* pHandle = m_vecHanle.value(key);
            if(pHandle)
            {
                if(pHandle->ifShow())
                {
                    if(bVisible)
                        pHandle->show();
                    else
                        pHandle->hide();
                }
            }
        }
    }
    virtual void setHandleVisible(int tag,bool bVisible)
    {
        foreach (int key, m_vecHanle.keys())
        {
            ItemControlHandle* pHandle = m_vecHanle.value(key);
            if(pHandle)
            {
                if(pHandle->getTag() == tag)
                {
                    if(bVisible)
                        pHandle->show();
                    else
                        pHandle->hide();
                    break;
                }
            }
        }
    }
    virtual void setHandleIfShow(int tag,bool bVisible)             //是否参与显示判断
    {
        foreach (int key, m_vecHanle.keys())
        {
            ItemControlHandle* pHandle = m_vecHanle.value(key);
            if(pHandle)
            {
                if(pHandle->getTag() == tag)
                {
                    pHandle->setIfShow(bVisible);
                    break;
                }
            }
        }
    }
    virtual void setFunctionHandleIfShow(bool bVisible)         //是否参与显示判断
    {
        foreach (int key, m_vecHanle.keys())
        {
            ItemControlHandle* pHandle = m_vecHanle.value(key);
            if(pHandle)
            {
                if(pHandle->getTag() < H_connect)
                {
                    pHandle->setIfShow(bVisible);
                }
            }
        }
    }
    virtual void setFunctionHandleEnaable(bool val)
    {
        foreach (int key, m_vecHanle.keys())
        {
            ItemControlHandle* pHandle = m_vecHanle.value(key);
            if(pHandle)
            {
                if(pHandle->getTag() < H_connect)      //设置port以外端点是否交互
                    pHandle->setEnable(val);
            }
        }
    }
    virtual void setHandleEnaable(HandleTag tag,bool val)
    {
        foreach (int key, m_vecHanle.keys())
        {
            ItemControlHandle* pHandle = m_vecHanle.value(key);
            if(pHandle)
            {
                if(pHandle->getTag() == tag)      //设置端点是否交互
                {
                    pHandle->setEnable(val);
                    break;
                }
            }
        }
    }
    virtual QPointF getSymmetricPointPos(int nHandle) //获取对称点的坐标位置，缩放的时候需要以对称点为锚点
    {
        QPointF pt;
        //handle的位置相对boundingRect会有一个向外的错位，因此直接采用bounddingRect的相应位置会更精准
        switch (nHandle)
        {
        case H_leftTop:
            //pt = m_vecHanle.at(H_rightBottom - 1)->pos();
            pt = m_boundingRect.bottomRight();
            break;
        case H_top:
            //pt = m_vecHanle.at(H_bottom - 1)->pos();
            pt = QPointF(m_boundingRect.width() * 0.5, m_boundingRect.bottom());
            break;
        case H_rightTop:
            //pt = m_vecHanle.at(H_leftBottom - 1)->pos();
            pt = m_boundingRect.bottomLeft();
            break;
        case H_right:
            //pt = m_vecHanle.at(H_left - 1)->pos();
            pt = QPointF(m_boundingRect.left(), m_boundingRect.height() * 0.5);
            break;
        case H_rightBottom:
            //pt = m_vecHanle.at(H_leftTop - 1)->pos();
            pt = m_boundingRect.topLeft();
            break;
        case H_bottom:
            //pt = m_vecHanle.at(H_top - 1)->pos();
            pt = QPointF(m_boundingRect.width() * 0.5, m_boundingRect.top());
            break;
        case H_leftBottom:
            //pt = m_vecHanle.at(H_rightTop - 1)->pos();
            pt = m_boundingRect.topRight();
            break;
        case H_left:
            //pt = m_vecHanle.at(H_right - 1)->pos();
            pt = QPointF(m_boundingRect.right(), m_boundingRect.height() * 0.5);
            break;
        default:
            break;
        }

        return pt;
    }
    virtual void updateHandles()
    {
        int nMargin = 5;
        const QRectF& boundRect = this->boundingRect();

        foreach (int key, m_vecHanle.keys())
        {
            ItemControlHandle* pHandle = m_vecHanle.value(key);
            if(pHandle)
            {
                switch (pHandle->getTag()) {
                case H_leftTop:
                    pHandle->move(boundRect.x() - nMargin, boundRect.y() - nMargin);
                    break;
                case H_top:
                    pHandle->move(boundRect.x() + boundRect.width() * 0.5, boundRect.y() - nMargin);
                    break;
                case H_rightTop:
                    pHandle->move(boundRect.x() + boundRect.width() + nMargin, boundRect.y() - nMargin);
                    break;
                case H_right:
                    pHandle->move(boundRect.x() + boundRect.width() + nMargin, boundRect.y() + boundRect.height() * 0.5 + nMargin);
                    break;
                case H_rightBottom:
                    pHandle->move(boundRect.x() + boundRect.width() + nMargin, boundRect.y() + boundRect.height() + nMargin);
                    break;
                case H_bottom:
                    pHandle->move(boundRect.x() + boundRect.width() * 0.5, boundRect.y() + boundRect.height()+ nMargin);
                    break;
                case H_leftBottom:
                    pHandle->move(boundRect.x() - nMargin, boundRect.y() + boundRect.height() + nMargin);
                    break;
                case H_left:
                    pHandle->move(boundRect.x() - nMargin, boundRect.y() + boundRect.height() * 0.5);
                    break;
                case H_rotate_leftTop:
                {
                    ItemControlHandle* handle = getHandle(H_leftTop);
                    if(handle)
                    {
                        int nSize = handle->getSize();
                        QPointF pt = handle->pos();
                        pHandle->move(pt.x() - nSize - 1, pt.y() - nSize - 1);
                    }
                    else
                        pHandle->setVisible(false);

                    break;
                }
                case H_rotate_rightTop:
                {
                    ItemControlHandle* handle = getHandle(H_rightTop);
                    if(handle)
                    {
                        int nSize = handle->getSize();
                        QPointF pt = handle->pos();
                        pHandle->move(pt.x() + nSize + 1, pt.y() - nSize - 1);
                    }
                    else
                        pHandle->setVisible(false);

                    break;
                }
                case H_rotate_rightBottom:
                {
                    ItemControlHandle* handle = getHandle(H_rightBottom);
                    if(handle)
                    {
                        int nSize = handle->getSize();
                        QPointF pt = handle->pos();
                        pHandle->move(pt.x() + nSize + 1, pt.y() + nSize + 1);
                    }
                    else
                        pHandle->setVisible(false);

                    break;
                }
                case H_rotate_leftBottom:
                {
                    ItemControlHandle* handle = getHandle(H_leftBottom);
                    if(handle)
                    {
                        int nSize = handle->getSize();
                        QPointF pt = handle->pos();
                        pHandle->move(pt.x() - nSize - 1, pt.y() + nSize + 1);
                    }
                    else
                        pHandle->setVisible(false);

                    break;
                }
                default:
                    break;
                }
            }
        }

        double dWidth = boundRect.width();
        double dHeight = boundRect.height();
        double dDiagonal = sqrt((dWidth* 0.5)*(dWidth * 0.5)+(dHeight*0.5)*(dHeight*0.5));
        double dParam = dWidth > dHeight ? (dDiagonal-dWidth*0.5):(dDiagonal-dHeight*0.5);
        m_boundingRect_selected = boundRect.adjusted(-dParam-nMargin, -dParam-nMargin, dParam+nMargin, dParam+nMargin);
    }

    virtual QRectF boundingRect() const { return m_boundingRect; }
    virtual QPainterPath shape()
    {
        QPainterPath path;
        return path;
    }

    virtual void syncRotationDataFromParent(const double&) {}
    virtual double getSyncRotationDataFromParent() {return m_dSyncRotationByParent;}
    virtual void setTouched(bool b){m_touched = b;}
    virtual void setLastPoint(QPointF p) {m_lastPoint = p;}
protected:
    ShapeType m_type;
    QPen      m_pen;
    QBrush    m_brush;
    double    m_dWidth;
    double    m_dHeight;
    QRectF    m_boundingRect;
    QRectF    m_boundingRect_selected; //选中矩形框
    bool      m_touched;               //被触碰状态

    double    m_dSyncRotationByParent; //父项(被加入到某一组)的旋转数据，因为加入某一组后，对该组进行旋转，自身的旋转数据不会同步更新

    QGraphicsPathItem*  m_pOperationCopy; //图元移动和旋转时的操作副本
    QPointF             m_movingIniPos; //移动副本开始移动初始
    QPointF             m_lastPoint;            //鼠标上次点击位置

    QMap<int,ItemControlHandle*> m_vecHanle;
};

class ItemPort;

typedef AbstractShapeType<QGraphicsItem> AbstractShape;

class GraphicsBaseItem : public QObject, public AbstractShapeType<QGraphicsItem>
{
     Q_OBJECT

public:
    GraphicsBaseItem(QGraphicsItem *parent);
    virtual ~GraphicsBaseItem();

    virtual void createOperationCopy();
    virtual void removeOperationCopy();
    virtual void moveOperationCopy(const QPointF&);
    virtual void rotateOperationCopy(const double&);
    virtual void syncRotationDataFromParent(const double&);
    //多边形、线段等点选创建的对象需要的函数
    virtual void addPoint(const QPointF&) {}
    virtual bool endDrawing() { return true; }


    virtual int addPort(PortState typ,QPointF vec);            //新建,返回-1失败
    virtual void addPort(PortState typ,int ntagId,QPointF vec); //载入  PortState为P_const时，QPointF中为(0~1，0~1)的相对位置;PortState为p_movable时，QPointF为坐标值
    virtual uint itemId() const {return m_itemId;}
    virtual void setItemId(uint n){m_itemId = n;}
    virtual void setState(ItemState s){m_state = s;}
    virtual void setBeginConnectPos(QPointF p){m_beginConnectPoint = p;}
    virtual void setEndConnectPos(QPointF p){m_endConnectPoint = p;}
    virtual void setItemType(GraphicsItemType type){m_Itemtype = type;}
    virtual GraphicsItemType getItemType() const {return m_Itemtype;}
    virtual void setLastPort(int n){_lastPort = n;}
    virtual int getLastPort() const {return _lastPort;}
    virtual ItemPort* getPortPtr(int) const;
    virtual ItemControlHandle* getHandlePtr(int) const;
    virtual QMap<QString,ItemPort*>& getPorts() {return m_mapPort;}
protected:
    virtual QVariant itemChange(QGraphicsItem::GraphicsItemChange, const QVariant&);
    virtual void contextMenuEvent(QGraphicsSceneContextMenuEvent*);

protected:
    uint                m_itemId;
    GraphicsItemType    m_Itemtype;
    ItemState           m_state;
    QMap<QString,ItemPort*>    m_mapPort;                   //单独存放port
    QPointF             m_beginConnectPoint;
    QPointF             m_endConnectPoint;
    uint                _portId;                                    //port编号
    int                 _lastPort;                          //最后触碰的port
};

#endif