Files
carbon-lang/toolchain/check/decl_introducer_state.h
T
Jon Ross-Perkins 2d3842fc06 Implement 'extern library' support for functions. (#4220)
Support for types (particularly classes) is left as a TODO.

There's also an issue I'm observing with a "define in impl" test, but
this is probably an issue with resolving the prior declaration which is
imported indirectly. The PR was already feeling big, so I'm choosing to
cut here.

Note, this does not implement the rule "The owning library's API file
must import the `extern` declaration, and must also contain a
declaration."
2024-08-19 22:12:21 +00:00

90 lines
3.1 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
#ifndef CARBON_TOOLCHAIN_CHECK_DECL_INTRODUCER_STATE_H_
#define CARBON_TOOLCHAIN_CHECK_DECL_INTRODUCER_STATE_H_
#include "toolchain/check/keyword_modifier_set.h"
#include "toolchain/lex/token_kind.h"
#include "toolchain/parse/node_ids.h"
namespace Carbon::Check {
// State stored for each declaration we are introducing: the kind of
// declaration and the keyword modifiers that apply to that declaration
// introducer.
struct DeclIntroducerState {
auto modifier_node_id(ModifierOrder order) -> Parse::NodeId {
return ordered_modifier_node_ids[static_cast<int8_t>(order)];
}
auto set_modifier_node_id(ModifierOrder order, Parse::NodeId node_id)
-> void {
ordered_modifier_node_ids[static_cast<int8_t>(order)] = node_id;
}
// The token kind of the introducer.
Lex::TokenKind kind;
// Nodes of modifiers on this declaration, in expected order. `Invalid` if no
// modifier of that kind is present.
Parse::NodeId
ordered_modifier_node_ids[static_cast<int8_t>(ModifierOrder::Decl) + 1] =
{Parse::NodeId::Invalid, Parse::NodeId::Invalid,
Parse::NodeId::Invalid};
// Invariant: contains just the modifiers represented by `saw_*_modifier`.
KeywordModifierSet modifier_set = KeywordModifierSet();
// If there's an `extern library` in use, the library name.
SemIR::LibraryNameId extern_library = SemIR::LibraryNameId::Invalid;
};
// Stack of `DeclIntroducerState` values, representing all the declaration
// introducers we are currently nested within. Commonly size 0 or 1, as nested
// introducers are rare.
class DeclIntroducerStateStack {
private:
// Returns true for introducer tokens. Supports Push/Pop `requires`.
template <Lex::TokenKind::RawEnumType Kind>
static constexpr auto IsDeclIntroducer() -> bool {
switch (Kind) {
#define CARBON_TOKEN(...)
#define CARBON_DECL_INTRODUCER_TOKEN(Name, ...) case Lex::TokenKind::Name:
#include "toolchain/lex/token_kind.def"
return true;
default:
return false;
}
}
public:
// Begins a declaration introducer `Kind`.
template <Lex::TokenKind::RawEnumType Kind>
requires(IsDeclIntroducer<Kind>())
auto Push() -> void {
stack_.push_back({.kind = Lex::TokenKind::Make(Kind)});
}
// Gets the state of declaration at the top of the stack -- the innermost
// declaration currently being processed.
auto innermost() -> DeclIntroducerState& { return stack_.back(); }
// Finishes a declaration introducer `Kind` and returns the produced state.
template <Lex::TokenKind::RawEnumType Kind>
requires(IsDeclIntroducer<Kind>())
auto Pop() -> DeclIntroducerState {
CARBON_CHECK(stack_.back().kind == Kind)
<< "Found: " << stack_.back().kind
<< " expected: " << Lex::TokenKind::Make(Kind);
return stack_.pop_back_val();
}
private:
llvm::SmallVector<DeclIntroducerState> stack_;
};
} // namespace Carbon::Check
#endif // CARBON_TOOLCHAIN_CHECK_DECL_INTRODUCER_STATE_H_