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

Refactor image sampling arguments, add textureLoad support to WGSL

Browse Source
This commit is contained in:
Dzmitry Malyshau
2020-09-18 09:57:12 -04:00
parent 0aa91927b2
commit d25ab1b1e8
5 changed files with 64 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/back/glsl.rs
View File

@@ -779,6 +779,9 @@ fn write_expression<'a, 'b>(
//TODO: handle MS
Cow::Owned(match level {
crate::SampleLevel::Auto => format!("texture({},{})", image_expr, coordinate_expr),
crate::SampleLevel::Zero => {
format!("textureLod({},{},0)", image_expr, coordinate_expr)
}
crate::SampleLevel::Exact(expr) => {
let level_expr = write_expression(&builder.expressions[expr], module, builder)?;
format!(
@@ -822,17 +825,12 @@ fn write_expression<'a, 'b>(
image_expr,
);
if !multi {
format!("texelFetch({},{})", sampler_constructor, coordinate_expr)
} else {
let index_expr =
write_expression(&builder.expressions[index], module, builder)?;
format!(
"texelFetch({},{},{})",
sampler_constructor, coordinate_expr, index_expr
)
}
let index_expr =
write_expression(&builder.expressions[index.unwrap()], module, builder)?;
format!(
"texelFetch({},{},{})",
sampler_constructor, coordinate_expr, index_expr
)
}
ImageClass::Storage(_) => format!("imageLoad({},{})", image_expr, coordinate_expr),
ImageClass::Depth => todo!(),

47
src/back/msl.rs
View File

@@ -515,16 +515,27 @@ impl<W: Write> Writer<W> {
sampler,
coordinate,
level,
depth_ref: None,
depth_ref,
} => {
let op = match depth_ref {
Some(_) => "sample_compare",
None => "sample",
};
//TODO: handle arrayed images
self.put_expression(image, function, module)?;
write!(self.out, ".sample(")?;
write!(self.out, ".{}(", op)?;
self.put_expression(sampler, function, module)?;
write!(self.out, ", ")?;
self.put_expression(coordinate, function, module)?;
if let Some(dref) = depth_ref {
write!(self.out, ", ")?;
self.put_expression(dref, function, module)?;
}
match level {
crate::SampleLevel::Auto => {}
crate::SampleLevel::Zero => {
write!(self.out, ", level(0)")?;
}
crate::SampleLevel::Exact(h) => {
write!(self.out, ", level(")?;
self.put_expression(h, function, module)?;
@@ -538,32 +549,6 @@ impl<W: Write> Writer<W> {
}
write!(self.out, ")")?;
}
crate::Expression::ImageSample {
image,
sampler,
coordinate,
level,
depth_ref: Some(dref),
} => {
//TODO: handle arrayed images
self.put_expression(image, function, module)?;
write!(self.out, ".sample_compare(")?;
self.put_expression(sampler, function, module)?;
write!(self.out, ", ")?;
self.put_expression(coordinate, function, module)?;
write!(self.out, ", ")?;
self.put_expression(dref, function, module)?;
match level {
crate::SampleLevel::Auto => {}
crate::SampleLevel::Exact(h) => {
write!(self.out, ", level(")?;
self.put_expression(h, function, module)?;
write!(self.out, ")")?;
}
crate::SampleLevel::Bias(_) => return Err(Error::UnexpectedSampleLevel(level)),
}
write!(self.out, ")")?;
}
crate::Expression::ImageLoad {
image,
coordinate,
@@ -573,8 +558,10 @@ impl<W: Write> Writer<W> {
self.put_expression(image, function, module)?;
write!(self.out, ".read(")?;
self.put_expression(coordinate, function, module)?;
write!(self.out, ", ")?;
self.put_expression(index, function, module)?;
if let Some(index) = index {
write!(self.out, ", ")?;
self.put_expression(index, function, module)?;
}
write!(self.out, ")")?;
}
crate::Expression::Call {