void GraphGadget::addConnectionGadget( Gaffer::Plug *dstPlug )
{
Gaffer::Plug *srcPlug = dstPlug->getInput<Gaffer::Plug>();
if( !srcPlug )
{
// there is no connection
return;
}
Gaffer::Node *dstNode = dstPlug->node();
NodeGadget *dstNodeGadget = findNodeGadget( dstNode );
if( !dstNodeGadget )
{
return;
}
Nodule *dstNodule = dstNodeGadget->nodule( dstPlug );
if( !dstNodule )
{
// the destination connection point is not represented in the graph
return;
}
Gaffer::Node *srcNode = srcPlug->node();
if( srcNode == dstNode )
{
// we don't want to visualise connections between plugs
// on the same node.
return;
}
// find the input nodule for the connection if there is one
NodeGadget *srcNodeGadget = findNodeGadget( srcNode );
Nodule *srcNodule = 0;
if( srcNodeGadget )
{
srcNodule = srcNodeGadget->nodule( srcPlug );
}
ConnectionGadgetPtr connection = new ConnectionGadget( srcNodule, dstNodule );
updateConnectionGadgetMinimisation( connection.get() );
addChild( connection );
m_connectionGadgets[dstPlug] = connection.get();
}
void GraphGadget::updateDragReconnectCandidate( const DragDropEvent &event )
{
m_dragReconnectCandidate = 0;
m_dragReconnectSrcNodule = 0;
m_dragReconnectDstNodule = 0;
if ( m_scriptNode->selection()->size() != 1 )
{
return;
}
// make sure the connection applies to this node.
Gaffer::Node *node = IECore::runTimeCast<Gaffer::Node>( m_scriptNode->selection()->member( 0 ) );
NodeGadget *selNodeGadget = nodeGadget( node );
if ( !node || !selNodeGadget )
{
return;
}
m_dragReconnectCandidate = reconnectionGadgetAt( selNodeGadget, event.line );
if ( !m_dragReconnectCandidate )
{
return;
}
// we don't want to reconnect the selected node to itself
Gaffer::Plug *srcPlug = m_dragReconnectCandidate->srcNodule()->plug();
Gaffer::Plug *dstPlug = m_dragReconnectCandidate->dstNodule()->plug();
if ( srcPlug->node() == node || dstPlug->node() == node )
{
m_dragReconnectCandidate = 0;
return;
}
Gaffer::DependencyNode *depNode = IECore::runTimeCast<Gaffer::DependencyNode>( node );
for ( Gaffer::RecursiveOutputPlugIterator cIt( node ); cIt != cIt.end(); ++cIt )
{
// must be compatible
Gaffer::Plug *p = cIt->get();
if ( !dstPlug->acceptsInput( p ) )
{
continue;
}
// must have a nodule
m_dragReconnectSrcNodule = selNodeGadget->nodule( p );
if ( !m_dragReconnectSrcNodule )
{
continue;
}
// must not be connected to a nodule
for ( Gaffer::Plug::OutputContainer::const_iterator oIt = p->outputs().begin(); oIt != p->outputs().end(); ++oIt )
{
NodeGadget *oNodeGadget = nodeGadget( (*oIt)->node() );
if ( oNodeGadget && oNodeGadget->nodule( *oIt ) )
{
m_dragReconnectSrcNodule = 0;
break;
}
}
if ( !m_dragReconnectSrcNodule )
{
continue;
}
if ( !depNode )
{
// found the best option
break;
}
// make sure its corresponding input is also free
if ( Gaffer::Plug *in = depNode->correspondingInput( p ) )
{
if ( in->getInput<Gaffer::Plug>() )
{
m_dragReconnectSrcNodule = 0;
continue;
}
m_dragReconnectDstNodule = selNodeGadget->nodule( in );
if ( m_dragReconnectDstNodule )
{
break;
}
}
}
// check input plugs on non-dependencyNodes
if ( !depNode && !m_dragReconnectDstNodule )
{
for ( Gaffer::RecursiveInputPlugIterator cIt( node ); cIt != cIt.end(); ++cIt )
{
Gaffer::Plug *p = cIt->get();
if ( !p->getInput<Gaffer::Plug>() && p->acceptsInput( srcPlug ) )
{
m_dragReconnectDstNodule = selNodeGadget->nodule( p );
if ( m_dragReconnectDstNodule )
{
break;
}
}
}
}
if ( !m_dragReconnectSrcNodule && !m_dragReconnectDstNodule )
{
m_dragReconnectCandidate = 0;
}
}