-
Notifications
You must be signed in to change notification settings - Fork 157
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
top-level: Add base
TopLevel
visitor
The `TopLevel` pass takes care of collecting definitions, placing them in the proper namespaces, and making them accessible for later resolution passes like `Early` and `Late`. It is meant to be run in a fixed point fashion, as import resolution, macro resolution and macro expansion may generate multiple new definitions. gcc/rust/ChangeLog: * Make-lang.in: Add new object file. * resolve/rust-toplevel-name-resolver-2.0.cc: New file. * resolve/rust-toplevel-name-resolver-2.0.h: New file.
- Loading branch information
1 parent
d18ff2f
commit 4103aa8
Showing
3 changed files
with
295 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,218 @@ | ||
// Copyright (C) 2020-2023 Free Software Foundation, Inc. | ||
|
||
// This file is part of GCC. | ||
|
||
// GCC is free software; you can redistribute it and/or modify it under | ||
// the terms of the GNU General Public License as published by the Free | ||
// Software Foundation; either version 3, or (at your option) any later | ||
// version. | ||
|
||
// GCC is distributed in the hope that it will be useful, but WITHOUT ANY | ||
// WARRANTY; without even the implied warranty of MERCHANTABILITY or | ||
// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | ||
// for more details. | ||
|
||
// You should have received a copy of the GNU General Public License | ||
// along with GCC; see the file COPYING3. If not see | ||
// <http://www.gnu.org/licenses/>. | ||
|
||
#include "rust-toplevel-name-resolver-2.0.h" | ||
#include "rust-ast-full.h" | ||
#include "rust-hir-map.h" | ||
|
||
namespace Rust { | ||
namespace Resolver2_0 { | ||
|
||
TopLevel::TopLevel (Resolver &resolver) : resolver (resolver) {} | ||
|
||
template <typename T> | ||
void | ||
TopLevel::insert_or_error_out (const Identifier &identifier, const T &node, | ||
Namespace ns) | ||
{ | ||
auto loc = node.get_locus (); | ||
auto node_id = node.get_node_id (); | ||
|
||
// keep track of each node's location to provide useful errors | ||
node_locations.emplace (node_id, loc); | ||
|
||
auto result = resolver.insert (identifier, node_id, ns); | ||
|
||
if (!result) | ||
{ | ||
rich_location rich_loc (line_table, loc); | ||
rich_loc.add_range (node_locations[result.error ().existing]); | ||
|
||
rust_error_at (rich_loc, "already defined", result.error ().name.c_str (), | ||
result.error ().existing); | ||
} | ||
} | ||
|
||
void | ||
TopLevel::go (AST::Crate &crate) | ||
{ | ||
for (auto &item : crate.items) | ||
item->accept_vis (*this); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::Module &module) | ||
{ | ||
// FIXME: Do we need to insert the module in the type namespace? | ||
|
||
auto sub_visitor = [this, &module] () { | ||
for (auto &item : module.get_items ()) | ||
item->accept_vis (*this); | ||
}; | ||
|
||
resolver.scoped (Rib::Kind::Module, module.get_node_id (), sub_visitor, | ||
module.get_name ()); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::MacroRulesDefinition ¯o) | ||
{ | ||
// FIXME: Do we want to insert macro rules here already? Probably, right? | ||
// So that we can easily resolve in `Early`? | ||
insert_or_error_out (macro.get_rule_name (), macro, Namespace::Macros); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::Function &function) | ||
{ | ||
insert_or_error_out (function.get_function_name (), function, | ||
Namespace::Values); | ||
|
||
auto def_fn | ||
= [this, &function] () { function.get_definition ()->accept_vis (*this); }; | ||
|
||
resolver.scoped (Rib::Kind::Function, function.get_node_id (), def_fn); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::Method &method) | ||
{ | ||
insert_or_error_out (method.get_method_name (), method, Namespace::Values); | ||
|
||
auto def_fn | ||
= [this, &method] () { method.get_definition ()->accept_vis (*this); }; | ||
|
||
resolver.scoped (Rib::Kind::Function, method.get_node_id (), def_fn); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::BlockExpr &expr) | ||
{ | ||
// extracting the lambda from the `scoped` call otherwise the code looks like | ||
// a hot turd thanks to our .clang-format | ||
|
||
auto sub_vis = [this, &expr] () { | ||
for (auto &stmt : expr.get_statements ()) | ||
stmt->accept_vis (*this); | ||
|
||
if (expr.has_tail_expr ()) | ||
expr.get_tail_expr ()->accept_vis (*this); | ||
}; | ||
|
||
resolver.scoped (Rib::Kind::Normal, expr.get_node_id (), sub_vis); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::StaticItem &static_item) | ||
{ | ||
auto sub_vis | ||
= [this, &static_item] () { static_item.get_expr ()->accept_vis (*this); }; | ||
|
||
resolver.scoped (Rib::Kind::Item, static_item.get_node_id (), sub_vis); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::TraitItemFunc &item) | ||
{ | ||
auto def_vis | ||
= [this, &item] () { item.get_definition ()->accept_vis (*this); }; | ||
|
||
if (item.has_definition ()) | ||
resolver.scoped (Rib::Kind::Function, item.get_node_id (), def_vis); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::StructStruct &struct_item) | ||
{ | ||
insert_or_error_out (struct_item.get_struct_name (), struct_item, | ||
Namespace::Types); | ||
|
||
// Do we need to insert the constructor in the value namespace as well? | ||
|
||
// Do we need to do anything if the struct is a unit struct? | ||
if (struct_item.is_unit_struct ()) | ||
insert_or_error_out (struct_item.get_struct_name (), struct_item, | ||
Namespace::Values); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::TupleStruct &tuple_struct) | ||
{ | ||
insert_or_error_out (tuple_struct.get_struct_name (), tuple_struct, | ||
Namespace::Types); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::EnumItem &variant) | ||
{ | ||
insert_or_error_out (variant.get_identifier (), variant, Namespace::Types); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::EnumItemTuple &variant) | ||
{ | ||
insert_or_error_out (variant.get_identifier (), variant, Namespace::Types); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::EnumItemStruct &variant) | ||
{ | ||
insert_or_error_out (variant.get_identifier (), variant, Namespace::Types); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::EnumItemDiscriminant &variant) | ||
{ | ||
insert_or_error_out (variant.get_identifier (), variant, Namespace::Types); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::Enum &enum_item) | ||
{ | ||
insert_or_error_out (enum_item.get_identifier (), enum_item, | ||
Namespace::Types); | ||
|
||
auto field_vis = [this, &enum_item] () { | ||
for (auto &variant : enum_item.get_variants ()) | ||
variant->accept_vis (*this); | ||
}; | ||
|
||
resolver.scoped (Rib::Kind::Item /* FIXME: Is that correct? */, | ||
enum_item.get_node_id (), field_vis, | ||
enum_item.get_identifier ()); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::Union &union_item) | ||
{ | ||
insert_or_error_out (union_item.get_identifier (), union_item, | ||
Namespace::Types); | ||
} | ||
|
||
void | ||
TopLevel::visit (AST::ConstantItem &const_item) | ||
{ | ||
auto expr_vis | ||
= [this, &const_item] () { const_item.get_expr ()->accept_vis (*this); }; | ||
|
||
resolver.scoped (Rib::Kind::ConstantItem, const_item.get_node_id (), | ||
expr_vis); | ||
} | ||
|
||
} // namespace Resolver2_0 | ||
} // namespace Rust |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,76 @@ | ||
// Copyright (C) 2020-2023 Free Software Foundation, Inc. | ||
|
||
// This file is part of GCC. | ||
|
||
// GCC is free software; you can redistribute it and/or modify it under | ||
// the terms of the GNU General Public License as published by the Free | ||
// Software Foundation; either version 3, or (at your option) any later | ||
// version. | ||
|
||
// GCC is distributed in the hope that it will be useful, but WITHOUT ANY | ||
// WARRANTY; without even the implied warranty of MERCHANTABILITY or | ||
// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | ||
// for more details. | ||
|
||
// You should have received a copy of the GNU General Public License | ||
// along with GCC; see the file COPYING3. If not see | ||
// <http://www.gnu.org/licenses/>. | ||
|
||
#ifndef RUST_TOPLEVEL_NAME_RESOLVER_2_0_H | ||
#define RUST_TOPLEVEL_NAME_RESOLVER_2_0_H | ||
|
||
#include "rust-ast-visitor.h" | ||
#include "rust-name-resolver-2.0.h" | ||
#include "rust-ast-resolve-base.h" | ||
|
||
namespace Rust { | ||
namespace Resolver2_0 { | ||
|
||
/** | ||
* The `TopLevel` visitor takes care of collecting all the definitions in a | ||
* crate, and inserting them into the proper namespaces. These definitions can | ||
* then be accessed by subsequent resolvers, such as `Early` or `Late`. | ||
*/ | ||
// TODO: Merge Resolver namespaces and use `public ResolverBase` | ||
class TopLevel : public ::Rust::Resolver::ResolverBase | ||
{ | ||
using ::Rust::Resolver::ResolverBase::visit; | ||
|
||
public: | ||
TopLevel (Resolver &resolver); | ||
|
||
void go (AST::Crate &crate); | ||
|
||
private: | ||
Resolver &resolver; | ||
|
||
// FIXME: Documentation | ||
template <typename T> | ||
void insert_or_error_out (const Identifier &identifier, const T &node, | ||
Namespace ns); | ||
|
||
// FIXME: Do we move these to our mappings? | ||
std::unordered_map<NodeId, location_t> node_locations; | ||
|
||
void visit (AST::Module &module) override; | ||
void visit (AST::MacroRulesDefinition ¯o) override; | ||
void visit (AST::Function &function) override; | ||
void visit (AST::Method &method) override; | ||
void visit (AST::BlockExpr &expr) override; | ||
void visit (AST::StaticItem &static_item) override; | ||
void visit (AST::TraitItemFunc &item) override; | ||
void visit (AST::StructStruct &struct_item) override; | ||
void visit (AST::TupleStruct &tuple_struct) override; | ||
void visit (AST::EnumItem &variant) override; | ||
void visit (AST::EnumItemTuple &variant) override; | ||
void visit (AST::EnumItemStruct &variant) override; | ||
void visit (AST::EnumItemDiscriminant &variant) override; | ||
void visit (AST::Enum &enum_item) override; | ||
void visit (AST::Union &union_item) override; | ||
void visit (AST::ConstantItem &const_item) override; | ||
}; | ||
|
||
} // namespace Resolver2_0 | ||
} // namespace Rust | ||
|
||
#endif // !RUST_TOPLEVEL_NAME_RESOLVER_2_0_H |