mod_base_visitors.hh

Go to the documentation of this file.

/*
 * LEGAL:   COPYRIGHT (C) 2007 JIM E. BROOKS
 *          THIS SOURCE CODE IS RELEASED UNDER THE TERMS
 *          OF THE GNU GENERAL PUBLIC LICENSE VERSION 2 (GPL 2).
 *****************************************************************************/

#if COMPILING_MODULE
#ifndef MOD_BASE_VISITORS_HH
#define MOD_BASE_VISITORS_HH 1

namespace mod_base {

class VisitorFindPartitionNode : public Visitor
{
public:
    VisitorFindPartitionNode( Graph& graph, const string& partitionName )
    :   Visitor(graph),
        mPartitionName(partitionName),
        mPartitionNode(NULL_NODE)
    {
    }

    void Visit( PartitionNode& partitionNode )
    {
        // Does this PartitionNode have the sought name?
        if ( partitionNode.GetName() == mPartitionName )
        {
            // Found match, stop traversing.
            mPartitionNode = &partitionNode;
            Stop();
        }
    }

    PartitionNode::Ptr GetPartitionNode( void )
    {
        return mPartitionNode;
    }

private:
    const string        mPartitionName;
    PartitionNode::Ptr  mPartitionNode;
};

template<typename NODE>
class VisitorFindNode : public Visitor
{
public:
    VisitorFindNode( Graph& graph, const string& partitionName )
    :   Visitor(graph),
        mNode(NULL_NODE)
    {
        // Find partition node (visitor-within-a-visitor).
        VisitorFindPartitionNode visitorFindPartitionNode( graph, partitionName );
        graph.Traverse( visitorFindPartitionNode );
        const PartitionNode::Ptr partitionNode = visitorFindPartitionNode.GetPartitionNode();

        // Was partition found?
        if ( partitionNode != NULL_NODE )
        {
            // Traversal will start at this PartitionNode.
            SetRootNode( partitionNode );
        }
        else
        {
            // Don't traverse.
            Stop();
        }
    }

    void Visit( NODE& node )
    {
        // Stop at the first type of node.
        mNode = &node;
        Stop();
    }

    typename NODE::Ptr GetNode( void )
    {
        return mNode;
    }

private:
    typename NODE::Ptr  mNode;  // matching node
};

} // namespace mod_base

#endif // MOD_BASE_VISITORS_HH
#endif // COMPILING_MODULE

Palomino 3D Engine documents generated by doxygen 1.5.3 on Fri Nov 23 11:26:12 2007