gccrs: Refactor PathProbe into cc file
authorPhilip Herron <herron.philip@googlemail.com>
Mon, 16 Jan 2023 17:08:14 +0000 (17:08 +0000)
committerArthur Cohen <arthur.cohen@embecosm.com>
Thu, 6 Apr 2023 08:47:20 +0000 (10:47 +0200)
Signed-off-by: Philip Herron <herron.philip@googlemail.com>
gcc/rust/ChangeLog:

* typecheck/rust-hir-path-probe.cc (PathProbeType::PathProbeType): refactor
(PathProbeType::Probe): likewise
(PathProbeType::visit): likewise
(PathProbeType::process_enum_item_for_candiates): likewise
(PathProbeType::process_impl_items_for_candidates): likewise
(PathProbeType::is_reciever_generic): likewise
(PathProbeImplTrait::PathProbeImplTrait): likewise
(PathProbeImplTrait::Probe): likewise
(PathProbeImplTrait::process_trait_impl_items_for_candidates): likewise
* typecheck/rust-hir-path-probe.h (struct PathProbeCandidate): likewise
* typecheck/rust-hir-trait-resolve.cc
(PathProbeImplTrait::process_trait_impl_items_for_candidates): likewise

gcc/rust/typecheck/rust-hir-path-probe.cc
gcc/rust/typecheck/rust-hir-path-probe.h
gcc/rust/typecheck/rust-hir-trait-resolve.cc

index cb3270d..06d8920 100644 (file)
 
 #include "rust-hir-path-probe.h"
 #include "rust-hir-type-check-item.h"
+#include "rust-hir-trait-resolve.h"
 
 namespace Rust {
 namespace Resolver {
 
+// PathProbeType
+
+PathProbeType::PathProbeType (const TyTy::BaseType *receiver,
+                             const HIR::PathIdentSegment &query,
+                             DefId specific_trait_id)
+  : TypeCheckBase (), receiver (receiver), search (query),
+    current_impl (nullptr), specific_trait_id (specific_trait_id)
+{}
+
+std::set<PathProbeCandidate>
+PathProbeType::Probe (const TyTy::BaseType *receiver,
+                     const HIR::PathIdentSegment &segment_name,
+                     bool probe_impls, bool probe_bounds,
+                     bool ignore_mandatory_trait_items,
+                     DefId specific_trait_id)
+{
+  PathProbeType probe (receiver, segment_name, specific_trait_id);
+  if (probe_impls)
+    {
+      if (receiver->get_kind () == TyTy::TypeKind::ADT)
+       {
+         const TyTy::ADTType *adt
+           = static_cast<const TyTy::ADTType *> (receiver);
+         if (adt->is_enum ())
+           probe.process_enum_item_for_candiates (adt);
+       }
+
+      probe.process_impl_items_for_candidates ();
+    }
+
+  if (!probe_bounds)
+    return probe.candidates;
+
+  if (!probe.is_reciever_generic ())
+    {
+      std::vector<std::pair<TraitReference *, HIR::ImplBlock *>> probed_bounds
+       = TypeBoundsProbe::Probe (receiver);
+      for (auto &candidate : probed_bounds)
+       {
+         const TraitReference *trait_ref = candidate.first;
+         if (specific_trait_id != UNKNOWN_DEFID)
+           {
+             if (trait_ref->get_mappings ().get_defid () != specific_trait_id)
+               continue;
+           }
+
+         HIR::ImplBlock *impl = candidate.second;
+         probe.process_associated_trait_for_candidates (
+           trait_ref, impl, ignore_mandatory_trait_items);
+       }
+    }
+
+  for (const TyTy::TypeBoundPredicate &predicate :
+       receiver->get_specified_bounds ())
+    {
+      const TraitReference *trait_ref = predicate.get ();
+      if (specific_trait_id != UNKNOWN_DEFID)
+       {
+         if (trait_ref->get_mappings ().get_defid () != specific_trait_id)
+           continue;
+       }
+
+      probe.process_predicate_for_candidates (predicate,
+                                             ignore_mandatory_trait_items);
+    }
+
+  return probe.candidates;
+}
+
+void
+PathProbeType::visit (HIR::TypeAlias &alias)
+{
+  Identifier name = alias.get_new_type_name ();
+  if (search.as_string ().compare (name) == 0)
+    {
+      HirId tyid = alias.get_mappings ().get_hirid ();
+      TyTy::BaseType *ty = nullptr;
+      bool ok = query_type (tyid, &ty);
+      rust_assert (ok);
+
+      PathProbeCandidate::ImplItemCandidate impl_item_candidate{&alias,
+                                                               current_impl};
+      PathProbeCandidate candidate{
+       PathProbeCandidate::CandidateType::IMPL_TYPE_ALIAS, ty,
+       alias.get_locus (), impl_item_candidate};
+      candidates.insert (std::move (candidate));
+    }
+}
+
+void
+PathProbeType::visit (HIR::ConstantItem &constant)
+{
+  Identifier name = constant.get_identifier ();
+  if (search.as_string ().compare (name) == 0)
+    {
+      HirId tyid = constant.get_mappings ().get_hirid ();
+      TyTy::BaseType *ty = nullptr;
+      bool ok = query_type (tyid, &ty);
+      rust_assert (ok);
+
+      PathProbeCandidate::ImplItemCandidate impl_item_candidate{&constant,
+                                                               current_impl};
+      PathProbeCandidate candidate{
+       PathProbeCandidate::CandidateType::IMPL_CONST, ty,
+       constant.get_locus (), impl_item_candidate};
+      candidates.insert (std::move (candidate));
+    }
+}
+
+void
+PathProbeType::visit (HIR::Function &function)
+{
+  Identifier name = function.get_function_name ();
+  if (search.as_string ().compare (name) == 0)
+    {
+      HirId tyid = function.get_mappings ().get_hirid ();
+      TyTy::BaseType *ty = nullptr;
+      bool ok = query_type (tyid, &ty);
+      rust_assert (ok);
+
+      PathProbeCandidate::ImplItemCandidate impl_item_candidate{&function,
+                                                               current_impl};
+      PathProbeCandidate candidate{PathProbeCandidate::CandidateType::IMPL_FUNC,
+                                  ty, function.get_locus (),
+                                  impl_item_candidate};
+      candidates.insert (std::move (candidate));
+    }
+}
+
+void
+PathProbeType::process_enum_item_for_candiates (const TyTy::ADTType *adt)
+{
+  if (specific_trait_id != UNKNOWN_DEFID)
+    return;
+
+  TyTy::VariantDef *v;
+  if (!adt->lookup_variant (search.as_string (), &v))
+    return;
+
+  PathProbeCandidate::EnumItemCandidate enum_item_candidate{adt, v};
+  PathProbeCandidate candidate{PathProbeCandidate::CandidateType::ENUM_VARIANT,
+                              receiver->clone (),
+                              mappings->lookup_location (adt->get_ty_ref ()),
+                              enum_item_candidate};
+  candidates.insert (std::move (candidate));
+}
+
+void
+PathProbeType::process_impl_items_for_candidates ()
+{
+  mappings->iterate_impl_items (
+    [&] (HirId id, HIR::ImplItem *item, HIR::ImplBlock *impl) mutable -> bool {
+      process_impl_item_candidate (id, item, impl);
+      return true;
+    });
+}
+
 void
 PathProbeType::process_impl_item_candidate (HirId id, HIR::ImplItem *item,
                                            HIR::ImplBlock *impl)
@@ -42,5 +200,191 @@ PathProbeType::process_impl_item_candidate (HirId id, HIR::ImplItem *item,
   item->accept_vis (*this);
 }
 
+void
+PathProbeType::process_associated_trait_for_candidates (
+  const TraitReference *trait_ref, HIR::ImplBlock *impl,
+  bool ignore_mandatory_trait_items)
+{
+  const TraitItemReference *trait_item_ref = nullptr;
+  if (!trait_ref->lookup_trait_item (search.as_string (), &trait_item_ref))
+    return;
+
+  bool trait_item_needs_implementation = !trait_item_ref->is_optional ();
+  if (ignore_mandatory_trait_items && trait_item_needs_implementation)
+    return;
+
+  PathProbeCandidate::CandidateType candidate_type;
+  switch (trait_item_ref->get_trait_item_type ())
+    {
+    case TraitItemReference::TraitItemType::FN:
+      candidate_type = PathProbeCandidate::CandidateType::TRAIT_FUNC;
+      break;
+    case TraitItemReference::TraitItemType::CONST:
+      candidate_type = PathProbeCandidate::CandidateType::TRAIT_ITEM_CONST;
+      break;
+    case TraitItemReference::TraitItemType::TYPE:
+      candidate_type = PathProbeCandidate::CandidateType::TRAIT_TYPE_ALIAS;
+      break;
+
+    case TraitItemReference::TraitItemType::ERROR:
+    default:
+      gcc_unreachable ();
+      break;
+    }
+
+  TyTy::BaseType *trait_item_tyty = trait_item_ref->get_tyty ();
+
+  // we can substitute the Self with the receiver here
+  if (trait_item_tyty->get_kind () == TyTy::TypeKind::FNDEF)
+    {
+      TyTy::FnType *fn = static_cast<TyTy::FnType *> (trait_item_tyty);
+      TyTy::SubstitutionParamMapping *param = nullptr;
+      for (auto &param_mapping : fn->get_substs ())
+       {
+         const HIR::TypeParam &type_param = param_mapping.get_generic_param ();
+         if (type_param.get_type_representation ().compare ("Self") == 0)
+           {
+             param = &param_mapping;
+             break;
+           }
+       }
+      rust_assert (param != nullptr);
+
+      std::vector<TyTy::SubstitutionArg> mappings;
+      mappings.push_back (TyTy::SubstitutionArg (param, receiver->clone ()));
+
+      Location locus; // FIXME
+      TyTy::SubstitutionArgumentMappings args (std::move (mappings), {}, locus);
+      trait_item_tyty = SubstMapperInternal::Resolve (trait_item_tyty, args);
+    }
+
+  PathProbeCandidate::TraitItemCandidate trait_item_candidate{trait_ref,
+                                                             trait_item_ref,
+                                                             impl};
+
+  PathProbeCandidate candidate{candidate_type, trait_item_tyty,
+                              trait_item_ref->get_locus (),
+                              trait_item_candidate};
+  candidates.insert (std::move (candidate));
+}
+
+void
+PathProbeType::process_predicate_for_candidates (
+  const TyTy::TypeBoundPredicate &predicate, bool ignore_mandatory_trait_items)
+{
+  const TraitReference *trait_ref = predicate.get ();
+
+  TyTy::TypeBoundPredicateItem item
+    = predicate.lookup_associated_item (search.as_string ());
+  if (item.is_error ())
+    return;
+
+  if (ignore_mandatory_trait_items && item.needs_implementation ())
+    return;
+
+  const TraitItemReference *trait_item_ref = item.get_raw_item ();
+  PathProbeCandidate::CandidateType candidate_type;
+  switch (trait_item_ref->get_trait_item_type ())
+    {
+    case TraitItemReference::TraitItemType::FN:
+      candidate_type = PathProbeCandidate::CandidateType::TRAIT_FUNC;
+      break;
+    case TraitItemReference::TraitItemType::CONST:
+      candidate_type = PathProbeCandidate::CandidateType::TRAIT_ITEM_CONST;
+      break;
+    case TraitItemReference::TraitItemType::TYPE:
+      candidate_type = PathProbeCandidate::CandidateType::TRAIT_TYPE_ALIAS;
+      break;
+
+    case TraitItemReference::TraitItemType::ERROR:
+    default:
+      gcc_unreachable ();
+      break;
+    }
+
+  TyTy::BaseType *trait_item_tyty = item.get_tyty_for_receiver (receiver);
+  PathProbeCandidate::TraitItemCandidate trait_item_candidate{trait_ref,
+                                                             trait_item_ref,
+                                                             nullptr};
+  PathProbeCandidate candidate{candidate_type, trait_item_tyty,
+                              trait_item_ref->get_locus (),
+                              trait_item_candidate};
+  candidates.insert (std::move (candidate));
+}
+
+std::vector<std::pair<const TraitReference *, HIR::ImplBlock *>>
+PathProbeType::union_bounds (
+  const std::vector<std::pair</*const*/ TraitReference *, HIR::ImplBlock *>> a,
+  const std::vector<std::pair<const TraitReference *, HIR::ImplBlock *>> b)
+  const
+{
+  std::map<DefId, std::pair<const TraitReference *, HIR::ImplBlock *>> mapper;
+  for (auto &ref : a)
+    {
+      mapper.insert ({ref.first->get_mappings ().get_defid (), ref});
+    }
+  for (auto &ref : b)
+    {
+      mapper.insert ({ref.first->get_mappings ().get_defid (), ref});
+    }
+
+  std::vector<std::pair<const TraitReference *, HIR::ImplBlock *>> union_set;
+  for (auto it = mapper.begin (); it != mapper.end (); it++)
+    {
+      union_set.push_back ({it->second.first, it->second.second});
+    }
+  return union_set;
+}
+
+bool
+PathProbeType::is_reciever_generic () const
+{
+  const TyTy::BaseType *root = receiver->get_root ();
+  bool receiver_is_type_param = root->get_kind () == TyTy::TypeKind::PARAM;
+  bool receiver_is_dyn = root->get_kind () == TyTy::TypeKind::DYNAMIC;
+  return receiver_is_type_param || receiver_is_dyn;
+}
+
+// PathProbImplTrait
+
+PathProbeImplTrait::PathProbeImplTrait (const TyTy::BaseType *receiver,
+                                       const HIR::PathIdentSegment &query,
+                                       const TraitReference *trait_reference)
+  : PathProbeType (receiver, query, UNKNOWN_DEFID),
+    trait_reference (trait_reference)
+{}
+
+std::set<PathProbeCandidate>
+PathProbeImplTrait::Probe (const TyTy::BaseType *receiver,
+                          const HIR::PathIdentSegment &segment_name,
+                          const TraitReference *trait_reference)
+{
+  PathProbeImplTrait probe (receiver, segment_name, trait_reference);
+  // iterate all impls for this trait and receiver
+  // then search for possible candidates using base class behaviours
+  probe.process_trait_impl_items_for_candidates ();
+  return probe.candidates;
+}
+
+void
+PathProbeImplTrait::process_trait_impl_items_for_candidates ()
+{
+  mappings->iterate_impl_items (
+    [&] (HirId id, HIR::ImplItem *item, HIR::ImplBlock *impl) mutable -> bool {
+      // just need to check if this is an impl block for this trait the next
+      // function checks the receiver
+      if (!impl->has_trait_ref ())
+       return true;
+
+      TraitReference *resolved
+       = TraitResolver::Lookup (*(impl->get_trait_ref ().get ()));
+      if (!trait_reference->is_equal (*resolved))
+       return true;
+
+      process_impl_item_candidate (id, item, impl);
+      return true;
+    });
+}
+
 } // namespace Resolver
 } // namespace Rust
index bb8698c..783282a 100644 (file)
@@ -152,7 +152,7 @@ struct PathProbeCandidate
     return UNKNOWN_DEFID;
   }
 
-  bool operator< (const PathProbeCandidate &c) const
+  bool operator<(const PathProbeCandidate &c) const
   {
     return get_defid () < c.get_defid ();
   }
@@ -165,144 +165,16 @@ public:
   Probe (const TyTy::BaseType *receiver,
         const HIR::PathIdentSegment &segment_name, bool probe_impls,
         bool probe_bounds, bool ignore_mandatory_trait_items,
-        DefId specific_trait_id = UNKNOWN_DEFID)
-  {
-    PathProbeType probe (receiver, segment_name, specific_trait_id);
-    if (probe_impls)
-      {
-       if (receiver->get_kind () == TyTy::TypeKind::ADT)
-         {
-           const TyTy::ADTType *adt
-             = static_cast<const TyTy::ADTType *> (receiver);
-           if (adt->is_enum ())
-             probe.process_enum_item_for_candiates (adt);
-         }
-
-       probe.process_impl_items_for_candidates ();
-      }
+        DefId specific_trait_id = UNKNOWN_DEFID);
 
-    if (!probe_bounds)
-      return probe.candidates;
-
-    if (!probe.is_reciever_generic ())
-      {
-       std::vector<std::pair<TraitReference *, HIR::ImplBlock *>> probed_bounds
-         = TypeBoundsProbe::Probe (receiver);
-       for (auto &candidate : probed_bounds)
-         {
-           const TraitReference *trait_ref = candidate.first;
-           if (specific_trait_id != UNKNOWN_DEFID)
-             {
-               if (trait_ref->get_mappings ().get_defid ()
-                   != specific_trait_id)
-                 continue;
-             }
-
-           HIR::ImplBlock *impl = candidate.second;
-           probe.process_associated_trait_for_candidates (
-             trait_ref, impl, ignore_mandatory_trait_items);
-         }
-      }
-
-    for (const TyTy::TypeBoundPredicate &predicate :
-        receiver->get_specified_bounds ())
-      {
-       const TraitReference *trait_ref = predicate.get ();
-       if (specific_trait_id != UNKNOWN_DEFID)
-         {
-           if (trait_ref->get_mappings ().get_defid () != specific_trait_id)
-             continue;
-         }
-
-       probe.process_predicate_for_candidates (predicate,
-                                               ignore_mandatory_trait_items);
-      }
-
-    return probe.candidates;
-  }
-
-  void visit (HIR::TypeAlias &alias) override
-  {
-    Identifier name = alias.get_new_type_name ();
-    if (search.as_string ().compare (name) == 0)
-      {
-       HirId tyid = alias.get_mappings ().get_hirid ();
-       TyTy::BaseType *ty = nullptr;
-       bool ok = query_type (tyid, &ty);
-       rust_assert (ok);
-
-       PathProbeCandidate::ImplItemCandidate impl_item_candidate{&alias,
-                                                                 current_impl};
-       PathProbeCandidate candidate{
-         PathProbeCandidate::CandidateType::IMPL_TYPE_ALIAS, ty,
-         alias.get_locus (), impl_item_candidate};
-       candidates.insert (std::move (candidate));
-      }
-  }
-
-  void visit (HIR::ConstantItem &constant) override
-  {
-    Identifier name = constant.get_identifier ();
-    if (search.as_string ().compare (name) == 0)
-      {
-       HirId tyid = constant.get_mappings ().get_hirid ();
-       TyTy::BaseType *ty = nullptr;
-       bool ok = query_type (tyid, &ty);
-       rust_assert (ok);
-
-       PathProbeCandidate::ImplItemCandidate impl_item_candidate{&constant,
-                                                                 current_impl};
-       PathProbeCandidate candidate{
-         PathProbeCandidate::CandidateType::IMPL_CONST, ty,
-         constant.get_locus (), impl_item_candidate};
-       candidates.insert (std::move (candidate));
-      }
-  }
-
-  void visit (HIR::Function &function) override
-  {
-    Identifier name = function.get_function_name ();
-    if (search.as_string ().compare (name) == 0)
-      {
-       HirId tyid = function.get_mappings ().get_hirid ();
-       TyTy::BaseType *ty = nullptr;
-       bool ok = query_type (tyid, &ty);
-       rust_assert (ok);
-
-       PathProbeCandidate::ImplItemCandidate impl_item_candidate{&function,
-                                                                 current_impl};
-       PathProbeCandidate candidate{
-         PathProbeCandidate::CandidateType::IMPL_FUNC, ty,
-         function.get_locus (), impl_item_candidate};
-       candidates.insert (std::move (candidate));
-      }
-  }
+  void visit (HIR::TypeAlias &alias) override;
+  void visit (HIR::ConstantItem &constant) override;
+  void visit (HIR::Function &function) override;
 
 protected:
-  void process_enum_item_for_candiates (const TyTy::ADTType *adt)
-  {
-    if (specific_trait_id != UNKNOWN_DEFID)
-      return;
-
-    TyTy::VariantDef *v;
-    if (!adt->lookup_variant (search.as_string (), &v))
-      return;
-
-    PathProbeCandidate::EnumItemCandidate enum_item_candidate{adt, v};
-    PathProbeCandidate candidate{
-      PathProbeCandidate::CandidateType::ENUM_VARIANT, receiver->clone (),
-      mappings->lookup_location (adt->get_ty_ref ()), enum_item_candidate};
-    candidates.insert (std::move (candidate));
-  }
+  void process_enum_item_for_candiates (const TyTy::ADTType *adt);
 
-  void process_impl_items_for_candidates ()
-  {
-    mappings->iterate_impl_items ([&] (HirId id, HIR::ImplItem *item,
-                                      HIR::ImplBlock *impl) mutable -> bool {
-      process_impl_item_candidate (id, item, impl);
-      return true;
-    });
-  }
+  void process_impl_items_for_candidates ();
 
   void process_impl_item_candidate (HirId id, HIR::ImplItem *item,
                                    HIR::ImplBlock *impl);
@@ -310,156 +182,24 @@ protected:
   void
   process_associated_trait_for_candidates (const TraitReference *trait_ref,
                                           HIR::ImplBlock *impl,
-                                          bool ignore_mandatory_trait_items)
-  {
-    const TraitItemReference *trait_item_ref = nullptr;
-    if (!trait_ref->lookup_trait_item (search.as_string (), &trait_item_ref))
-      return;
-
-    bool trait_item_needs_implementation = !trait_item_ref->is_optional ();
-    if (ignore_mandatory_trait_items && trait_item_needs_implementation)
-      return;
-
-    PathProbeCandidate::CandidateType candidate_type;
-    switch (trait_item_ref->get_trait_item_type ())
-      {
-      case TraitItemReference::TraitItemType::FN:
-       candidate_type = PathProbeCandidate::CandidateType::TRAIT_FUNC;
-       break;
-      case TraitItemReference::TraitItemType::CONST:
-       candidate_type = PathProbeCandidate::CandidateType::TRAIT_ITEM_CONST;
-       break;
-      case TraitItemReference::TraitItemType::TYPE:
-       candidate_type = PathProbeCandidate::CandidateType::TRAIT_TYPE_ALIAS;
-       break;
-
-      case TraitItemReference::TraitItemType::ERROR:
-      default:
-       gcc_unreachable ();
-       break;
-      }
-
-    TyTy::BaseType *trait_item_tyty = trait_item_ref->get_tyty ();
-
-    // we can substitute the Self with the receiver here
-    if (trait_item_tyty->get_kind () == TyTy::TypeKind::FNDEF)
-      {
-       TyTy::FnType *fn = static_cast<TyTy::FnType *> (trait_item_tyty);
-       TyTy::SubstitutionParamMapping *param = nullptr;
-       for (auto &param_mapping : fn->get_substs ())
-         {
-           const HIR::TypeParam &type_param
-             = param_mapping.get_generic_param ();
-           if (type_param.get_type_representation ().compare ("Self") == 0)
-             {
-               param = &param_mapping;
-               break;
-             }
-         }
-       rust_assert (param != nullptr);
-
-       std::vector<TyTy::SubstitutionArg> mappings;
-       mappings.push_back (TyTy::SubstitutionArg (param, receiver->clone ()));
-
-       Location locus; // FIXME
-       TyTy::SubstitutionArgumentMappings args (std::move (mappings), {},
-                                                locus);
-       trait_item_tyty = SubstMapperInternal::Resolve (trait_item_tyty, args);
-      }
-
-    PathProbeCandidate::TraitItemCandidate trait_item_candidate{trait_ref,
-                                                               trait_item_ref,
-                                                               impl};
-
-    PathProbeCandidate candidate{candidate_type, trait_item_tyty,
-                                trait_item_ref->get_locus (),
-                                trait_item_candidate};
-    candidates.insert (std::move (candidate));
-  }
+                                          bool ignore_mandatory_trait_items);
 
   void
   process_predicate_for_candidates (const TyTy::TypeBoundPredicate &predicate,
-                                   bool ignore_mandatory_trait_items)
-  {
-    const TraitReference *trait_ref = predicate.get ();
-
-    TyTy::TypeBoundPredicateItem item
-      = predicate.lookup_associated_item (search.as_string ());
-    if (item.is_error ())
-      return;
-
-    if (ignore_mandatory_trait_items && item.needs_implementation ())
-      return;
-
-    const TraitItemReference *trait_item_ref = item.get_raw_item ();
-    PathProbeCandidate::CandidateType candidate_type;
-    switch (trait_item_ref->get_trait_item_type ())
-      {
-      case TraitItemReference::TraitItemType::FN:
-       candidate_type = PathProbeCandidate::CandidateType::TRAIT_FUNC;
-       break;
-      case TraitItemReference::TraitItemType::CONST:
-       candidate_type = PathProbeCandidate::CandidateType::TRAIT_ITEM_CONST;
-       break;
-      case TraitItemReference::TraitItemType::TYPE:
-       candidate_type = PathProbeCandidate::CandidateType::TRAIT_TYPE_ALIAS;
-       break;
-
-      case TraitItemReference::TraitItemType::ERROR:
-      default:
-       gcc_unreachable ();
-       break;
-      }
-
-    TyTy::BaseType *trait_item_tyty = item.get_tyty_for_receiver (receiver);
-    PathProbeCandidate::TraitItemCandidate trait_item_candidate{trait_ref,
-                                                               trait_item_ref,
-                                                               nullptr};
-    PathProbeCandidate candidate{candidate_type, trait_item_tyty,
-                                trait_item_ref->get_locus (),
-                                trait_item_candidate};
-    candidates.insert (std::move (candidate));
-  }
+                                   bool ignore_mandatory_trait_items);
 
 protected:
   PathProbeType (const TyTy::BaseType *receiver,
-                const HIR::PathIdentSegment &query, DefId specific_trait_id)
-    : TypeCheckBase (), receiver (receiver), search (query),
-      current_impl (nullptr), specific_trait_id (specific_trait_id)
-  {}
+                const HIR::PathIdentSegment &query, DefId specific_trait_id);
 
   std::vector<std::pair<const TraitReference *, HIR::ImplBlock *>>
   union_bounds (
     const std::vector<std::pair</*const*/ TraitReference *, HIR::ImplBlock *>>
       a,
     const std::vector<std::pair<const TraitReference *, HIR::ImplBlock *>> b)
-    const
-  {
-    std::map<DefId, std::pair<const TraitReference *, HIR::ImplBlock *>> mapper;
-    for (auto &ref : a)
-      {
-       mapper.insert ({ref.first->get_mappings ().get_defid (), ref});
-      }
-    for (auto &ref : b)
-      {
-       mapper.insert ({ref.first->get_mappings ().get_defid (), ref});
-      }
-
-    std::vector<std::pair<const TraitReference *, HIR::ImplBlock *>> union_set;
-    for (auto it = mapper.begin (); it != mapper.end (); it++)
-      {
-       union_set.push_back ({it->second.first, it->second.second});
-      }
-    return union_set;
-  }
+    const;
 
-  bool is_reciever_generic () const
-  {
-    const TyTy::BaseType *root = receiver->get_root ();
-    bool receiver_is_type_param = root->get_kind () == TyTy::TypeKind::PARAM;
-    bool receiver_is_dyn = root->get_kind () == TyTy::TypeKind::DYNAMIC;
-    return receiver_is_type_param || receiver_is_dyn;
-  }
+  bool is_reciever_generic () const;
 
   const TyTy::BaseType *receiver;
   const HIR::PathIdentSegment &search;
@@ -489,24 +229,14 @@ public:
   static std::set<PathProbeCandidate>
   Probe (const TyTy::BaseType *receiver,
         const HIR::PathIdentSegment &segment_name,
-        const TraitReference *trait_reference)
-  {
-    PathProbeImplTrait probe (receiver, segment_name, trait_reference);
-    // iterate all impls for this trait and receiver
-    // then search for possible candidates using base class behaviours
-    probe.process_trait_impl_items_for_candidates ();
-    return probe.candidates;
-  }
+        const TraitReference *trait_reference);
 
 private:
-  void process_trait_impl_items_for_candidates ();
-
   PathProbeImplTrait (const TyTy::BaseType *receiver,
                      const HIR::PathIdentSegment &query,
-                     const TraitReference *trait_reference)
-    : PathProbeType (receiver, query, UNKNOWN_DEFID),
-      trait_reference (trait_reference)
-  {}
+                     const TraitReference *trait_reference);
+
+  void process_trait_impl_items_for_candidates ();
 
   const TraitReference *trait_reference;
 };
index 85cad8e..19f95ca 100644 (file)
@@ -594,27 +594,5 @@ TraitItemReference::is_object_safe () const
   return false;
 }
 
-// rust-hir-path-probe.h
-
-void
-PathProbeImplTrait::process_trait_impl_items_for_candidates ()
-{
-  mappings->iterate_impl_items (
-    [&] (HirId id, HIR::ImplItem *item, HIR::ImplBlock *impl) mutable -> bool {
-      // just need to check if this is an impl block for this trait the next
-      // function checks the receiver
-      if (!impl->has_trait_ref ())
-       return true;
-
-      TraitReference *resolved
-       = TraitResolver::Lookup (*(impl->get_trait_ref ().get ()));
-      if (!trait_reference->is_equal (*resolved))
-       return true;
-
-      process_impl_item_candidate (id, item, impl);
-      return true;
-    });
-}
-
 } // namespace Resolver
 } // namespace Rust