Reused private code in 'remove_virtual_functions.cpp' by making it public.

Author: marek-trtik

src/goto-programs/remove_virtual_functions.cpp

@@ -11,7 +11,6 @@ Author: Daniel Kroening, [email protected]
 #include <algorithm>
 
 #include "remove_virtual_functions.h"
-#include "class_hierarchy.h"
 #include "class_identifier.h"
 
 #include <goto-programs/resolve_concrete_function_call.h>
@@ -24,7 +23,8 @@ class remove_virtual_functionst
 {
 public:
   remove_virtual_functionst(
-    const symbol_tablet &_symbol_table);
+    const symbol_tablet &_symbol_table,
+    const class_hierarchyt &_class_hierarchy);
 
   void operator()(goto_functionst &goto_functions);
 
@@ -40,13 +40,14 @@ class remove_virtual_functionst
   const namespacet ns;
   const symbol_tablet &symbol_table;
 
-  class_hierarchyt class_hierarchy;
+  const class_hierarchyt &class_hierarchy;
 
   void remove_virtual_function(
     goto_programt &goto_program,
     goto_programt::targett target);
-
+public:
   void get_functions(const exprt &, dispatch_table_entriest &);
+protected:
   typedef std::function<
     resolve_concrete_function_callt::concrete_function_callt(
       const irep_idt &,
@@ -64,11 +65,12 @@ class remove_virtual_functionst
 };
 
 remove_virtual_functionst::remove_virtual_functionst(
-  const symbol_tablet &_symbol_table):
+  const symbol_tablet &_symbol_table,
+  const class_hierarchyt &_class_hierarchy):
   ns(_symbol_table),
-  symbol_table(_symbol_table)
+  symbol_table(_symbol_table),
+  class_hierarchy(_class_hierarchy)
 {
-  class_hierarchy(symbol_table);
 }
 
 void remove_virtual_functionst::remove_virtual_function(
@@ -434,7 +436,9 @@ void remove_virtual_functions(
   const symbol_tablet &symbol_table,
   goto_functionst &goto_functions)
 {
-  remove_virtual_functionst rvf(symbol_table);
+  class_hierarchyt class_hierarchy;
+  class_hierarchy(symbol_table);
+  remove_virtual_functionst rvf(symbol_table, class_hierarchy);
   rvf(goto_functions);
 }
 
@@ -446,7 +450,9 @@ void remove_virtual_functions(goto_modelt &goto_model)
 
 void remove_virtual_functions(goto_model_functiont &function)
 {
-  remove_virtual_functionst rvf(function.get_symbol_table());
+  class_hierarchyt class_hierarchy;
+  class_hierarchy(function.get_symbol_table());
+  remove_virtual_functionst rvf(function.get_symbol_table(), class_hierarchy);
   bool changed = rvf.remove_virtual_functions(
     function.get_goto_function().body);
   // Give fresh location numbers to `function`, in case it has grown:
@@ -461,8 +467,20 @@ void remove_virtual_function(
   const dispatch_table_entriest &dispatch_table,
   virtual_dispatch_fallback_actiont fallback_action)
 {
-  remove_virtual_functionst rvf(goto_model.symbol_table);
+  class_hierarchyt class_hierarchy;
+  class_hierarchy(goto_model.symbol_table);
+  remove_virtual_functionst rvf(goto_model.symbol_table, class_hierarchy);
 
   rvf.remove_virtual_function(
     goto_program, instruction, dispatch_table, fallback_action);
 }
+
+void get_overridden_functions(
+  const exprt &function,
+  const symbol_tablet &symbol_table,
+  const class_hierarchyt &class_hierarchy,
+  dispatch_table_entriest &overridden_functions)
+{
+  remove_virtual_functionst instance(symbol_table, class_hierarchy);
+  instance.get_functions(function, overridden_functions);
+}

src/goto-programs/remove_virtual_functions.h

@@ -15,6 +15,7 @@ Date: April 2016
 #define CPROVER_GOTO_PROGRAMS_REMOVE_VIRTUAL_FUNCTIONS_H
 
 #include "goto_model.h"
+#include "class_hierarchy.h"
 
 // remove virtual function calls
 // and replace by case-split
@@ -64,4 +65,19 @@ void remove_virtual_function(
   const dispatch_table_entriest &dispatch_table,
   virtual_dispatch_fallback_actiont fallback_action);
 
+/// Given a function expression representing a virtual method of a class,
+/// the function computes all overridden methods of that virtual method.
+/// \param function: The virtual function expression for which the overridden
+///                  methods will be searched for.
+/// \param symbol_table: A symbol table.
+/// \param class_hierarchy: A class hierarchy.
+/// \param [out] overridden_functions: Output collection into which all
+///                                    overridden functions will be stored.
+void get_overridden_functions(
+  const exprt &function,
+  const symbol_tablet &symbol_table,
+  const class_hierarchyt &class_hierarchy,
+  dispatch_table_entriest &overridden_functions);
+
+
 #endif // CPROVER_GOTO_PROGRAMS_REMOVE_VIRTUAL_FUNCTIONS_H