make_connected.hpp 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
  1. //=======================================================================
  2. // Copyright 2007 Aaron Windsor
  3. //
  4. // Distributed under the Boost Software License, Version 1.0. (See
  5. // accompanying file LICENSE_1_0.txt or copy at
  6. // http://www.boost.org/LICENSE_1_0.txt)
  7. //=======================================================================
  8. #ifndef __MAKE_CONNECTED_HPP__
  9. #define __MAKE_CONNECTED_HPP__
  10. #include <boost/config.hpp>
  11. #include <boost/next_prior.hpp>
  12. #include <boost/tuple/tuple.hpp> //for tie
  13. #include <boost/graph/connected_components.hpp>
  14. #include <boost/property_map/property_map.hpp>
  15. #include <vector>
  16. #include <boost/graph/planar_detail/add_edge_visitors.hpp>
  17. #include <boost/graph/planar_detail/bucket_sort.hpp>
  18. namespace boost
  19. {
  20. template <typename Graph,
  21. typename VertexIndexMap,
  22. typename AddEdgeVisitor
  23. >
  24. void make_connected(Graph& g, VertexIndexMap vm, AddEdgeVisitor& vis)
  25. {
  26. typedef typename graph_traits<Graph>::vertex_iterator vertex_iterator_t;
  27. typedef typename graph_traits<Graph>::vertex_descriptor vertex_t;
  28. typedef typename graph_traits<Graph>::vertices_size_type v_size_t;
  29. typedef iterator_property_map< typename std::vector<v_size_t>::iterator,
  30. VertexIndexMap
  31. > vertex_to_v_size_map_t;
  32. std::vector<v_size_t> component_vector(num_vertices(g));
  33. vertex_to_v_size_map_t component(component_vector.begin(), vm);
  34. std::vector<vertex_t> vertices_by_component(num_vertices(g));
  35. v_size_t num_components = connected_components(g, component);
  36. if (num_components < 2)
  37. return;
  38. vertex_iterator_t vi, vi_end;
  39. boost::tie(vi,vi_end) = vertices(g);
  40. std::copy(vi, vi_end, vertices_by_component.begin());
  41. bucket_sort(vertices_by_component.begin(),
  42. vertices_by_component.end(),
  43. component,
  44. num_components
  45. );
  46. typedef typename std::vector<vertex_t>::iterator vec_of_vertices_itr_t;
  47. vec_of_vertices_itr_t ci_end = vertices_by_component.end();
  48. vec_of_vertices_itr_t ci_prev = vertices_by_component.begin();
  49. if (ci_prev == ci_end)
  50. return;
  51. for(vec_of_vertices_itr_t ci = boost::next(ci_prev);
  52. ci != ci_end; ci_prev = ci, ++ci
  53. )
  54. {
  55. if (component[*ci_prev] != component[*ci])
  56. vis.visit_vertex_pair(*ci_prev, *ci, g);
  57. }
  58. }
  59. template <typename Graph, typename VertexIndexMap>
  60. inline void make_connected(Graph& g, VertexIndexMap vm)
  61. {
  62. default_add_edge_visitor vis;
  63. make_connected(g, vm, vis);
  64. }
  65. template <typename Graph>
  66. inline void make_connected(Graph& g)
  67. {
  68. make_connected(g, get(vertex_index,g));
  69. }
  70. } // namespace boost
  71. #endif //__MAKE_CONNECTED_HPP__