gccrs: Initial Type resolution for closures

gcc/rust/ChangeLog:

	* hir/tree/rust-hir-expr.h: Add `get_params` method.
	* typecheck/rust-hir-type-check-expr.cc (TypeCheckExpr::visit): Typecheck closure nodes.
	(TypeCheckExpr::resolve_fn_trait_call): New function.
	* typecheck/rust-hir-type-check-expr.h: Declare `resolve_fn_trait_call` and
	`resolve_possible_fn_trait_call_method_name`.
	* typecheck/rust-hir-type-check.h: Declare `get_context_type`.
	* typecheck/rust-tyctx.cc (TypeCheckContextItem::get_context_type): New function.
	* typecheck/rust-tyty-cmp.h: Visit closures properly.
	* typecheck/rust-tyty-rules.h: Likewise.
	* typecheck/rust-tyty.cc (BaseType::bounds_compatible): Add commented out assertin.
	(ClosureType::as_string): Implement it.
	(ClosureType::clone): Fix closure cloning.
	(ClosureType::setup_fn_once_output): New function.
	* typecheck/rust-tyty.h: Improve `ClosureType` class and declare `setup_fn_once_output`.
This commit is contained in:
Philip Herron
2022-10-18 13:21:40 +01:00
committed by Arthur Cohen
parent 898240946c
commit 740a199722
9 changed files with 479 additions and 31 deletions

View File

@@ -2131,6 +2131,8 @@ public:
};
std::unique_ptr<Expr> &get_expr () { return expr; }
std::vector<ClosureParam> &get_params () { return params; }
void accept_vis (HIRFullVisitor &vis) override;
void accept_vis (HIRExpressionVisitor &vis) override;

View File

@@ -178,16 +178,6 @@ TypeCheckExpr::visit (HIR::CallExpr &expr)
{
TyTy::BaseType *function_tyty = TypeCheckExpr::Resolve (expr.get_fnexpr ());
bool valid_tyty = function_tyty->get_kind () == TyTy::TypeKind::ADT
|| function_tyty->get_kind () == TyTy::TypeKind::FNDEF
|| function_tyty->get_kind () == TyTy::TypeKind::FNPTR;
if (!valid_tyty)
{
rust_error_at (expr.get_locus (),
"Failed to resolve expression of function call");
return;
}
rust_debug_loc (expr.get_locus (), "resolved_call_expr to: {%s}",
function_tyty->get_name ().c_str ());
@@ -214,6 +204,24 @@ TypeCheckExpr::visit (HIR::CallExpr &expr)
rust_assert (adt->number_of_variants () == 1);
variant = *adt->get_variants ().at (0);
}
infered
= TyTy::TypeCheckCallExpr::go (function_tyty, expr, variant, context);
return;
}
bool resolved_fn_trait_call
= resolve_fn_trait_call (expr, function_tyty, &infered);
if (resolved_fn_trait_call)
return;
bool valid_tyty = function_tyty->get_kind () == TyTy::TypeKind::FNDEF
|| function_tyty->get_kind () == TyTy::TypeKind::FNPTR;
if (!valid_tyty)
{
rust_error_at (expr.get_locus (),
"Failed to resolve expression of function call");
return;
}
infered = TyTy::TypeCheckCallExpr::go (function_tyty, expr, variant, context);
@@ -1422,7 +1430,123 @@ TypeCheckExpr::visit (HIR::MatchExpr &expr)
void
TypeCheckExpr::visit (HIR::ClosureExpr &expr)
{
gcc_unreachable ();
TypeCheckContextItem &current_context = context->peek_context ();
TyTy::FnType *current_context_fndecl = current_context.get_context_type ();
HirId ref = expr.get_mappings ().get_hirid ();
DefId id = expr.get_mappings ().get_defid ();
RustIdent ident{current_context_fndecl->get_ident ().path, expr.get_locus ()};
// get from parent context
std::vector<TyTy::SubstitutionParamMapping> subst_refs
= current_context_fndecl->clone_substs ();
std::vector<TyTy::TyVar> parameter_types;
for (auto &p : expr.get_params ())
{
if (p.has_type_given ())
{
TyTy::BaseType *param_tyty
= TypeCheckType::Resolve (p.get_type ().get ());
TyTy::TyVar param_ty (param_tyty->get_ref ());
parameter_types.push_back (param_ty);
TypeCheckPattern::Resolve (p.get_pattern ().get (),
param_ty.get_tyty ());
}
else
{
TyTy::TyVar param_ty
= TyTy::TyVar::get_implicit_infer_var (p.get_locus ());
parameter_types.push_back (param_ty);
TypeCheckPattern::Resolve (p.get_pattern ().get (),
param_ty.get_tyty ());
}
}
// we generate an implicit hirid for the closure args
HirId implicit_args_id = mappings->get_next_hir_id ();
TyTy::TupleType *closure_args
= new TyTy::TupleType (implicit_args_id, expr.get_locus (),
parameter_types);
context->insert_implicit_type (closure_args);
Location result_type_locus = expr.has_return_type ()
? expr.get_return_type ()->get_locus ()
: expr.get_locus ();
TyTy::TyVar result_type
= expr.has_return_type ()
? TyTy::TyVar (
TypeCheckType::Resolve (expr.get_return_type ().get ())->get_ref ())
: TyTy::TyVar::get_implicit_infer_var (expr.get_locus ());
// resolve the block
Location closure_expr_locus = expr.get_expr ()->get_locus ();
TyTy::BaseType *closure_expr_ty
= TypeCheckExpr::Resolve (expr.get_expr ().get ());
coercion_site (expr.get_mappings ().get_hirid (),
TyTy::TyWithLocation (result_type.get_tyty (),
result_type_locus),
TyTy::TyWithLocation (closure_expr_ty, closure_expr_locus),
expr.get_locus ());
// generate the closure type
infered = new TyTy::ClosureType (ref, id, ident, closure_args, result_type,
subst_refs);
// FIXME
// all closures automatically inherit the appropriate fn trait. Lets just
// assume FnOnce for now. I think this is based on the return type of the
// closure
Analysis::RustLangItem::ItemType lang_item_type
= Analysis::RustLangItem::ItemType::FN_ONCE;
DefId respective_lang_item_id = UNKNOWN_DEFID;
bool lang_item_defined
= mappings->lookup_lang_item (lang_item_type, &respective_lang_item_id);
if (!lang_item_defined)
{
// FIXME
// we need to have a unified way or error'ing when we are missing lang
// items that is useful
rust_fatal_error (
expr.get_locus (), "unable to find lang item: %<%s%>",
Analysis::RustLangItem::ToString (lang_item_type).c_str ());
}
rust_assert (lang_item_defined);
// these lang items are always traits
HIR::Item *item = mappings->lookup_defid (respective_lang_item_id);
rust_assert (item->get_item_kind () == HIR::Item::ItemKind::Trait);
HIR::Trait *trait_item = static_cast<HIR::Trait *> (item);
TraitReference *trait = TraitResolver::Resolve (*trait_item);
rust_assert (!trait->is_error ());
TyTy::TypeBoundPredicate predicate (*trait, expr.get_locus ());
// resolve the trait bound where the <(Args)> are the parameter tuple type
HIR::GenericArgs args = HIR::GenericArgs::create_empty (expr.get_locus ());
// lets generate an implicit Type so that it resolves to the implict tuple
// type we have created
auto crate_num = mappings->get_current_crate ();
Analysis::NodeMapping mapping (crate_num, expr.get_mappings ().get_nodeid (),
implicit_args_id, UNKNOWN_LOCAL_DEFID);
HIR::TupleType *implicit_tuple
= new HIR::TupleType (mapping,
{} // we dont need to fill this out because it will
// auto resolve because the hir id's match
,
expr.get_locus ());
args.get_type_args ().push_back (std::unique_ptr<HIR::Type> (implicit_tuple));
// apply the arguments
predicate.apply_generic_arguments (&args);
// finally inherit the trait bound
infered->inherit_bounds ({predicate});
}
bool
@@ -1630,6 +1754,168 @@ TypeCheckExpr::resolve_operator_overload (
return true;
}
HIR::PathIdentSegment
TypeCheckExpr::resolve_possible_fn_trait_call_method_name (
const TyTy::BaseType &receiver)
{
// Question
// do we need to probe possible bounds here? I think not, i think when we
// support Fn traits they are explicitly specified
// FIXME
// the logic to map the FnTrait to their respective call trait-item is
// duplicated over in the backend/rust-compile-expr.cc
for (const auto &bound : receiver.get_specified_bounds ())
{
bool found_fn = bound.get_name ().compare ("Fn") == 0;
bool found_fn_mut = bound.get_name ().compare ("FnMut") == 0;
bool found_fn_once = bound.get_name ().compare ("FnOnce") == 0;
if (found_fn)
{
return HIR::PathIdentSegment ("call");
}
else if (found_fn_mut)
{
return HIR::PathIdentSegment ("call_mut");
}
else if (found_fn_once)
{
return HIR::PathIdentSegment ("call_once");
}
}
// nothing
return HIR::PathIdentSegment ("");
}
bool
TypeCheckExpr::resolve_fn_trait_call (HIR::CallExpr &expr,
TyTy::BaseType *receiver_tyty,
TyTy::BaseType **result)
{
// we turn this into a method call expr
HIR::PathIdentSegment method_name
= resolve_possible_fn_trait_call_method_name (*receiver_tyty);
if (method_name.is_error ())
return false;
auto candidates = MethodResolver::Probe (receiver_tyty, method_name);
if (candidates.empty ())
return false;
if (candidates.size () > 1)
{
RichLocation r (expr.get_locus ());
for (auto &c : candidates)
r.add_range (c.candidate.locus);
rust_error_at (
r, "multiple candidates found for function trait method call %<%s%>",
method_name.as_string ().c_str ());
return false;
}
if (receiver_tyty->get_kind () == TyTy::TypeKind::CLOSURE)
{
const TyTy::ClosureType &closure
= static_cast<TyTy::ClosureType &> (*receiver_tyty);
closure.setup_fn_once_output ();
}
auto candidate = *candidates.begin ();
rust_debug_loc (expr.get_locus (),
"resolved call-expr to fn trait: {%u} {%s}",
candidate.candidate.ty->get_ref (),
candidate.candidate.ty->debug_str ().c_str ());
// Get the adjusted self
Adjuster adj (receiver_tyty);
TyTy::BaseType *adjusted_self = adj.adjust_type (candidate.adjustments);
// store the adjustments for code-generation to know what to do which must be
// stored onto the receiver to so as we don't trigger duplicate deref mappings
// ICE when an argument is a method call
HirId autoderef_mappings_id = expr.get_mappings ().get_hirid ();
context->insert_autoderef_mappings (autoderef_mappings_id,
std::move (candidate.adjustments));
context->insert_receiver (expr.get_mappings ().get_hirid (), receiver_tyty);
PathProbeCandidate &resolved_candidate = candidate.candidate;
TyTy::BaseType *lookup_tyty = candidate.candidate.ty;
NodeId resolved_node_id
= resolved_candidate.is_impl_candidate ()
? resolved_candidate.item.impl.impl_item->get_impl_mappings ()
.get_nodeid ()
: resolved_candidate.item.trait.item_ref->get_mappings ().get_nodeid ();
if (lookup_tyty->get_kind () != TyTy::TypeKind::FNDEF)
{
RichLocation r (expr.get_locus ());
r.add_range (resolved_candidate.locus);
rust_error_at (r, "associated impl item is not a method");
return false;
}
TyTy::BaseType *lookup = lookup_tyty;
TyTy::FnType *fn = static_cast<TyTy::FnType *> (lookup);
if (!fn->is_method ())
{
RichLocation r (expr.get_locus ());
r.add_range (resolved_candidate.locus);
rust_error_at (r, "associated function is not a method");
return false;
}
// fn traits only support tuple argument passing so we need to implicitly set
// this up to get the same type checking we get in the rest of the pipeline
std::vector<TyTy::TyVar> call_args;
for (auto &arg : expr.get_arguments ())
{
TyTy::BaseType *a = TypeCheckExpr::Resolve (arg.get ());
call_args.push_back (TyTy::TyVar (a->get_ref ()));
}
// crate implicit tuple
HirId implicit_arg_id = mappings->get_next_hir_id ();
Analysis::NodeMapping mapping (mappings->get_current_crate (), UNKNOWN_NODEID,
implicit_arg_id, UNKNOWN_LOCAL_DEFID);
TyTy::TupleType *tuple
= new TyTy::TupleType (implicit_arg_id, expr.get_locus (), call_args);
context->insert_implicit_type (implicit_arg_id, tuple);
std::vector<TyTy::Argument> args;
TyTy::Argument a (mapping, tuple,
expr.get_locus () /*FIXME is there a better location*/);
args.push_back (std::move (a));
TyTy::BaseType *function_ret_tyty
= TyTy::TypeCheckMethodCallExpr::go (fn, expr.get_mappings (), args,
expr.get_locus (), expr.get_locus (),
adjusted_self, context);
if (function_ret_tyty == nullptr
|| function_ret_tyty->get_kind () == TyTy::TypeKind::ERROR)
{
rust_error_at (expr.get_locus (),
"failed check fn trait call-expr MethodCallExpr");
return false;
}
// store the expected fntype
context->insert_operator_overload (expr.get_mappings ().get_hirid (), fn);
// set up the resolved name on the path
resolver->insert_resolved_name (expr.get_mappings ().get_nodeid (),
resolved_node_id);
// return the result of the function back
*result = function_ret_tyty;
return true;
}
bool
TypeCheckExpr::validate_arithmetic_type (
const TyTy::BaseType *tyty, HIR::ArithmeticOrLogicalExpr::ExprType expr_type)

View File

@@ -103,6 +103,13 @@ protected:
HIR::OperatorExprMeta expr, TyTy::BaseType *lhs,
TyTy::BaseType *rhs);
bool resolve_fn_trait_call (HIR::CallExpr &expr,
TyTy::BaseType *function_tyty,
TyTy::BaseType **result);
HIR::PathIdentSegment
resolve_possible_fn_trait_call_method_name (const TyTy::BaseType &receiver);
private:
TypeCheckExpr ();

View File

@@ -70,6 +70,8 @@ public:
return item.trait_item;
}
TyTy::FnType *get_context_type ();
private:
union Item
{

View File

@@ -154,5 +154,37 @@ TypeCheckContext::peek_context ()
return return_type_stack.back ().first;
}
// TypeCheckContextItem
TyTy::FnType *
TypeCheckContextItem::get_context_type ()
{
auto &context = *TypeCheckContext::get ();
HirId reference = UNKNOWN_HIRID;
switch (get_type ())
{
case ITEM:
reference = get_item ()->get_mappings ().get_hirid ();
break;
case IMPL_ITEM:
reference = get_impl_item ().second->get_mappings ().get_hirid ();
break;
case TRAIT_ITEM:
reference = get_trait_item ()->get_mappings ().get_hirid ();
break;
}
rust_assert (reference != UNKNOWN_HIRID);
TyTy::BaseType *lookup = nullptr;
bool ok = context.lookup_type (reference, &lookup);
rust_assert (ok);
rust_assert (lookup->get_kind () == TyTy::TypeKind::FNDEF);
return static_cast<TyTy::FnType *> (lookup);
}
} // namespace Resolver
} // namespace Rust

View File

@@ -871,6 +871,29 @@ public:
ok = true;
}
void visit (const ClosureType &type) override
{
if (base->get_def_id () != type.get_def_id ())
{
BaseCmp::visit (type);
return;
}
if (!base->get_parameters ().can_eq (&type.get_parameters (), false))
{
BaseCmp::visit (type);
return;
}
if (!base->get_result_type ().can_eq (&type.get_result_type (), false))
{
BaseCmp::visit (type);
return;
}
ok = true;
}
private:
const BaseType *get_base () const override { return base; }
const ClosureType *base;

View File

@@ -624,7 +624,45 @@ class ClosureRules : public BaseRules
public:
ClosureRules (ClosureType *base) : BaseRules (base), base (base) {}
// TODO
void visit (InferType &type) override
{
if (type.get_infer_kind () != InferType::InferTypeKind::GENERAL)
{
BaseRules::visit (type);
return;
}
resolved = base->clone ();
resolved->set_ref (type.get_ref ());
}
void visit (ClosureType &type) override
{
if (base->get_def_id () != type.get_def_id ())
{
BaseRules::visit (type);
return;
}
TyTy::BaseType *args_res
= base->get_parameters ().unify (&type.get_parameters ());
if (args_res == nullptr || args_res->get_kind () == TypeKind::ERROR)
{
BaseRules::visit (type);
return;
}
TyTy::BaseType *res
= base->get_result_type ().unify (&type.get_result_type ());
if (res == nullptr || res->get_kind () == TypeKind::ERROR)
{
BaseRules::visit (type);
return;
}
resolved = base->clone ();
resolved->set_ref (type.get_ref ());
}
private:
BaseType *get_base () override { return base; }

View File

@@ -27,6 +27,7 @@
#include "rust-substitution-mapper.h"
#include "rust-hir-trait-ref.h"
#include "rust-hir-type-bounds.h"
#include "rust-hir-trait-resolve.h"
#include "options.h"
namespace Rust {
@@ -200,6 +201,7 @@ BaseType::bounds_compatible (const BaseType &other, Location locus,
rust_error_at (r,
"bounds not satisfied for %s %<%s%> is not satisfied",
other.get_name ().c_str (), missing_preds.c_str ());
// rust_assert (!emit_error);
}
}
@@ -1672,7 +1674,9 @@ ClosureType::accept_vis (TyConstVisitor &vis) const
std::string
ClosureType::as_string () const
{
return "TODO";
std::string params_buf = parameters->as_string ();
return "|" + params_buf + "| {" + result_type.get_tyty ()->as_string ()
+ "} {" + raw_bounds_as_string () + "}";
}
BaseType *
@@ -1699,8 +1703,10 @@ ClosureType::is_equal (const BaseType &other) const
BaseType *
ClosureType::clone () const
{
return new ClosureType (get_ref (), get_ty_ref (), ident, id, parameter_types,
result_type, clone_substs (), get_combined_refs ());
return new ClosureType (get_ref (), get_ty_ref (), ident, id,
(TyTy::TupleType *) parameters->clone (), result_type,
clone_substs (), get_combined_refs (),
specified_bounds);
}
BaseType *
@@ -1716,6 +1722,52 @@ ClosureType::handle_substitions (SubstitutionArgumentMappings mappings)
return nullptr;
}
void
ClosureType::setup_fn_once_output () const
{
// lookup the lang items
auto fn_once_lang_item = Analysis::RustLangItem::ItemType::FN_ONCE;
auto fn_once_output_lang_item
= Analysis::RustLangItem::ItemType::FN_ONCE_OUTPUT;
DefId trait_id = UNKNOWN_DEFID;
bool trait_lang_item_defined
= mappings->lookup_lang_item (fn_once_lang_item, &trait_id);
rust_assert (trait_lang_item_defined);
DefId trait_item_id = UNKNOWN_DEFID;
bool trait_item_lang_item_defined
= mappings->lookup_lang_item (fn_once_output_lang_item, &trait_item_id);
rust_assert (trait_item_lang_item_defined);
// resolve to the trait
HIR::Item *item = mappings->lookup_defid (trait_id);
rust_assert (item->get_item_kind () == HIR::Item::ItemKind::Trait);
HIR::Trait *trait = static_cast<HIR::Trait *> (item);
Resolver::TraitReference *trait_ref
= Resolver::TraitResolver::Resolve (*trait);
rust_assert (!trait_ref->is_error ());
// resolve to trait item
HIR::TraitItem *trait_item
= mappings->lookup_trait_item_defid (trait_item_id);
rust_assert (trait_item != nullptr);
rust_assert (trait_item->get_item_kind ()
== HIR::TraitItem::TraitItemKind::TYPE);
std::string item_identifier = trait_item->trait_identifier ();
// setup associated types #[lang = "fn_once_output"]
Resolver::TraitItemReference *item_reference = nullptr;
bool found = trait_ref->lookup_trait_item_by_type (
item_identifier, Resolver::TraitItemReference::TraitItemType::TYPE,
&item_reference);
rust_assert (found);
// setup
item_reference->associated_type_set (&get_result_type ());
}
void
ArrayType::accept_vis (TyVisitor &vis)
{

View File

@@ -1626,31 +1626,35 @@ class ClosureType : public BaseType, public SubstitutionRef
{
public:
ClosureType (HirId ref, DefId id, RustIdent ident,
std::vector<TyVar> parameter_types, TyVar result_type,
TyTy::TupleType *parameters, TyVar result_type,
std::vector<SubstitutionParamMapping> subst_refs,
std::set<HirId> refs = std::set<HirId> ())
std::set<HirId> refs = std::set<HirId> (),
std::vector<TypeBoundPredicate> specified_bounds
= std::vector<TypeBoundPredicate> ())
: BaseType (ref, ref, TypeKind::CLOSURE, ident, refs),
SubstitutionRef (std::move (subst_refs),
SubstitutionArgumentMappings::error ()),
parameter_types (std::move (parameter_types)),
result_type (std::move (result_type)), id (id)
parameters (parameters), result_type (std::move (result_type)), id (id)
{
LocalDefId local_def_id = id.localDefId;
rust_assert (local_def_id != UNKNOWN_LOCAL_DEFID);
inherit_bounds (specified_bounds);
}
ClosureType (HirId ref, HirId ty_ref, RustIdent ident, DefId id,
std::vector<TyVar> parameter_types, TyVar result_type,
TyTy::TupleType *parameters, TyVar result_type,
std::vector<SubstitutionParamMapping> subst_refs,
std::set<HirId> refs = std::set<HirId> ())
std::set<HirId> refs = std::set<HirId> (),
std::vector<TypeBoundPredicate> specified_bounds
= std::vector<TypeBoundPredicate> ())
: BaseType (ref, ty_ref, TypeKind::CLOSURE, ident, refs),
SubstitutionRef (std::move (subst_refs),
SubstitutionArgumentMappings::error ()),
parameter_types (std::move (parameter_types)),
result_type (std::move (result_type)), id (id)
parameters (parameters), result_type (std::move (result_type)), id (id)
{
LocalDefId local_def_id = id.localDefId;
rust_assert (local_def_id != UNKNOWN_LOCAL_DEFID);
inherit_bounds (specified_bounds);
}
void accept_vis (TyVisitor &vis) override;
@@ -1669,13 +1673,8 @@ public:
bool is_concrete () const override final
{
for (auto &param : parameter_types)
{
auto p = param.get_tyty ();
if (!p->is_concrete ())
return false;
}
return result_type.get_tyty ()->is_concrete ();
return parameters->is_concrete ()
&& result_type.get_tyty ()->is_concrete ();
}
bool needs_generic_substitutions () const override final
@@ -1693,8 +1692,15 @@ public:
ClosureType *
handle_substitions (SubstitutionArgumentMappings mappings) override final;
TyTy::TupleType &get_parameters () const { return *parameters; }
TyTy::BaseType &get_result_type () const { return *result_type.get_tyty (); }
DefId get_def_id () const { return id; }
void setup_fn_once_output () const;
private:
std::vector<TyVar> parameter_types;
TyTy::TupleType *parameters;
TyVar result_type;
DefId id;
};