smt2 id_map: Do not use nil_typet to detect map insert

Use std::map's emplace API for that purpose, and while at it also refactor all
inserts into id_map.

Author: tautschnig

src/solvers/smt2/smt2_parser.cpp

@@ -118,22 +118,33 @@ exprt::operandst smt2_parsert::operands()
   return result;
 }
 
-irep_idt smt2_parsert::get_fresh_id(const irep_idt &id)
+irep_idt smt2_parsert::add_fresh_id(const irep_idt &id, const exprt &expr)
 {
-  if(id_map[id].type.is_nil())
+  if(id_map
+       .emplace(
+         std::piecewise_construct,
+         std::forward_as_tuple(id),
+         std::forward_as_tuple(expr))
+       .second)
+  {
     return id; // id not yet used
+  }
 
   auto &count=renaming_counters[id];
   irep_idt new_id;
   do
   {
     new_id=id2string(id)+'#'+std::to_string(count);
     count++;
-  }
-  while(id_map.find(new_id)!=id_map.end());
+  } while(!id_map
+             .emplace(
+               std::piecewise_construct,
+               std::forward_as_tuple(new_id),
+               std::forward_as_tuple(expr))
+             .second);
 
   // record renaming
-  renaming_map[id]=new_id;
+  renaming_map[id] = new_id;
 
   return new_id;
 }
@@ -184,10 +195,7 @@ exprt smt2_parsert::let_expression()
   for(auto &b : bindings)
   {
     // get a fresh id for it
-    b.first=get_fresh_id(b.first);
-    auto &entry=id_map[b.first];
-    entry.type=b.second.type();
-    entry.definition=b.second;
+    b.first = add_fresh_id(b.first, b.second);
   }
 
   exprt expr=expression();
@@ -246,9 +254,9 @@ exprt smt2_parsert::quantifier_expression(irep_idt id)
   // go forwards, add to id_map
   for(const auto &b : bindings)
   {
-    auto &entry=id_map[b.get_identifier()];
-    entry.type=b.type();
-    entry.definition=nil_exprt();
+    const irep_idt id =
+      add_fresh_id(b.get_identifier(), exprt(ID_nil, b.type()));
+    CHECK_RETURN(id == b.get_identifier());
   }
 
   exprt expr=expression();
@@ -1134,9 +1142,8 @@ smt2_parsert::function_signature_definition()
     parameters.push_back(id);
     domain.push_back(sort());
 
-    auto &entry=id_map[id];
-    entry.type = domain.back();
-    entry.definition=nil_exprt();
+    if(add_fresh_id(id, exprt(ID_nil, domain.back())) != id)
+      CHECK_RETURN(false);
 
     if(next_token() != smt2_tokenizert::CLOSE)
       throw error("expected ')' at end of parameter");
@@ -1196,12 +1203,8 @@ void smt2_parsert::command(const std::string &c)
     irep_idt id = smt2_tokenizer.get_buffer();
     auto type = sort();
 
-    if(id_map.find(id)!=id_map.end())
+    if(add_fresh_id(id, exprt(ID_nil, type)) != id)
       throw error() << "identifier `" << id << "' defined twice";
-
-    auto &entry = id_map[id];
-    entry.type = type;
-    entry.definition = nil_exprt();
   }
   else if(c=="declare-fun")
   {
@@ -1211,12 +1214,8 @@ void smt2_parsert::command(const std::string &c)
     irep_idt id = smt2_tokenizer.get_buffer();
     auto type = function_signature_declaration();
 
-    if(id_map.find(id)!=id_map.end())
+    if(add_fresh_id(id, exprt(ID_nil, type)) != id)
       throw error() << "identifier `" << id << "' defined twice";
-
-    auto &entry = id_map[id];
-    entry.type = type;
-    entry.definition = nil_exprt();
   }
   else if(c == "define-const")
   {
@@ -1225,9 +1224,6 @@ void smt2_parsert::command(const std::string &c)
 
     const irep_idt id = smt2_tokenizer.get_buffer();
 
-    if(id_map.find(id) != id_map.end())
-      throw error() << "identifier `" << id << "' defined twice";
-
     const auto type = sort();
     const auto value = expression();
 
@@ -1240,9 +1236,8 @@ void smt2_parsert::command(const std::string &c)
     }
 
     // create the entry
-    auto &entry = id_map[id];
-    entry.type = type;
-    entry.definition = value;
+    if(add_fresh_id(id, value) != id)
+      throw error() << "identifier `" << id << "' defined twice";
   }
   else if(c=="define-fun")
   {
@@ -1251,9 +1246,6 @@ void smt2_parsert::command(const std::string &c)
 
     const irep_idt id = smt2_tokenizer.get_buffer();
 
-    if(id_map.find(id)!=id_map.end())
-      throw error() << "identifier `" << id << "' defined twice";
-
     const auto signature = function_signature_definition();
     const auto body = expression();
 
@@ -1276,10 +1268,10 @@ void smt2_parsert::command(const std::string &c)
     }
 
     // create the entry
-    auto &entry = id_map[id];
-    entry.type = signature.type;
-    entry.parameters = signature.parameters;
-    entry.definition = body;
+    if(add_fresh_id(id, body) != id)
+      throw error() << "identifier `" << id << "' defined twice";
+
+    id_map.at(id).parameters = signature.parameters;
   }
   else if(c=="exit")
   {

src/solvers/smt2/smt2_parser.h

@@ -33,7 +33,7 @@ class smt2_parsert
 
   struct idt
   {
-    idt():type(nil_typet())
+    explicit idt(const exprt &expr) : type(expr.type()), definition(expr)
     {
     }
 
@@ -84,7 +84,7 @@ class smt2_parsert
   renaming_mapt renaming_map;
   using renaming_counterst=std::map<irep_idt, unsigned>;
   renaming_counterst renaming_counters;
-  irep_idt get_fresh_id(const irep_idt &);
+  irep_idt add_fresh_id(const irep_idt &, const exprt &);
   irep_idt rename_id(const irep_idt &) const;
 
   struct signature_with_parameter_idst