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(); }