mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-06-05 06:03:34 +00:00
d734dd0c10
Change-Id: I0bc5645d65d7b53058bda7b343a991a9614741be Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/34271 Reviewed-by: dan sinclair <dsinclair@chromium.org>
119 lines
3.1 KiB
C++
119 lines
3.1 KiB
C++
// 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/ast/module.h"
|
|
|
|
#include <sstream>
|
|
|
|
#include "src/ast/type/struct_type.h"
|
|
|
|
namespace tint {
|
|
namespace ast {
|
|
|
|
Module::Module() = default;
|
|
|
|
Module::Module(Module&&) = default;
|
|
|
|
Module::~Module() = default;
|
|
|
|
Function* Module::FindFunctionByName(const std::string& name) const {
|
|
for (auto* func : functions_) {
|
|
if (func->name() == name) {
|
|
return func;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
Function* Module::FindFunctionByNameAndStage(const std::string& name,
|
|
ast::PipelineStage stage) const {
|
|
for (auto* func : functions_) {
|
|
if (func->name() == name && func->pipeline_stage() == stage) {
|
|
return func;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
bool Module::IsValid() const {
|
|
for (auto* var : global_variables_) {
|
|
if (var == nullptr || !var->IsValid()) {
|
|
return false;
|
|
}
|
|
}
|
|
for (auto* const ty : constructed_types_) {
|
|
if (ty == nullptr) {
|
|
return false;
|
|
}
|
|
if (ty->Is<ast::type::AliasType>()) {
|
|
auto* alias = ty->As<ast::type::AliasType>();
|
|
if (alias->type() == nullptr) {
|
|
return false;
|
|
}
|
|
if (alias->type()->Is<ast::type::StructType>() &&
|
|
alias->type()->As<ast::type::StructType>()->name().empty()) {
|
|
return false;
|
|
}
|
|
} else if (ty->Is<ast::type::StructType>()) {
|
|
auto* str = ty->As<ast::type::StructType>();
|
|
if (str->name().empty()) {
|
|
return false;
|
|
}
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
for (auto* func : functions_) {
|
|
if (func == nullptr || !func->IsValid()) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
std::string Module::to_str() const {
|
|
std::ostringstream out;
|
|
|
|
out << "Module{" << std::endl;
|
|
const auto indent = 2;
|
|
for (auto* const ty : constructed_types_) {
|
|
for (size_t i = 0; i < indent; ++i) {
|
|
out << " ";
|
|
}
|
|
if (ty->Is<ast::type::AliasType>()) {
|
|
auto* alias = ty->As<ast::type::AliasType>();
|
|
out << alias->name() << " -> " << alias->type()->type_name() << std::endl;
|
|
if (alias->type()->Is<ast::type::StructType>()) {
|
|
alias->type()->As<ast::type::StructType>()->impl()->to_str(out, indent);
|
|
}
|
|
} else if (ty->Is<ast::type::StructType>()) {
|
|
auto* str = ty->As<ast::type::StructType>();
|
|
out << str->name() << " ";
|
|
str->impl()->to_str(out, indent);
|
|
}
|
|
}
|
|
for (auto* var : global_variables_) {
|
|
var->to_str(out, indent);
|
|
}
|
|
for (auto* func : functions_) {
|
|
func->to_str(out, indent);
|
|
}
|
|
out << "}" << std::endl;
|
|
|
|
return out.str();
|
|
}
|
|
|
|
} // namespace ast
|
|
} // namespace tint
|