mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 22:02:23 +01:00
Richard SmithandJon Ross-Perkins
fcfb1345d5
For an expression such as `(Type as Interface).AssocFn()`, track the `Self` type `Type` in the result of the member access so that it's available when checking the function call. This introduces a new kind of type, `ImplFunctionType`, that represents the type of a function that is expected within an impl, modeled as the type of the function within the interface plus a value to use as `Self`. Calls to values of this type behave like calls to the underlying function except that the `Self` parameter is pre-bound to the self type from the facet. In order to support this, fix an issue where the imported list of generic bindings lost their association with their enclosing generic. This adds a little complexity to `import_ref`, including a new recursive cycle that I intend to address in a follow-up PR. --------- Co-authored-by: Jon Ross-Perkins <jperkins@google.com>
121 lines
4.0 KiB
C++
121 lines
4.0 KiB
C++
// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
|
|
// Exceptions. See /LICENSE for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
#include "toolchain/sem_ir/function.h"
|
|
|
|
#include "toolchain/sem_ir/file.h"
|
|
#include "toolchain/sem_ir/generic.h"
|
|
#include "toolchain/sem_ir/ids.h"
|
|
|
|
namespace Carbon::SemIR {
|
|
|
|
auto GetCalleeFunction(const File& sem_ir, InstId callee_id) -> CalleeFunction {
|
|
CalleeFunction result = {.function_id = FunctionId::None,
|
|
.enclosing_specific_id = SpecificId::None,
|
|
.resolved_specific_id = SpecificId::None,
|
|
.self_type_id = InstId::None,
|
|
.self_id = InstId::None,
|
|
.is_error = false};
|
|
|
|
if (auto specific_function =
|
|
sem_ir.insts().TryGetAs<SpecificFunction>(callee_id)) {
|
|
result.resolved_specific_id = specific_function->specific_id;
|
|
callee_id = specific_function->callee_id;
|
|
}
|
|
|
|
if (auto bound_method = sem_ir.insts().TryGetAs<BoundMethod>(callee_id)) {
|
|
result.self_id = bound_method->object_id;
|
|
callee_id = bound_method->function_decl_id;
|
|
}
|
|
|
|
// Identify the function we're calling by its type.
|
|
auto val_id = sem_ir.constant_values().GetConstantInstId(callee_id);
|
|
if (!val_id.has_value()) {
|
|
return result;
|
|
}
|
|
auto fn_type_inst =
|
|
sem_ir.types().GetAsInst(sem_ir.insts().Get(val_id).type_id());
|
|
|
|
if (auto impl_fn_type = fn_type_inst.TryAs<FunctionTypeWithSelfType>()) {
|
|
// Combine the associated function's `Self` with the interface function
|
|
// data.
|
|
result.self_type_id = impl_fn_type->self_id;
|
|
fn_type_inst = sem_ir.insts().Get(impl_fn_type->interface_function_type_id);
|
|
}
|
|
|
|
auto fn_type = fn_type_inst.TryAs<FunctionType>();
|
|
if (!fn_type) {
|
|
result.is_error = fn_type_inst.Is<SemIR::ErrorInst>();
|
|
return result;
|
|
}
|
|
|
|
result.function_id = fn_type->function_id;
|
|
result.enclosing_specific_id = fn_type->specific_id;
|
|
return result;
|
|
}
|
|
|
|
auto Function::ParamPatternInfo::GetNameId(const File& sem_ir) -> NameId {
|
|
return sem_ir.entity_names().Get(entity_name_id).name_id;
|
|
}
|
|
|
|
auto Function::GetParamPatternInfoFromPatternId(const File& sem_ir,
|
|
InstId pattern_id)
|
|
-> ParamPatternInfo {
|
|
auto inst_id = pattern_id;
|
|
auto inst = sem_ir.insts().Get(inst_id);
|
|
|
|
if (auto addr_pattern = inst.TryAs<SemIR::AddrPattern>()) {
|
|
inst_id = addr_pattern->inner_id;
|
|
inst = sem_ir.insts().Get(inst_id);
|
|
}
|
|
|
|
auto param_pattern_id = inst_id;
|
|
auto param_pattern_inst = inst.As<SemIR::AnyParamPattern>();
|
|
|
|
inst_id = param_pattern_inst.subpattern_id;
|
|
inst = sem_ir.insts().Get(inst_id);
|
|
|
|
auto binding_pattern = inst.As<AnyBindingPattern>();
|
|
return {.inst_id = param_pattern_id,
|
|
.inst = param_pattern_inst,
|
|
.entity_name_id = binding_pattern.entity_name_id};
|
|
}
|
|
|
|
auto Function::GetNameFromPatternId(const File& sem_ir, InstId pattern_id)
|
|
-> SemIR::NameId {
|
|
auto inst_id = pattern_id;
|
|
auto inst = sem_ir.insts().Get(inst_id);
|
|
|
|
if (auto addr_pattern = inst.TryAs<SemIR::AddrPattern>()) {
|
|
inst_id = addr_pattern->inner_id;
|
|
inst = sem_ir.insts().Get(inst_id);
|
|
}
|
|
|
|
if (inst_id == SemIR::ErrorInst::SingletonInstId) {
|
|
return SemIR::NameId::None;
|
|
}
|
|
|
|
auto param_pattern_inst = inst.As<SemIR::AnyParamPattern>();
|
|
|
|
inst_id = param_pattern_inst.subpattern_id;
|
|
inst = sem_ir.insts().Get(inst_id);
|
|
|
|
if (inst.Is<ReturnSlotPattern>()) {
|
|
return SemIR::NameId::ReturnSlot;
|
|
}
|
|
auto binding_pattern = inst.As<AnyBindingPattern>();
|
|
return sem_ir.entity_names().Get(binding_pattern.entity_name_id).name_id;
|
|
}
|
|
|
|
auto Function::GetDeclaredReturnType(const File& file,
|
|
SpecificId specific_id) const -> TypeId {
|
|
if (!return_slot_pattern_id.has_value()) {
|
|
return TypeId::None;
|
|
}
|
|
return GetTypeInSpecific(file, specific_id,
|
|
file.insts().Get(return_slot_pattern_id).type_id());
|
|
}
|
|
|
|
} // namespace Carbon::SemIR
|