mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 21:20:11 +01:00
Geoff Romer
4c9049346d
See [here](https://docs.google.com/document/d/1rWcueFwIfZox6GKVGxiUG4cBzjrZ6djXiIDGyJDtrE4/edit?tab=t.0) for the design doc. This also removes the default value of the `result_type_inst_id` parameter of `HandleAction`, moves it before the action in the parameter list, and documents it. This solves two problems: - The default made it easy to forget, leading to unnecessary `TypeOfInst` instructions. - When it was present, putting it after the fairly "bulky" action argument tended to make the callsite harder to read.
91 lines
3.0 KiB
C++
91 lines
3.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/pattern.h"
|
|
|
|
#include "toolchain/base/kind_switch.h"
|
|
#include "toolchain/sem_ir/ids.h"
|
|
#include "toolchain/sem_ir/typed_insts.h"
|
|
|
|
namespace Carbon::SemIR {
|
|
|
|
// Returns the pattern instruction corresponding to the given ID, after
|
|
// unwrapping any simple pattern operators such as `var`.
|
|
static auto GetUnwrapped(const File& sem_ir, InstId pattern_id)
|
|
-> std::pair<InstId, Inst> {
|
|
auto inst_id = pattern_id;
|
|
auto inst = sem_ir.insts().Get(inst_id);
|
|
|
|
if (auto var_pattern = inst.TryAs<AnyVarPattern>()) {
|
|
inst_id = var_pattern->subpattern_id;
|
|
inst = sem_ir.insts().Get(inst_id);
|
|
}
|
|
|
|
return {inst_id, inst};
|
|
}
|
|
|
|
// Returns the name and entity name introduced by the given instruction if it is
|
|
// a binding pattern, or otherwise `{None, None}`.
|
|
static auto GetBoundEntityName(const File& sem_ir, Inst inst)
|
|
-> std::pair<NameId, EntityNameId> {
|
|
if (auto binding_pattern = inst.TryAs<AnyBindingPattern>()) {
|
|
return {sem_ir.entity_names().Get(binding_pattern->entity_name_id).name_id,
|
|
binding_pattern->entity_name_id};
|
|
}
|
|
return {NameId::None, EntityNameId::None};
|
|
}
|
|
|
|
auto IsSelfPattern(const File& sem_ir, InstId pattern_id) -> bool {
|
|
auto [_, inst] = GetUnwrapped(sem_ir, pattern_id);
|
|
auto [name_id, entity_name_id] = GetBoundEntityName(sem_ir, inst);
|
|
return name_id == NameId::SelfValue;
|
|
}
|
|
|
|
auto GetFirstBindingNameFromPatternId(const File& sem_ir, InstId pattern_id)
|
|
-> EntityNameId {
|
|
llvm::SmallVector<InstId> work_list = {pattern_id};
|
|
while (!work_list.empty()) {
|
|
auto [_, inst] = GetUnwrapped(sem_ir, work_list.pop_back_val());
|
|
if (auto tuple_patt = inst.TryAs<TuplePattern>()) {
|
|
auto block = sem_ir.inst_blocks().Get(tuple_patt->elements_id);
|
|
work_list.append(block.rbegin(), block.rend());
|
|
continue;
|
|
}
|
|
|
|
// TODO: Look through struct patterns.
|
|
|
|
if (auto ref_pattern = inst.TryAs<RefParamPattern>()) {
|
|
// TODO: This introduces a name, but we don't model it as a binding.
|
|
return EntityNameId::None;
|
|
}
|
|
|
|
auto [name_id, entity_name_id] = GetBoundEntityName(sem_ir, inst);
|
|
CARBON_CHECK(entity_name_id.has_value(), "Unhandled pattern inst kind {0}",
|
|
inst);
|
|
|
|
// Skip unnamed bindings.
|
|
if (name_id != NameId::Underscore) {
|
|
return entity_name_id;
|
|
}
|
|
}
|
|
return EntityNameId::None;
|
|
}
|
|
|
|
auto GetPrettyNameFromPatternId(const File& sem_ir, InstId pattern_id)
|
|
-> NameId {
|
|
auto [inst_id, inst] = GetUnwrapped(sem_ir, pattern_id);
|
|
CARBON_KIND_SWITCH(inst) {
|
|
case CARBON_KIND_ANY(SemIR::AnyLeafParamPattern, param_pattern): {
|
|
return param_pattern.pretty_name_id;
|
|
}
|
|
case CARBON_KIND_ANY(SemIR::AnyBindingPattern, _): {
|
|
return GetBoundEntityName(sem_ir, inst).first;
|
|
}
|
|
default:
|
|
return NameId::None;
|
|
}
|
|
}
|
|
|
|
} // namespace Carbon::SemIR
|