github/wgpu
1
1
Fork 0
mirror of https://github.com/gfx-rs/wgpu.git synced 2026-04-22 03:02:01 -04:00
Code Issues Packages Projects Releases Wiki Activity

[glsl-in] Implement textureSize

Browse Source
This commit is contained in:
João Capucho
2021-06-20 10:44:10 +01:00
committed by Dzmitry Malyshau
parent fd042e6f31
commit 68609cdff8
1 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/front/glsl/functions.rs
View File

@@ -1,8 +1,8 @@
use crate::{
proc::ensure_block_returns, Arena, BinaryOperator, Block, EntryPoint, Expression, Function,
FunctionArgument, FunctionResult, Handle, LocalVariable, MathFunction, RelationalFunction,
SampleLevel, ScalarKind, Statement, StructMember, SwizzleComponent, Type, TypeInner,
VectorSize,
FunctionArgument, FunctionResult, Handle, ImageQuery, LocalVariable, MathFunction,
RelationalFunction, SampleLevel, ScalarKind, Statement, StructMember, SwizzleComponent, Type,
TypeInner, VectorSize,
};
use super::{ast::*, error::ErrorKind, SourceMetadata};
@@ -204,6 +204,21 @@ impl Program<'_> {
))
}
}
"textureSize" => {
if !(1..=2).contains(&args.len()) {
return Err(ErrorKind::wrong_function_args(name, 1, args.len(), meta));
}
Ok(Some(ctx.add_expression(
Expression::ImageQuery {
image: args[0].0,
query: ImageQuery::Size {
level: args.get(1).map(|e| e.0),
},
},
body,
)))
}
"texelFetch" => {
if args.len() != 3 {
return Err(ErrorKind::wrong_function_args(name, 3, args.len(), meta));