value.cpp 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887
  1. // Part of the Carbon Language project, under the Apache License v2.0 with LLVM
  2. // Exceptions. See /LICENSE for license information.
  3. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  4. #include "explorer/interpreter/value.h"
  5. #include <algorithm>
  6. #include "common/check.h"
  7. #include "explorer/common/arena.h"
  8. #include "explorer/common/error_builders.h"
  9. #include "explorer/interpreter/action.h"
  10. #include "llvm/ADT/StringExtras.h"
  11. #include "llvm/Support/Casting.h"
  12. #include "llvm/Support/Error.h"
  13. namespace Carbon {
  14. using llvm::cast;
  15. using llvm::dyn_cast;
  16. auto StructValue::FindField(const std::string& name) const
  17. -> std::optional<Nonnull<const Value*>> {
  18. for (const NamedValue& element : elements_) {
  19. if (element.name == name) {
  20. return element.value;
  21. }
  22. }
  23. return std::nullopt;
  24. }
  25. static auto GetMember(Nonnull<Arena*> arena, Nonnull<const Value*> v,
  26. const FieldPath::Component& field,
  27. SourceLocation source_loc, Nonnull<const Value*> me_value)
  28. -> ErrorOr<Nonnull<const Value*>> {
  29. const std::string& f = field.name();
  30. if (field.witness().has_value()) {
  31. Nonnull<const Witness*> witness = cast<Witness>(*field.witness());
  32. switch (witness->kind()) {
  33. case Value::Kind::ImplWitness: {
  34. auto* impl_witness = cast<ImplWitness>(witness);
  35. if (std::optional<Nonnull<const Declaration*>> mem_decl =
  36. FindMember(f, impl_witness->declaration().members());
  37. mem_decl.has_value()) {
  38. const auto& fun_decl = cast<FunctionDeclaration>(**mem_decl);
  39. if (fun_decl.is_method()) {
  40. return arena->New<BoundMethodValue>(&fun_decl, v,
  41. impl_witness->type_args(),
  42. impl_witness->witnesses());
  43. } else {
  44. // Class function.
  45. auto* fun = cast<FunctionValue>(*fun_decl.constant_value());
  46. return arena->New<FunctionValue>(&fun->declaration(),
  47. impl_witness->type_args(),
  48. impl_witness->witnesses());
  49. }
  50. } else {
  51. return CompilationError(source_loc)
  52. << "member " << f << " not in " << *witness;
  53. }
  54. }
  55. case Value::Kind::SymbolicWitness: {
  56. return RuntimeError(source_loc)
  57. << "member lookup for " << f << " in symbolic " << *witness
  58. << " not implemented yet";
  59. }
  60. default:
  61. CARBON_FATAL() << "expected Witness, not " << *witness;
  62. }
  63. }
  64. switch (v->kind()) {
  65. case Value::Kind::StructValue: {
  66. std::optional<Nonnull<const Value*>> field =
  67. cast<StructValue>(*v).FindField(f);
  68. if (field == std::nullopt) {
  69. return RuntimeError(source_loc) << "member " << f << " not in " << *v;
  70. }
  71. return *field;
  72. }
  73. case Value::Kind::NominalClassValue: {
  74. const auto& object = cast<NominalClassValue>(*v);
  75. // Look for a field.
  76. // Note that the value representation of an empty class is a
  77. // `StructType`, not a `StructValue`.
  78. std::optional<Nonnull<const Value*>> field;
  79. if (auto* struct_value = dyn_cast<StructValue>(&object.inits())) {
  80. field = struct_value->FindField(f);
  81. }
  82. if (field.has_value()) {
  83. return *field;
  84. } else {
  85. // Look for a method in the object's class
  86. const auto& class_type = cast<NominalClassType>(object.type());
  87. std::optional<Nonnull<const FunctionValue*>> func =
  88. class_type.FindFunction(f);
  89. if (func == std::nullopt) {
  90. return RuntimeError(source_loc) << "member " << f << " not in " << *v
  91. << " or its " << class_type;
  92. } else if ((*func)->declaration().is_method()) {
  93. // Found a method. Turn it into a bound method.
  94. const FunctionValue& m = cast<FunctionValue>(**func);
  95. return arena->New<BoundMethodValue>(&m.declaration(), me_value,
  96. class_type.type_args(),
  97. class_type.witnesses());
  98. } else {
  99. // Found a class function
  100. return arena->New<FunctionValue>(&(*func)->declaration(),
  101. class_type.type_args(),
  102. class_type.witnesses());
  103. }
  104. }
  105. }
  106. case Value::Kind::ChoiceType: {
  107. const auto& choice = cast<ChoiceType>(*v);
  108. if (!choice.FindAlternative(f)) {
  109. return RuntimeError(source_loc)
  110. << "alternative " << f << " not in " << *v;
  111. }
  112. return arena->New<AlternativeConstructorValue>(f, choice.name());
  113. }
  114. case Value::Kind::NominalClassType: {
  115. // Access a class function.
  116. const NominalClassType& class_type = cast<NominalClassType>(*v);
  117. std::optional<Nonnull<const FunctionValue*>> fun =
  118. class_type.FindFunction(f);
  119. if (fun == std::nullopt) {
  120. return RuntimeError(source_loc)
  121. << "class function " << f << " not in " << *v;
  122. }
  123. return arena->New<FunctionValue>(&(*fun)->declaration(),
  124. class_type.type_args(),
  125. class_type.witnesses());
  126. }
  127. default:
  128. CARBON_FATAL() << "field access not allowed for value " << *v;
  129. }
  130. }
  131. auto Value::GetMember(Nonnull<Arena*> arena, const FieldPath& path,
  132. SourceLocation source_loc,
  133. Nonnull<const Value*> me_value) const
  134. -> ErrorOr<Nonnull<const Value*>> {
  135. Nonnull<const Value*> value(this);
  136. for (const FieldPath::Component& field : path.components_) {
  137. CARBON_ASSIGN_OR_RETURN(
  138. value, Carbon::GetMember(arena, value, field, source_loc, me_value));
  139. }
  140. return value;
  141. }
  142. static auto SetFieldImpl(
  143. Nonnull<Arena*> arena, Nonnull<const Value*> value,
  144. std::vector<FieldPath::Component>::const_iterator path_begin,
  145. std::vector<FieldPath::Component>::const_iterator path_end,
  146. Nonnull<const Value*> field_value, SourceLocation source_loc)
  147. -> ErrorOr<Nonnull<const Value*>> {
  148. if (path_begin == path_end) {
  149. return field_value;
  150. }
  151. switch (value->kind()) {
  152. case Value::Kind::StructValue: {
  153. std::vector<NamedValue> elements = cast<StructValue>(*value).elements();
  154. auto it = std::find_if(elements.begin(), elements.end(),
  155. [path_begin](const NamedValue& element) {
  156. return element.name == (*path_begin).name();
  157. });
  158. if (it == elements.end()) {
  159. return RuntimeError(source_loc)
  160. << "field " << (*path_begin).name() << " not in " << *value;
  161. }
  162. CARBON_ASSIGN_OR_RETURN(
  163. it->value, SetFieldImpl(arena, it->value, path_begin + 1, path_end,
  164. field_value, source_loc));
  165. return arena->New<StructValue>(elements);
  166. }
  167. case Value::Kind::NominalClassValue: {
  168. const NominalClassValue& object = cast<NominalClassValue>(*value);
  169. CARBON_ASSIGN_OR_RETURN(Nonnull<const Value*> inits,
  170. SetFieldImpl(arena, &object.inits(), path_begin,
  171. path_end, field_value, source_loc));
  172. return arena->New<NominalClassValue>(&object.type(), inits);
  173. }
  174. case Value::Kind::TupleValue: {
  175. std::vector<Nonnull<const Value*>> elements =
  176. cast<TupleValue>(*value).elements();
  177. // TODO(geoffromer): update FieldPath to hold integers as well as strings.
  178. int index = std::stoi((*path_begin).name());
  179. if (index < 0 || static_cast<size_t>(index) >= elements.size()) {
  180. return RuntimeError(source_loc) << "index " << (*path_begin).name()
  181. << " out of range in " << *value;
  182. }
  183. CARBON_ASSIGN_OR_RETURN(
  184. elements[index], SetFieldImpl(arena, elements[index], path_begin + 1,
  185. path_end, field_value, source_loc));
  186. return arena->New<TupleValue>(elements);
  187. }
  188. default:
  189. CARBON_FATAL() << "field access not allowed for value " << *value;
  190. }
  191. }
  192. auto Value::SetField(Nonnull<Arena*> arena, const FieldPath& path,
  193. Nonnull<const Value*> field_value,
  194. SourceLocation source_loc) const
  195. -> ErrorOr<Nonnull<const Value*>> {
  196. return SetFieldImpl(arena, Nonnull<const Value*>(this),
  197. path.components_.begin(), path.components_.end(),
  198. field_value, source_loc);
  199. }
  200. static auto PrintNameWithBindings(llvm::raw_ostream& out,
  201. Nonnull<const Declaration*> declaration,
  202. const BindingMap& args) {
  203. out << GetName(*declaration).value_or("(anonymous)");
  204. // TODO: Print '()' if declaration is parameterized but no args are provided.
  205. if (!args.empty()) {
  206. out << "(";
  207. llvm::ListSeparator sep;
  208. for (const auto& [bind, val] : args) {
  209. out << sep << bind->name() << " = " << *val;
  210. }
  211. out << ")";
  212. }
  213. }
  214. void Value::Print(llvm::raw_ostream& out) const {
  215. switch (kind()) {
  216. case Value::Kind::AlternativeConstructorValue: {
  217. const auto& alt = cast<AlternativeConstructorValue>(*this);
  218. out << alt.choice_name() << "." << alt.alt_name();
  219. break;
  220. }
  221. case Value::Kind::BindingPlaceholderValue: {
  222. const auto& placeholder = cast<BindingPlaceholderValue>(*this);
  223. out << "Placeholder<";
  224. if (placeholder.value_node().has_value()) {
  225. out << (*placeholder.value_node());
  226. } else {
  227. out << "_";
  228. }
  229. out << ">";
  230. break;
  231. }
  232. case Value::Kind::AddrValue: {
  233. const auto& addr = cast<AddrValue>(*this);
  234. out << "Addr<" << addr.pattern() << ">";
  235. break;
  236. }
  237. case Value::Kind::AlternativeValue: {
  238. const auto& alt = cast<AlternativeValue>(*this);
  239. out << "alt " << alt.choice_name() << "." << alt.alt_name() << " "
  240. << alt.argument();
  241. break;
  242. }
  243. case Value::Kind::StructValue: {
  244. const auto& struct_val = cast<StructValue>(*this);
  245. out << "{";
  246. llvm::ListSeparator sep;
  247. for (const NamedValue& element : struct_val.elements()) {
  248. out << sep << "." << element.name << " = " << *element.value;
  249. }
  250. out << "}";
  251. break;
  252. }
  253. case Value::Kind::NominalClassValue: {
  254. const auto& s = cast<NominalClassValue>(*this);
  255. out << cast<NominalClassType>(s.type()).declaration().name() << s.inits();
  256. break;
  257. }
  258. case Value::Kind::TupleValue: {
  259. out << "(";
  260. llvm::ListSeparator sep;
  261. for (Nonnull<const Value*> element : cast<TupleValue>(*this).elements()) {
  262. out << sep << *element;
  263. }
  264. out << ")";
  265. break;
  266. }
  267. case Value::Kind::IntValue:
  268. out << cast<IntValue>(*this).value();
  269. break;
  270. case Value::Kind::BoolValue:
  271. out << (cast<BoolValue>(*this).value() ? "true" : "false");
  272. break;
  273. case Value::Kind::FunctionValue: {
  274. const FunctionValue& fun = cast<FunctionValue>(*this);
  275. out << "fun<" << fun.declaration().name() << ">";
  276. if (!fun.type_args().empty()) {
  277. out << "[";
  278. llvm::ListSeparator sep;
  279. for (const auto& [ty_var, ty_arg] : fun.type_args()) {
  280. out << sep << *ty_var << "=" << *ty_arg;
  281. }
  282. out << "]";
  283. }
  284. if (!fun.witnesses().empty()) {
  285. out << "{|";
  286. llvm::ListSeparator sep;
  287. for (const auto& [impl_bind, witness] : fun.witnesses()) {
  288. out << sep << *witness;
  289. }
  290. out << "|}";
  291. }
  292. break;
  293. }
  294. case Value::Kind::BoundMethodValue: {
  295. const BoundMethodValue& method = cast<BoundMethodValue>(*this);
  296. out << "bound_method<" << method.declaration().name() << ">";
  297. if (!method.type_args().empty()) {
  298. out << "[";
  299. llvm::ListSeparator sep;
  300. for (const auto& [ty_var, ty_arg] : method.type_args()) {
  301. out << sep << *ty_var << "=" << *ty_arg;
  302. }
  303. out << "]";
  304. }
  305. if (!method.witnesses().empty()) {
  306. out << "{|";
  307. llvm::ListSeparator sep;
  308. for (const auto& [impl_bind, witness] : method.witnesses()) {
  309. out << sep << *witness;
  310. }
  311. out << "|}";
  312. }
  313. break;
  314. }
  315. case Value::Kind::PointerValue:
  316. out << "ptr<" << cast<PointerValue>(*this).address() << ">";
  317. break;
  318. case Value::Kind::LValue:
  319. out << "lval<" << cast<LValue>(*this).address() << ">";
  320. break;
  321. case Value::Kind::BoolType:
  322. out << "Bool";
  323. break;
  324. case Value::Kind::IntType:
  325. out << "i32";
  326. break;
  327. case Value::Kind::TypeType:
  328. out << "Type";
  329. break;
  330. case Value::Kind::AutoType:
  331. out << "auto";
  332. break;
  333. case Value::Kind::ContinuationType:
  334. out << "Continuation";
  335. break;
  336. case Value::Kind::PointerType:
  337. out << cast<PointerType>(*this).type() << "*";
  338. break;
  339. case Value::Kind::FunctionType: {
  340. const auto& fn_type = cast<FunctionType>(*this);
  341. out << "fn ";
  342. if (!fn_type.deduced_bindings().empty()) {
  343. out << "[";
  344. llvm::ListSeparator sep;
  345. for (Nonnull<const GenericBinding*> deduced :
  346. fn_type.deduced_bindings()) {
  347. out << sep << *deduced;
  348. }
  349. out << "]";
  350. }
  351. out << fn_type.parameters() << " -> " << fn_type.return_type();
  352. break;
  353. }
  354. case Value::Kind::StructType: {
  355. out << "{";
  356. llvm::ListSeparator sep;
  357. for (const auto& [name, type] : cast<StructType>(*this).fields()) {
  358. out << sep << "." << name << ": " << *type;
  359. }
  360. out << "}";
  361. break;
  362. }
  363. case Value::Kind::NominalClassType: {
  364. const auto& class_type = cast<NominalClassType>(*this);
  365. out << "class ";
  366. PrintNameWithBindings(out, &class_type.declaration(),
  367. class_type.type_args());
  368. if (!class_type.witnesses().empty()) {
  369. out << " witnesses ";
  370. llvm::ListSeparator sep;
  371. for (const auto& [impl_bind, witness] : class_type.witnesses()) {
  372. out << sep << *witness;
  373. }
  374. }
  375. break;
  376. }
  377. case Value::Kind::InterfaceType: {
  378. const auto& iface_type = cast<InterfaceType>(*this);
  379. out << "interface ";
  380. PrintNameWithBindings(out, &iface_type.declaration(), iface_type.args());
  381. break;
  382. }
  383. case Value::Kind::ConstraintType: {
  384. const auto& constraint = cast<ConstraintType>(*this);
  385. out << "constraint ";
  386. llvm::ListSeparator combine(" & ");
  387. for (const ConstraintType::LookupContext& ctx :
  388. constraint.lookup_contexts()) {
  389. out << combine << *ctx.context;
  390. }
  391. out << " where ";
  392. llvm::ListSeparator sep;
  393. for (const ConstraintType::ImplConstraint& impl :
  394. constraint.impl_constraints()) {
  395. // TODO: Skip cases where `impl.type` is `.Self` and the interface is
  396. // in `lookup_contexts()`.
  397. out << sep << *impl.type << " is " << *impl.interface;
  398. }
  399. for (const ConstraintType::EqualityConstraint& equality :
  400. constraint.equality_constraints()) {
  401. out << sep;
  402. llvm::ListSeparator equal(" == ");
  403. for (Nonnull<const Value*> value : equality.values) {
  404. out << equal << *value;
  405. }
  406. }
  407. break;
  408. }
  409. case Value::Kind::ImplWitness: {
  410. const auto& witness = cast<ImplWitness>(*this);
  411. out << "witness " << *witness.declaration().impl_type() << " as "
  412. << witness.declaration().interface();
  413. break;
  414. }
  415. case Value::Kind::SymbolicWitness: {
  416. const auto& witness = cast<SymbolicWitness>(*this);
  417. out << "witness " << witness.impl_expression();
  418. break;
  419. }
  420. case Value::Kind::ParameterizedEntityName:
  421. out << *GetName(cast<ParameterizedEntityName>(*this).declaration());
  422. break;
  423. case Value::Kind::MemberName: {
  424. const auto& member_name = cast<MemberName>(*this);
  425. if (member_name.base_type().has_value()) {
  426. out << *member_name.base_type().value();
  427. }
  428. if (member_name.base_type().has_value() &&
  429. member_name.interface().has_value()) {
  430. out << "(";
  431. }
  432. if (member_name.interface().has_value()) {
  433. out << *member_name.interface().value();
  434. }
  435. out << "." << member_name.name();
  436. if (member_name.base_type().has_value() &&
  437. member_name.interface().has_value()) {
  438. out << ")";
  439. }
  440. break;
  441. }
  442. case Value::Kind::ChoiceType:
  443. out << "choice " << cast<ChoiceType>(*this).name();
  444. break;
  445. case Value::Kind::VariableType:
  446. out << cast<VariableType>(*this).binding();
  447. break;
  448. case Value::Kind::ContinuationValue: {
  449. out << cast<ContinuationValue>(*this).stack();
  450. break;
  451. }
  452. case Value::Kind::StringType:
  453. out << "String";
  454. break;
  455. case Value::Kind::StringValue:
  456. out << "\"";
  457. out.write_escaped(cast<StringValue>(*this).value());
  458. out << "\"";
  459. break;
  460. case Value::Kind::TypeOfClassType:
  461. out << "typeof(" << cast<TypeOfClassType>(*this).class_type() << ")";
  462. break;
  463. case Value::Kind::TypeOfInterfaceType:
  464. out << "typeof("
  465. << cast<TypeOfInterfaceType>(*this)
  466. .interface_type()
  467. .declaration()
  468. .name()
  469. << ")";
  470. break;
  471. case Value::Kind::TypeOfConstraintType:
  472. out << "typeof(" << cast<TypeOfConstraintType>(*this).constraint_type()
  473. << ")";
  474. break;
  475. case Value::Kind::TypeOfChoiceType:
  476. out << "typeof(" << cast<TypeOfChoiceType>(*this).choice_type().name()
  477. << ")";
  478. break;
  479. case Value::Kind::TypeOfParameterizedEntityName:
  480. out << "parameterized entity name "
  481. << cast<TypeOfParameterizedEntityName>(*this).name();
  482. break;
  483. case Value::Kind::TypeOfMemberName: {
  484. out << "member name " << cast<TypeOfMemberName>(*this).member().name();
  485. break;
  486. }
  487. case Value::Kind::StaticArrayType: {
  488. const auto& array_type = cast<StaticArrayType>(*this);
  489. out << "[" << array_type.element_type() << "; " << array_type.size()
  490. << "]";
  491. break;
  492. }
  493. }
  494. }
  495. ContinuationValue::StackFragment::~StackFragment() {
  496. CARBON_CHECK(reversed_todo_.empty())
  497. << "All StackFragments must be empty before the Carbon program ends.";
  498. }
  499. void ContinuationValue::StackFragment::StoreReversed(
  500. std::vector<std::unique_ptr<Action>> reversed_todo) {
  501. CARBON_CHECK(reversed_todo_.empty());
  502. reversed_todo_ = std::move(reversed_todo);
  503. }
  504. void ContinuationValue::StackFragment::RestoreTo(
  505. Stack<std::unique_ptr<Action>>& todo) {
  506. while (!reversed_todo_.empty()) {
  507. todo.Push(std::move(reversed_todo_.back()));
  508. reversed_todo_.pop_back();
  509. }
  510. }
  511. void ContinuationValue::StackFragment::Clear() {
  512. // We destroy the underlying Actions explicitly to ensure they're
  513. // destroyed in the correct order.
  514. for (auto& action : reversed_todo_) {
  515. action.reset();
  516. }
  517. reversed_todo_.clear();
  518. }
  519. void ContinuationValue::StackFragment::Print(llvm::raw_ostream& out) const {
  520. out << "{";
  521. llvm::ListSeparator sep(" :: ");
  522. for (const std::unique_ptr<Action>& action : reversed_todo_) {
  523. out << sep << *action;
  524. }
  525. out << "}";
  526. }
  527. // Check whether two binding maps, which are assumed to have the same keys, are
  528. // equal.
  529. static auto BindingMapEqual(const BindingMap& map1, const BindingMap& map2)
  530. -> bool {
  531. CARBON_CHECK(map1.size() == map2.size()) << "maps should have same keys";
  532. for (const auto& [key, value] : map1) {
  533. if (!ValueEqual(value, map2.at(key))) {
  534. return false;
  535. }
  536. }
  537. return true;
  538. }
  539. auto TypeEqual(Nonnull<const Value*> t1, Nonnull<const Value*> t2) -> bool {
  540. if (t1->kind() != t2->kind()) {
  541. return false;
  542. }
  543. switch (t1->kind()) {
  544. case Value::Kind::PointerType:
  545. return TypeEqual(&cast<PointerType>(*t1).type(),
  546. &cast<PointerType>(*t2).type());
  547. case Value::Kind::FunctionType: {
  548. const auto& fn1 = cast<FunctionType>(*t1);
  549. const auto& fn2 = cast<FunctionType>(*t2);
  550. return TypeEqual(&fn1.parameters(), &fn2.parameters()) &&
  551. TypeEqual(&fn1.return_type(), &fn2.return_type());
  552. }
  553. case Value::Kind::StructType: {
  554. const auto& struct1 = cast<StructType>(*t1);
  555. const auto& struct2 = cast<StructType>(*t2);
  556. if (struct1.fields().size() != struct2.fields().size()) {
  557. return false;
  558. }
  559. for (size_t i = 0; i < struct1.fields().size(); ++i) {
  560. if (struct1.fields()[i].name != struct2.fields()[i].name ||
  561. !TypeEqual(struct1.fields()[i].value, struct2.fields()[i].value)) {
  562. return false;
  563. }
  564. }
  565. return true;
  566. }
  567. case Value::Kind::NominalClassType: {
  568. const auto& class1 = cast<NominalClassType>(*t1);
  569. const auto& class2 = cast<NominalClassType>(*t2);
  570. return class1.declaration().name() == class2.declaration().name() &&
  571. BindingMapEqual(class1.type_args(), class2.type_args());
  572. }
  573. case Value::Kind::InterfaceType: {
  574. const auto& iface1 = cast<InterfaceType>(*t1);
  575. const auto& iface2 = cast<InterfaceType>(*t2);
  576. return iface1.declaration().name() == iface2.declaration().name() &&
  577. BindingMapEqual(iface1.args(), iface2.args());
  578. }
  579. case Value::Kind::ConstraintType: {
  580. const auto& constraint1 = cast<ConstraintType>(*t1);
  581. const auto& constraint2 = cast<ConstraintType>(*t2);
  582. if (constraint1.impl_constraints().size() !=
  583. constraint2.impl_constraints().size() ||
  584. constraint1.equality_constraints().size() !=
  585. constraint2.equality_constraints().size() ||
  586. constraint1.lookup_contexts().size() !=
  587. constraint2.lookup_contexts().size()) {
  588. return false;
  589. }
  590. for (size_t i = 0; i < constraint1.impl_constraints().size(); ++i) {
  591. const auto& impl1 = constraint1.impl_constraints()[i];
  592. const auto& impl2 = constraint2.impl_constraints()[i];
  593. if (!TypeEqual(impl1.type, impl2.type) ||
  594. !TypeEqual(impl1.interface, impl2.interface)) {
  595. return false;
  596. }
  597. }
  598. for (size_t i = 0; i < constraint1.equality_constraints().size(); ++i) {
  599. const auto& equality1 = constraint1.equality_constraints()[i];
  600. const auto& equality2 = constraint2.equality_constraints()[i];
  601. if (equality1.values.size() != equality2.values.size()) {
  602. return false;
  603. }
  604. for (size_t j = 0; j < equality1.values.size(); ++j) {
  605. if (!ValueEqual(equality1.values[i], equality2.values[i])) {
  606. return false;
  607. }
  608. }
  609. }
  610. for (size_t i = 0; i < constraint1.lookup_contexts().size(); ++i) {
  611. const auto& context1 = constraint1.lookup_contexts()[i];
  612. const auto& context2 = constraint2.lookup_contexts()[i];
  613. if (!TypeEqual(context1.context, context2.context)) {
  614. return false;
  615. }
  616. }
  617. return true;
  618. }
  619. case Value::Kind::ChoiceType:
  620. return cast<ChoiceType>(*t1).name() == cast<ChoiceType>(*t2).name();
  621. case Value::Kind::TupleValue: {
  622. const auto& tup1 = cast<TupleValue>(*t1);
  623. const auto& tup2 = cast<TupleValue>(*t2);
  624. if (tup1.elements().size() != tup2.elements().size()) {
  625. return false;
  626. }
  627. for (size_t i = 0; i < tup1.elements().size(); ++i) {
  628. if (!TypeEqual(tup1.elements()[i], tup2.elements()[i])) {
  629. return false;
  630. }
  631. }
  632. return true;
  633. }
  634. case Value::Kind::IntType:
  635. case Value::Kind::BoolType:
  636. case Value::Kind::ContinuationType:
  637. case Value::Kind::TypeType:
  638. case Value::Kind::StringType:
  639. return true;
  640. case Value::Kind::VariableType:
  641. return &cast<VariableType>(*t1).binding() ==
  642. &cast<VariableType>(*t2).binding();
  643. case Value::Kind::TypeOfClassType:
  644. return TypeEqual(&cast<TypeOfClassType>(*t1).class_type(),
  645. &cast<TypeOfClassType>(*t2).class_type());
  646. case Value::Kind::TypeOfInterfaceType:
  647. return TypeEqual(&cast<TypeOfInterfaceType>(*t1).interface_type(),
  648. &cast<TypeOfInterfaceType>(*t2).interface_type());
  649. case Value::Kind::TypeOfConstraintType:
  650. return TypeEqual(&cast<TypeOfConstraintType>(*t1).constraint_type(),
  651. &cast<TypeOfConstraintType>(*t2).constraint_type());
  652. case Value::Kind::TypeOfChoiceType:
  653. return TypeEqual(&cast<TypeOfChoiceType>(*t1).choice_type(),
  654. &cast<TypeOfChoiceType>(*t2).choice_type());
  655. case Value::Kind::StaticArrayType: {
  656. const auto& array1 = cast<StaticArrayType>(*t1);
  657. const auto& array2 = cast<StaticArrayType>(*t2);
  658. return TypeEqual(&array1.element_type(), &array2.element_type()) &&
  659. array1.size() == array2.size();
  660. }
  661. case Value::Kind::IntValue:
  662. case Value::Kind::BoolValue:
  663. case Value::Kind::FunctionValue:
  664. case Value::Kind::BoundMethodValue:
  665. case Value::Kind::StructValue:
  666. case Value::Kind::NominalClassValue:
  667. case Value::Kind::AlternativeValue:
  668. case Value::Kind::AlternativeConstructorValue:
  669. case Value::Kind::StringValue:
  670. case Value::Kind::PointerValue:
  671. case Value::Kind::LValue:
  672. case Value::Kind::BindingPlaceholderValue:
  673. case Value::Kind::AddrValue:
  674. case Value::Kind::ContinuationValue:
  675. case Value::Kind::ParameterizedEntityName:
  676. case Value::Kind::MemberName:
  677. case Value::Kind::TypeOfParameterizedEntityName:
  678. case Value::Kind::TypeOfMemberName:
  679. CARBON_FATAL() << "TypeEqual used to compare non-type values\n"
  680. << *t1 << "\n"
  681. << *t2;
  682. case Value::Kind::ImplWitness:
  683. case Value::Kind::SymbolicWitness:
  684. CARBON_FATAL() << "TypeEqual: unexpected Witness";
  685. break;
  686. case Value::Kind::AutoType:
  687. CARBON_FATAL() << "TypeEqual: unexpected AutoType";
  688. break;
  689. }
  690. }
  691. // Returns true if the two values are equal and returns false otherwise.
  692. //
  693. // This function implements the `==` operator of Carbon.
  694. auto ValueEqual(Nonnull<const Value*> v1, Nonnull<const Value*> v2) -> bool {
  695. if (v1->kind() != v2->kind()) {
  696. return false;
  697. }
  698. switch (v1->kind()) {
  699. case Value::Kind::IntValue:
  700. return cast<IntValue>(*v1).value() == cast<IntValue>(*v2).value();
  701. case Value::Kind::BoolValue:
  702. return cast<BoolValue>(*v1).value() == cast<BoolValue>(*v2).value();
  703. case Value::Kind::FunctionValue: {
  704. std::optional<Nonnull<const Statement*>> body1 =
  705. cast<FunctionValue>(*v1).declaration().body();
  706. std::optional<Nonnull<const Statement*>> body2 =
  707. cast<FunctionValue>(*v2).declaration().body();
  708. return body1.has_value() == body2.has_value() &&
  709. (!body1.has_value() || *body1 == *body2);
  710. }
  711. case Value::Kind::BoundMethodValue: {
  712. const auto& m1 = cast<BoundMethodValue>(*v1);
  713. const auto& m2 = cast<BoundMethodValue>(*v2);
  714. std::optional<Nonnull<const Statement*>> body1 = m1.declaration().body();
  715. std::optional<Nonnull<const Statement*>> body2 = m2.declaration().body();
  716. return ValueEqual(m1.receiver(), m2.receiver()) &&
  717. body1.has_value() == body2.has_value() &&
  718. (!body1.has_value() || *body1 == *body2);
  719. }
  720. case Value::Kind::TupleValue: {
  721. const std::vector<Nonnull<const Value*>>& elements1 =
  722. cast<TupleValue>(*v1).elements();
  723. const std::vector<Nonnull<const Value*>>& elements2 =
  724. cast<TupleValue>(*v2).elements();
  725. if (elements1.size() != elements2.size()) {
  726. return false;
  727. }
  728. for (size_t i = 0; i < elements1.size(); ++i) {
  729. if (!ValueEqual(elements1[i], elements2[i])) {
  730. return false;
  731. }
  732. }
  733. return true;
  734. }
  735. case Value::Kind::StructValue: {
  736. const auto& struct_v1 = cast<StructValue>(*v1);
  737. const auto& struct_v2 = cast<StructValue>(*v2);
  738. CARBON_CHECK(struct_v1.elements().size() == struct_v2.elements().size());
  739. for (size_t i = 0; i < struct_v1.elements().size(); ++i) {
  740. CARBON_CHECK(struct_v1.elements()[i].name ==
  741. struct_v2.elements()[i].name);
  742. if (!ValueEqual(struct_v1.elements()[i].value,
  743. struct_v2.elements()[i].value)) {
  744. return false;
  745. }
  746. }
  747. return true;
  748. }
  749. case Value::Kind::StringValue:
  750. return cast<StringValue>(*v1).value() == cast<StringValue>(*v2).value();
  751. case Value::Kind::ParameterizedEntityName: {
  752. std::optional<std::string> name1 =
  753. GetName(cast<ParameterizedEntityName>(v1)->declaration());
  754. std::optional<std::string> name2 =
  755. GetName(cast<ParameterizedEntityName>(v2)->declaration());
  756. CARBON_CHECK(name1.has_value() && name2.has_value())
  757. << "parameterized name refers to unnamed declaration";
  758. return *name1 == *name2;
  759. }
  760. case Value::Kind::IntType:
  761. case Value::Kind::BoolType:
  762. case Value::Kind::TypeType:
  763. case Value::Kind::FunctionType:
  764. case Value::Kind::PointerType:
  765. case Value::Kind::AutoType:
  766. case Value::Kind::StructType:
  767. case Value::Kind::NominalClassType:
  768. case Value::Kind::InterfaceType:
  769. case Value::Kind::ConstraintType:
  770. case Value::Kind::ImplWitness:
  771. case Value::Kind::SymbolicWitness:
  772. case Value::Kind::ChoiceType:
  773. case Value::Kind::ContinuationType:
  774. case Value::Kind::VariableType:
  775. case Value::Kind::StringType:
  776. case Value::Kind::TypeOfClassType:
  777. case Value::Kind::TypeOfInterfaceType:
  778. case Value::Kind::TypeOfConstraintType:
  779. case Value::Kind::TypeOfChoiceType:
  780. case Value::Kind::TypeOfParameterizedEntityName:
  781. case Value::Kind::TypeOfMemberName:
  782. case Value::Kind::StaticArrayType:
  783. return TypeEqual(v1, v2);
  784. case Value::Kind::NominalClassValue:
  785. case Value::Kind::AlternativeValue:
  786. case Value::Kind::BindingPlaceholderValue:
  787. case Value::Kind::AddrValue:
  788. case Value::Kind::AlternativeConstructorValue:
  789. case Value::Kind::ContinuationValue:
  790. case Value::Kind::PointerValue:
  791. case Value::Kind::LValue:
  792. case Value::Kind::MemberName:
  793. // TODO: support pointer comparisons once we have a clearer distinction
  794. // between pointers and lvalues.
  795. CARBON_FATAL() << "ValueEqual does not support this kind of value: "
  796. << *v1;
  797. }
  798. }
  799. auto ChoiceType::FindAlternative(std::string_view name) const
  800. -> std::optional<Nonnull<const Value*>> {
  801. for (const NamedValue& alternative : alternatives_) {
  802. if (alternative.name == name) {
  803. return alternative.value;
  804. }
  805. }
  806. return std::nullopt;
  807. }
  808. auto NominalClassType::FindFunction(const std::string& name) const
  809. -> std::optional<Nonnull<const FunctionValue*>> {
  810. for (const auto& member : declaration().members()) {
  811. switch (member->kind()) {
  812. case DeclarationKind::FunctionDeclaration: {
  813. const auto& fun = cast<FunctionDeclaration>(*member);
  814. if (fun.name() == name) {
  815. return &cast<FunctionValue>(**fun.constant_value());
  816. }
  817. break;
  818. }
  819. default:
  820. break;
  821. }
  822. }
  823. return std::nullopt;
  824. }
  825. auto FindMember(const std::string& name,
  826. llvm::ArrayRef<Nonnull<Declaration*>> members)
  827. -> std::optional<Nonnull<const Declaration*>> {
  828. for (Nonnull<const Declaration*> member : members) {
  829. if (std::optional<std::string> mem_name = GetName(*member);
  830. mem_name.has_value()) {
  831. if (*mem_name == name) {
  832. return member;
  833. }
  834. }
  835. }
  836. return std::nullopt;
  837. }
  838. auto Member::name() const -> std::string {
  839. if (const Declaration* decl = member_.dyn_cast<const Declaration*>()) {
  840. return GetName(*decl).value();
  841. } else {
  842. return member_.get<const NamedValue*>()->name;
  843. }
  844. }
  845. auto Member::type() const -> const Value& {
  846. if (const Declaration* decl = member_.dyn_cast<const Declaration*>()) {
  847. return decl->static_type();
  848. } else {
  849. return *member_.get<const NamedValue*>()->value;
  850. }
  851. }
  852. void ImplBinding::Print(llvm::raw_ostream& out) const {
  853. out << "impl binding " << *type_var_ << " as " << *iface_;
  854. }
  855. void ImplBinding::PrintID(llvm::raw_ostream& out) const {
  856. out << *type_var_ << " as " << *iface_;
  857. }
  858. } // namespace Carbon