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] Add support for derivative functions

Browse Source
This commit is contained in:
João Capucho
2021-08-22 17:29:09 +01:00
committed by Dzmitry Malyshau
parent 222e0949bd
commit b0569d9593
1 changed files with 23 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/front/glsl/builtins.rs
View File

@@ -4,9 +4,9 @@ use super::{
Error, ErrorKind, Parser, Result, SourceMetadata,
};
use crate::{
BinaryOperator, Block, Constant, ConstantInner, Expression, Handle, ImageClass, ImageDimension,
ImageQuery, MathFunction, Module, RelationalFunction, SampleLevel, ScalarKind as Sk,
ScalarValue, Span, Type, TypeInner, VectorSize,
BinaryOperator, Block, Constant, ConstantInner, DerivativeAxis, Expression, Handle, ImageClass,
ImageDimension, ImageQuery, MathFunction, Module, RelationalFunction, SampleLevel,
ScalarKind as Sk, ScalarValue, Span, Type, TypeInner, VectorSize,
};
impl Module {
@@ -452,7 +452,8 @@ pub fn inject_builtin(declaration: &mut FunctionDeclaration, module: &mut Module
}
"sin" | "exp" | "exp2" | "sinh" | "cos" | "cosh" | "tan" | "tanh" | "acos" | "asin"
| "log" | "log2" | "radians" | "degrees" | "asinh" | "acosh" | "atanh"
| "floatBitsToInt" | "floatBitsToUint" => {
| "floatBitsToInt" | "floatBitsToUint" | "dFdx" | "dFdxFine" | "dFdxCoarse" | "dFdy"
| "dFdyFine" | "dFdyCoarse" | "fwidth" | "fwidthFine" | "fwidthCoarse" => {
// bits layout
// bit 0 trough 1 - dims
for bits in 0..(0b100) {
@@ -489,6 +490,15 @@ pub fn inject_builtin(declaration: &mut FunctionDeclaration, module: &mut Module
"degrees" => MacroCall::ConstMultiply(180.0 / std::f64::consts::PI),
"floatBitsToInt" => MacroCall::BitCast(Sk::Sint),
"floatBitsToUint" => MacroCall::BitCast(Sk::Uint),
"dFdx" | "dFdxFine" | "dFdxCoarse" => {
MacroCall::Derivate(DerivativeAxis::X)
}
"dFdy" | "dFdyFine" | "dFdyCoarse" => {
MacroCall::Derivate(DerivativeAxis::Y)
}
"fwidth" | "fwidthFine" | "fwidthCoarse" => {
MacroCall::Derivate(DerivativeAxis::Width)
}
_ => unreachable!(),
},
))
@@ -1395,6 +1405,7 @@ pub enum MacroCall {
Clamp(Option<VectorSize>),
ConstMultiply(f64),
BitCast(Sk),
Derivate(DerivativeAxis),
}
impl MacroCall {
@@ -1639,6 +1650,14 @@ impl MacroCall {
SourceMetadata::none(),
body,
)),
MacroCall::Derivate(axis) => Ok(ctx.add_expression(
Expression::Derivative {
axis,
expr: args[0],
},
SourceMetadata::none(),
body,
)),
}
}
}