page_rank.hpp 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160
  1. // Copyright 2004-5 The Trustees of Indiana University.
  2. // Copyright 2002 Brad King and Douglas Gregor
  3. // Distributed under the Boost Software License, Version 1.0.
  4. // (See accompanying file LICENSE_1_0.txt or copy at
  5. // http://www.boost.org/LICENSE_1_0.txt)
  6. // Authors: Douglas Gregor
  7. // Andrew Lumsdaine
  8. #ifndef BOOST_GRAPH_PAGE_RANK_HPP
  9. #define BOOST_GRAPH_PAGE_RANK_HPP
  10. #include <boost/property_map/property_map.hpp>
  11. #include <boost/graph/graph_traits.hpp>
  12. #include <boost/graph/properties.hpp>
  13. #include <boost/graph/iteration_macros.hpp>
  14. #include <boost/graph/overloading.hpp>
  15. #include <boost/graph/detail/mpi_include.hpp>
  16. #include <vector>
  17. namespace boost { namespace graph {
  18. struct n_iterations
  19. {
  20. explicit n_iterations(std::size_t n) : n(n) { }
  21. template<typename RankMap, typename Graph>
  22. bool
  23. operator()(const RankMap&, const Graph&)
  24. {
  25. return n-- == 0;
  26. }
  27. private:
  28. std::size_t n;
  29. };
  30. namespace detail {
  31. template<typename Graph, typename RankMap, typename RankMap2>
  32. void page_rank_step(const Graph& g, RankMap from_rank, RankMap2 to_rank,
  33. typename property_traits<RankMap>::value_type damping,
  34. incidence_graph_tag)
  35. {
  36. typedef typename property_traits<RankMap>::value_type rank_type;
  37. // Set new rank maps
  38. BGL_FORALL_VERTICES_T(v, g, Graph) put(to_rank, v, rank_type(1 - damping));
  39. BGL_FORALL_VERTICES_T(u, g, Graph) {
  40. rank_type u_rank_out = damping * get(from_rank, u) / out_degree(u, g);
  41. BGL_FORALL_ADJ_T(u, v, g, Graph)
  42. put(to_rank, v, get(to_rank, v) + u_rank_out);
  43. }
  44. }
  45. template<typename Graph, typename RankMap, typename RankMap2>
  46. void page_rank_step(const Graph& g, RankMap from_rank, RankMap2 to_rank,
  47. typename property_traits<RankMap>::value_type damping,
  48. bidirectional_graph_tag)
  49. {
  50. typedef typename property_traits<RankMap>::value_type damping_type;
  51. BGL_FORALL_VERTICES_T(v, g, Graph) {
  52. typename property_traits<RankMap>::value_type rank(0);
  53. BGL_FORALL_INEDGES_T(v, e, g, Graph)
  54. rank += get(from_rank, source(e, g)) / out_degree(source(e, g), g);
  55. put(to_rank, v, (damping_type(1) - damping) + damping * rank);
  56. }
  57. }
  58. } // end namespace detail
  59. template<typename Graph, typename RankMap, typename Done, typename RankMap2>
  60. void
  61. page_rank(const Graph& g, RankMap rank_map, Done done,
  62. typename property_traits<RankMap>::value_type damping,
  63. typename graph_traits<Graph>::vertices_size_type n,
  64. RankMap2 rank_map2
  65. BOOST_GRAPH_ENABLE_IF_MODELS_PARM(Graph, vertex_list_graph_tag))
  66. {
  67. typedef typename property_traits<RankMap>::value_type rank_type;
  68. rank_type initial_rank = rank_type(rank_type(1) / n);
  69. BGL_FORALL_VERTICES_T(v, g, Graph) put(rank_map, v, initial_rank);
  70. bool to_map_2 = true;
  71. while ((to_map_2 && !done(rank_map, g)) ||
  72. (!to_map_2 && !done(rank_map2, g))) {
  73. typedef typename graph_traits<Graph>::traversal_category category;
  74. if (to_map_2) {
  75. detail::page_rank_step(g, rank_map, rank_map2, damping, category());
  76. } else {
  77. detail::page_rank_step(g, rank_map2, rank_map, damping, category());
  78. }
  79. to_map_2 = !to_map_2;
  80. }
  81. if (!to_map_2) {
  82. BGL_FORALL_VERTICES_T(v, g, Graph) put(rank_map, v, get(rank_map2, v));
  83. }
  84. }
  85. template<typename Graph, typename RankMap, typename Done>
  86. void
  87. page_rank(const Graph& g, RankMap rank_map, Done done,
  88. typename property_traits<RankMap>::value_type damping,
  89. typename graph_traits<Graph>::vertices_size_type n)
  90. {
  91. typedef typename property_traits<RankMap>::value_type rank_type;
  92. std::vector<rank_type> ranks2(num_vertices(g));
  93. page_rank(g, rank_map, done, damping, n,
  94. make_iterator_property_map(ranks2.begin(), get(vertex_index, g)));
  95. }
  96. template<typename Graph, typename RankMap, typename Done>
  97. inline void
  98. page_rank(const Graph& g, RankMap rank_map, Done done,
  99. typename property_traits<RankMap>::value_type damping = 0.85)
  100. {
  101. page_rank(g, rank_map, done, damping, num_vertices(g));
  102. }
  103. template<typename Graph, typename RankMap>
  104. inline void
  105. page_rank(const Graph& g, RankMap rank_map)
  106. {
  107. page_rank(g, rank_map, n_iterations(20));
  108. }
  109. // TBD: this could be _much_ more efficient, using a queue to store
  110. // the vertices that should be reprocessed and keeping track of which
  111. // vertices are in the queue with a property map. Baah, this only
  112. // applies when we have a bidirectional graph.
  113. template<typename MutableGraph>
  114. void
  115. remove_dangling_links(MutableGraph& g
  116. BOOST_GRAPH_ENABLE_IF_MODELS_PARM(MutableGraph,
  117. vertex_list_graph_tag))
  118. {
  119. typename graph_traits<MutableGraph>::vertices_size_type old_n;
  120. do {
  121. old_n = num_vertices(g);
  122. typename graph_traits<MutableGraph>::vertex_iterator vi, vi_end;
  123. for (boost::tie(vi, vi_end) = vertices(g); vi != vi_end; /* in loop */) {
  124. typename graph_traits<MutableGraph>::vertex_descriptor v = *vi++;
  125. if (out_degree(v, g) == 0) {
  126. clear_vertex(v, g);
  127. remove_vertex(v, g);
  128. }
  129. }
  130. } while (num_vertices(g) < old_n);
  131. }
  132. } } // end namespace boost::graph
  133. #include BOOST_GRAPH_MPI_INCLUDE(<boost/graph/distributed/page_rank.hpp>)
  134. #endif // BOOST_GRAPH_PAGE_RANK_HPP