mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-08-31 07:50:24 +00:00
Move the bulk of the constant evaulation logic out of transform::FoldConstants and into Resolver and sem::Expression. transform::FoldConstants now replace TypeConstructor nodes that have a constant value on the expression. This is ground work to: * Cleaning up the HLSL uniform buffer indexing, which is `/` and `%` arithmatic heavy * Prepares us to handle `constexpr` when it lands in the spec * Provide a centralized place to do constant evaluation, instead of the having similar logic scattered around the codebase. Change-Id: I3e2f542be692046a8d243b62a82556db519953e7 Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/57426 Kokoro: Kokoro <noreply+kokoro@google.com> Reviewed-by: Antonio Maiorano <amaiorano@google.com> Reviewed-by: James Price <jrprice@google.com>
129 lines
4.3 KiB
C++
129 lines
4.3 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/writer/append_vector.h"
|
|
|
|
#include <utility>
|
|
|
|
#include "src/sem/expression.h"
|
|
|
|
namespace tint {
|
|
namespace writer {
|
|
|
|
namespace {
|
|
|
|
ast::TypeConstructorExpression* AsVectorConstructor(ProgramBuilder* b,
|
|
ast::Expression* expr) {
|
|
if (auto* constructor = expr->As<ast::TypeConstructorExpression>()) {
|
|
if (b->TypeOf(constructor)->Is<sem::Vector>()) {
|
|
return constructor;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
ast::TypeConstructorExpression* AppendVector(ProgramBuilder* b,
|
|
ast::Expression* vector,
|
|
ast::Expression* scalar) {
|
|
uint32_t packed_size;
|
|
const sem::Type* packed_el_sem_ty;
|
|
auto* vector_sem = b->Sem().Get(vector);
|
|
auto* vector_ty = vector_sem->Type()->UnwrapRef();
|
|
if (auto* vec = vector_ty->As<sem::Vector>()) {
|
|
packed_size = vec->size() + 1;
|
|
packed_el_sem_ty = vec->type();
|
|
} else {
|
|
packed_size = 2;
|
|
packed_el_sem_ty = vector_ty;
|
|
}
|
|
|
|
ast::Type* packed_el_ty = nullptr;
|
|
if (packed_el_sem_ty->Is<sem::I32>()) {
|
|
packed_el_ty = b->create<ast::I32>();
|
|
} else if (packed_el_sem_ty->Is<sem::U32>()) {
|
|
packed_el_ty = b->create<ast::U32>();
|
|
} else if (packed_el_sem_ty->Is<sem::F32>()) {
|
|
packed_el_ty = b->create<ast::F32>();
|
|
} else if (packed_el_sem_ty->Is<sem::Bool>()) {
|
|
packed_el_ty = b->create<ast::Bool>();
|
|
} else {
|
|
TINT_UNREACHABLE(Writer, b->Diagnostics())
|
|
<< "unsupported vector element type: "
|
|
<< packed_el_sem_ty->TypeInfo().name;
|
|
}
|
|
|
|
auto* statement = vector_sem->Stmt();
|
|
|
|
auto* packed_ty = b->create<ast::Vector>(packed_el_ty, packed_size);
|
|
auto* packed_sem_ty = b->create<sem::Vector>(packed_el_sem_ty, packed_size);
|
|
|
|
// If the coordinates are already passed in a vector constructor, extract
|
|
// the elements into the new vector instead of nesting a vector-in-vector.
|
|
ast::ExpressionList packed;
|
|
if (auto* vc = AsVectorConstructor(b, vector)) {
|
|
packed = vc->values();
|
|
if (packed.size() == 0) {
|
|
// Zero-value vector constructor. Populate with zeros
|
|
auto buildZero = [&]() -> ast::ScalarConstructorExpression* {
|
|
if (packed_el_sem_ty->Is<sem::I32>()) {
|
|
return b->Expr(0);
|
|
} else if (packed_el_sem_ty->Is<sem::U32>()) {
|
|
return b->Expr(0u);
|
|
} else if (packed_el_sem_ty->Is<sem::F32>()) {
|
|
return b->Expr(0.0f);
|
|
} else if (packed_el_sem_ty->Is<sem::Bool>()) {
|
|
return b->Expr(false);
|
|
} else {
|
|
TINT_UNREACHABLE(Writer, b->Diagnostics())
|
|
<< "unsupported vector element type: "
|
|
<< packed_el_sem_ty->TypeInfo().name;
|
|
}
|
|
return nullptr;
|
|
};
|
|
|
|
for (uint32_t i = 0; i < packed_size - 1; i++) {
|
|
auto* zero = buildZero();
|
|
b->Sem().Add(
|
|
zero, b->create<sem::Expression>(zero, packed_el_sem_ty, statement,
|
|
sem::Constant{}));
|
|
packed.emplace_back(zero);
|
|
}
|
|
}
|
|
} else {
|
|
packed.emplace_back(vector);
|
|
}
|
|
if (packed_el_sem_ty != b->TypeOf(scalar)->UnwrapRef()) {
|
|
// Cast scalar to the vector element type
|
|
auto* scalar_cast = b->Construct(packed_el_ty, scalar);
|
|
b->Sem().Add(scalar_cast,
|
|
b->create<sem::Expression>(scalar_cast, packed_el_sem_ty,
|
|
statement, sem::Constant{}));
|
|
packed.emplace_back(scalar_cast);
|
|
} else {
|
|
packed.emplace_back(scalar);
|
|
}
|
|
|
|
auto* constructor = b->Construct(packed_ty, std::move(packed));
|
|
b->Sem().Add(constructor,
|
|
b->create<sem::Expression>(constructor, packed_sem_ty, statement,
|
|
sem::Constant{}));
|
|
|
|
return constructor;
|
|
}
|
|
|
|
} // namespace writer
|
|
} // namespace tint
|