lookup_edge.hpp 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. //=======================================================================
  2. // Copyright 2009 Trustees of Indiana University
  3. // Author: Jeremiah Willcock
  4. //
  5. // Distributed under the Boost Software License, Version 1.0. (See
  6. // accompanying file LICENSE_1_0.txt or copy at
  7. // http://www.boost.org/LICENSE_1_0.txt)
  8. //=======================================================================
  9. #ifndef BOOST_GRAPH_LOOKUP_EDGE_HPP
  10. #define BOOST_GRAPH_LOOKUP_EDGE_HPP
  11. #include <utility>
  12. #include <boost/config.hpp>
  13. #include <boost/utility/enable_if.hpp>
  14. #include <boost/graph/graph_traits.hpp>
  15. // lookup_edge: a function that acts like edge() but falls back to out_edges()
  16. // and a search when edge() is not provided.
  17. namespace boost {
  18. template <typename Graph>
  19. std::pair<typename boost::graph_traits<Graph>::edge_descriptor, bool>
  20. lookup_edge(typename boost::graph_traits<Graph>::vertex_descriptor src,
  21. typename boost::graph_traits<Graph>::vertex_descriptor tgt,
  22. const Graph& g,
  23. typename boost::enable_if<is_adjacency_matrix<Graph>, int>::type = 0) {
  24. return edge(src, tgt, g);
  25. }
  26. template <typename Graph>
  27. std::pair<typename boost::graph_traits<Graph>::edge_descriptor, bool>
  28. lookup_edge(typename boost::graph_traits<Graph>::vertex_descriptor src,
  29. typename boost::graph_traits<Graph>::vertex_descriptor tgt,
  30. const Graph& g,
  31. typename boost::disable_if<is_adjacency_matrix<Graph>, int>::type = 0) {
  32. typedef typename boost::graph_traits<Graph>::out_edge_iterator it;
  33. typedef typename boost::graph_traits<Graph>::edge_descriptor edesc;
  34. std::pair<it, it> oe = out_edges(src, g);
  35. for (; oe.first != oe.second; ++oe.first) {
  36. edesc e = *oe.first;
  37. if (target(e, g) == tgt) return std::make_pair(e, true);
  38. }
  39. return std::make_pair(edesc(), false);
  40. }
  41. }
  42. #endif // BOOST_GRAPH_LOOKUP_EDGE_HPP