graph/example/visitor.cpp
Jeremy Siek 60a5e33e58 cycle_edge -> non_tree_edge
[SVN r8804]
2001-01-28 02:34:55 +00:00

123 lines
3.8 KiB
C++

//=======================================================================
// Copyright 1997, 1998, 1999, 2000 University of Notre Dame.
// Authors: Andrew Lumsdaine, Lie-Quan Lee, Jeremy G. Siek
//
// This file is part of the Boost Graph Library
//
// You should have received a copy of the License Agreement for the
// Boost Graph Library along with the software; see the file LICENSE.
// If not, contact Office of Research, University of Notre Dame, Notre
// Dame, IN 46556.
//
// Permission to modify the code and to distribute modified code is
// granted, provided the text of this NOTICE is retained, a notice that
// the code was modified is included with the above COPYRIGHT NOTICE and
// with the COPYRIGHT NOTICE in the LICENSE file, and that the LICENSE
// file is distributed with the modified code.
//
// LICENSOR MAKES NO REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED.
// By way of example, but not limitation, Licensor MAKES NO
// REPRESENTATIONS OR WARRANTIES OF MERCHANTABILITY OR FITNESS FOR ANY
// PARTICULAR PURPOSE OR THAT THE USE OF THE LICENSED SOFTWARE COMPONENTS
// OR DOCUMENTATION WILL NOT INFRINGE ANY PATENTS, COPYRIGHTS, TRADEMARKS
// OR OTHER RIGHTS.
//=======================================================================
//
// Sample output
//
// DFS categorized directed graph
// tree: 0 --> 2
// tree: 2 --> 1
// back: 1 --> 1
// tree: 1 --> 3
// back: 3 --> 1
// tree: 3 --> 4
// back: 4 --> 0
// back: 4 --> 1
// forward or cross: 2 --> 3
// BFS categorized directed graph
// tree: 0 --> 2
// tree: 2 --> 1
// tree: 2 --> 3
// cycle: 1 --> 1
// cycle: 1 --> 3
// cycle: 3 --> 1
// tree: 3 --> 4
// cycle: 4 --> 0
// cycle: 4 --> 1
#include <boost/config.hpp>
#include <iostream>
#include <vector>
#include <algorithm>
#include <utility>
#include <string>
#include <boost/graph/visitors.hpp>
#include <boost/graph/adjacency_list.hpp>
#include <boost/graph/breadth_first_search.hpp>
#include <boost/graph/depth_first_search.hpp>
using namespace boost;
using namespace std;
template <class Tag>
struct edge_printer : public base_visitor<edge_printer<Tag> > {
typedef Tag event_filter;
edge_printer(std::string edge_t) : m_edge_type(edge_t) { }
template <class Edge, class Graph>
void operator()(Edge e, Graph& G) {
std::cout << m_edge_type << ": " << source(e, G)
<< " --> " << target(e, G) << std::endl;
}
std::string m_edge_type;
};
template <class Tag>
edge_printer<Tag> print_edge(std::string type, Tag) {
return edge_printer<Tag>(type);
}
int
main(int, char*[])
{
using namespace boost;
typedef adjacency_list<> Graph;
typedef std::pair<int,int> E;
E edges[] = { E(0, 2),
E(1, 1), E(1, 3),
E(2, 1), E(2, 3),
E(3, 1), E(3, 4),
E(4, 0), E(4, 1) };
Graph G(5, edges, edges + sizeof(edges)/sizeof(E));
typedef boost::graph_traits<Graph>::vertex_descriptor Vertex;
typedef boost::graph_traits<Graph>::vertices_size_type size_type;
boost::property_map<Graph, vertex_index_t>::type
vertex_id = get(vertex_index, G);
std::vector<default_color_type> c(num_vertices(G));
std::vector<size_type> d(num_vertices(G));
std::vector<size_type> f(num_vertices(G));
cout << "DFS categorized directed graph" << endl;
depth_first_search(G, make_dfs_visitor(
std::make_pair(print_edge("tree", on_tree_edge()),
std::make_pair(print_edge("back", on_back_edge()),
print_edge("forward or cross", on_forward_or_cross_edge())
))),
make_iterator_property_map(c.begin(), vertex_id, c[0]));
cout << endl << "BFS categorized directed graph" << endl;
boost::breadth_first_search
(G, vertex(0, G), make_bfs_visitor(
std::make_pair(print_edge("tree", on_tree_edge()),
print_edge("cycle", on_non_tree_edge()))),
make_iterator_property_map(c.begin(), vertex_id, c[0]));
return 0;
}