stringify.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729
  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 "toolchain/sem_ir/stringify.h"
  5. #include <optional>
  6. #include <string>
  7. #include <utility>
  8. #include <variant>
  9. #include "common/raw_string_ostream.h"
  10. #include "common/variant_helpers.h"
  11. #include "toolchain/base/kind_switch.h"
  12. #include "toolchain/sem_ir/entity_with_params_base.h"
  13. #include "toolchain/sem_ir/ids.h"
  14. #include "toolchain/sem_ir/inst_kind.h"
  15. #include "toolchain/sem_ir/singleton_insts.h"
  16. #include "toolchain/sem_ir/struct_type_field.h"
  17. #include "toolchain/sem_ir/type_info.h"
  18. #include "toolchain/sem_ir/typed_insts.h"
  19. namespace Carbon::SemIR {
  20. // Map an instruction kind representing an expression into an integer describing
  21. // the precedence of that expression's syntax. Higher numbers correspond to
  22. // higher precedence.
  23. static auto GetPrecedence(InstKind kind) -> int {
  24. if (kind == ConstType::Kind) {
  25. return -1;
  26. }
  27. if (kind == PointerType::Kind) {
  28. return -2;
  29. }
  30. // TODO: Handle other kinds of expressions with precedence.
  31. return 0;
  32. }
  33. namespace {
  34. // Contains the stack of steps for `Stringify`.
  35. //
  36. // Note that when pushing items onto the stack, they're printed in the reverse
  37. // order of when they were pushed. All reference lifetimes must match the
  38. // lifetime of `Stringify`.
  39. class StepStack {
  40. public:
  41. // An individual step in the stack, which stringifies some component of a type
  42. // name.
  43. using Step = std::variant<InstId, llvm::StringRef, NameId, ElementIndex>;
  44. // Support `Push` for a qualified name. e.g., `A.B.C`.
  45. using QualifiedNameItem = std::pair<NameScopeId, NameId>;
  46. // Support `Push` for a qualified entity name. e.g., `A.B.C`.
  47. using EntityNameItem = std::pair<const EntityWithParamsBase&, SpecificId>;
  48. // The full set of things which can be pushed, including all members of
  49. // `Step`.
  50. using PushItem =
  51. std::variant<InstId, llvm::StringRef, NameId, ElementIndex,
  52. QualifiedNameItem, EntityNameItem, EntityNameId, TypeId,
  53. SpecificInterface, llvm::ListSeparator*>;
  54. // Starts a new stack, which always contains the first instruction to
  55. // stringify.
  56. explicit StepStack(const SemIR::File* file) : sem_ir_(file) {}
  57. // These push basic entries onto the stack.
  58. auto PushInstId(InstId inst_id) -> void { steps_.push_back(inst_id); }
  59. auto PushString(llvm::StringRef string) -> void { steps_.push_back(string); }
  60. auto PushNameId(NameId name_id) -> void { steps_.push_back(name_id); }
  61. auto PushElementIndex(ElementIndex element_index) -> void {
  62. steps_.push_back(element_index);
  63. }
  64. // Pushes all components of a qualified name (`A.B.C`) onto the stack.
  65. auto PushQualifiedName(NameScopeId name_scope_id, NameId name_id) -> void {
  66. PushNameId(name_id);
  67. while (name_scope_id.has_value() && name_scope_id != NameScopeId::Package) {
  68. const auto& name_scope = sem_ir_->name_scopes().Get(name_scope_id);
  69. // TODO: Decide how to print unnamed scopes.
  70. if (name_scope.name_id().has_value()) {
  71. PushString(".");
  72. // TODO: For a generic scope, pass a SpecificId to this function and
  73. // include the relevant arguments.
  74. PushNameId(name_scope.name_id());
  75. }
  76. name_scope_id = name_scope.parent_scope_id();
  77. }
  78. }
  79. // Pushes a specific's entity name onto the stack, such as `A.B(T)`.
  80. auto PushEntityName(const EntityWithParamsBase& entity,
  81. SpecificId specific_id) -> void {
  82. PushSpecificId(entity, specific_id);
  83. PushQualifiedName(entity.parent_scope_id, entity.name_id);
  84. }
  85. // Pushes a entity name onto the stack, such as `A.B`.
  86. auto PushEntityNameId(EntityNameId entity_name_id) -> void {
  87. const auto& entity_name = sem_ir_->entity_names().Get(entity_name_id);
  88. PushQualifiedName(entity_name.parent_scope_id, entity_name.name_id);
  89. }
  90. // Pushes an instruction by its TypeId.
  91. auto PushTypeId(TypeId type_id) -> void {
  92. PushInstId(sem_ir_->types().GetInstId(type_id));
  93. }
  94. // Pushes a specific interface.
  95. auto PushSpecificInterface(SpecificInterface specific_interface) -> void {
  96. PushEntityName(sem_ir_->interfaces().Get(specific_interface.interface_id),
  97. specific_interface.specific_id);
  98. }
  99. // Pushes a sequence of items onto the stack. This handles reversal, such that
  100. // the caller can pass items in print order instead of stack order.
  101. //
  102. // Note that with `ListSeparator`, the object's reference isn't stored, but
  103. // the separator `StringRef` will be. That should be a constant though, so is
  104. // safe.
  105. auto PushArray(llvm::ArrayRef<PushItem> items) -> void {
  106. for (auto item : llvm::reverse(items)) {
  107. VariantMatch(
  108. item, [&](InstId inst_id) { PushInstId(inst_id); },
  109. [&](llvm::StringRef string) { PushString(string); },
  110. [&](NameId name_id) { PushNameId(name_id); },
  111. [&](ElementIndex element_index) { PushElementIndex(element_index); },
  112. [&](QualifiedNameItem qualified_name) {
  113. PushQualifiedName(qualified_name.first, qualified_name.second);
  114. },
  115. [&](EntityNameItem entity_name) {
  116. PushEntityName(entity_name.first, entity_name.second);
  117. },
  118. [&](EntityNameId entity_name_id) {
  119. PushEntityNameId(entity_name_id);
  120. },
  121. [&](TypeId type_id) { PushTypeId(type_id); },
  122. [&](SpecificInterface specific_interface) {
  123. PushSpecificInterface(specific_interface);
  124. },
  125. [&](llvm::ListSeparator* sep) { PushString(*sep); });
  126. }
  127. }
  128. // Wraps `PushArray` without requiring `{}` for arguments.
  129. template <typename... T>
  130. auto Push(T... items) -> void {
  131. PushArray({items...});
  132. }
  133. auto empty() const -> bool { return steps_.empty(); }
  134. auto Pop() -> Step { return steps_.pop_back_val(); }
  135. private:
  136. // Handles the generic portion of a specific entity name, such as `(T)` in
  137. // `A.B(T)`.
  138. auto PushSpecificId(const EntityWithParamsBase& entity,
  139. SpecificId specific_id) -> void {
  140. if (!entity.param_patterns_id.has_value()) {
  141. return;
  142. }
  143. int num_params =
  144. sem_ir_->inst_blocks().Get(entity.param_patterns_id).size();
  145. if (!num_params) {
  146. PushString("()");
  147. return;
  148. }
  149. if (!specific_id.has_value()) {
  150. // The name of the generic was used within the generic itself.
  151. // TODO: Should we print the names of the generic parameters in this
  152. // case?
  153. return;
  154. }
  155. const auto& specific = sem_ir_->specifics().Get(specific_id);
  156. auto args =
  157. sem_ir_->inst_blocks().Get(specific.args_id).take_back(num_params);
  158. bool last = true;
  159. for (auto arg : llvm::reverse(args)) {
  160. PushString(last ? ")" : ", ");
  161. PushInstId(arg);
  162. last = false;
  163. }
  164. PushString("(");
  165. }
  166. const SemIR::File* sem_ir_;
  167. // Remaining steps to take.
  168. llvm::SmallVector<Step> steps_;
  169. };
  170. // Provides `StringifyInst` overloads for each instruction.
  171. class Stringifier {
  172. public:
  173. explicit Stringifier(const SemIR::File* sem_ir, StepStack* step_stack,
  174. llvm::raw_ostream* out)
  175. : sem_ir_(sem_ir), step_stack_(step_stack), out_(out) {}
  176. // By default try to print a constant, but otherwise may fail to
  177. // stringify.
  178. auto StringifyInstDefault(SemIR::InstId inst_id, Inst inst) -> void {
  179. // We don't know how to print this instruction, but it might have a
  180. // constant value that we can print.
  181. auto const_inst_id = sem_ir_->constant_values().GetConstantInstId(inst_id);
  182. if (const_inst_id.has_value() && const_inst_id != inst_id) {
  183. step_stack_->PushInstId(const_inst_id);
  184. return;
  185. }
  186. // We don't need to handle stringification for instructions that don't
  187. // show up in errors, but make it clear what's going on so that it's
  188. // clearer when stringification is needed.
  189. *out_ << "<cannot stringify " << inst_id << ": " << inst << ">";
  190. }
  191. template <typename InstT>
  192. auto StringifyInst(SemIR::InstId inst_id, InstT inst) -> void {
  193. // This doesn't use requires so that more specific overloads are chosen when
  194. // provided.
  195. static_assert(InstT::Kind.is_type() != InstIsType::Always ||
  196. std::same_as<InstT, WhereExpr>,
  197. "Types should have a dedicated overload");
  198. // TODO: We should have Stringify support for all types where
  199. // InstT::Kind.constant_kind() is neither Never nor Indirect.
  200. StringifyInstDefault(inst_id, inst);
  201. }
  202. // Singleton instructions use their IR name as a label.
  203. template <typename InstT>
  204. requires(IsSingletonInstKind(InstT::Kind))
  205. auto StringifyInst(SemIR::InstId /*inst_id*/, InstT /*inst*/) -> void {
  206. *out_ << InstT::Kind.ir_name();
  207. }
  208. auto StringifyInst(SemIR::InstId /*inst_id*/, ArrayType inst) -> void {
  209. *out_ << "array(";
  210. step_stack_->Push(inst.element_type_inst_id, ", ", inst.bound_id, ")");
  211. }
  212. auto StringifyInst(SemIR::InstId /*inst_id*/, AssociatedConstantDecl inst)
  213. -> void {
  214. const auto& assoc_const =
  215. sem_ir_->associated_constants().Get(inst.assoc_const_id);
  216. step_stack_->PushQualifiedName(assoc_const.parent_scope_id,
  217. assoc_const.name_id);
  218. }
  219. auto StringifyInst(SemIR::InstId /*inst_id*/, AssociatedEntityType inst)
  220. -> void {
  221. *out_ << "<associated entity in ";
  222. step_stack_->Push(">");
  223. step_stack_->PushSpecificInterface(
  224. SpecificInterface{inst.interface_id, inst.interface_specific_id});
  225. }
  226. auto StringifyInst(SemIR::InstId /*inst_id*/, BoolLiteral inst) -> void {
  227. step_stack_->Push(inst.value.ToBool() ? "true" : "false");
  228. }
  229. template <typename InstT>
  230. requires(std::same_as<InstT, BindAlias> ||
  231. std::same_as<InstT, BindSymbolicName> ||
  232. std::same_as<InstT, ExportDecl>)
  233. auto StringifyInst(SemIR::InstId /*inst_id*/, InstT inst) -> void {
  234. step_stack_->PushEntityNameId(inst.entity_name_id);
  235. }
  236. auto StringifyInst(SemIR::InstId /*inst_id*/, ClassType inst) -> void {
  237. const auto& class_info = sem_ir_->classes().Get(inst.class_id);
  238. if (auto literal_info = NumericTypeLiteralInfo::ForType(*sem_ir_, inst);
  239. literal_info.is_valid()) {
  240. literal_info.PrintLiteral(*sem_ir_, *out_);
  241. return;
  242. }
  243. step_stack_->PushEntityName(class_info, inst.specific_id);
  244. }
  245. auto StringifyInst(SemIR::InstId /*inst_id*/, ConstType inst) -> void {
  246. *out_ << "const ";
  247. // Add parentheses if required.
  248. if (GetPrecedence(sem_ir_->insts().Get(inst.inner_id).kind()) <
  249. GetPrecedence(SemIR::ConstType::Kind)) {
  250. *out_ << "(";
  251. // Note the `inst.inner_id` ends up here.
  252. step_stack_->PushString(")");
  253. }
  254. step_stack_->PushInstId(inst.inner_id);
  255. }
  256. auto StringifyInst(SemIR::InstId /*inst_id*/, FacetAccessType inst) -> void {
  257. // Given `T:! I`, print `T as type` as simply `T`.
  258. step_stack_->PushInstId(inst.facet_value_inst_id);
  259. }
  260. auto StringifyInst(SemIR::InstId /*inst_id*/, FacetType inst) -> void {
  261. const FacetTypeInfo& facet_type_info =
  262. sem_ir_->facet_types().Get(inst.facet_type_id);
  263. // Output `where` restrictions.
  264. bool some_where = false;
  265. if (facet_type_info.other_requirements) {
  266. step_stack_->PushString("...");
  267. some_where = true;
  268. }
  269. for (auto rewrite : llvm::reverse(facet_type_info.rewrite_constraints)) {
  270. if (some_where) {
  271. step_stack_->PushString(" and");
  272. }
  273. step_stack_->Push(" ", rewrite.lhs_id, " = ", rewrite.rhs_id);
  274. some_where = true;
  275. }
  276. if (!facet_type_info.self_impls_constraints.empty()) {
  277. if (some_where) {
  278. step_stack_->PushString(" and");
  279. }
  280. llvm::ListSeparator sep(" & ");
  281. for (auto impls : llvm::reverse(facet_type_info.self_impls_constraints)) {
  282. step_stack_->Push(impls, &sep);
  283. }
  284. step_stack_->PushString(" .Self impls ");
  285. some_where = true;
  286. }
  287. // TODO: Other restrictions from facet_type_info.
  288. if (some_where) {
  289. step_stack_->PushString(" where");
  290. }
  291. // Output extend interface requirements.
  292. if (facet_type_info.extend_constraints.empty()) {
  293. step_stack_->PushString("type");
  294. return;
  295. }
  296. llvm::ListSeparator sep(" & ");
  297. for (auto impls : llvm::reverse(facet_type_info.extend_constraints)) {
  298. step_stack_->Push(impls, &sep);
  299. }
  300. }
  301. auto StringifyInst(SemIR::InstId /*inst_id*/, FacetValue inst) -> void {
  302. // No need to output the witness.
  303. step_stack_->Push(inst.type_inst_id, " as ", inst.type_id);
  304. }
  305. auto StringifyInst(SemIR::InstId /*inst_id*/, FloatType inst) -> void {
  306. // TODO: Is this okay?
  307. if (auto width_value =
  308. sem_ir_->insts().TryGetAs<IntValue>(inst.bit_width_id)) {
  309. *out_ << "f";
  310. sem_ir_->ints().Get(width_value->int_id).print(*out_, /*isSigned=*/false);
  311. } else {
  312. *out_ << "Core.Float(";
  313. step_stack_->Push(inst.bit_width_id, ")");
  314. }
  315. }
  316. auto StringifyInst(SemIR::InstId /*inst_id*/, FunctionType inst) -> void {
  317. const auto& fn = sem_ir_->functions().Get(inst.function_id);
  318. *out_ << "<type of ";
  319. step_stack_->Push(
  320. StepStack::QualifiedNameItem{fn.parent_scope_id, fn.name_id}, ">");
  321. }
  322. auto StringifyInst(SemIR::InstId /*inst_id*/, FunctionTypeWithSelfType inst)
  323. -> void {
  324. StepStack::PushItem fn_name = SemIR::InstId::None;
  325. if (auto fn_inst = sem_ir_->insts().TryGetAs<FunctionType>(
  326. inst.interface_function_type_id)) {
  327. const auto& fn = sem_ir_->functions().Get(fn_inst->function_id);
  328. fn_name = StepStack::QualifiedNameItem(fn.parent_scope_id, fn.name_id);
  329. } else {
  330. fn_name = inst.interface_function_type_id;
  331. }
  332. *out_ << "<type of ";
  333. step_stack_->Push(fn_name, " in ", inst.self_id, ">");
  334. }
  335. auto StringifyInst(SemIR::InstId /*inst_id*/, GenericClassType inst) -> void {
  336. const auto& class_info = sem_ir_->classes().Get(inst.class_id);
  337. *out_ << "<type of ";
  338. step_stack_->Push(StepStack::QualifiedNameItem{class_info.parent_scope_id,
  339. class_info.name_id},
  340. ">");
  341. }
  342. auto StringifyInst(SemIR::InstId /*inst_id*/, GenericInterfaceType inst)
  343. -> void {
  344. const auto& interface = sem_ir_->interfaces().Get(inst.interface_id);
  345. *out_ << "<type of ";
  346. step_stack_->Push(StepStack::QualifiedNameItem{interface.parent_scope_id,
  347. interface.name_id},
  348. ">");
  349. }
  350. // Determine the specific interface that an impl witness instruction provides
  351. // an implementation of.
  352. // TODO: Should we track this in the type?
  353. auto TryGetSpecificInterfaceForImplWitness(SemIR::InstId impl_witness_id)
  354. -> std::optional<SpecificInterface> {
  355. if (auto lookup = sem_ir_->insts().TryGetAs<SemIR::LookupImplWitness>(
  356. impl_witness_id)) {
  357. return sem_ir_->specific_interfaces().Get(
  358. lookup->query_specific_interface_id);
  359. }
  360. // TODO: Handle ImplWitness.
  361. return std::nullopt;
  362. }
  363. auto StringifyInst(SemIR::InstId /*inst_id*/, ImplWitnessAccess inst)
  364. -> void {
  365. auto witness_inst_id =
  366. sem_ir_->constant_values().GetConstantInstId(inst.witness_id);
  367. if (auto specific_interface =
  368. TryGetSpecificInterfaceForImplWitness(witness_inst_id)) {
  369. const auto& interface =
  370. sem_ir_->interfaces().Get(specific_interface->interface_id);
  371. auto entities =
  372. sem_ir_->inst_blocks().Get(interface.associated_entities_id);
  373. size_t index = inst.index.index;
  374. CARBON_CHECK(index < entities.size(), "Access out of bounds.");
  375. auto entity_inst_id = entities[index];
  376. step_stack_->PushString(")");
  377. if (auto associated_const =
  378. sem_ir_->insts().TryGetAs<AssociatedConstantDecl>(
  379. entity_inst_id)) {
  380. step_stack_->PushNameId(sem_ir_->associated_constants()
  381. .Get(associated_const->assoc_const_id)
  382. .name_id);
  383. } else if (auto function_decl =
  384. sem_ir_->insts().TryGetAs<FunctionDecl>(entity_inst_id)) {
  385. const auto& function =
  386. sem_ir_->functions().Get(function_decl->function_id);
  387. step_stack_->PushNameId(function.name_id);
  388. } else {
  389. step_stack_->PushInstId(entity_inst_id);
  390. }
  391. step_stack_->Push(
  392. ".(",
  393. StepStack::EntityNameItem{interface, specific_interface->specific_id},
  394. ".");
  395. } else {
  396. step_stack_->Push(".(TODO: element ", inst.index, " in ", witness_inst_id,
  397. ")");
  398. }
  399. if (auto lookup =
  400. sem_ir_->insts().TryGetAs<LookupImplWitness>(witness_inst_id)) {
  401. bool period_self = false;
  402. if (auto sym_name = sem_ir_->insts().TryGetAs<BindSymbolicName>(
  403. lookup->query_self_inst_id)) {
  404. auto name_id =
  405. sem_ir_->entity_names().Get(sym_name->entity_name_id).name_id;
  406. period_self = (name_id == SemIR::NameId::PeriodSelf);
  407. }
  408. if (!period_self) {
  409. step_stack_->PushInstId(lookup->query_self_inst_id);
  410. }
  411. } else {
  412. // TODO: Omit parens if not needed for precedence.
  413. step_stack_->Push("(", witness_inst_id, ")");
  414. }
  415. }
  416. auto StringifyInst(SemIR::InstId /*inst_id*/, ImportRefUnloaded inst)
  417. -> void {
  418. if (inst.entity_name_id.has_value()) {
  419. step_stack_->PushEntityNameId(inst.entity_name_id);
  420. } else {
  421. *out_ << "<import ref unloaded invalid entity name>";
  422. }
  423. }
  424. auto StringifyInst(SemIR::InstId /*inst_id*/, IntType inst) -> void {
  425. *out_ << "<builtin ";
  426. step_stack_->PushString(">");
  427. if (auto width_value =
  428. sem_ir_->insts().TryGetAs<IntValue>(inst.bit_width_id)) {
  429. *out_ << (inst.int_kind.is_signed() ? "i" : "u");
  430. sem_ir_->ints().Get(width_value->int_id).print(*out_, /*isSigned=*/false);
  431. } else {
  432. *out_ << (inst.int_kind.is_signed() ? "Int(" : "UInt(");
  433. step_stack_->Push(inst.bit_width_id, ")");
  434. }
  435. }
  436. auto StringifyInst(SemIR::InstId /*inst_id*/, IntValue inst) -> void {
  437. sem_ir_->ints().Get(inst.int_id).print(*out_, /*isSigned=*/true);
  438. }
  439. auto StringifyInst(SemIR::InstId /*inst_id*/, LookupImplWitness inst)
  440. -> void {
  441. step_stack_->Push(
  442. inst.query_self_inst_id, " as ",
  443. sem_ir_->specific_interfaces().Get(inst.query_specific_interface_id));
  444. }
  445. auto StringifyInst(SemIR::InstId /*inst_id*/, NameRef inst) -> void {
  446. *out_ << sem_ir_->names().GetFormatted(inst.name_id);
  447. }
  448. auto StringifyInst(SemIR::InstId /*inst_id*/, Namespace inst) -> void {
  449. const auto& name_scope = sem_ir_->name_scopes().Get(inst.name_scope_id);
  450. step_stack_->PushQualifiedName(name_scope.parent_scope_id(),
  451. name_scope.name_id());
  452. }
  453. auto StringifyInst(SemIR::InstId /*inst_id*/, PointerType inst) -> void {
  454. step_stack_->Push(inst.pointee_id, "*");
  455. }
  456. auto StringifyInst(SemIR::InstId /*inst_id*/, SpecificFunction inst) -> void {
  457. auto callee = SemIR::GetCalleeFunction(*sem_ir_, inst.callee_id);
  458. if (callee.function_id.has_value()) {
  459. step_stack_->PushEntityName(sem_ir_->functions().Get(callee.function_id),
  460. inst.specific_id);
  461. } else {
  462. step_stack_->PushString("<invalid specific function>");
  463. }
  464. }
  465. auto StringifyInst(SemIR::InstId /*inst_id*/, SpecificImplFunction inst)
  466. -> void {
  467. auto callee = SemIR::GetCalleeFunction(*sem_ir_, inst.callee_id);
  468. if (callee.function_id.has_value()) {
  469. // TODO: The specific_id here is for the interface member, but the
  470. // entity we're passing is the impl member. This might result in
  471. // strange output once we render specific arguments properly.
  472. step_stack_->PushEntityName(sem_ir_->functions().Get(callee.function_id),
  473. inst.specific_id);
  474. } else {
  475. step_stack_->PushString("<invalid specific function>");
  476. }
  477. }
  478. auto StringifyInst(SemIR::InstId /*inst_id*/, StructType inst) -> void {
  479. auto fields = sem_ir_->struct_type_fields().Get(inst.fields_id);
  480. if (fields.empty()) {
  481. *out_ << "{}";
  482. return;
  483. }
  484. *out_ << "{";
  485. step_stack_->PushString("}");
  486. llvm::ListSeparator sep;
  487. for (auto field : llvm::reverse(fields)) {
  488. step_stack_->Push(".", field.name_id, ": ", field.type_inst_id, &sep);
  489. }
  490. }
  491. auto StringifyInst(SemIR::InstId /*inst_id*/, StructValue inst) -> void {
  492. auto field_values = sem_ir_->inst_blocks().Get(inst.elements_id);
  493. if (field_values.empty()) {
  494. *out_ << "{}";
  495. return;
  496. }
  497. auto struct_type = sem_ir_->types().GetAs<StructType>(
  498. sem_ir_->types().GetObjectRepr(inst.type_id));
  499. auto fields = sem_ir_->struct_type_fields().Get(struct_type.fields_id);
  500. if (fields.size() != field_values.size()) {
  501. *out_ << "{<struct value type length mismatch>}";
  502. return;
  503. }
  504. *out_ << "{";
  505. step_stack_->PushString("}");
  506. llvm::ListSeparator sep;
  507. for (auto [field, value_inst_id] :
  508. llvm::reverse(llvm::zip(fields, field_values))) {
  509. step_stack_->Push(".", field.name_id, " = ", value_inst_id, &sep);
  510. }
  511. }
  512. auto StringifyInst(SemIR::InstId /*inst_id*/, TupleType inst) -> void {
  513. auto refs = sem_ir_->inst_blocks().Get(inst.type_elements_id);
  514. if (refs.empty()) {
  515. *out_ << "()";
  516. return;
  517. }
  518. *out_ << "(";
  519. step_stack_->PushString(")");
  520. // A tuple of one element has a comma to disambiguate from an
  521. // expression.
  522. if (refs.size() == 1) {
  523. step_stack_->PushString(",");
  524. }
  525. llvm::ListSeparator sep;
  526. for (auto ref : llvm::reverse(refs)) {
  527. step_stack_->Push(ref, &sep);
  528. }
  529. }
  530. auto StringifyInst(SemIR::InstId /*inst_id*/, TupleValue inst) -> void {
  531. auto refs = sem_ir_->inst_blocks().Get(inst.elements_id);
  532. if (refs.empty()) {
  533. *out_ << "()";
  534. return;
  535. }
  536. *out_ << "(";
  537. step_stack_->PushString(")");
  538. // A tuple of one element has a comma to disambiguate from an
  539. // expression.
  540. if (refs.size() == 1) {
  541. step_stack_->PushString(",");
  542. }
  543. llvm::ListSeparator sep;
  544. for (auto ref : llvm::reverse(refs)) {
  545. step_stack_->Push(ref, &sep);
  546. }
  547. }
  548. auto StringifyInst(SemIR::InstId inst_id, TypeOfInst /*inst*/) -> void {
  549. // Print the constant value if we've already computed the inst.
  550. auto const_inst_id = sem_ir_->constant_values().GetConstantInstId(inst_id);
  551. if (const_inst_id.has_value() && const_inst_id != inst_id) {
  552. step_stack_->PushInstId(const_inst_id);
  553. return;
  554. }
  555. *out_ << "<dependent type>";
  556. }
  557. auto StringifyInst(SemIR::InstId /*inst_id*/, UnboundElementType inst)
  558. -> void {
  559. *out_ << "<unbound element of class ";
  560. step_stack_->Push(inst.class_type_inst_id, ">");
  561. }
  562. auto StringifyInst(SemIR::InstId /*inst_id*/, VtablePtr /*inst*/) -> void {
  563. *out_ << "<vtable ptr>";
  564. }
  565. private:
  566. const SemIR::File* sem_ir_;
  567. StepStack* step_stack_;
  568. llvm::raw_ostream* out_;
  569. };
  570. } // namespace
  571. static auto Stringify(const SemIR::File& sem_ir, StepStack& step_stack)
  572. -> std::string {
  573. RawStringOstream out;
  574. Stringifier stringifier(&sem_ir, &step_stack, &out);
  575. while (!step_stack.empty()) {
  576. auto step = step_stack.Pop();
  577. VariantMatch(
  578. step,
  579. [&](InstId inst_id) {
  580. if (!inst_id.has_value()) {
  581. out << "<invalid>";
  582. return;
  583. }
  584. auto untyped_inst = sem_ir.insts().Get(inst_id);
  585. CARBON_KIND_SWITCH(untyped_inst) {
  586. #define CARBON_SEM_IR_INST_KIND(InstT) \
  587. case CARBON_KIND(InstT typed_inst): { \
  588. stringifier.StringifyInst(inst_id, typed_inst); \
  589. break; \
  590. }
  591. #include "toolchain/sem_ir/inst_kind.def"
  592. }
  593. },
  594. [&](llvm::StringRef string) { out << string; },
  595. [&](NameId name_id) { out << sem_ir.names().GetFormatted(name_id); },
  596. [&](ElementIndex element_index) { out << element_index.index; });
  597. }
  598. return out.TakeStr();
  599. }
  600. auto StringifyConstantInst(const SemIR::File& sem_ir, InstId outer_inst_id)
  601. -> std::string {
  602. StepStack step_stack(&sem_ir);
  603. step_stack.PushInstId(outer_inst_id);
  604. return Stringify(sem_ir, step_stack);
  605. }
  606. auto StringifySpecific(const File& sem_ir, SpecificId specific_id)
  607. -> std::string {
  608. StepStack step_stack(&sem_ir);
  609. const auto& specific = sem_ir.specifics().Get(specific_id);
  610. const auto& generic = sem_ir.generics().Get(specific.generic_id);
  611. auto decl = sem_ir.insts().Get(generic.decl_id);
  612. CARBON_KIND_SWITCH(decl) {
  613. case CARBON_KIND(SemIR::ClassDecl class_decl): {
  614. // Print `Core.Int(N)` as `iN`.
  615. // TODO: This duplicates work done in StringifyInst for ClassType.
  616. const auto& class_info = sem_ir.classes().Get(class_decl.class_id);
  617. if (auto literal_info = NumericTypeLiteralInfo::ForType(
  618. sem_ir, SemIR::ClassType{.type_id = SemIR::TypeType::TypeId,
  619. .class_id = class_decl.class_id,
  620. .specific_id = specific_id});
  621. literal_info.is_valid()) {
  622. RawStringOstream out;
  623. literal_info.PrintLiteral(sem_ir, out);
  624. return out.TakeStr();
  625. }
  626. step_stack.PushEntityName(class_info, specific_id);
  627. break;
  628. }
  629. case CARBON_KIND(SemIR::FunctionDecl function_decl): {
  630. step_stack.PushEntityName(
  631. sem_ir.functions().Get(function_decl.function_id), specific_id);
  632. break;
  633. }
  634. case CARBON_KIND(SemIR::ImplDecl impl_decl): {
  635. step_stack.PushEntityName(sem_ir.impls().Get(impl_decl.impl_id),
  636. specific_id);
  637. break;
  638. }
  639. case CARBON_KIND(SemIR::InterfaceDecl interface_decl): {
  640. step_stack.PushEntityName(
  641. sem_ir.interfaces().Get(interface_decl.interface_id), specific_id);
  642. break;
  643. }
  644. default: {
  645. // TODO: Include the specific arguments here.
  646. step_stack.PushInstId(generic.decl_id);
  647. break;
  648. }
  649. }
  650. return Stringify(sem_ir, step_stack);
  651. }
  652. } // namespace Carbon::SemIR