Resolve symbol types in symbol table

Author: danpoe

src/json-symtab-language/json_symtab_language.cpp

@@ -11,6 +11,7 @@ Author: Chris Smowton, [email protected]
 #include "json_symtab_language.h"
 #include <json/json_parser.h>
 #include <util/json_symbol_table.h>
+#include <util/namespace.h>
 
 bool json_symtab_languaget::parse(
   std::istream &instream,
@@ -30,6 +31,7 @@ bool json_symtab_languaget::typecheck(
   try
   {
     symbol_table_from_json(parsed_json_file, symbol_table);
+    follow_symbol_types(symbol_table);
     return false;
   }
   catch(const std::string &str)
@@ -39,6 +41,65 @@ bool json_symtab_languaget::typecheck(
   }
 }
 
+void json_symtab_languaget::follow_symbol_types(
+  irept &irep,
+  const namespacet &ns)
+{
+  if(irep.id() == ID_symbol)
+  {
+    const symbolt &symbol = ns.lookup(irep);
+
+    if(symbol.is_type)
+    {
+      ns.follow_symbol(irep);
+    }
+  }
+
+  for(irept &sub : irep.get_sub())
+  {
+    follow_symbol_types(sub, ns);
+  }
+
+  for(auto &entry : irep.get_named_sub())
+  {
+    irept &sub = entry.second;
+
+    follow_symbol_types(sub, ns);
+  }
+}
+
+void json_symtab_languaget::follow_symbol_types(symbol_tablet &symbol_table)
+{
+  const namespacet ns(symbol_table);
+
+  typedef symbol_tablet::symbolst symbolst;
+  symbolst &symbols = symbol_table.symbols;
+
+  for(auto &entry : symbols)
+  {
+    symbolt &symbol = entry.second;
+
+    // Modify entries in place
+    follow_symbol_types(symbol.type, ns);
+    follow_symbol_types(symbol.value, ns);
+  }
+
+  // Remove type entries from the symbol table
+  for(symbolst::iterator it = symbols.begin(); it != symbols.end();)
+  {
+    const symbolt &symbol = it->second;
+
+    if(symbol.is_type)
+    {
+      it = symbols.erase(it);
+    }
+    else
+    {
+      it++;
+    }
+  }
+}
+
 void json_symtab_languaget::show_parse(std::ostream &out)
 {
   parsed_json_file.output(out);

src/json-symtab-language/json_symtab_language.h

@@ -60,6 +60,9 @@ class json_symtab_languaget:public languaget
   }
 
  protected:
+  void follow_symbol_types(symbol_tablet &symbol_table);
+  void follow_symbol_types(irept &irep, const namespacet &ns);
+
   jsont parsed_json_file;
 };