1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use type_fields::Field;

use crate::signed_distance_field::{attributes::distance::Distance, SignedDistanceField};

use super::{Operator, SignedDistanceOperator};

#[derive(Debug, Default, Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Field)]
pub struct ConditionalOp<Op, const CONDITION: bool> {
    conditional_op: Op,
}

impl<Op, Dim, const CONDITION: bool> SignedDistanceOperator<Dim, Distance>
    for ConditionalOp<Op, CONDITION>
where
    Op: SignedDistanceOperator<Dim, Distance>,
{
    fn operator<Sdf>(&self, sdf: &Sdf, p: Dim) -> Distance
    where
        Sdf: SignedDistanceField<Dim, Distance>,
        Dim: Clone,
    {
        if CONDITION {
            self.conditional_op.operator(sdf, p)
        } else {
            sdf.evaluate(p)
        }
    }
}

pub type Conditional<Op, Sdf, const CONDITION: bool> = Operator<ConditionalOp<Op, CONDITION>, Sdf>;

impl<Op, Sdf, const CONDITION: bool> Conditional<Op, Sdf, CONDITION> {
    pub fn conditional_op(&mut self) -> &mut Op {
        &mut self.op.conditional_op
    }
}

#[cfg(test)]
pub mod test {
    use rust_gpu_bridge::prelude::Vec3;
    use type_fields::field::Field;

    use crate::{prelude::StretchDistOp, signed_distance_field::shapes::composite::Sphere};

    use super::Conditional;

    #[test]
    pub fn test_conditional() {
        Conditional::<StretchDistOp<Vec3>, Sphere, true>::default()
            .with(Conditional::conditional_op, StretchDistOp::default());
    }
}