call graph helper interface to depth limited search

Gets all functions reachable within n steps, by calling depth limited search

Author: polgreen

src/analyses/call_graph_helpers.cpp

@@ -70,3 +70,34 @@ std::set<irep_idt> get_reaching_functions(
 {
   return get_connected_functions(graph, function, false);
 }
+
+std::set<irep_idt> get_functions_reachable_within_n_steps(
+  const call_grapht::directed_grapht &graph,
+  const std::set<irep_idt> &start_functions,
+  std::size_t &n,
+  bool forwards)
+{
+  std::vector<std::size_t> start_indices;
+  std::set<irep_idt> result;
+
+  for(const auto &func : start_functions)
+    start_indices.push_back(*(graph.get_node_index(func)));
+
+  for(const auto &index :
+      graph.depth_limited_search(start_indices, n, forwards))
+    result.insert(graph[index].function);
+
+  return result;
+}
+
+std::set<irep_idt> get_functions_reachable_within_n_steps(
+  const call_grapht::directed_grapht &graph,
+  const irep_idt &start_function,
+  std::size_t &n,
+  bool forwards)
+{
+  std::set<irep_idt> start_functions;
+  start_functions.insert(start_function);
+  return get_functions_reachable_within_n_steps(
+    graph, start_functions, n, forwards);
+}

src/analyses/call_graph_helpers.h

@@ -49,4 +49,28 @@ std::set<irep_idt> get_reachable_functions(
 std::set<irep_idt> get_reaching_functions(
   const call_grapht::directed_grapht &graph, const irep_idt &function);
 
+/// Get either callers or callees reachable from a given
+/// list of functions within N steps
+/// \param graph: call graph
+/// \param start_functions: set of start functions
+/// \param forwards: if true, get callees; otherwise get callers
+/// \param n: number of steps to consider
+std::set<irep_idt> get_functions_reachable_within_n_steps(
+  const call_grapht::directed_grapht &graph,
+  const std::set<irep_idt> &start_functions,
+  std::size_t &n,
+  bool forwards);
+
+/// Get either callers or callees reachable from a given
+/// list of functions within N steps
+/// \param graph: call graph
+/// \param start_function: single start function
+/// \param forwards: if true, get callees; otherwise get callers
+/// \param n: number of steps to consider
+std::set<irep_idt> get_functions_reachable_within_n_steps(
+  const call_grapht::directed_grapht &graph,
+  const irep_idt &start_function,
+  std::size_t &n,
+  bool forwards);
+
 #endif