void draw_edges_visitor::visit( node_t *n)
{
Imath::V2f p0( generic_output_location( n));
input_location_node_visitor visitor;
for( int i = 0; i < n->num_outputs(); ++i)
{
node_t *dst = boost::get<0>( n->output_plug().connections()[i]);
visitor.index = boost::get<2>( n->output_plug().connections()[i]);
dst->accept( visitor);
view_.draw_bezier_edge( painter_, p0, visitor.loc);
}
}
void pick_edge_visitor::visit( node_t *n)
{
Imath::V2f p0( generic_output_location( n));
input_location_node_visitor visitor;
for( int i = 0; i < n->num_outputs(); ++i)
{
node_t *target = boost::get<0>( n->output_plug().connections()[i]);
visitor.index = boost::get<2>( n->output_plug().connections()[i]);
target->accept( visitor);
if( view_.pick_bezier_edge( p0, visitor.loc, p_))
{
src = n;
dst = target;
port = visitor.index;
return;
}
}
}
void pick_generic_node( node_t *n, const Imath::V2f& p, const composition_view_t& view, pick_result_t& result)
{
result.node = 0;
result.component = pick_result_t::no_pick;
result.plug_num = -1;
if( ( p.x < n->location().x) || (p.x > n->location().x + generic_node_width( n)))
return;
if( ( p.y < n->location().y - 6.0) || ( p.y > n->location().y + generic_node_height() + 5))
return;
result.node = n;
result.component = pick_result_t::body_picked;
Imath::V2f q( generic_output_location( n));
Imath::V2i pi( view.world_to_screen( p));
Imath::V2i qi( view.world_to_screen( q));
if( ( pi - qi).length2() < 25)
{
result.component = pick_result_t::output_picked;
result.plug_num = 0;
return;
}
for( unsigned int i=0;i<n->num_inputs();++i)
{
q = generic_input_location( n, i);
qi = view.world_to_screen( q);
if( ( pi - qi).length2() < 25)
{
result.component = pick_result_t::input_picked;
result.plug_num = i;
return;
}
}
}
void draw_generic_node( QPainter& painter, const node_t *n)
{
Imath::V2f p = n->location();
QPen pen;
QBrush brush;
brush.setStyle( Qt::SolidPattern);
brush.setColor( palette_t::instance().qcolor( "node_shadow"));
painter.setPen( Qt::NoPen);
painter.setBrush( brush);
painter.drawRoundedRect( QRectF( p.x + shadow_offset, p.y + shadow_offset, generic_node_width( n), generic_node_height()), radius, radius);
for( unsigned int i=0;i<n->num_inputs();++i)
{
brush.setColor( QColor( n->input_plugs()[i].color().x, n->input_plugs()[i].color().y, n->input_plugs()[i].color().z));
painter.setBrush( brush);
Imath::V2f q = generic_input_location( n, i);
painter.drawRect( q.x - 3, q.y - 3, 6, 6);
}
if( n->has_output_plug())
{
// TODO: use the output plug's color here.
brush.setColor( palette_t::instance().qcolor( "out plug"));
painter.setBrush( brush);
Imath::V2f q = generic_output_location( n);
painter.drawRect( q.x - 3, q.y - 3, 6, 6);
}
if( n->selected())
brush.setColor( palette_t::instance().qcolor( "node_picked_bg"));
else
{
if( n->plugin_error())
brush.setColor( palette_t::instance().qcolor( "node_error_bg"));
else
brush.setColor( palette_t::instance().qcolor( "node_bg"));
}
painter.setBrush( brush);
painter.drawRoundedRect( QRectF( p.x, p.y, generic_node_width( n), generic_node_height()), radius, radius);
painter.setBrush( Qt::NoBrush);
if( app().ui()->active_node() == n)
pen.setColor( palette_t::instance().qcolor( "node_active_out"));
else
{
if( app().ui()->context_node() == n)
pen.setColor( palette_t::instance().qcolor( "node_context_out"));
else
pen.setColor( palette_t::instance().qcolor( "text"));
}
painter.setPen( pen);
painter.drawRoundedRect( QRectF( p.x, p.y, generic_node_width( n), generic_node_height()), radius, radius);
pen.setColor( palette_t::instance().qcolor( "text"));
painter.setPen( pen);
painter.drawText( QRectF( p.x, p.y + 3, generic_node_width( n), generic_node_height() - 3), Qt::AlignCenter, QString( n->name().c_str()));
if( n->ignored())
{
pen.setColor( palette_t::instance().qcolor( "box_pick"));
pen.setWidth(2);
painter.setPen( pen);
painter.drawLine( QPointF( p.x - 3, p.y - 2), QPointF( p.x + generic_node_width( n) + 3, p.y + generic_node_height() + 2));
}
}
void output_location_node_visitor::visit( node_t *n)
{
RAMEN_ASSERT( index == 0);
loc = generic_output_location( n);
}