Blame view

3rdparty/boost_1_81_0/libs/graph/example/print-in-edges.cpp 3.86 KB
977ed18d   Hu Chunming   提交三方库
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
  //=======================================================================
  // Copyright 2001 Jeremy G. Siek, Andrew Lumsdaine, Lie-Quan Lee,
  //
  // Distributed under the Boost Software License, Version 1.0. (See
  // accompanying file LICENSE_1_0.txt or copy at
  // http://www.boost.org/LICENSE_1_0.txt)
  //=======================================================================
  #include <boost/config.hpp>
  #include <iostream>
  #include <fstream>
  #include <string>
  #include <boost/graph/adjacency_list.hpp>
  
  using namespace boost;
  
  template < typename Graph, typename VertexNamePropertyMap >
  void read_graph_file(std::istream& graph_in, std::istream& name_in, Graph& g,
      VertexNamePropertyMap name_map)
  {
      typedef typename graph_traits< Graph >::vertices_size_type size_type;
      size_type n_vertices;
      typename graph_traits< Graph >::vertex_descriptor u;
      typename property_traits< VertexNamePropertyMap >::value_type name;
  
      graph_in >> n_vertices; // read in number of vertices
      for (size_type i = 0; i < n_vertices; ++i)
      { // Add n vertices to the graph
          u = add_vertex(g);
          name_in >> name;
          put(name_map, u, name); // ** Attach name property to vertex u **
      }
      size_type src, targ;
      while (graph_in >> src) // Read in edges
          if (graph_in >> targ)
              add_edge(src, targ, g); // add an edge to the graph
          else
              break;
  }
  
  template < typename Graph, typename VertexNameMap >
  void output_in_edges(std::ostream& out, const Graph& g,
      typename graph_traits< Graph >::vertex_descriptor v, VertexNameMap name_map)
  {
      typename graph_traits< Graph >::in_edge_iterator ei, ei_end;
      for (boost::tie(ei, ei_end) = in_edges(v, g); ei != ei_end; ++ei)
          out << get(name_map, source(*ei, g)) << " -> "
              << get(name_map, target(*ei, g)) << std::endl;
  }
  
  template < typename NameMap > class name_equals_t
  {
  public:
      name_equals_t(const std::string& n, NameMap map)
      : m_name(n), m_name_map(map)
      {
      }
      template < typename Vertex > bool operator()(Vertex u) const
      {
          return get(m_name_map, u) == m_name;
      }
  
  private:
      std::string m_name;
      NameMap m_name_map;
  };
  
  // object generator function
  template < typename NameMap >
  inline name_equals_t< NameMap > name_equals(
      const std::string& str, NameMap name)
  {
      return name_equals_t< NameMap >(str, name);
  }
  
  int main(int argc, const char** argv)
  {
      typedef adjacency_list< listS, // Store out-edges of each vertex in a
                                     // std::list
          vecS, // Store vertex set in a std::vector
          bidirectionalS, // The graph is directed, with both out-edges and
                          // in-edges
          property< vertex_name_t, std::string > // Add a vertex property
          >
          graph_type;
  
      graph_type g; // use default constructor to create empty graph
      const char* dep_file_name
          = argc >= 2 ? argv[1] : "makefile-dependencies.dat";
      const char* target_file_name
          = argc >= 3 ? argv[2] : "makefile-target-names.dat";
      std::ifstream file_in(dep_file_name), name_in(target_file_name);
      if (!file_in)
      {
          std::cerr << "** Error: could not open file " << dep_file_name
                    << std::endl;
          return -1;
      }
      if (!name_in)
      {
          std::cerr << "** Error: could not open file " << target_file_name
                    << std::endl;
          return -1;
      }
  
      // Obtain internal property map from the graph
      property_map< graph_type, vertex_name_t >::type name_map
          = get(vertex_name, g);
      read_graph_file(file_in, name_in, g, name_map);
  
      graph_traits< graph_type >::vertex_iterator i, end;
      boost::tie(i, end) = vertices(g);
      i = std::find_if(i, end, name_equals("libzigzag.a", get(vertex_name, g)));
      output_in_edges(std::cout, g, *i, get(vertex_name, g));
      assert(num_vertices(g) == 15);
      assert(num_edges(g) == 19);
      return 0;
  }