size_t GraphGadget::connectionGadgets( const Gaffer::Plug *plug, std::vector<ConnectionGadget *> &connections, const Gaffer::Set *excludedNodes )
{
if( plug->direction() == Gaffer::Plug::In )
{
const Gaffer::Plug *input = plug->getInput<Gaffer::Plug>();
if( input )
{
if( !excludedNodes || !excludedNodes->contains( input->node() ) )
{
ConnectionGadget *connection = connectionGadget( plug );
if( connection && connection->srcNodule() )
{
connections.push_back( connection );
}
}
}
}
else
{
const Gaffer::Plug::OutputContainer &outputs = plug->outputs();
for( Gaffer::Plug::OutputContainer::const_iterator it = outputs.begin(), eIt = outputs.end(); it != eIt; ++it )
{
if( excludedNodes && excludedNodes->contains( (*it)->node() ) )
{
continue;
}
ConnectionGadget *connection = connectionGadget( *it );
if( connection && connection->srcNodule() )
{
connections.push_back( connection );
}
}
}
return connections.size();
}
void GraphGadget::addConnectionGadgets( Gaffer::GraphComponent *plugParent )
{
/// \todo I think this could be faster if we could iterate over just the nodules rather than all the plugs. Perhaps
/// we could make it easy to recurse over all the Nodules of a NodeGadget if we had a RecursiveChildIterator for GraphComponents?
Gaffer::Node *node = plugParent->isInstanceOf( Gaffer::Node::staticTypeId() ) ? static_cast<Gaffer::Node *>( plugParent ) : plugParent->ancestor<Gaffer::Node>();
NodeGadget *nodeGadget = findNodeGadget( node );
if( !nodeGadget )
{
return;
}
for( Gaffer::PlugIterator pIt( plugParent->children().begin(), plugParent->children().end() ); pIt!=pIt.end(); pIt++ )
{
if( (*pIt)->direction() == Gaffer::Plug::In )
{
// add connections for input plugs
if( !findConnectionGadget( pIt->get() ) )
{
addConnectionGadget( pIt->get() );
}
}
else
{
// reconnect any old output connections which may have been dangling
Nodule *srcNodule = nodeGadget->nodule( *pIt );
if( srcNodule )
{
for( Gaffer::Plug::OutputContainer::const_iterator oIt( (*pIt)->outputs().begin() ); oIt!= (*pIt)->outputs().end(); oIt++ )
{
ConnectionGadget *connection = findConnectionGadget( *oIt );
if( connection && !connection->srcNodule() )
{
assert( connection->dstNodule()->plug()->getInput<Gaffer::Plug>() == *pIt );
connection->setNodules( srcNodule, connection->dstNodule() );
}
}
}
}
if( (*pIt)->isInstanceOf( Gaffer::CompoundPlug::staticTypeId() ) )
{
addConnectionGadgets( pIt->get() );
}
}
}
static NodulePtr srcNodule( ConnectionGadget &c )
{
return c.srcNodule();
}
