value.cpp 31 KB

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