oak-team/oak-editor
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

more work

Browse Source
This commit is contained in:
itsmattkc
2021-11-16 23:58:34 -08:00
parent 854ca0a7a9
commit 93be94bc2c
39 changed files with 493 additions and 1357 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/widget/nodeview/nodeviewcontext.cpp
+100 -62
View File
@@ -19,37 +19,29 @@ namespace olive {
#define super QGraphicsRectItem
NodeViewContext::NodeViewContext(QGraphicsItem *item) :
super(item)
NodeViewContext::NodeViewContext(Node *context, QGraphicsItem *item) :
super(item),
context_(context)
{
// Set default label text
SetContext(nullptr);
}
void NodeViewContext::SetContext(Node *node)
{
context_ = node;
if (context_) {
if (Block *block = dynamic_cast<Block*>(node)) {
rational timebase = block->track()->sequence()->GetVideoParams().frame_rate_as_time_base();
lbl_ = QCoreApplication::translate("NodeViewContext",
"%1 [%2] :: %3 - %4").arg(block->GetLabelAndName(),
Track::Reference::TypeToTranslatedString(block->track()->type()),
Timecode::time_to_timecode(block->in(), timebase, Core::instance()->GetTimecodeDisplay()),
Timecode::time_to_timecode(block->out(), timebase, Core::instance()->GetTimecodeDisplay()));
} else {
lbl_ = node->GetLabelAndName();
}
Color c = node->color();
setPen(QPen(c.toQColor(), 2));
c.set_alpha(0.5f);
setBrush(c.toQColor());
if (Block *block = dynamic_cast<Block*>(context_)) {
rational timebase = block->track()->sequence()->GetVideoParams().frame_rate_as_time_base();
lbl_ = QCoreApplication::translate("NodeViewContext",
"%1 [%2] :: %3 - %4").arg(block->GetLabelAndName(),
Track::Reference::TypeToTranslatedString(block->track()->type()),
Timecode::time_to_timecode(block->in(), timebase, Core::instance()->GetTimecodeDisplay()),
Timecode::time_to_timecode(block->out(), timebase, Core::instance()->GetTimecodeDisplay()));
} else {
lbl_ = QCoreApplication::translate("NodeViewContext", "(None)");
lbl_ = context_->GetLabelAndName();
}
const Node::PositionMap &map = context_->GetContextPositions();
for (auto it=map.cbegin(); it!=map.cend(); it++) {
AddChild(it.key());
}
connect(context_, &Node::NodeAddedToContext, this, &NodeViewContext::AddChild, Qt::DirectConnection);
connect(context_, &Node::NodePositionInContextChanged, this, &NodeViewContext::SetChildPosition, Qt::DirectConnection);
connect(context_, &Node::NodeRemovedFromContext, this, &NodeViewContext::RemoveChild, Qt::DirectConnection);
}
void NodeViewContext::AddChild(Node *node)
@@ -59,33 +51,73 @@ void NodeViewContext::AddChild(Node *node)
}
NodeViewItem *item = new NodeViewItem(this);
item->SetNode(node);
item->SetNodePosition(context_->parent()->GetNodesForContext(context_).value(node));
item->SetNode(node, context_);
item->SetNodePosition(context_->GetNodePositionInContext(node));
item->SetFlowDirection(flow_dir_);
connect(node, &Node::InputConnected, this, &NodeViewContext::ChildInputConnected);
connect(node, &Node::InputDisconnected, this, &NodeViewContext::ChildInputDisconnected);
item_map_.insert(node, item);
if (node == context_) {
item->SetLabelAsOutput(true);
}
for (auto it=node->output_connections().cbegin(); it!=node->output_connections().cend(); it++) {
foreach (auto child, childItems()) {
if (NodeViewItem *other_item = dynamic_cast<NodeViewItem*>(child)) {
if (other_item->GetNode() == it->second.node()) {
AddEdgeInternal(node, it->second, item, other_item);
}
if (!it->second.IsHidden()) {
if (NodeViewItem *other_item = item_map_.value(it->second.node())) {
AddEdgeInternal(node, it->second, item, other_item);
}
}
}
for (auto it=node->input_connections().cbegin(); it!=node->input_connections().cend(); it++) {
foreach (auto child, childItems()) {
if (NodeViewItem *other_item = dynamic_cast<NodeViewItem*>(child)) {
if (it->second == other_item->GetNode()) {
AddEdgeInternal(it->second, it->first, other_item, item);
}
if (!it->first.IsHidden()) {
if (NodeViewItem *other_item = item_map_.value(it->second)) {
AddEdgeInternal(it->second, it->first, other_item, item);
}
}
}
UpdateRect();
}
void NodeViewContext::SetChildPosition(Node *node, const QPointF &pos)
{
item_map_.value(node)->SetNodePosition(pos);
}
void NodeViewContext::RemoveChild(Node *node)
{
disconnect(node, &Node::InputConnected, this, &NodeViewContext::ChildInputConnected);
disconnect(node, &Node::InputDisconnected, this, &NodeViewContext::ChildInputDisconnected);
delete item_map_.take(node);
}
void NodeViewContext::ChildInputConnected(Node *output, const NodeInput &input)
{
// Add edge
if (!input.IsHidden()) {
if (NodeViewItem* output_item = item_map_.value(output)) {
AddEdgeInternal(output, input, output_item, item_map_.value(input.node()));
}
}
}
bool NodeViewContext::ChildInputDisconnected(Node *output, const NodeInput &input)
{
// Remove edge
for (int i=0; i<edges_.size(); i++) {
if (edges_.at(i)->output() == output && edges_.at(i)->input() == input) {
delete edges_.at(i);
edges_.removeAt(i);
return true;
}
}
return false;
}
qreal GetTextOffset(const QFontMetricsF &fm)
@@ -105,23 +137,19 @@ void NodeViewContext::UpdateRect()
rect.adjust(-pad, - lbl_offset*2 - fm.height() - pad, pad, pad);
setRect(rect);
last_titlebar_height_ = rect.y() + (cbr.y() - rect.y());
last_titlebar_height_ = rect.y() + (cbr.y() - rect.y()) - pad;
}
void NodeViewContext::SetFlowDirection(NodeViewCommon::FlowDirection dir)
{
flow_dir_ = dir;
foreach (auto child, childItems()) {
if (NodeViewItem *item = dynamic_cast<NodeViewItem*>(child)) {
item->SetFlowDirection(dir);
}
foreach (NodeViewItem *item, item_map_) {
item->SetFlowDirection(dir);
}
foreach (auto child, childItems()) {
if (NodeViewEdge *edge = dynamic_cast<NodeViewEdge*>(child)) {
edge->SetFlowDirection(dir);
}
foreach (NodeViewEdge *edge, edges_) {
edge->SetFlowDirection(dir);
}
}
@@ -129,29 +157,40 @@ void NodeViewContext::SetCurvedEdges(bool e)
{
curved_edges_ = e;
const QList<QGraphicsItem*> &children = childItems();
foreach (auto child, children) {
if (NodeViewEdge *edge = dynamic_cast<NodeViewEdge*>(child)) {
edge->SetCurved(e);
}
foreach (NodeViewEdge *edge, edges_) {
edge->SetCurved(e);
}
}
void NodeViewContext::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget)
{
QPen p = pen();
// Set pen and brush
Color color = context_->color();
QColor c = color.toQColor();
QPen pen(c, 2);
if (option->state & QStyle::State_Selected) {
p.setStyle(Qt::DotLine);
pen.setStyle(Qt::DotLine);
}
painter->setPen(pen);
painter->setPen(p);
painter->setBrush(brush());
QColor bg = c;
bg.setAlpha(128);
painter->setBrush(bg);
// Draw semi-transparent rect for whole item
int rounded = painter->fontMetrics().height();
painter->drawRoundedRect(rect(), rounded, rounded);
painter->setPen(widget->palette().text().color());
// Draw solid background for titlebar
QRectF titlebar_rect = rect();
titlebar_rect.setHeight(last_titlebar_height_ - rect().top());
painter->setClipRect(titlebar_rect);
painter->setBrush(c);
painter->drawRoundedRect(rect(), rounded, rounded);
painter->setClipping(false);
// Draw titlebar text
painter->setPen(ColorCoding::GetUISelectorColor(color));
int offset = GetTextOffset(painter->fontMetrics());
@@ -182,8 +221,7 @@ NodeViewEdge* NodeViewContext::AddEdgeInternal(Node *output, const NodeInput& in
edge_ui->SetFlowDirection(flow_dir_);
edge_ui->SetCurved(curved_edges_);
from->AddEdge(edge_ui);
to->AddEdge(edge_ui);
edges_.append(edge_ui);
return edge_ui;
}