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] Handle Pow in constant evaluation

Browse Source
This commit is contained in:
João Capucho
2021-08-01 18:35:46 +01:00
committed by Dzmitry Malyshau
parent da77e1f847
commit c084e2cfa8
1 changed files with 44 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
src/front/glsl/constants.rs
View File

@@ -49,6 +49,8 @@ pub enum ConstantSolvingError {
InvalidUnaryOpArg,
#[error("Cannot apply the binary op to the arguments")]
InvalidBinaryOpArgs,
#[error("Cannot apply math function to type")]
InvalidMathArg,
#[error("Splat/swizzle type is not registered")]
DestinationTypeNotFound,
#[error("Not implemented: {0}")]
@@ -154,8 +156,48 @@ impl<'a> ConstantSolver<'a> {
self.binary_op(op, left_constant, right_constant)
}
Expression::Math { fun, .. } => {
Err(ConstantSolvingError::NotImplemented(format!("{:?}", fun)))
Expression::Math { fun, arg, arg1, .. } => {
let arg = self.solve(arg)?;
let arg1 = arg1.map(|arg| self.solve(arg)).transpose()?;
let const0 = &self.constants[arg].inner;
let const1 = arg1.map(|arg| &self.constants[arg].inner);
match fun {
crate::MathFunction::Pow => {
let (value, width) = match (const0, const1.unwrap()) {
(
&ConstantInner::Scalar {
width,
value: value0,
},
&ConstantInner::Scalar { value: value1, .. },
) => (
match (value0, value1) {
(ScalarValue::Sint(a), ScalarValue::Sint(b)) => {
ScalarValue::Sint(a.pow(b as u32))
}
(ScalarValue::Uint(a), ScalarValue::Uint(b)) => {
ScalarValue::Uint(a.pow(b as u32))
}
(ScalarValue::Float(a), ScalarValue::Float(b)) => {
ScalarValue::Float(a.powf(b))
}
_ => return Err(ConstantSolvingError::InvalidMathArg),
},
width,
),
_ => return Err(ConstantSolvingError::InvalidMathArg),
};
Ok(self.constants.fetch_or_append(Constant {
name: None,
specialization: None,
inner: ConstantInner::Scalar { width, value },
}))
}
_ => Err(ConstantSolvingError::NotImplemented(format!("{:?}", fun))),
}
}
Expression::As {
convert,