| // Copyright 2020 The Tint Authors. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| #include "src/tint/utils/generator/text_generator.h" |
| |
| #include <algorithm> |
| #include <limits> |
| |
| #include "src/tint/utils/containers/map.h" |
| #include "src/tint/utils/ice/ice.h" |
| |
| namespace tint { |
| |
| TextGenerator::TextGenerator() = default; |
| |
| TextGenerator::~TextGenerator() = default; |
| |
| TextGenerator::LineWriter::LineWriter(TextBuffer* buf) : buffer(buf) {} |
| |
| TextGenerator::LineWriter::LineWriter(LineWriter&& other) { |
| buffer = other.buffer; |
| other.buffer = nullptr; |
| } |
| |
| TextGenerator::LineWriter::~LineWriter() { |
| if (buffer) { |
| buffer->Append(os.str()); |
| } |
| } |
| |
| TextGenerator::TextBuffer::TextBuffer() = default; |
| TextGenerator::TextBuffer::~TextBuffer() = default; |
| |
| void TextGenerator::TextBuffer::IncrementIndent() { |
| current_indent += 2; |
| } |
| |
| void TextGenerator::TextBuffer::DecrementIndent() { |
| current_indent = std::max(2u, current_indent) - 2u; |
| } |
| |
| void TextGenerator::TextBuffer::Append(const std::string& line) { |
| lines.emplace_back(LineInfo{current_indent, line}); |
| } |
| |
| void TextGenerator::TextBuffer::Insert(const std::string& line, size_t before, uint32_t indent) { |
| if (TINT_UNLIKELY(before >= lines.size())) { |
| TINT_ICE() << "TextBuffer::Insert() called with before >= lines.size()\n" |
| << " before:" << before << "\n" |
| << " lines.size(): " << lines.size(); |
| return; |
| } |
| using DT = decltype(lines)::difference_type; |
| lines.insert(lines.begin() + static_cast<DT>(before), LineInfo{indent, line}); |
| } |
| |
| void TextGenerator::TextBuffer::Append(const TextBuffer& tb) { |
| for (auto& line : tb.lines) { |
| // TODO(bclayton): inefficient, consider optimizing |
| lines.emplace_back(LineInfo{current_indent + line.indent, line.content}); |
| } |
| } |
| |
| void TextGenerator::TextBuffer::Insert(const TextBuffer& tb, size_t before, uint32_t indent) { |
| if (TINT_UNLIKELY(before >= lines.size())) { |
| TINT_ICE() << "TextBuffer::Insert() called with before >= lines.size()\n" |
| << " before:" << before << "\n" |
| << " lines.size(): " << lines.size(); |
| return; |
| } |
| size_t idx = 0; |
| for (auto& line : tb.lines) { |
| // TODO(bclayton): inefficient, consider optimizing |
| using DT = decltype(lines)::difference_type; |
| lines.insert(lines.begin() + static_cast<DT>(before + idx), |
| LineInfo{indent + line.indent, line.content}); |
| idx++; |
| } |
| } |
| |
| std::string TextGenerator::TextBuffer::String(uint32_t indent /* = 0 */) const { |
| StringStream ss; |
| for (auto& line : lines) { |
| if (!line.content.empty()) { |
| for (uint32_t i = 0; i < indent + line.indent; i++) { |
| ss << " "; |
| } |
| ss << line.content; |
| } |
| ss << std::endl; |
| } |
| return ss.str(); |
| } |
| |
| TextGenerator::ScopedParen::ScopedParen(StringStream& stream) : s(stream) { |
| s << "("; |
| } |
| |
| TextGenerator::ScopedParen::~ScopedParen() { |
| s << ")"; |
| } |
| |
| TextGenerator::ScopedIndent::ScopedIndent(TextGenerator* generator) |
| : ScopedIndent(generator->current_buffer_) {} |
| |
| TextGenerator::ScopedIndent::ScopedIndent(TextBuffer* buffer) : buffer_(buffer) { |
| buffer_->IncrementIndent(); |
| } |
| TextGenerator::ScopedIndent::~ScopedIndent() { |
| buffer_->DecrementIndent(); |
| } |
| |
| } // namespace tint |