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

Add support for relational functions to glsl-in

Browse Source
This commit is contained in:
João Capucho
2021-02-06 20:24:07 +00:00
committed by Dzmitry Malyshau
parent da8f9ecbb9
commit 708751a805
1 changed files with 25 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,6 +1,7 @@
use crate::{
proc::{ensure_block_returns, Typifier},
BinaryOperator, Block, Expression, Function, MathFunction, SampleLevel, TypeInner,
BinaryOperator, Block, Expression, Function, MathFunction, RelationalFunction, SampleLevel,
TypeInner,
};
use super::{ast::*, error::ErrorKind};
@@ -171,6 +172,10 @@ impl Program {
statements: fc.args.into_iter().flat_map(|a| a.statements).collect(),
})
}
"isinf" => self.parse_relational_fun(name, fc.args, RelationalFunction::IsInf),
"isnan" => self.parse_relational_fun(name, fc.args, RelationalFunction::IsNan),
"all" => self.parse_relational_fun(name, fc.args, RelationalFunction::All),
"any" => self.parse_relational_fun(name, fc.args, RelationalFunction::Any),
func_name => {
let function = *self.lookup_function.get(func_name).ok_or_else(|| {
ErrorKind::SemanticError(
@@ -191,6 +196,25 @@ impl Program {
}
}
pub fn parse_relational_fun(
&mut self,
name: String,
args: Vec<ExpressionRule>,
fun: RelationalFunction,
) -> Result<ExpressionRule, ErrorKind> {
if args.len() != 1 {
return Err(ErrorKind::WrongNumberArgs(name, 1, args.len()));
}
Ok(ExpressionRule {
expression: self.context.expressions.append(Expression::Relational {
fun,
argument: args[0].expression,
}),
sampler: None,
statements: args.into_iter().flat_map(|a| a.statements).collect(),
})
}
pub fn add_function_prelude(&mut self) {
for (var_handle, var) in self.module.global_variables.iter() {
if let Some(name) = var.name.as_ref() {