SimpleDrawingScene.h

#ifndef SIMPLEDRAWINGSCENE_H
#define SIMPLEDRAWINGSCENE_H
#include <QGraphicsScene>
#include <QPainterPath>
 
class SimpleDrawingScene : public QGraphicsScene {
public:
    enum class DrawingMode {
        DrawingMode, 
        EditMode 
    };
 
    enum class SupportItemItem {
        Line, 
        Rect, 
        Ellipse, 
        FreeHand 
    };
 
    explicit SimpleDrawingScene(QObject* parent = nullptr);
 
    // Property setters/getters
    inline void setColor(const QColor& color) { currentColor = color; }
 
    inline QColor getColor() const { return currentColor; }
 
    inline void setPenWidth(const short penWidth) { this->penWidth = penWidth; }
 
    inline short getPenWidth() const { return penWidth; }
 
    inline void setItemType(const SupportItemItem i) { itemType = i; }
 
    void setMode(const DrawingMode m);
 
    QImage exportImage();
 
public slots:
    void undo();
 
    void redo();
 
protected:
    void mousePressEvent(QGraphicsSceneMouseEvent* event) override;
 
    void mouseMoveEvent(QGraphicsSceneMouseEvent* event) override;
 
    void mouseReleaseEvent(QGraphicsSceneMouseEvent* event) override;
 
private:
    QGraphicsItem* create_from_types(QPen& pen);
 
    void process_drawing_moving(const QPointF& currentPos);
 
    QPainterPath currentPath;
 
    QGraphicsPathItem* pathItem = nullptr;
 
    QColor currentColor;
 
    short penWidth;
 
    QGraphicsItem* tempItem { nullptr };
 
    QPointF startPos {};
 
    DrawingMode mode { DrawingMode::DrawingMode };
 
    SupportItemItem itemType { SupportItemItem::Line };
 
    QList<QGraphicsItem*> itemsStack;
 
    QList<QGraphicsItem*> redoStack;
};
 
#endif // SIMPLEDRAWINGSCENE_H