nordic-dev.net/wgpu
nordic-dev.net/wgpu
2
0
Fork 0
mirror of https://github.com/gfx-rs/wgpu.git synced 2024-11-29 18:23:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

impl len and distance

Browse Source 
Signed-off-by: sagudev <16504129+sagudev@users.noreply.github.com>
This commit is contained in:
sagudev 2024-09-06 15:53:46 +02:00
parent 2aae42d5c3
commit ec7ceacf57
1 changed files with 51 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
naga/src/proc/constant_evaluator.rs
View File

@ -1572,6 +1572,57 @@ impl<'a> ConstantEvaluator<'a> {
Err(ConstantEvaluatorError::InvalidMathArg)
}
}
crate::MathFunction::Length => {
let e1 = LiteralVector::from_expr(arg, self, span, true)?;
fn float_length<F, const CAP: usize>(e: ArrayVec<F, CAP>) -> F
where
F: std::ops::Mul<F>,
F: num_traits::Float + std::iter::Sum,
{
e.iter().map(|&ei| ei * ei).sum::<F>().sqrt()
}
LiteralVector::from_literal(match e1 {
LiteralVector::AbstractFloat(a) => Literal::AbstractFloat(float_length(a)),
LiteralVector::F32(a) => Literal::F32(float_length(a)),
_ => return Err(ConstantEvaluatorError::InvalidMathArg),
})
.handle(self, span)
}
crate::MathFunction::Distance => {
let e1 = LiteralVector::from_expr(arg, self, span, true)?;
let e2 = LiteralVector::from_expr(arg1.unwrap(), self, span, true)?;
if e1.len() != e2.len() {
return Err(ConstantEvaluatorError::InvalidMathArg);
}
fn float_distance<F, const CAP: usize>(
a: ArrayVec<F, CAP>,
b: ArrayVec<F, CAP>,
) -> F
where
F: std::ops::Mul<F>,
F: num_traits::Float + std::iter::Sum + std::ops::Sub,
{
a.iter()
.zip(b.iter())
.map(|(&aa, &bb)| aa - bb)
.map(|ei| ei * ei)
.sum::<F>()
.sqrt()
}
LiteralVector::from_literal(match (e1, e2) {
(LiteralVector::AbstractFloat(a), LiteralVector::AbstractFloat(b)) => {
Literal::AbstractFloat(float_distance(a, b))
}
(LiteralVector::F32(a), LiteralVector::F32(b)) => {
Literal::F32(float_distance(a, b))
}
_ => return Err(ConstantEvaluatorError::InvalidMathArg),
})
.handle(self, span)
}
// computational
crate::MathFunction::Sign => {
component_wise_signed!(self, span, [arg], |e| { Ok([e.signum()]) })