node.h

#ifndef NODE_H
#define NODE_H

#define NODE_TYPE_NEURON 0
#define NODE_TYPE_LAYER 1

#include <QGraphicsItem>
#include <QList>

class Edge;
class GraphWidget;
QT_BEGIN_NAMESPACE
class QGraphicsSceneMouseEvent;
QT_END_NAMESPACE

//! [0]
class Node : public QGraphicsItem
{
public:
    Node(QString name, GraphWidget *graphWidget, int nodeType=NODE_TYPE_NEURON);


    void addEdge(Edge *edge);
    void resetGroundTruth();
    QList<Edge *> edges() const;

    enum { Type = UserType + 1 };
    int type() const override { return Type; }

    void calculateForces();
    bool advancePosition();

    QRectF boundingRect() const override;
    QPainterPath shape() const override;
    void paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget) override;

signals:
    void showlayer(int layer);

protected:
    QVariant itemChange(GraphicsItemChange change, const QVariant &value) override;

    void mousePressEvent(QGraphicsSceneMouseEvent *event) override;
    void mouseReleaseEvent(QGraphicsSceneMouseEvent *event) override;

private:
    QList<Edge *> edgeList;
    QPointF newPos;
    GraphWidget *graph;
    bool isGroundtruth;
    int nodeType;

    QString name;

};
//! [0]

#endif // NODE_H