Add missing ToWgslString impls for vecs (#242)

Implement `ToWgslString` for the following vector types:
- `IVec2`, `IVec3`, `IVec4`
- `UVec2`, `UVec3`, `UVec4`
- `BVec2`, `BVec3`, `BVec4`

CHANGELOG.md

@@ -22,6 +22,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Added `EvalContext::push_stmt()` to emit a single statement prepended to the currently evaluating expression. This is useful to define temporary local variables for storing expressions with a side-effect.
 - Added `EvalContext::make_fn()` to create a function with a dedicated `EvalContext`, allowing to properly scope local variables and stored expression side effects.
 - Added `Module::try_get()`, similar to `Module::get()` but returning a `Result<&Expr, ExprError>` instead for convenience.
+- Added implementations of `ToWgslString` for the missing vector types (`UVec2/3/4`, `IVec2/3/4`, `BVec2/3/4`).
 
 ### Changed
 

src/lib.rs

@@ -297,18 +297,117 @@ impl ToWgslString for bool {
     }
 }
 
+impl ToWgslString for BVec2 {
+    fn to_wgsl_string(&self) -> String {
+        format!(
+            "vec2<bool>({0},{1})",
+            self.x.to_wgsl_string(),
+            self.y.to_wgsl_string()
+        )
+    }
+}
+
+impl ToWgslString for BVec3 {
+    fn to_wgsl_string(&self) -> String {
+        format!(
+            "vec3<bool>({0},{1},{2})",
+            self.x.to_wgsl_string(),
+            self.y.to_wgsl_string(),
+            self.z.to_wgsl_string()
+        )
+    }
+}
+
+impl ToWgslString for BVec4 {
+    fn to_wgsl_string(&self) -> String {
+        format!(
+            "vec4<bool>({0},{1},{2},{3})",
+            self.x.to_wgsl_string(),
+            self.y.to_wgsl_string(),
+            self.z.to_wgsl_string(),
+            self.w.to_wgsl_string()
+        )
+    }
+}
+
 impl ToWgslString for i32 {
     fn to_wgsl_string(&self) -> String {
         format!("{}", self)
     }
 }
 
+impl ToWgslString for IVec2 {
+    fn to_wgsl_string(&self) -> String {
+        format!(
+            "vec2<i32>({0},{1})",
+            self.x.to_wgsl_string(),
+            self.y.to_wgsl_string()
+        )
+    }
+}
+
+impl ToWgslString for IVec3 {
+    fn to_wgsl_string(&self) -> String {
+        format!(
+            "vec3<i32>({0},{1},{2})",
+            self.x.to_wgsl_string(),
+            self.y.to_wgsl_string(),
+            self.z.to_wgsl_string()
+        )
+    }
+}
+
+impl ToWgslString for IVec4 {
+    fn to_wgsl_string(&self) -> String {
+        format!(
+            "vec4<i32>({0},{1},{2},{3})",
+            self.x.to_wgsl_string(),
+            self.y.to_wgsl_string(),
+            self.z.to_wgsl_string(),
+            self.w.to_wgsl_string()
+        )
+    }
+}
+
 impl ToWgslString for u32 {
     fn to_wgsl_string(&self) -> String {
         format!("{}", self)
     }
 }
 
+impl ToWgslString for UVec2 {
+    fn to_wgsl_string(&self) -> String {
+        format!(
+            "vec2<u32>({0},{1})",
+            self.x.to_wgsl_string(),
+            self.y.to_wgsl_string()
+        )
+    }
+}
+
+impl ToWgslString for UVec3 {
+    fn to_wgsl_string(&self) -> String {
+        format!(
+            "vec3<u32>({0},{1},{2})",
+            self.x.to_wgsl_string(),
+            self.y.to_wgsl_string(),
+            self.z.to_wgsl_string()
+        )
+    }
+}
+
+impl ToWgslString for UVec4 {
+    fn to_wgsl_string(&self) -> String {
+        format!(
+            "vec4<u32>({0},{1},{2},{3})",
+            self.x.to_wgsl_string(),
+            self.y.to_wgsl_string(),
+            self.z.to_wgsl_string(),
+            self.w.to_wgsl_string()
+        )
+    }
+}
+
 impl ToWgslString for CpuValue<f32> {
     fn to_wgsl_string(&self) -> String {
         match self {
@@ -1481,6 +1580,36 @@ mod tests {
         assert_eq!(s, "vec4<f32>(1.,2.,-1.,2.)");
     }
 
+    #[test]
+    fn to_wgsl_ivec() {
+        let s = IVec2::new(1, 2).to_wgsl_string();
+        assert_eq!(s, "vec2<i32>(1,2)");
+        let s = IVec3::new(1, 2, -1).to_wgsl_string();
+        assert_eq!(s, "vec3<i32>(1,2,-1)");
+        let s = IVec4::new(1, 2, -1, 2).to_wgsl_string();
+        assert_eq!(s, "vec4<i32>(1,2,-1,2)");
+    }
+
+    #[test]
+    fn to_wgsl_uvec() {
+        let s = UVec2::new(1, 2).to_wgsl_string();
+        assert_eq!(s, "vec2<u32>(1,2)");
+        let s = UVec3::new(1, 2, 42).to_wgsl_string();
+        assert_eq!(s, "vec3<u32>(1,2,42)");
+        let s = UVec4::new(1, 2, 42, 5).to_wgsl_string();
+        assert_eq!(s, "vec4<u32>(1,2,42,5)");
+    }
+
+    #[test]
+    fn to_wgsl_bvec() {
+        let s = BVec2::new(false, true).to_wgsl_string();
+        assert_eq!(s, "vec2<bool>(false,true)");
+        let s = BVec3::new(false, true, true).to_wgsl_string();
+        assert_eq!(s, "vec3<bool>(false,true,true)");
+        let s = BVec4::new(false, true, true, false).to_wgsl_string();
+        assert_eq!(s, "vec4<bool>(false,true,true,false)");
+    }
+
     #[test]
     fn to_wgsl_value_f32() {
         let s = CpuValue::Single(1.0_f32).to_wgsl_string();