mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-05-31 11:41:35 +00:00
97a7d266e0
Adds a transform that adds an additional 'level' parameter to textureLoad calls when used with a texture_external. Adds a test for the textureLoad transform. Additionally adds a test for calls to textureDimensions with a texture_external. Bug: dawn:728 Change-Id: I613ce8185e9a4c49529fd8e48323e586c95dde04 Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/50420 Commit-Queue: Ben Clayton <bclayton@google.com> Kokoro: Kokoro <noreply+kokoro@google.com> Reviewed-by: Ben Clayton <bclayton@google.com>
109 lines
4.4 KiB
C++
109 lines
4.4 KiB
C++
// 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/transform/external_texture_transform.h"
|
|
|
|
#include "src/program_builder.h"
|
|
#include "src/sem/call.h"
|
|
#include "src/sem/variable.h"
|
|
|
|
namespace tint {
|
|
namespace transform {
|
|
|
|
ExternalTextureTransform::ExternalTextureTransform() = default;
|
|
ExternalTextureTransform::~ExternalTextureTransform() = default;
|
|
|
|
Output ExternalTextureTransform::Run(const Program* in, const DataMap&) {
|
|
ProgramBuilder out;
|
|
CloneContext ctx(&out, in);
|
|
auto& sem = ctx.src->Sem();
|
|
|
|
// Within this transform, usages of texture_external are replaced with a
|
|
// texture_2d<f32>, which will allow us perform operations on a
|
|
// texture_external without maintaining texture_external-specific code
|
|
// generation paths in the backends.
|
|
|
|
// When replacing instances of texture_external with texture_2d<f32> we must
|
|
// also modify calls to the texture_external overload of textureLoad, which
|
|
// unlike its texture_2d<f32> overload does not require a level parameter.
|
|
// To do this we identify calls to textureLoad that use texture_external as
|
|
// the first parameter and add a parameter for the level (which is always 0).
|
|
|
|
// Scan the AST nodes for calls to textureLoad.
|
|
for (auto* node : ctx.src->ASTNodes().Objects()) {
|
|
if (auto* call_expr = node->As<ast::CallExpression>()) {
|
|
if (auto* intrinsic =
|
|
sem.Get(call_expr)->Target()->As<sem::Intrinsic>()) {
|
|
if (intrinsic->Type() == sem::IntrinsicType::kTextureLoad) {
|
|
// When a textureLoad has been identified, check if the first
|
|
// parameter is an external texture.
|
|
if (auto* var =
|
|
sem.Get(call_expr->params()[0])->As<sem::VariableUser>()) {
|
|
if (var->Variable()->Type()->Is<sem::ExternalTexture>()) {
|
|
if (call_expr->params().size() != 2) {
|
|
TINT_ICE(ctx.dst->Diagnostics())
|
|
<< "expected TextureLoad call with a texture_external to "
|
|
"have 2 parameters, found "
|
|
<< call_expr->params().size() << " parameters";
|
|
}
|
|
// Replace the textureLoad call with another that has the same
|
|
// parameters in addition to a signed integer literal for the
|
|
// level parameter.
|
|
auto* exp = ctx.Clone(call_expr->func());
|
|
|
|
auto* externalTextureParam = ctx.Clone(call_expr->params()[0]);
|
|
auto* coordsParam = ctx.Clone(call_expr->params()[1]);
|
|
// Level is always 0 for an external texture.
|
|
auto* levelParam = ctx.dst->Expr(0);
|
|
|
|
ast::ExpressionList params = {externalTextureParam, coordsParam,
|
|
levelParam};
|
|
|
|
auto* newCall = ctx.dst->create<ast::CallExpression>(exp, params);
|
|
ctx.Replace(call_expr, newCall);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Scan the AST nodes for external texture declarations.
|
|
for (auto* node : ctx.src->ASTNodes().Objects()) {
|
|
if (auto* var = node->As<ast::Variable>()) {
|
|
if (Is<ast::ExternalTexture>(var->type())) {
|
|
// Replace a single-plane external texture with a 2D, f32 sampled
|
|
// texture.
|
|
auto newType = ctx.dst->ty.sampled_texture(ast::TextureDimension::k2d,
|
|
ctx.dst->ty.f32());
|
|
auto clonedSrc = ctx.Clone(var->source());
|
|
auto clonedSym = ctx.Clone(var->symbol());
|
|
auto* clonedConstructor = ctx.Clone(var->constructor());
|
|
auto clonedDecorations = ctx.Clone(var->decorations());
|
|
auto* newVar = ctx.dst->create<ast::Variable>(
|
|
clonedSrc, clonedSym, var->declared_storage_class(), newType,
|
|
var->is_const(), clonedConstructor, clonedDecorations);
|
|
|
|
ctx.Replace(var, newVar);
|
|
}
|
|
}
|
|
}
|
|
|
|
ctx.Clone();
|
|
return Output{Program(std::move(out))};
|
|
}
|
|
|
|
} // namespace transform
|
|
} // namespace tint
|