topo-sort-file-dep.cpp 3.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  1. //=======================================================================
  2. // Copyright 2001 Jeremy G. Siek, Andrew Lumsdaine, Lie-Quan Lee,
  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. #include <boost/config.hpp>
  9. #include <fstream>
  10. #include <iostream>
  11. #include <string>
  12. #include <boost/graph/adjacency_list.hpp>
  13. #include <boost/graph/graph_utility.hpp>
  14. using namespace boost;
  15. namespace std
  16. {
  17. template < typename T >
  18. std::istream & operator >> (std::istream & in, std::pair < T, T > &p)
  19. {
  20. in >> p.first >> p.second;
  21. return in;
  22. }
  23. }
  24. typedef adjacency_list < listS, // Store out-edges of each vertex in a std::list
  25. vecS, // Store vertex set in a std::vector
  26. directedS // The file dependency graph is directed
  27. > file_dep_graph;
  28. typedef graph_traits < file_dep_graph >::vertex_descriptor vertex_t;
  29. typedef graph_traits < file_dep_graph >::edge_descriptor edge_t;
  30. void
  31. topo_sort_dfs(const file_dep_graph & g, vertex_t u, vertex_t * &topo_order,
  32. int *mark)
  33. {
  34. mark[u] = 1; // 1 means visited, 0 means not yet visited
  35. graph_traits < file_dep_graph >::adjacency_iterator vi, vi_end;
  36. for (boost::tie(vi, vi_end) = adjacent_vertices(u, g); vi != vi_end; ++vi)
  37. if (mark[*vi] == 0)
  38. topo_sort_dfs(g, *vi, topo_order, mark);
  39. *--topo_order = u;
  40. }
  41. void
  42. topo_sort(const file_dep_graph & g, vertex_t * topo_order)
  43. {
  44. std::vector < int >mark(num_vertices(g), 0);
  45. graph_traits < file_dep_graph >::vertex_iterator vi, vi_end;
  46. for (boost::tie(vi, vi_end) = vertices(g); vi != vi_end; ++vi)
  47. if (mark[*vi] == 0)
  48. topo_sort_dfs(g, *vi, topo_order, &mark[0]);
  49. }
  50. int
  51. main(int argc, const char** argv)
  52. {
  53. std::ifstream file_in(argc >= 2 ? argv[1] : "makefile-dependencies.dat");
  54. typedef graph_traits < file_dep_graph >::vertices_size_type size_type;
  55. size_type n_vertices;
  56. file_in >> n_vertices; // read in number of vertices
  57. std::istream_iterator < std::pair < size_type, size_type > >
  58. input_begin(file_in), input_end;
  59. #if defined(BOOST_MSVC) && BOOST_MSVC <= 1300
  60. // VC++ can't handle the iterator constructor
  61. file_dep_graph g(n_vertices);
  62. while (input_begin != input_end) {
  63. size_type i, j;
  64. boost::tie(i, j) = *input_begin++;
  65. add_edge(i, j, g);
  66. }
  67. #else
  68. file_dep_graph g(input_begin, input_end, n_vertices);
  69. #endif
  70. std::vector < std::string > name(num_vertices(g));
  71. std::ifstream name_in(argc >= 3 ? argv[2] : "makefile-target-names.dat");
  72. graph_traits < file_dep_graph >::vertex_iterator vi, vi_end;
  73. for (boost::tie(vi, vi_end) = vertices(g); vi != vi_end; ++vi)
  74. name_in >> name[*vi];
  75. std::vector < vertex_t > order(num_vertices(g));
  76. topo_sort(g, &order[0] + num_vertices(g));
  77. for (size_type i = 0; i < num_vertices(g); ++i)
  78. std::cout << name[order[i]] << std::endl;
  79. return 0;
  80. }