mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-25 13:20:12 +01:00
Note this makes some structural choices that I'm not sure how popular they'll be... Most notably, I could remove `DeclarationKind` because I'm using `std::visit` and `Declaration::Visitor` in reality. I could go to `switch(tag())` instead, but I'm wondering if this approach will be well received for the separation of ownership. Alternatively, I could move `Declaration` around so that it actually implements the things like `TopLevel` -- I did feel weird with the old structure of typecheck.cpp implementing members of declaration.h, though.
109 lines
3.5 KiB
C++
109 lines
3.5 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 "executable_semantics/ast/declaration.h"
|
|
|
|
#include <iostream>
|
|
|
|
namespace Carbon {
|
|
|
|
auto Declaration::MakeFunctionDeclaration(FunctionDefinition definition)
|
|
-> const Declaration {
|
|
Declaration d;
|
|
d.value = FunctionDeclaration({.definition = definition});
|
|
return d;
|
|
}
|
|
|
|
auto Declaration::MakeStructDeclaration(int line_num, std::string name,
|
|
std::list<Member*> members)
|
|
-> const Declaration {
|
|
Declaration d;
|
|
d.value = StructDeclaration(
|
|
{.definition = StructDefinition({.line_num = line_num,
|
|
.name = std::move(name),
|
|
.members = std::move(members)})});
|
|
return d;
|
|
}
|
|
|
|
auto Declaration::MakeChoiceDeclaration(
|
|
int line_num, std::string name,
|
|
std::list<std::pair<std::string, const Expression*>> alternatives)
|
|
-> const Declaration {
|
|
Declaration d;
|
|
d.value = ChoiceDeclaration({.line_num = line_num,
|
|
.name = std::move(name),
|
|
.alternatives = std::move(alternatives)});
|
|
return d;
|
|
}
|
|
|
|
auto Declaration::MakeVariableDeclaration(int source_location, std::string name,
|
|
const Expression* type,
|
|
const Expression* initializer)
|
|
-> const Declaration {
|
|
Declaration d;
|
|
d.value = VariableDeclaration({.source_location = source_location,
|
|
.name = std::move(name),
|
|
.type = type,
|
|
.initializer = initializer});
|
|
return d;
|
|
}
|
|
|
|
auto Declaration::GetFunctionDeclaration() const -> const FunctionDeclaration& {
|
|
return std::get<FunctionDeclaration>(value);
|
|
}
|
|
|
|
auto Declaration::GetStructDeclaration() const -> const StructDeclaration& {
|
|
return std::get<StructDeclaration>(value);
|
|
}
|
|
|
|
auto Declaration::GetChoiceDeclaration() const -> const ChoiceDeclaration& {
|
|
return std::get<ChoiceDeclaration>(value);
|
|
}
|
|
|
|
auto Declaration::GetVariableDeclaration() const -> const VariableDeclaration& {
|
|
return std::get<VariableDeclaration>(value);
|
|
}
|
|
|
|
void Declaration::Print() const {
|
|
switch (tag()) {
|
|
case DeclarationKind::FunctionDeclaration:
|
|
GetFunctionDeclaration().definition.Print();
|
|
break;
|
|
|
|
case DeclarationKind::StructDeclaration: {
|
|
const StructDefinition& struct_def = GetStructDeclaration().definition;
|
|
std::cout << "struct " << struct_def.name << " {" << std::endl;
|
|
for (Member* m : struct_def.members) {
|
|
m->Print();
|
|
}
|
|
std::cout << "}" << std::endl;
|
|
break;
|
|
}
|
|
|
|
case DeclarationKind::ChoiceDeclaration: {
|
|
const auto& choice = GetChoiceDeclaration();
|
|
std::cout << "choice " << choice.name << " {" << std::endl;
|
|
for (const auto& [name, signature] : choice.alternatives) {
|
|
std::cout << "alt " << name << " ";
|
|
PrintExp(signature);
|
|
std::cout << ";" << std::endl;
|
|
}
|
|
std::cout << "}" << std::endl;
|
|
break;
|
|
}
|
|
|
|
case DeclarationKind::VariableDeclaration: {
|
|
const auto& var = GetVariableDeclaration();
|
|
std::cout << "var ";
|
|
PrintExp(var.type);
|
|
std::cout << " : " << var.name << " = ";
|
|
PrintExp(var.initializer);
|
|
std::cout << std::endl;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace Carbon
|