Make class hierarchy output with parents optional

Author: peterschrammel

jbmc/unit/java_bytecode/goto-programs/class_hierarchy_output.cpp

@@ -47,7 +47,7 @@ SCENARIO(
   std::stringstream output_dot_stream;
 
   hierarchy(symbol_table);
-  hierarchy.output(output_stream);
+  hierarchy.output(output_stream, false);
   hierarchy.output_dot(output_dot_stream);
 
   std::string output = output_stream.str();

src/goto-instrument/goto_instrument_parse_options.cpp

@@ -659,7 +659,7 @@ int goto_instrument_parse_optionst::doit()
       if(cmdline.isset("dot"))
         hierarchy.output_dot(std::cout);
       else
-        hierarchy.output(std::cout);
+        hierarchy.output(std::cout, false);
 
       return 0;
     }

src/goto-programs/class_hierarchy.cpp

@@ -30,6 +30,9 @@ void class_hierarchyt::operator()(const symbol_tablet &symbol_table)
     {
       const struct_typet &struct_type = to_struct_type(symbol_pair.second.type);
 
+      class_map[symbol_pair.first].is_abstract =
+        struct_type.get_bool(ID_abstract);
+
       const irept::subt &bases=
         struct_type.find(ID_bases).get_sub();
 
@@ -123,17 +126,20 @@ void class_hierarchyt::get_parents_trans_rec(
     get_parents_trans_rec(child, dest);
 }
 
-void class_hierarchyt::output(std::ostream &out) const
+/// Output the class hierarchy in plain text
+/// \param out: the output stream
+/// \param children_only: print the children only and do not print the parents
+void class_hierarchyt::output(std::ostream &out, bool children_only) const
 {
   for(const auto &c : class_map)
   {
-    for(const auto &pa : c.second.parents)
-      out << "Parent of " << c.first << ": "
-          << pa << '\n';
-
+    if(!children_only)
+    {
+      for(const auto &pa : c.second.parents)
+        out << "Parent of " << c.first << ": " << pa << '\n';
+    }
     for(const auto &ch : c.second.children)
-      out << "Child of " << c.first << ": "
-          << ch << '\n';
+      out << "Child of " << c.first << ": " << ch << '\n';
   }
 }
 

src/goto-programs/class_hierarchy.h

@@ -32,6 +32,7 @@ class class_hierarchyt
   {
   public:
     idst parents, children;
+    bool is_abstract;
   };
 
   typedef std::map<irep_idt, entryt> class_mapt;
@@ -55,7 +56,7 @@ class class_hierarchyt
     return result;
   }
 
-  void output(std::ostream &) const;
+  void output(std::ostream &, bool children_only) const;
   void output_dot(std::ostream &) const;
 
 protected: