123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616 |
- #ifndef BOOST_GRAPH_BRANDES_BETWEENNESS_CENTRALITY_HPP
- #define BOOST_GRAPH_BRANDES_BETWEENNESS_CENTRALITY_HPP
- #include <stack>
- #include <vector>
- #include <boost/graph/overloading.hpp>
- #include <boost/graph/dijkstra_shortest_paths.hpp>
- #include <boost/graph/breadth_first_search.hpp>
- #include <boost/graph/relax.hpp>
- #include <boost/graph/graph_traits.hpp>
- #include <boost/tuple/tuple.hpp>
- #include <boost/type_traits/is_convertible.hpp>
- #include <boost/type_traits/is_same.hpp>
- #include <boost/mpl/if.hpp>
- #include <boost/property_map/property_map.hpp>
- #include <boost/graph/named_function_params.hpp>
- #include <algorithm>
- namespace boost {
- namespace detail { namespace graph {
-
- template<typename Graph, typename WeightMap, typename IncomingMap,
- typename DistanceMap, typename PathCountMap>
- struct brandes_dijkstra_visitor : public bfs_visitor<>
- {
- typedef typename graph_traits<Graph>::vertex_descriptor vertex_descriptor;
- typedef typename graph_traits<Graph>::edge_descriptor edge_descriptor;
- brandes_dijkstra_visitor(std::stack<vertex_descriptor>& ordered_vertices,
- WeightMap weight,
- IncomingMap incoming,
- DistanceMap distance,
- PathCountMap path_count)
- : ordered_vertices(ordered_vertices), weight(weight),
- incoming(incoming), distance(distance),
- path_count(path_count)
- { }
-
- void edge_relaxed(edge_descriptor e, const Graph& g)
- {
- vertex_descriptor v = source(e, g), w = target(e, g);
- incoming[w].clear();
- incoming[w].push_back(e);
- put(path_count, w, get(path_count, v));
- }
-
- void edge_not_relaxed(edge_descriptor e, const Graph& g)
- {
- typedef typename property_traits<WeightMap>::value_type weight_type;
- typedef typename property_traits<DistanceMap>::value_type distance_type;
- vertex_descriptor v = source(e, g), w = target(e, g);
- distance_type d_v = get(distance, v), d_w = get(distance, w);
- weight_type w_e = get(weight, e);
- closed_plus<distance_type> combine;
- if (d_w == combine(d_v, w_e)) {
- put(path_count, w, get(path_count, w) + get(path_count, v));
- incoming[w].push_back(e);
- }
- }
-
- void examine_vertex(vertex_descriptor w, const Graph&)
- {
- ordered_vertices.push(w);
- }
- private:
- std::stack<vertex_descriptor>& ordered_vertices;
- WeightMap weight;
- IncomingMap incoming;
- DistanceMap distance;
- PathCountMap path_count;
- };
-
- template<typename WeightMap>
- struct brandes_dijkstra_shortest_paths
- {
- brandes_dijkstra_shortest_paths(WeightMap weight_map)
- : weight_map(weight_map) { }
- template<typename Graph, typename IncomingMap, typename DistanceMap,
- typename PathCountMap, typename VertexIndexMap>
- void
- operator()(Graph& g,
- typename graph_traits<Graph>::vertex_descriptor s,
- std::stack<typename graph_traits<Graph>::vertex_descriptor>& ov,
- IncomingMap incoming,
- DistanceMap distance,
- PathCountMap path_count,
- VertexIndexMap vertex_index)
- {
- typedef brandes_dijkstra_visitor<Graph, WeightMap, IncomingMap,
- DistanceMap, PathCountMap> visitor_type;
- visitor_type visitor(ov, weight_map, incoming, distance, path_count);
- dijkstra_shortest_paths(g, s,
- boost::weight_map(weight_map)
- .vertex_index_map(vertex_index)
- .distance_map(distance)
- .visitor(visitor));
- }
- private:
- WeightMap weight_map;
- };
-
- struct brandes_unweighted_shortest_paths
- {
-
- template<typename Graph, typename IncomingMap, typename DistanceMap,
- typename PathCountMap>
- struct visitor_type : public bfs_visitor<>
- {
- typedef typename graph_traits<Graph>::edge_descriptor edge_descriptor;
- typedef typename graph_traits<Graph>::vertex_descriptor
- vertex_descriptor;
-
- visitor_type(IncomingMap incoming, DistanceMap distance,
- PathCountMap path_count,
- std::stack<vertex_descriptor>& ordered_vertices)
- : incoming(incoming), distance(distance),
- path_count(path_count), ordered_vertices(ordered_vertices) { }
-
- void examine_vertex(vertex_descriptor v, Graph&)
- {
- ordered_vertices.push(v);
- }
-
- void tree_edge(edge_descriptor e, Graph& g)
- {
- vertex_descriptor v = source(e, g);
- vertex_descriptor w = target(e, g);
- put(distance, w, get(distance, v) + 1);
-
- put(path_count, w, get(path_count, v));
- incoming[w].push_back(e);
- }
-
- void non_tree_edge(edge_descriptor e, Graph& g)
- {
- vertex_descriptor v = source(e, g);
- vertex_descriptor w = target(e, g);
- if (get(distance, w) == get(distance, v) + 1) {
- put(path_count, w, get(path_count, w) + get(path_count, v));
- incoming[w].push_back(e);
- }
- }
- private:
- IncomingMap incoming;
- DistanceMap distance;
- PathCountMap path_count;
- std::stack<vertex_descriptor>& ordered_vertices;
- };
- template<typename Graph, typename IncomingMap, typename DistanceMap,
- typename PathCountMap, typename VertexIndexMap>
- void
- operator()(Graph& g,
- typename graph_traits<Graph>::vertex_descriptor s,
- std::stack<typename graph_traits<Graph>::vertex_descriptor>& ov,
- IncomingMap incoming,
- DistanceMap distance,
- PathCountMap path_count,
- VertexIndexMap vertex_index)
- {
- typedef typename graph_traits<Graph>::vertex_descriptor
- vertex_descriptor;
- visitor_type<Graph, IncomingMap, DistanceMap, PathCountMap>
- visitor(incoming, distance, path_count, ov);
-
- std::vector<default_color_type>
- colors(num_vertices(g), color_traits<default_color_type>::white());
- boost::queue<vertex_descriptor> Q;
- breadth_first_visit(g, s, Q, visitor,
- make_iterator_property_map(colors.begin(),
- vertex_index));
- }
- };
-
-
- template<typename Iter>
- inline void
- init_centrality_map(std::pair<Iter, Iter>, dummy_property_map) { }
-
-
- template<typename Iter, typename Centrality>
- void
- init_centrality_map(std::pair<Iter, Iter> keys, Centrality centrality_map)
- {
- typedef typename property_traits<Centrality>::value_type
- centrality_type;
- while (keys.first != keys.second) {
- put(centrality_map, *keys.first, centrality_type(0));
- ++keys.first;
- }
- }
-
-
- template<typename Key, typename T>
- inline void
- update_centrality(dummy_property_map, const Key&, const T&) { }
-
- template<typename CentralityMap, typename Key, typename T>
- inline void
- update_centrality(CentralityMap centrality_map, Key k, const T& x)
- { put(centrality_map, k, get(centrality_map, k) + x); }
- template<typename Iter>
- inline void
- divide_centrality_by_two(std::pair<Iter, Iter>, dummy_property_map) {}
- template<typename Iter, typename CentralityMap>
- inline void
- divide_centrality_by_two(std::pair<Iter, Iter> keys,
- CentralityMap centrality_map)
- {
- typename property_traits<CentralityMap>::value_type two(2);
- while (keys.first != keys.second) {
- put(centrality_map, *keys.first, get(centrality_map, *keys.first) / two);
- ++keys.first;
- }
- }
- template<typename Graph, typename CentralityMap, typename EdgeCentralityMap,
- typename IncomingMap, typename DistanceMap,
- typename DependencyMap, typename PathCountMap,
- typename VertexIndexMap, typename ShortestPaths>
- void
- brandes_betweenness_centrality_impl(const Graph& g,
- CentralityMap centrality,
- EdgeCentralityMap edge_centrality_map,
- IncomingMap incoming,
- DistanceMap distance,
- DependencyMap dependency,
- PathCountMap path_count,
- VertexIndexMap vertex_index,
- ShortestPaths shortest_paths)
- {
- typedef typename graph_traits<Graph>::vertex_iterator vertex_iterator;
- typedef typename graph_traits<Graph>::vertex_descriptor vertex_descriptor;
-
- init_centrality_map(vertices(g), centrality);
- init_centrality_map(edges(g), edge_centrality_map);
- std::stack<vertex_descriptor> ordered_vertices;
- vertex_iterator s, s_end;
- for (boost::tie(s, s_end) = vertices(g); s != s_end; ++s) {
-
- vertex_iterator w, w_end;
- for (boost::tie(w, w_end) = vertices(g); w != w_end; ++w) {
- incoming[*w].clear();
- put(path_count, *w, 0);
- put(dependency, *w, 0);
- }
- put(path_count, *s, 1);
-
-
-
-
- shortest_paths(g, *s, ordered_vertices, incoming, distance,
- path_count, vertex_index);
-
- while (!ordered_vertices.empty()) {
- vertex_descriptor w = ordered_vertices.top();
- ordered_vertices.pop();
-
- typedef typename property_traits<IncomingMap>::value_type
- incoming_type;
- typedef typename incoming_type::iterator incoming_iterator;
- typedef typename property_traits<DependencyMap>::value_type
- dependency_type;
-
- for (incoming_iterator vw = incoming[w].begin();
- vw != incoming[w].end(); ++vw) {
- vertex_descriptor v = source(*vw, g);
- dependency_type factor = dependency_type(get(path_count, v))
- / dependency_type(get(path_count, w));
- factor *= (dependency_type(1) + get(dependency, w));
- put(dependency, v, get(dependency, v) + factor);
- update_centrality(edge_centrality_map, *vw, factor);
- }
-
- if (w != *s) {
- update_centrality(centrality, w, get(dependency, w));
- }
- }
- }
- typedef typename graph_traits<Graph>::directed_category directed_category;
- const bool is_undirected =
- is_convertible<directed_category*, undirected_tag*>::value;
- if (is_undirected) {
- divide_centrality_by_two(vertices(g), centrality);
- divide_centrality_by_two(edges(g), edge_centrality_map);
- }
- }
- } }
- template<typename Graph, typename CentralityMap, typename EdgeCentralityMap,
- typename IncomingMap, typename DistanceMap,
- typename DependencyMap, typename PathCountMap,
- typename VertexIndexMap>
- void
- brandes_betweenness_centrality(const Graph& g,
- CentralityMap centrality,
- EdgeCentralityMap edge_centrality_map,
- IncomingMap incoming,
- DistanceMap distance,
- DependencyMap dependency,
- PathCountMap path_count,
- VertexIndexMap vertex_index
- BOOST_GRAPH_ENABLE_IF_MODELS_PARM(Graph,vertex_list_graph_tag))
- {
- detail::graph::brandes_unweighted_shortest_paths shortest_paths;
- detail::graph::brandes_betweenness_centrality_impl(g, centrality,
- edge_centrality_map,
- incoming, distance,
- dependency, path_count,
- vertex_index,
- shortest_paths);
- }
- template<typename Graph, typename CentralityMap, typename EdgeCentralityMap,
- typename IncomingMap, typename DistanceMap,
- typename DependencyMap, typename PathCountMap,
- typename VertexIndexMap, typename WeightMap>
- void
- brandes_betweenness_centrality(const Graph& g,
- CentralityMap centrality,
- EdgeCentralityMap edge_centrality_map,
- IncomingMap incoming,
- DistanceMap distance,
- DependencyMap dependency,
- PathCountMap path_count,
- VertexIndexMap vertex_index,
- WeightMap weight_map
- BOOST_GRAPH_ENABLE_IF_MODELS_PARM(Graph,vertex_list_graph_tag))
- {
- detail::graph::brandes_dijkstra_shortest_paths<WeightMap>
- shortest_paths(weight_map);
- detail::graph::brandes_betweenness_centrality_impl(g, centrality,
- edge_centrality_map,
- incoming, distance,
- dependency, path_count,
- vertex_index,
- shortest_paths);
- }
- namespace detail { namespace graph {
- template<typename Graph, typename CentralityMap, typename EdgeCentralityMap,
- typename WeightMap, typename VertexIndexMap>
- void
- brandes_betweenness_centrality_dispatch2(const Graph& g,
- CentralityMap centrality,
- EdgeCentralityMap edge_centrality_map,
- WeightMap weight_map,
- VertexIndexMap vertex_index)
- {
- typedef typename graph_traits<Graph>::degree_size_type degree_size_type;
- typedef typename graph_traits<Graph>::edge_descriptor edge_descriptor;
- typedef typename mpl::if_c<(is_same<CentralityMap,
- dummy_property_map>::value),
- EdgeCentralityMap,
- CentralityMap>::type a_centrality_map;
- typedef typename property_traits<a_centrality_map>::value_type
- centrality_type;
- typename graph_traits<Graph>::vertices_size_type V = num_vertices(g);
-
- std::vector<std::vector<edge_descriptor> > incoming(V);
- std::vector<centrality_type> distance(V);
- std::vector<centrality_type> dependency(V);
- std::vector<degree_size_type> path_count(V);
- brandes_betweenness_centrality(
- g, centrality, edge_centrality_map,
- make_iterator_property_map(incoming.begin(), vertex_index),
- make_iterator_property_map(distance.begin(), vertex_index),
- make_iterator_property_map(dependency.begin(), vertex_index),
- make_iterator_property_map(path_count.begin(), vertex_index),
- vertex_index,
- weight_map);
- }
-
- template<typename Graph, typename CentralityMap, typename EdgeCentralityMap,
- typename VertexIndexMap>
- void
- brandes_betweenness_centrality_dispatch2(const Graph& g,
- CentralityMap centrality,
- EdgeCentralityMap edge_centrality_map,
- VertexIndexMap vertex_index)
- {
- typedef typename graph_traits<Graph>::degree_size_type degree_size_type;
- typedef typename graph_traits<Graph>::edge_descriptor edge_descriptor;
- typedef typename mpl::if_c<(is_same<CentralityMap,
- dummy_property_map>::value),
- EdgeCentralityMap,
- CentralityMap>::type a_centrality_map;
- typedef typename property_traits<a_centrality_map>::value_type
- centrality_type;
- typename graph_traits<Graph>::vertices_size_type V = num_vertices(g);
-
- std::vector<std::vector<edge_descriptor> > incoming(V);
- std::vector<centrality_type> distance(V);
- std::vector<centrality_type> dependency(V);
- std::vector<degree_size_type> path_count(V);
- brandes_betweenness_centrality(
- g, centrality, edge_centrality_map,
- make_iterator_property_map(incoming.begin(), vertex_index),
- make_iterator_property_map(distance.begin(), vertex_index),
- make_iterator_property_map(dependency.begin(), vertex_index),
- make_iterator_property_map(path_count.begin(), vertex_index),
- vertex_index);
- }
- template<typename WeightMap>
- struct brandes_betweenness_centrality_dispatch1
- {
- template<typename Graph, typename CentralityMap,
- typename EdgeCentralityMap, typename VertexIndexMap>
- static void
- run(const Graph& g, CentralityMap centrality,
- EdgeCentralityMap edge_centrality_map, VertexIndexMap vertex_index,
- WeightMap weight_map)
- {
- brandes_betweenness_centrality_dispatch2(g, centrality, edge_centrality_map,
- weight_map, vertex_index);
- }
- };
- template<>
- struct brandes_betweenness_centrality_dispatch1<param_not_found>
- {
- template<typename Graph, typename CentralityMap,
- typename EdgeCentralityMap, typename VertexIndexMap>
- static void
- run(const Graph& g, CentralityMap centrality,
- EdgeCentralityMap edge_centrality_map, VertexIndexMap vertex_index,
- param_not_found)
- {
- brandes_betweenness_centrality_dispatch2(g, centrality, edge_centrality_map,
- vertex_index);
- }
- };
- template <typename T>
- struct is_bgl_named_params {
- BOOST_STATIC_CONSTANT(bool, value = false);
- };
- template <typename Param, typename Tag, typename Rest>
- struct is_bgl_named_params<bgl_named_params<Param, Tag, Rest> > {
- BOOST_STATIC_CONSTANT(bool, value = true);
- };
- } }
- template<typename Graph, typename Param, typename Tag, typename Rest>
- void
- brandes_betweenness_centrality(const Graph& g,
- const bgl_named_params<Param,Tag,Rest>& params
- BOOST_GRAPH_ENABLE_IF_MODELS_PARM(Graph,vertex_list_graph_tag))
- {
- typedef bgl_named_params<Param,Tag,Rest> named_params;
- typedef typename get_param_type<edge_weight_t, named_params>::type ew;
- detail::graph::brandes_betweenness_centrality_dispatch1<ew>::run(
- g,
- choose_param(get_param(params, vertex_centrality),
- dummy_property_map()),
- choose_param(get_param(params, edge_centrality),
- dummy_property_map()),
- choose_const_pmap(get_param(params, vertex_index), g, vertex_index),
- get_param(params, edge_weight));
- }
- template<typename Graph, typename CentralityMap>
- typename disable_if<detail::graph::is_bgl_named_params<CentralityMap>,
- void>::type
- brandes_betweenness_centrality(const Graph& g, CentralityMap centrality
- BOOST_GRAPH_ENABLE_IF_MODELS_PARM(Graph,vertex_list_graph_tag))
- {
- detail::graph::brandes_betweenness_centrality_dispatch2(
- g, centrality, dummy_property_map(), get(vertex_index, g));
- }
- template<typename Graph, typename CentralityMap, typename EdgeCentralityMap>
- void
- brandes_betweenness_centrality(const Graph& g, CentralityMap centrality,
- EdgeCentralityMap edge_centrality_map
- BOOST_GRAPH_ENABLE_IF_MODELS_PARM(Graph,vertex_list_graph_tag))
- {
- detail::graph::brandes_betweenness_centrality_dispatch2(
- g, centrality, edge_centrality_map, get(vertex_index, g));
- }
- template<typename Graph, typename CentralityMap>
- void
- relative_betweenness_centrality(const Graph& g, CentralityMap centrality)
- {
- typedef typename graph_traits<Graph>::vertex_iterator vertex_iterator;
- typedef typename property_traits<CentralityMap>::value_type centrality_type;
- typename graph_traits<Graph>::vertices_size_type n = num_vertices(g);
- centrality_type factor = centrality_type(2)/centrality_type(n*n - 3*n + 2);
- vertex_iterator v, v_end;
- for (boost::tie(v, v_end) = vertices(g); v != v_end; ++v) {
- put(centrality, *v, factor * get(centrality, *v));
- }
- }
- template<typename Graph, typename CentralityMap>
- typename property_traits<CentralityMap>::value_type
- central_point_dominance(const Graph& g, CentralityMap centrality
- BOOST_GRAPH_ENABLE_IF_MODELS_PARM(Graph,vertex_list_graph_tag))
- {
- using std::max;
- typedef typename graph_traits<Graph>::vertex_iterator vertex_iterator;
- typedef typename property_traits<CentralityMap>::value_type centrality_type;
- typename graph_traits<Graph>::vertices_size_type n = num_vertices(g);
-
- centrality_type max_centrality(0);
- vertex_iterator v, v_end;
- for (boost::tie(v, v_end) = vertices(g); v != v_end; ++v) {
- max_centrality = (max)(max_centrality, get(centrality, *v));
- }
-
- centrality_type sum(0);
- for (boost::tie(v, v_end) = vertices(g); v != v_end; ++v) {
- sum += (max_centrality - get(centrality, *v));
- }
- return sum/(n-1);
- }
- }
- #endif
|