blob: 34e59ebb32a66bf4cef20752a3dff4ee92be34b3 [file] [log] [blame]
/*
* Copyright 2019 Google LLC
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "src/sksl/SkSLASTNode.h"
#include "src/sksl/SkSLCompiler.h"
#include "src/sksl/SkSLString.h"
namespace SkSL {
String ASTNode::description() const {
switch (fKind) {
case Kind::kNull: return "";
case Kind::kBinary:
return "(" + this->begin()->description() + " " +
Compiler::OperatorName(getToken().fKind) + " " +
(this->begin() + 1)->description() + ")";
case Kind::kBlock: {
String result = "{\n";
for (const auto& c : *this) {
result += c.description();
result += "\n";
}
result += "}";
return result;
}
case Kind::kBool:
return getBool() ? "true" : "false";
case Kind::kBreak:
return "break";
case Kind::kCall: {
auto iter = this->begin();
String result = iter->description();
result += "(";
const char* separator = "";
while (iter != this->end()) {
result += separator;
result += (iter++)->description();
separator = ",";
}
result += ")";
return result;
}
case Kind::kContinue:
return "continue";
case Kind::kDiscard:
return "discard";
case Kind::kDo:
return "do " + this->begin()->description() + " while (" +
(this->begin() + 1)->description() + ")";
case Kind::kEnum: {
String result = "enum ";
result += getString();
result += " {\n";
for (const auto& c : *this) {
result += c.description();
result += "\n";
}
result += "};";
return result;
}
case Kind::kEnumCase:
if (this->begin() != this->end()) {
return String(getString()) + " = " + this->begin()->description();
}
return getString();
case Kind::kExtension:
return "#extension " + getString();
case Kind::kField:
return this->begin()->description() + "." + getString();
case Kind::kFile: {
String result;
for (const auto& c : *this) {
result += c.description();
result += "\n";
}
return result;
}
case Kind::kFloat:
return to_string(getFloat());
case Kind::kFor:
return "for (" + this->begin()->description() + "; " +
(this->begin() + 1)->description() + "; " + (this->begin() + 2)->description() +
") " + (this->begin() + 3)->description();
case Kind::kFunction: {
FunctionData fd = getFunctionData();
String result = fd.fModifiers.description();
if (result.size()) {
result += " ";
}
auto iter = this->begin();
result += (iter++)->description() + " " + fd.fName + "(";
const char* separator = "";
for (size_t i = 0; i < fd.fParameterCount; ++i) {
result += separator;
result += (iter++)->description();
separator = ", ";
}
result += ")";
if (iter != this->end()) {
result += " " + (iter++)->description();
SkASSERT(iter == this->end());
}
else {
result += ";";
}
return result;
}
case Kind::kIdentifier:
return getString();
case Kind::kIndex:
return this->begin()->description() + "[" + (this->begin() + 1)->description() + "]";
case Kind::kIf: {
String result;
if (getBool()) {
result = "@";
}
auto iter = this->begin();
result += "if (" + (iter++)->description() + ") ";
result += (iter++)->description();
if (iter != this->end()) {
result += " else " + (iter++)->description();
SkASSERT(iter == this->end());
}
return result;
}
case Kind::kInt:
return to_string(getInt());
case Kind::kInterfaceBlock: {
InterfaceBlockData id = getInterfaceBlockData();
String result = id.fModifiers.description() + " " + id.fTypeName + " {\n";
auto iter = this->begin();
for (size_t i = 0; i < id.fDeclarationCount; ++i) {
result += (iter++)->description() + "\n";
}
result += "} ";
result += id.fInstanceName;
for (size_t i = 0; i < id.fSizeCount; ++i) {
result += "[" + (iter++)->description() + "]";
}
SkASSERT(iter == this->end());
result += ";";
return result;
}
case Kind::kModifiers:
return getModifiers().description();
case Kind::kParameter: {
ParameterData pd = getParameterData();
auto iter = this->begin();
String result = (iter++)->description() + " " + pd.fName;
for (size_t i = 0; i < pd.fSizeCount; ++i) {
result += "[" + (iter++)->description() + "]";
}
if (iter != this->end()) {
result += " = " + (iter++)->description();
SkASSERT(iter == this->end());
}
return result;
}
case Kind::kPostfix:
return this->begin()->description() + Compiler::OperatorName(getToken().fKind);
case Kind::kPrefix:
return Compiler::OperatorName(getToken().fKind) + this->begin()->description();
case Kind::kReturn:
if (this->begin() != this->end()) {
return "return " + this->begin()->description() + ";";
}
return "return;";
case Kind::kSection:
return "@section { ... }";
case Kind::kSwitchCase: {
auto iter = this->begin();
String result;
if (*iter) {
result.appendf("case %s:\n", iter->description().c_str());
} else {
result = "default:\n";
}
for (++iter; iter != this->end(); ++iter) {
result += "\n" + iter->description();
}
return result;
}
case Kind::kSwitch: {
auto iter = this->begin();
String result;
if (getBool()) {
result = "@";
}
result += "switch (" + (iter++)->description() + ") {";
for (; iter != this->end(); ++iter) {
result += iter->description() + "\n";
}
result += "}";
return result;
}
case Kind::kTernary:
return "(" + this->begin()->description() + " ? " + (this->begin() + 1)->description() +
" : " + (this->begin() + 2)->description() + ")";
case Kind::kType:
return String(getTypeData().fName);
case Kind::kVarDeclaration: {
VarData vd = getVarData();
String result = vd.fName;
auto iter = this->begin();
for (size_t i = 0; i < vd.fSizeCount; ++i) {
result += "[" + (iter++)->description() + "]";
}
if (iter != this->end()) {
result += " = " + (iter++)->description();
SkASSERT(iter == this->end());
}
return result;
}
case Kind::kVarDeclarations: {
auto iter = this->begin();
String result = (iter++)->description();
if (result.size()) {
result += " ";
}
result += (iter++)->description();
const char* separator = " ";
for (; iter != this->end(); ++iter) {
result += separator + iter->description();
separator = ", ";
}
return result;
}
default:
SkASSERT(false);
return "<error>";
}
}
} // namespace