GraphDisplayWidget.h

Go to the documentation of this file.

/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2025 Univ. Grenoble Alpes, CNRS, Grenoble INP - UGA, TIMC, 38000 Grenoble, France
 *
 * Visit http://camitk.imag.fr for more information
 *
 * This file is part of CamiTK.
 *
 * CamiTK is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License version 3
 * only, as published by the Free Software Foundation.
 *
 * CamiTK is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License version 3 for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * version 3 along with CamiTK.  If not, see <http://www.gnu.org/licenses/>.
 *
 * $CAMITK_LICENCE_END$
 ****************************************************************************/
 
 
#ifndef GRAPHDISPLAYWIDGET_H
#define GRAPHDISPLAYWIDGET_H
 
#include "CamiTKAPI.h"
 
// Qt Stuff
#include <QObject>
#include <QGraphicsView>
#include <QGraphicsScene>
#include <QGraphicsItem>
#include <QColor>
#include <QPen>
#include <QVector>
#include <QMap>
 
namespace camitk {
 
class CAMITK_API GraphDisplayWidget : public QGraphicsView {
    Q_OBJECT
 
public:
    GraphDisplayWidget(QWidget* parent = nullptr);
 
    virtual ~GraphDisplayWidget();
 
    void clear();
    void addNode(const void* ptr, QString tooltip, QColor color, QColor lineColor, QString internalText = "");
    void addLink(const void* ptr, const void* from, const void* to, QString tooltip, const QPen& pen, bool arrowHead);
 
    void refresh();
 
    void autoScale();
 
    void setNodeDiameter(float d);
 
signals:
    void nodeSelectionChanged(const void* node);
 
    void linkSelectionChanged(const void* link);
 
    void nodeDoubleClicked(const void* node);
 
    void linkDoubleClicked(const void* node);
 
public slots:
    void setSelectedNode(const void* node);
    void setSelectedLink(const void* link);
 
    void updateView();
 
protected:
    void mouseDoubleClickEvent(QMouseEvent*) override;
 
    void resizeEvent(QResizeEvent*) override;
 
    // make sure the scene fits all the potentially moved nodes
    void mouseReleaseEvent(QMouseEvent* e) override;
 
private:
    double diameter;
 
    double halfPenWidth;
 
    QMap<const void*, QGraphicsEllipseItem*> node2item;
    // same but for internal text decoration
    QMap<const void*, QGraphicsSimpleTextItem*> node2textItem;
    QMap<QGraphicsEllipseItem*, const void*> item2node;
    QMap<QGraphicsSimpleTextItem*, const void*> textItem2node;
 
    QMap<const void*, QGraphicsLineItem*> link2item;
    // same but for the arrow heads
    QMap<const void*, QGraphicsPolygonItem*> link2arrowHead;
    QMap<QGraphicsLineItem*, const void*> item2link;
 
    // Selections
    const void* currentSelectedNode;
    const void* currentSelectedLink;
 
    // Pens to manage to selected/unselected
    QPen selectedPen;
    QMap<const void*, QPen> unselectedPen;
 
    // For each link, from and to nodes
    QMap<const void*, std::pair<const void*, const void*>> links;
    // nodeArity[nodeObjectPtr] is the number of links from and to the node
    QMap<const void*, int> nodeArity;
 
    // Move the graphical representation of a node in the canvas
    void moveNode(const void* ptr, float posX, float posY);
    // Move the graphical representation of a link in the canvas
    void moveLink(const void* ptr, float posX1, float posY1, float posX2, float posY2);
 
 
 
    // update link position using the node (new) center, called when nodes are moved
    void updateLinkPosition();
 
    // currently being update, n
};
 
} // namespace camitk
#endif // GRAPHDISPLAYWIDGET_H