117 lines
2.8 KiB
C++
117 lines
2.8 KiB
C++
#include "Gui/Network/SocketGraphic.h"
|
|
#include <QTextDocument>
|
|
#include <iostream>
|
|
#include <qgraphicsitem.h>
|
|
#include "Gui/Network/NodeEdgeGraphic.h"
|
|
|
|
SocketGraphic::SocketGraphic(enzo::nt::SocketIOType type, enzo::nt::OpId opId, unsigned int socketIndex, QGraphicsItem *parent)
|
|
: QGraphicsItem(parent), type_{type}, opId_{opId}, socketIndex_{socketIndex}
|
|
{
|
|
brushActive_ = QBrush("white");
|
|
brushInactive_ = QBrush("#9f9f9f");
|
|
socketSize_ = 3;
|
|
setAcceptHoverEvents(true);
|
|
initBoundingBox();
|
|
}
|
|
|
|
unsigned int SocketGraphic::getIndex() const
|
|
{
|
|
return socketIndex_;
|
|
}
|
|
|
|
|
|
void SocketGraphic::addEdge(NodeEdgeGraphic* edge)
|
|
{
|
|
std::cout << "adding edge\n";
|
|
edges_.insert(edge);
|
|
}
|
|
|
|
void SocketGraphic::removeEdge(NodeEdgeGraphic* edge)
|
|
{
|
|
std::cout << "removing edge\n";
|
|
std::cout << "before size: " << edges_.size() << "\n";
|
|
edges_.erase(edge);
|
|
std::cout << "after size: " << edges_.size() << "\n";
|
|
// auto it = find(edges_.begin(), edges_.end(), edge);
|
|
// if(it!=edges_.end())
|
|
// {
|
|
// edges_.erase(it);
|
|
// }
|
|
}
|
|
|
|
void SocketGraphic::initBoundingBox()
|
|
{
|
|
boundRect_ = QRect(
|
|
-socketSize_/2.0f*paddingScale_,
|
|
-socketSize_/2.0f*paddingScale_,
|
|
socketSize_*paddingScale_,
|
|
socketSize_*paddingScale_
|
|
);
|
|
|
|
}
|
|
|
|
void SocketGraphic::posChanged(QPointF pos)
|
|
{
|
|
for(auto* edge : edges_)
|
|
{
|
|
// edge->setPos(startSocket_->scenePos(), socket->scenePos());
|
|
if(type_==enzo::nt::SocketIOType::Input)
|
|
{
|
|
edge->setStartPos(pos);
|
|
}
|
|
else if(type_==enzo::nt::SocketIOType::Output)
|
|
{
|
|
edge->setEndPos(pos);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
enzo::nt::OpId SocketGraphic::getOpId() const
|
|
{
|
|
return opId_;
|
|
}
|
|
|
|
QRectF SocketGraphic::boundingRect() const
|
|
{
|
|
return boundRect_;
|
|
}
|
|
|
|
QPainterPath SocketGraphic::shape() const{
|
|
QPainterPath path;
|
|
QPointF startPt(boundRect_.center().x(), type_==enzo::nt::SocketIOType::Input ? boundRect_.top() : boundRect_.bottom());
|
|
path.moveTo(startPt);
|
|
path.arcTo(boundRect_, 0, type_==enzo::nt::SocketIOType::Input ? 180 : -180);
|
|
path.lineTo(boundRect_.right(), boundRect_.center().y());
|
|
path.closeSubpath();
|
|
|
|
return path;
|
|
}
|
|
|
|
enzo::nt::SocketIOType SocketGraphic::getIO() { return type_; }
|
|
|
|
|
|
void SocketGraphic::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget)
|
|
{
|
|
painter->setPen(Qt::NoPen);
|
|
painter->setBrush(hovered_ ? brushActive_ : brushInactive_);
|
|
painter->drawEllipse(QPoint(0,0), socketSize_, socketSize_);
|
|
|
|
// painter->drawRect(boundRect_);
|
|
|
|
|
|
}
|
|
|
|
void SocketGraphic::hoverEnterEvent(QGraphicsSceneHoverEvent *event)
|
|
{
|
|
hovered_ = true;
|
|
update();
|
|
}
|
|
|
|
|
|
void SocketGraphic::hoverLeaveEvent(QGraphicsSceneHoverEvent *event)
|
|
{
|
|
hovered_ = false;
|
|
update();
|
|
}
|