blob: 2e95c2c1702bb0a2d88d937ccb0417b971eba8ba [file] [log] [blame]
// Copyright 2022 The Dawn & Tint Authors
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
// 1. Redistributions of source code must retain the above copyright notice, this
// list of conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright notice,
// this list of conditions and the following disclaimer in the documentation
// and/or other materials provided with the distribution.
//
// 3. Neither the name of the copyright holder nor the names of its
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
// FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
// DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
// CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
// OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#include "src/tint/lang/core/ir/disassembler.h"
#include "src//tint/lang/core/ir/unary.h"
#include "src/tint/lang/core/constant/composite.h"
#include "src/tint/lang/core/constant/scalar.h"
#include "src/tint/lang/core/constant/splat.h"
#include "src/tint/lang/core/fluent_types.h"
#include "src/tint/lang/core/ir/access.h"
#include "src/tint/lang/core/ir/binary.h"
#include "src/tint/lang/core/ir/bitcast.h"
#include "src/tint/lang/core/ir/block.h"
#include "src/tint/lang/core/ir/block_param.h"
#include "src/tint/lang/core/ir/break_if.h"
#include "src/tint/lang/core/ir/construct.h"
#include "src/tint/lang/core/ir/continue.h"
#include "src/tint/lang/core/ir/convert.h"
#include "src/tint/lang/core/ir/core_builtin_call.h"
#include "src/tint/lang/core/ir/discard.h"
#include "src/tint/lang/core/ir/exit_if.h"
#include "src/tint/lang/core/ir/exit_loop.h"
#include "src/tint/lang/core/ir/exit_switch.h"
#include "src/tint/lang/core/ir/if.h"
#include "src/tint/lang/core/ir/instruction_result.h"
#include "src/tint/lang/core/ir/let.h"
#include "src/tint/lang/core/ir/load.h"
#include "src/tint/lang/core/ir/load_vector_element.h"
#include "src/tint/lang/core/ir/loop.h"
#include "src/tint/lang/core/ir/multi_in_block.h"
#include "src/tint/lang/core/ir/next_iteration.h"
#include "src/tint/lang/core/ir/return.h"
#include "src/tint/lang/core/ir/store.h"
#include "src/tint/lang/core/ir/store_vector_element.h"
#include "src/tint/lang/core/ir/switch.h"
#include "src/tint/lang/core/ir/swizzle.h"
#include "src/tint/lang/core/ir/terminate_invocation.h"
#include "src/tint/lang/core/ir/unreachable.h"
#include "src/tint/lang/core/ir/user_call.h"
#include "src/tint/lang/core/ir/var.h"
#include "src/tint/lang/core/type/struct.h"
#include "src/tint/lang/core/type/type.h"
#include "src/tint/utils/ice/ice.h"
#include "src/tint/utils/macros/scoped_assignment.h"
#include "src/tint/utils/rtti/switch.h"
#include "src/tint/utils/text/string.h"
using namespace tint::core::fluent_types; // NOLINT
namespace tint::core::ir {
namespace {
class ScopedIndent {
public:
explicit ScopedIndent(uint32_t& indent) : indent_(indent) { indent_ += 2; }
~ScopedIndent() { indent_ -= 2; }
private:
uint32_t& indent_;
};
} // namespace
std::string Disassemble(const Module& mod) {
return Disassembler{mod}.Disassemble();
}
Disassembler::Disassembler(const Module& mod) : mod_(mod) {}
Disassembler::~Disassembler() = default;
StringStream& Disassembler::Indent() {
for (uint32_t i = 0; i < indent_size_; i++) {
out_ << " ";
}
return out_;
}
void Disassembler::EmitLine() {
out_ << std::endl;
current_output_line_ += 1;
current_output_start_pos_ = out_.tellp();
}
size_t Disassembler::IdOf(const Block* node) {
TINT_ASSERT(node);
return block_ids_.GetOrAdd(node, [&] { return block_ids_.Count(); });
}
std::string Disassembler::IdOf(const Value* value) {
TINT_ASSERT(value);
return value_ids_.GetOrAdd(value, [&] {
if (auto sym = mod_.NameOf(value)) {
if (ids_.Add(sym.Name())) {
return sym.Name();
}
auto prefix = sym.Name() + "_";
for (size_t i = 1;; i++) {
auto name = prefix + std::to_string(i);
if (ids_.Add(name)) {
return name;
}
}
}
return std::to_string(value_ids_.Count());
});
}
std::string Disassembler::NameOf(const If* inst) {
if (!inst) {
return "undef";
}
return if_names_.GetOrAdd(inst, [&] { return "if_" + std::to_string(if_names_.Count()); });
}
std::string Disassembler::NameOf(const Loop* inst) {
if (!inst) {
return "undef";
}
return loop_names_.GetOrAdd(inst,
[&] { return "loop_" + std::to_string(loop_names_.Count()); });
}
std::string Disassembler::NameOf(const Switch* inst) {
if (!inst) {
return "undef";
}
return switch_names_.GetOrAdd(
inst, [&] { return "switch_" + std::to_string(switch_names_.Count()); });
}
Source::Location Disassembler::MakeCurrentLocation() {
return Source::Location{current_output_line_, out_.tellp() - current_output_start_pos_ + 1};
}
std::string Disassembler::Disassemble() {
for (auto* ty : mod_.Types()) {
if (auto* str = ty->As<core::type::Struct>()) {
EmitStructDecl(str);
}
}
if (!mod_.root_block->IsEmpty()) {
EmitBlock(mod_.root_block, "root");
EmitLine();
}
for (auto& func : mod_.functions) {
EmitFunction(func);
}
return out_.str();
}
void Disassembler::EmitBlock(const Block* blk, std::string_view comment /* = "" */) {
Indent();
SourceMarker sm(this);
out_ << "%b" << IdOf(blk) << " = block";
if (auto* merge = blk->As<MultiInBlock>()) {
if (!merge->Params().IsEmpty()) {
out_ << " (";
EmitValueList(merge->Params().Slice());
out_ << ")";
}
}
sm.Store(blk);
out_ << " {";
if (!comment.empty()) {
out_ << " # " << comment;
}
EmitLine();
{
ScopedIndent si(indent_size_);
for (auto* inst : *blk) {
Indent();
EmitInstruction(inst);
}
}
Indent() << "}";
EmitLine();
}
void Disassembler::EmitBindingPoint(BindingPoint p) {
out_ << "@binding_point(" << p.group << ", " << p.binding << ")";
}
void Disassembler::EmitLocation(Location loc) {
out_ << "@location(" << loc.value << ")";
if (loc.interpolation.has_value()) {
out_ << ", @interpolate(";
out_ << loc.interpolation->type;
if (loc.interpolation->sampling != core::InterpolationSampling::kUndefined) {
out_ << ", ";
out_ << loc.interpolation->sampling;
}
out_ << ")";
}
}
void Disassembler::EmitParamAttributes(const FunctionParam* p) {
if (!p->Invariant() && !p->Location().has_value() && !p->BindingPoint().has_value() &&
!p->Builtin().has_value()) {
return;
}
out_ << " [";
bool need_comma = false;
auto comma = [&] {
if (need_comma) {
out_ << ", ";
}
};
if (p->Invariant()) {
comma();
out_ << "@invariant";
need_comma = true;
}
if (p->Location().has_value()) {
EmitLocation(p->Location().value());
need_comma = true;
}
if (p->BindingPoint().has_value()) {
comma();
EmitBindingPoint(p->BindingPoint().value());
need_comma = true;
}
if (p->Builtin().has_value()) {
comma();
out_ << "@" << p->Builtin().value();
need_comma = true;
}
out_ << "]";
}
void Disassembler::EmitReturnAttributes(const Function* func) {
if (!func->ReturnInvariant() && !func->ReturnLocation().has_value() &&
!func->ReturnBuiltin().has_value()) {
return;
}
out_ << " [";
bool need_comma = false;
auto comma = [&] {
if (need_comma) {
out_ << ", ";
}
};
if (func->ReturnInvariant()) {
comma();
out_ << "@invariant";
need_comma = true;
}
if (func->ReturnLocation().has_value()) {
comma();
EmitLocation(func->ReturnLocation().value());
need_comma = true;
}
if (func->ReturnBuiltin().has_value()) {
comma();
out_ << "@" << func->ReturnBuiltin().value();
need_comma = true;
}
out_ << "]";
}
void Disassembler::EmitFunction(const Function* func) {
in_function_ = true;
std::string fn_id = IdOf(func);
Indent() << "%" << fn_id << " =";
if (func->Stage() != Function::PipelineStage::kUndefined) {
out_ << " @" << func->Stage();
}
if (func->WorkgroupSize()) {
auto arr = func->WorkgroupSize().value();
out_ << " @workgroup_size(" << arr[0] << ", " << arr[1] << ", " << arr[2] << ")";
}
out_ << " func(";
for (auto* p : func->Params()) {
if (p != func->Params().Front()) {
out_ << ", ";
}
out_ << "%" << IdOf(p) << ":" << p->Type()->FriendlyName();
EmitParamAttributes(p);
}
out_ << "):" << func->ReturnType()->FriendlyName();
EmitReturnAttributes(func);
out_ << " -> %b" << IdOf(func->Block()) << " {";
{ // Add a comment if the function IDs or parameter IDs doesn't match their name
Vector<std::string, 4> names;
if (auto name = mod_.NameOf(func); name.IsValid()) {
if (name.NameView() != fn_id) {
names.Push("%" + std::string(fn_id) + ": '" + name.Name() + "'");
}
}
for (auto* p : func->Params()) {
if (auto name = mod_.NameOf(p); name.IsValid()) {
auto id = IdOf(p);
if (name.NameView() != id) {
names.Push("%" + std::string(id) + ": '" + name.Name() + "'");
}
}
}
if (!names.IsEmpty()) {
out_ << " # " << tint::Join(names, ", ");
}
}
EmitLine();
{
ScopedIndent si(indent_size_);
EmitBlock(func->Block());
}
Indent() << "}";
EmitLine();
}
void Disassembler::EmitValueWithType(const Instruction* val) {
SourceMarker sm(this);
if (val->Result(0)) {
EmitValueWithType(val->Result(0));
} else {
out_ << "undef";
}
sm.StoreResult(IndexedValue{val, 0});
}
void Disassembler::EmitValueWithType(const Value* val) {
if (!val) {
out_ << "undef";
return;
}
EmitValue(val);
out_ << ":" << val->Type()->FriendlyName();
}
void Disassembler::EmitValue(const Value* val) {
tint::Switch(
val,
[&](const ir::Constant* constant) {
std::function<void(const core::constant::Value*)> emit =
[&](const core::constant::Value* c) {
tint::Switch(
c,
[&](const core::constant::Scalar<AFloat>* scalar) {
out_ << scalar->ValueAs<AFloat>().value;
},
[&](const core::constant::Scalar<AInt>* scalar) {
out_ << scalar->ValueAs<AInt>().value;
},
[&](const core::constant::Scalar<i32>* scalar) {
out_ << scalar->ValueAs<i32>().value << "i";
},
[&](const core::constant::Scalar<u32>* scalar) {
out_ << scalar->ValueAs<u32>().value << "u";
},
[&](const core::constant::Scalar<f32>* scalar) {
out_ << scalar->ValueAs<f32>().value << "f";
},
[&](const core::constant::Scalar<f16>* scalar) {
out_ << scalar->ValueAs<f16>().value << "h";
},
[&](const core::constant::Scalar<bool>* scalar) {
out_ << (scalar->ValueAs<bool>() ? "true" : "false");
},
[&](const core::constant::Splat* splat) {
out_ << splat->Type()->FriendlyName() << "(";
emit(splat->Index(0));
out_ << ")";
},
[&](const core::constant::Composite* composite) {
out_ << composite->Type()->FriendlyName() << "(";
bool need_comma = false;
for (const auto* elem : composite->elements) {
if (need_comma) {
out_ << ", ";
}
emit(elem);
need_comma = true;
}
out_ << ")";
});
};
emit(constant->Value());
},
[&](const ir::InstructionResult* rv) { out_ << "%" << IdOf(rv); },
[&](const ir::BlockParam* p) {
out_ << "%" << IdOf(p) << ":" << p->Type()->FriendlyName();
},
[&](const ir::FunctionParam* p) { out_ << "%" << IdOf(p); },
[&](const ir::Function* f) { out_ << "%" << IdOf(f); },
[&](Default) {
if (val == nullptr) {
out_ << "undef";
} else {
out_ << "Unknown value: " << val->TypeInfo().name;
}
});
}
void Disassembler::EmitInstructionName(const Instruction* inst) {
SourceMarker sm(this);
out_ << inst->FriendlyName();
sm.Store(inst);
}
void Disassembler::EmitInstruction(const Instruction* inst) {
TINT_DEFER(EmitLine());
if (!inst->Alive()) {
SourceMarker sm(this);
out_ << "<destroyed " << inst->TypeInfo().name << " " << tint::ToString(inst) << ">";
sm.Store(inst);
return;
}
tint::Switch(
inst, //
[&](const Switch* s) { EmitSwitch(s); }, //
[&](const If* i) { EmitIf(i); }, //
[&](const Loop* l) { EmitLoop(l); }, //
[&](const Binary* b) { EmitBinary(b); }, //
[&](const Unary* u) { EmitUnary(u); }, //
[&](const Discard* d) { EmitInstructionName(d); },
[&](const Store* s) {
EmitInstructionName(s);
out_ << " ";
EmitOperand(s, Store::kToOperandOffset);
out_ << ", ";
EmitOperand(s, Store::kFromOperandOffset);
},
[&](const StoreVectorElement* s) {
EmitInstructionName(s);
out_ << " ";
EmitOperandList(s);
},
[&](const UserCall* uc) {
EmitValueWithType(uc);
out_ << " = ";
EmitInstructionName(uc);
out_ << " ";
EmitOperand(uc, UserCall::kFunctionOperandOffset);
if (!uc->Args().IsEmpty()) {
out_ << ", ";
}
EmitOperandList(uc, UserCall::kArgsOperandOffset);
},
[&](const Var* v) {
EmitValueWithType(v);
out_ << " = ";
EmitInstructionName(v);
if (v->Initializer()) {
out_ << ", ";
EmitOperand(v, Var::kInitializerOperandOffset);
}
if (v->BindingPoint().has_value()) {
out_ << " ";
EmitBindingPoint(v->BindingPoint().value());
}
if (v->Attributes().invariant) {
out_ << " @invariant";
}
if (v->Attributes().location.has_value()) {
out_ << " @location(" << v->Attributes().location.value() << ")";
}
if (v->Attributes().blend_src.has_value()) {
out_ << " @blend_src(" << v->Attributes().blend_src.value() << ")";
}
if (v->Attributes().interpolation.has_value()) {
auto& interp = v->Attributes().interpolation.value();
out_ << " @interpolate(" << interp.type;
if (interp.sampling != core::InterpolationSampling::kUndefined) {
out_ << ", " << interp.sampling;
}
out_ << ")";
}
if (v->Attributes().builtin.has_value()) {
out_ << " @builtin(" << v->Attributes().builtin.value() << ")";
}
},
[&](const Swizzle* s) {
EmitValueWithType(s);
out_ << " = ";
EmitInstructionName(s);
out_ << " ";
EmitValue(s->Object());
out_ << ", ";
for (auto idx : s->Indices()) {
switch (idx) {
case 0:
out_ << "x";
break;
case 1:
out_ << "y";
break;
case 2:
out_ << "z";
break;
case 3:
out_ << "w";
break;
}
}
},
[&](const Terminator* b) { EmitTerminator(b); },
[&](Default) {
EmitValueWithType(inst);
out_ << " = ";
EmitInstructionName(inst);
if (!inst->Operands().IsEmpty()) {
out_ << " ";
EmitOperandList(inst);
}
});
{ // Add a comment if the result IDs don't match their names
Vector<std::string, 4> names;
for (auto* result : inst->Results()) {
if (result) {
if (auto name = mod_.NameOf(result); name.IsValid()) {
auto id = IdOf(result);
if (name.NameView() != id) {
names.Push("%" + std::string(id) + ": '" + name.Name() + "'");
}
}
}
}
if (!names.IsEmpty()) {
out_ << " # " << tint::Join(names, ", ");
}
}
}
void Disassembler::EmitOperand(const Instruction* inst, size_t index) {
SourceMarker marker(this);
EmitValue(inst->Operands()[index]);
marker.Store(IndexedValue{inst, static_cast<uint32_t>(index)});
}
void Disassembler::EmitOperandList(const Instruction* inst, size_t start_index /* = 0 */) {
for (size_t i = start_index, n = inst->Operands().Length(); i < n; i++) {
if (i != start_index) {
out_ << ", ";
}
EmitOperand(inst, i);
}
}
void Disassembler::EmitIf(const If* if_) {
SourceMarker sm(this);
if (auto results = if_->Results(); !results.IsEmpty()) {
for (size_t i = 0; i < results.Length(); ++i) {
if (i > 0) {
out_ << ", ";
}
SourceMarker rs(this);
EmitValueWithType(results[i]);
rs.StoreResult(IndexedValue{if_, i});
}
out_ << " = ";
}
out_ << "if ";
EmitOperand(if_, If::kConditionOperandOffset);
bool has_false = !if_->False()->IsEmpty();
out_ << " [t: %b" << IdOf(if_->True());
if (has_false) {
out_ << ", f: %b" << IdOf(if_->False());
}
out_ << "]";
sm.Store(if_);
out_ << " { # " << NameOf(if_);
EmitLine();
// True block is assumed to have instructions
{
ScopedIndent si(indent_size_);
EmitBlock(if_->True(), "true");
}
if (has_false) {
ScopedIndent si(indent_size_);
EmitBlock(if_->False(), "false");
} else if (auto results = if_->Results(); !results.IsEmpty()) {
ScopedIndent si(indent_size_);
Indent();
out_ << "# implicit false block: exit_if undef";
for (size_t v = 1; v < if_->Results().Length(); v++) {
out_ << ", undef";
}
EmitLine();
}
Indent();
out_ << "}";
}
void Disassembler::EmitLoop(const Loop* l) {
Vector<std::string, 3> parts;
if (!l->Initializer()->IsEmpty()) {
parts.Push("i: %b" + std::to_string(IdOf(l->Initializer())));
}
parts.Push("b: %b" + std::to_string(IdOf(l->Body())));
if (!l->Continuing()->IsEmpty()) {
parts.Push("c: %b" + std::to_string(IdOf(l->Continuing())));
}
SourceMarker sm(this);
if (auto results = l->Results(); !results.IsEmpty()) {
for (size_t i = 0; i < results.Length(); ++i) {
if (i > 0) {
out_ << ", ";
}
SourceMarker rs(this);
EmitValueWithType(results[i]);
rs.StoreResult(IndexedValue{l, i});
}
out_ << " = ";
}
out_ << "loop [" << tint::Join(parts, ", ") << "]";
sm.Store(l);
out_ << " { # " << NameOf(l);
EmitLine();
if (!l->Initializer()->IsEmpty()) {
ScopedIndent si(indent_size_);
EmitBlock(l->Initializer(), "initializer");
}
// Loop is assumed to always have a body
{
ScopedIndent si(indent_size_);
EmitBlock(l->Body(), "body");
}
if (!l->Continuing()->IsEmpty()) {
ScopedIndent si(indent_size_);
EmitBlock(l->Continuing(), "continuing");
}
Indent();
out_ << "}";
}
void Disassembler::EmitSwitch(const Switch* s) {
SourceMarker sm(this);
if (auto results = s->Results(); !results.IsEmpty()) {
for (size_t i = 0; i < results.Length(); ++i) {
if (i > 0) {
out_ << ", ";
}
SourceMarker rs(this);
EmitValueWithType(results[i]);
rs.StoreResult(IndexedValue{s, i});
}
out_ << " = ";
}
out_ << "switch ";
EmitValue(s->Condition());
out_ << " [";
for (auto& c : s->Cases()) {
if (&c != &s->Cases().Front()) {
out_ << ", ";
}
out_ << "c: (";
for (auto& selector : c.selectors) {
if (&selector != &c.selectors.Front()) {
out_ << " ";
}
if (selector.IsDefault()) {
out_ << "default";
} else {
EmitValue(selector.val);
}
}
out_ << ", %b" << IdOf(c.block) << ")";
}
out_ << "]";
sm.Store(s);
out_ << " { # " << NameOf(s);
EmitLine();
for (auto& c : s->Cases()) {
ScopedIndent si(indent_size_);
EmitBlock(c.block, "case");
}
Indent();
out_ << "}";
}
void Disassembler::EmitTerminator(const Terminator* b) {
SourceMarker sm(this);
size_t args_offset = 0;
tint::Switch(
b,
[&](const ir::Return*) {
out_ << "ret";
args_offset = ir::Return::kArgsOperandOffset;
},
[&](const ir::Continue* cont) {
out_ << "continue %b" << IdOf(cont->Loop()->Continuing());
args_offset = ir::Continue::kArgsOperandOffset;
},
[&](const ir::ExitIf*) {
out_ << "exit_if";
args_offset = ir::ExitIf::kArgsOperandOffset;
},
[&](const ir::ExitSwitch*) {
out_ << "exit_switch";
args_offset = ir::ExitSwitch::kArgsOperandOffset;
},
[&](const ir::ExitLoop*) {
out_ << "exit_loop";
args_offset = ir::ExitLoop::kArgsOperandOffset;
},
[&](const ir::NextIteration* ni) {
out_ << "next_iteration %b" << IdOf(ni->Loop()->Body());
args_offset = ir::NextIteration::kArgsOperandOffset;
},
[&](const ir::Unreachable*) { out_ << "unreachable"; },
[&](const ir::BreakIf* bi) {
out_ << "break_if ";
EmitValue(bi->Condition());
out_ << " %b" << IdOf(bi->Loop()->Body());
args_offset = ir::BreakIf::kArgsOperandOffset;
},
[&](const ir::TerminateInvocation*) { out_ << "terminate_invocation"; },
[&](Default) { out_ << "unknown terminator " << b->TypeInfo().name; });
if (!b->Args().IsEmpty()) {
out_ << " ";
EmitOperandList(b, args_offset);
}
sm.Store(b);
tint::Switch(
b, //
[&](const ir::ExitIf* e) { out_ << " # " << NameOf(e->If()); }, //
[&](const ir::ExitSwitch* e) { out_ << " # " << NameOf(e->Switch()); }, //
[&](const ir::ExitLoop* e) { out_ << " # " << NameOf(e->Loop()); } //
);
}
void Disassembler::EmitValueList(tint::Slice<const Value* const> values) {
for (size_t i = 0, n = values.Length(); i < n; i++) {
if (i > 0) {
out_ << ", ";
}
EmitValue(values[i]);
}
}
void Disassembler::EmitBinary(const Binary* b) {
SourceMarker sm(this);
EmitValueWithType(b);
out_ << " = ";
switch (b->Op()) {
case BinaryOp::kAdd:
out_ << "add";
break;
case BinaryOp::kSubtract:
out_ << "sub";
break;
case BinaryOp::kMultiply:
out_ << "mul";
break;
case BinaryOp::kDivide:
out_ << "div";
break;
case BinaryOp::kModulo:
out_ << "mod";
break;
case BinaryOp::kAnd:
out_ << "and";
break;
case BinaryOp::kOr:
out_ << "or";
break;
case BinaryOp::kXor:
out_ << "xor";
break;
case BinaryOp::kEqual:
out_ << "eq";
break;
case BinaryOp::kNotEqual:
out_ << "neq";
break;
case BinaryOp::kLessThan:
out_ << "lt";
break;
case BinaryOp::kGreaterThan:
out_ << "gt";
break;
case BinaryOp::kLessThanEqual:
out_ << "lte";
break;
case BinaryOp::kGreaterThanEqual:
out_ << "gte";
break;
case BinaryOp::kShiftLeft:
out_ << "shl";
break;
case BinaryOp::kShiftRight:
out_ << "shr";
break;
case BinaryOp::kLogicalAnd:
out_ << "logical-and";
break;
case BinaryOp::kLogicalOr:
out_ << "logical-or";
break;
}
out_ << " ";
EmitOperandList(b);
sm.Store(b);
}
void Disassembler::EmitUnary(const Unary* u) {
SourceMarker sm(this);
EmitValueWithType(u);
out_ << " = ";
switch (u->Op()) {
case UnaryOp::kComplement:
out_ << "complement";
break;
case UnaryOp::kNegation:
out_ << "negation";
break;
case UnaryOp::kAddressOf:
out_ << "ref-to-ptr";
break;
case UnaryOp::kIndirection:
out_ << "ptr-to-ref";
break;
case UnaryOp::kNot:
out_ << "not";
break;
}
out_ << " ";
EmitOperandList(u);
sm.Store(u);
}
void Disassembler::EmitStructDecl(const core::type::Struct* str) {
out_ << str->Name().Name() << " = struct @align(" << str->Align() << ")";
if (str->StructFlags().Contains(core::type::StructFlag::kBlock)) {
out_ << ", @block";
}
out_ << " {";
EmitLine();
for (auto* member : str->Members()) {
out_ << " " << member->Name().Name() << ":" << member->Type()->FriendlyName();
out_ << " @offset(" << member->Offset() << ")";
if (member->Attributes().invariant) {
out_ << ", @invariant";
}
if (member->Attributes().location.has_value()) {
out_ << ", @location(" << member->Attributes().location.value() << ")";
}
if (member->Attributes().interpolation.has_value()) {
auto& interp = member->Attributes().interpolation.value();
out_ << ", @interpolate(" << interp.type;
if (interp.sampling != core::InterpolationSampling::kUndefined) {
out_ << ", " << interp.sampling;
}
out_ << ")";
}
if (member->Attributes().builtin.has_value()) {
out_ << ", @builtin(" << member->Attributes().builtin.value() << ")";
}
EmitLine();
}
out_ << "}";
EmitLine();
EmitLine();
}
} // namespace tint::core::ir