Merge #669

669: Fix bug with monomorphizing of trait items r=philberty a=philberty

When we monomorphize functions we need to check if we have already
generated this function already. All function items have a DefId which is
unique for the crate and HIR item this means for generic implementations
of an item we end up with mappings of:

  DefId -> [ (concete-tyty-fntype, GCC::Function), ... ]

So for any function we can lookup for that DefId is there a suitable
version of this function already compiled. This was working untill we have
generic trait items which also need to be handled in the same way.

Fixes #668


Co-authored-by: Philip Herron <philip.herron@embecosm.com>

1 files changed, 32 insertions, 22 deletions

diff --git a/gcc/rust/backend/rust-compile-context.h b/gcc/rust/backend/rust-compile-context.h
index 05c15e3..3a92c32 100644
--- a/gcc/rust/backend/rust-compile-context.h
+++ b/gcc/rust/backend/rust-compile-context.h
@@ -160,32 +160,41 @@ public:
     return true;
   }
 
-  void insert_function_decl (HirId id, ::Bfunction *fn,
-			     const TyTy::BaseType *ref)
+  void insert_function_decl (const TyTy::FnType *ref, ::Bfunction *fn)
   {
+    auto id = ref->get_ty_ref ();
+    auto dId = ref->get_id ();
+
     rust_assert (compiled_fn_map.find (id) == compiled_fn_map.end ());
     compiled_fn_map[id] = fn;
-    if (ref != nullptr)
-      {
-	std::pair<HirId, ::Bfunction *> elem (id, fn);
-	mono_fns[ref] = std::move (elem);
-      }
+
+    auto it = mono_fns.find (dId);
+    if (it == mono_fns.end ())
+      mono_fns[dId] = {};
+
+    mono_fns[dId].push_back ({ref, fn});
   }
 
   bool lookup_function_decl (HirId id, ::Bfunction **fn,
+			     DefId dId = UNKNOWN_DEFID,
 			     const TyTy::BaseType *ref = nullptr)
   {
     // for for any monomorphized fns
     if (ref != nullptr)
       {
-	for (auto it = mono_fns.begin (); it != mono_fns.end (); it++)
+	rust_assert (dId != UNKNOWN_DEFID);
+
+	auto it = mono_fns.find (dId);
+	if (it == mono_fns.end ())
+	  return false;
+
+	for (auto &e : mono_fns[dId])
 	  {
-	    std::pair<HirId, ::Bfunction *> &val = it->second;
-	    const TyTy::BaseType *r = it->first;
+	    const TyTy::BaseType *r = e.first;
+	    ::Bfunction *f = e.second;
 	    if (ref->is_equal (*r))
 	      {
-		*fn = val.second;
-
+		*fn = f;
 		return true;
 	      }
 	  }
@@ -316,18 +325,19 @@ private:
   std::map<HirId, ::Bfunction *> compiled_fn_map;
   std::map<HirId, ::Bexpression *> compiled_consts;
   std::map<HirId, ::Blabel *> compiled_labels;
-  std::vector< ::std::vector<Bstatement *> > statements;
-  std::vector< ::Bblock *> scope_stack;
-  std::vector< ::Bvariable *> loop_value_stack;
-  std::vector< ::Blabel *> loop_begin_labels;
-  std::map<const TyTy::BaseType *, std::pair<HirId, ::Btype *> > mono;
-  std::map<const TyTy::BaseType *, std::pair<HirId, ::Bfunction *> > mono_fns;
+  std::vector<::std::vector<Bstatement *>> statements;
+  std::vector<::Bblock *> scope_stack;
+  std::vector<::Bvariable *> loop_value_stack;
+  std::vector<::Blabel *> loop_begin_labels;
+  std::map<const TyTy::BaseType *, std::pair<HirId, ::Btype *>> mono;
+  std::map<DefId, std::vector<std::pair<const TyTy::BaseType *, ::Bfunction *>>>
+    mono_fns;
 
   // To GCC middle-end
-  std::vector< ::Btype *> type_decls;
-  std::vector< ::Bvariable *> var_decls;
-  std::vector< ::Bexpression *> const_decls;
-  std::vector< ::Bfunction *> func_decls;
+  std::vector<::Btype *> type_decls;
+  std::vector<::Bvariable *> var_decls;
+  std::vector<::Bexpression *> const_decls;
+  std::vector<::Bfunction *> func_decls;
 };
 
 class TyTyResolveCompile : public TyTy::TyVisitor