blob: d665a10e38c39f759e27c38712e68a6419a8862d [file] [log] [blame]
// Copyright 2021 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/lang/wgsl/ast/builder.h"
using namespace tint::core::number_suffixes; // NOLINT
namespace tint::ast {
Builder::VarOptions::~VarOptions() = default;
Builder::LetOptions::~LetOptions() = default;
Builder::ConstOptions::~ConstOptions() = default;
Builder::OverrideOptions::~OverrideOptions() = default;
Builder::Builder()
: id_(GenerationID::New()),
ast_(ast_nodes_.Create<ast::Module>(id_, AllocateNodeID(), Source{})) {}
Builder::Builder(Builder&& rhs)
: id_(std::move(rhs.id_)),
last_ast_node_id_(std::move(rhs.last_ast_node_id_)),
ast_nodes_(std::move(rhs.ast_nodes_)),
ast_(rhs.ast_),
symbols_(std::move(rhs.symbols_)),
diagnostics_(std::move(rhs.diagnostics_)) {
rhs.MarkAsMoved();
}
Builder::~Builder() = default;
Builder& Builder::operator=(Builder&& rhs) {
rhs.MarkAsMoved();
AssertNotMoved();
id_ = std::move(rhs.id_);
last_ast_node_id_ = std::move(rhs.last_ast_node_id_);
ast_nodes_ = std::move(rhs.ast_nodes_);
ast_ = std::move(rhs.ast_);
symbols_ = std::move(rhs.symbols_);
diagnostics_ = std::move(rhs.diagnostics_);
return *this;
}
bool Builder::IsValid() const {
return !diagnostics_.contains_errors();
}
void Builder::MarkAsMoved() {
AssertNotMoved();
moved_ = true;
}
void Builder::AssertNotMoved() const {
if (TINT_UNLIKELY(moved_)) {
TINT_ICE() << "Attempting to use Builder after it has been moved";
}
}
Builder::TypesBuilder::TypesBuilder(Builder* pb) : builder(pb) {}
const ast::Statement* Builder::WrapInStatement(const ast::Expression* expr) {
// Create a temporary variable of inferred type from expr.
return Decl(Let(symbols_.New(), expr));
}
const ast::VariableDeclStatement* Builder::WrapInStatement(const ast::Variable* v) {
return create<ast::VariableDeclStatement>(v);
}
const ast::Statement* Builder::WrapInStatement(const ast::Statement* stmt) {
return stmt;
}
const ast::Function* Builder::WrapInFunction(VectorRef<const ast::Statement*> stmts) {
return Func("test_function", {}, ty.void_(), std::move(stmts),
Vector{
create<ast::StageAttribute>(ast::PipelineStage::kCompute),
WorkgroupSize(1_i, 1_i, 1_i),
});
}
} // namespace tint::ast