143 lines
3.8 KiB
C++
143 lines
3.8 KiB
C++
/***
|
|
|
|
Olive - Non-Linear Video Editor
|
|
Copyright (C) 2019 Olive Team
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
***/
|
|
|
|
#include "nodeviewedge.h"
|
|
|
|
#include <QApplication>
|
|
#include <QDebug>
|
|
#include <QGraphicsSceneMouseEvent>
|
|
|
|
#include "common/clamp.h"
|
|
#include "common/lerp.h"
|
|
#include "nodeview.h"
|
|
#include "nodeviewscene.h"
|
|
|
|
NodeViewEdge::NodeViewEdge(QGraphicsItem *parent) :
|
|
QGraphicsPathItem(parent),
|
|
edge_(nullptr),
|
|
color_group_(QPalette::Active),
|
|
color_role_(QPalette::Text)
|
|
{
|
|
// Ensures this UI object is drawn behind other objects
|
|
setZValue(-1);
|
|
|
|
// Use font metrics to set edge width for basic high DPI support
|
|
edge_width_ = QFontMetrics(QFont()).height() / 12;
|
|
}
|
|
|
|
void NodeViewEdge::SetEdge(NodeEdgePtr edge)
|
|
{
|
|
SetConnected(true);
|
|
|
|
// Set the new edge pointer
|
|
edge_ = edge;
|
|
|
|
// Re-adjust the line positioning for this new edge
|
|
Adjust();
|
|
}
|
|
|
|
NodeEdgePtr NodeViewEdge::edge()
|
|
{
|
|
return edge_;
|
|
}
|
|
|
|
qreal CalculateEdgeYPoint(NodeViewItem *item, NodeParam* param, NodeViewItem *opposing)
|
|
{
|
|
if (item->IsExpanded()) {
|
|
return item->pos().y() + item->GetParameterConnectorRect(param).center().y();
|
|
} else {
|
|
qreal max_height = qMax(opposing->rect().height(), item->rect().height());
|
|
|
|
// Calculate the Y distance between the two nodes and create a 0.0-1.0 range for lerping
|
|
qreal input_value = clamp(0.5 + ((opposing->pos().y() + opposing->rect().top()) - (item->pos().y() + item->rect().top())) / max_height / 4, 0.0, 1.0);
|
|
|
|
// Use a lerp function to draw the line between the two corners
|
|
qreal input_y = item->pos().y() + item->rect().top() + lerp(0.0, item->rect().height(), input_value);
|
|
|
|
// Set Y values according to calculations
|
|
return input_y;
|
|
}
|
|
}
|
|
|
|
void NodeViewEdge::Adjust()
|
|
{
|
|
if (!edge_ || !scene()) {
|
|
return;
|
|
}
|
|
|
|
// Get the UI objects of both nodes that this edge connects
|
|
NodeViewItem* output = static_cast<NodeViewScene*>(scene())->NodeToUIObject(edge_->output()->parentNode());
|
|
NodeViewItem* input = static_cast<NodeViewScene*>(scene())->NodeToUIObject(edge_->input()->parentNode());
|
|
|
|
if (!output || !input) {
|
|
return;
|
|
}
|
|
|
|
// Create initial values
|
|
QPointF output_point = QPointF(output->pos().x() + output->rect().left() + output->rect().width(), 0);
|
|
QPointF input_point = QPointF(input->pos().x() + output->rect().left(), 0);
|
|
|
|
// Calculate output/input points
|
|
output_point.setY(CalculateEdgeYPoint(output, edge_->output(), input));
|
|
input_point.setY(CalculateEdgeYPoint(input, edge_->input(), output));
|
|
|
|
// Draw a line between the two
|
|
SetPoints(output_point, input_point);
|
|
}
|
|
|
|
void NodeViewEdge::SetConnected(bool c)
|
|
{
|
|
if (c) {
|
|
color_group_ = QPalette::Active;
|
|
} else {
|
|
color_group_ = QPalette::Disabled;
|
|
}
|
|
|
|
UpdatePen();
|
|
}
|
|
|
|
void NodeViewEdge::SetHighlighted(bool e)
|
|
{
|
|
if (e) {
|
|
color_role_ = QPalette::Highlight;
|
|
} else {
|
|
color_role_ = QPalette::Text;
|
|
}
|
|
|
|
UpdatePen();
|
|
}
|
|
|
|
void NodeViewEdge::SetPoints(const QPointF &start, const QPointF &end)
|
|
{
|
|
QPainterPath path;
|
|
double half_x = lerp(start.x(), end.x(), 0.5);
|
|
path.moveTo(start);
|
|
path.cubicTo(QPointF(half_x, start.y()), QPointF(half_x, end.y()), end);
|
|
setPath(path);
|
|
}
|
|
|
|
void NodeViewEdge::UpdatePen()
|
|
{
|
|
setPen(QPen(qApp->palette().color(color_group_, color_role_), edge_width_));
|
|
|
|
//update();
|
|
}
|
|
|