Make Gen.unitVector available on non-Apple platforms

Author: x-sheep

Sources/PropertyBased/Gen+SIMD.swift

@@ -5,6 +5,10 @@
 //  Created by Lennard Sprong on 19/05/2025.
 //
 
+#if canImport(simd)
+import simd
+#endif
+
 extension Generator {
     /// Produces a generator that creates a pair of values.
     ///
@@ -51,56 +55,55 @@ extension Generator where ResultValue: SIMDScalar & Sendable {
     }
 }
 
-#if canImport(simd)
-import simd
-extension Gen where Value == simd_float2 {
+extension Gen<SIMD2<Float>> {
     /// A generator of vectors with length 1.
-    public static var unitVector: Generator<simd_float2, Shrink.None<(Float, Float)>> {
+    public static var unitVector: Generator<SIMD2<Float>, Shrink.None<(Float, Float)>> {
         let gen = Gen<Float>.float(in: 0...1)
         return gen.simd2.map { normalize($0) }.filter { $0.x.isFinite }.withoutShrink()
     }
 }
 
-extension Gen where Value == simd_float3 {
+extension Gen<SIMD3<Float>> {
     /// A generator of vectors with length 1.
-    public static var unitVector: Generator<simd_float3, Shrink.None<(Float, Float, Float)>> {
+    public static var unitVector: Generator<SIMD3<Float>, Shrink.None<(Float, Float, Float)>> {
         let gen = Gen<Float>.float(in: 0...1)
         return gen.simd3.map { normalize($0) }.filter { $0.x.isFinite }.withoutShrink()
     }
 }
 
-extension Gen where Value == simd_float4 {
+extension Gen<SIMD4<Float>> {
     /// A generator of vectors with length 1.
-    public static var unitVector: Generator<simd_float4, Shrink.None<(Float, Float, Float, Float)>> {
+    public static var unitVector: Generator<SIMD4<Float>, Shrink.None<(Float, Float, Float, Float)>> {
         let gen = Gen<Float>.float(in: 0...1)
         return gen.simd4.map { normalize($0) }.filter { $0.x.isFinite }.withoutShrink()
     }
 }
 
-extension Gen where Value == simd_double2 {
+extension Gen<SIMD2<Double>> {
     /// A generator of vectors with length 1.
-    public static var unitVector: Generator<simd_double2, Shrink.None<(Double, Double)>> {
+    public static var unitVector: Generator<SIMD2<Double>, Shrink.None<(Double, Double)>> {
         let gen = Gen<Double>.double(in: 0...1)
         return gen.simd2.map { normalize($0) }.filter { $0.x.isFinite }.withoutShrink()
     }
 }
 
-extension Gen where Value == simd_double3 {
+extension Gen<SIMD3<Double>> {
     /// A generator of vectors with length 1.
-    public static var unitVector: Generator<simd_double3, Shrink.None<(Double, Double, Double)>> {
+    public static var unitVector: Generator<SIMD3<Double>, Shrink.None<(Double, Double, Double)>> {
         let gen = Gen<Double>.double(in: 0...1)
         return gen.simd3.map { normalize($0) }.filter { $0.x.isFinite }.withoutShrink()
     }
 }
 
-extension Gen where Value == simd_double4 {
+extension Gen<SIMD4<Double>> {
     /// A generator of vectors with length 1.
-    public static var unitVector: Generator<simd_double4, Shrink.None<(Double, Double, Double, Double)>> {
+    public static var unitVector: Generator<SIMD4<Double>, Shrink.None<(Double, Double, Double, Double)>> {
         let gen = Gen<Double>.double(in: 0...1)
         return gen.simd4.map { normalize($0) }.filter { $0.x.isFinite }.withoutShrink()
     }
 }
 
+#if canImport(simd)
 extension Gen where Value == simd_quatf {
     /// A generator of rotation quaternions with length 1 and a random angle.
     public static var simd_quatf: Generator<simd_quatf, Shrink.Tuple<(Float, (Float, Float, Float))>> {
@@ -118,4 +121,11 @@ extension Gen where Value == simd_quatd {
         return zip(angle, vector).map { t in simd.simd_quatd(angle: t.0, axis: t.1) }
     }
 }
+#else
+func length<F: FloatingPoint, T: SIMD<F>>(_ item: T) -> F {
+    (item * item).sum().squareRoot()
+}
+func normalize<F: FloatingPoint, T: SIMD<F>>(_ item: T) -> T {
+    return item / length(item)
+}
 #endif

Tests/PropertyBasedTests/GenTests+SIMD.swift

@@ -5,9 +5,10 @@
 //  Created by Lennard Sprong on 21/05/2025.
 //
 
-import PropertyBased
 import Testing
 
+@testable import PropertyBased
+
 #if canImport(simd)
 import simd
 #endif
@@ -46,38 +47,38 @@ import simd
         }
     }
 
-    #if canImport(simd)
     @Test func testUnitVectorFloat2() async {
-        await propertyCheck(input: Gen<simd_float2>.unitVector) { vec in
+        await propertyCheck(input: Gen<SIMD2<Float>>.unitVector) { vec in
             #expect(abs(length(vec) - 1) < 0.0001)
         }
     }
     @Test func testUnitVectorFloat3() async {
-        await propertyCheck(input: Gen<simd_float3>.unitVector) { vec in
+        await propertyCheck(input: Gen<SIMD3<Float>>.unitVector) { vec in
             #expect(abs(length(vec) - 1) < 0.0001)
         }
     }
     @Test func testUnitVectorFloat4() async {
-        await propertyCheck(input: Gen<simd_float4>.unitVector) { vec in
+        await propertyCheck(input: Gen<SIMD4<Float>>.unitVector) { vec in
             #expect(abs(length(vec) - 1) < 0.0001)
         }
     }
     @Test func testUnitVectorDouble2() async {
-        await propertyCheck(input: Gen<simd_double2>.unitVector) { vec in
+        await propertyCheck(input: Gen<SIMD2<Double>>.unitVector) { vec in
             #expect(abs(length(vec) - 1) < 0.0001)
         }
     }
     @Test func testUnitVectorDouble3() async {
-        await propertyCheck(input: Gen<simd_double3>.unitVector) { vec in
+        await propertyCheck(input: Gen<SIMD3<Double>>.unitVector) { vec in
             #expect(abs(length(vec) - 1) < 0.0001)
         }
     }
     @Test func testUnitVectorDouble4() async {
-        await propertyCheck(input: Gen<simd_double4>.unitVector) { vec in
+        await propertyCheck(input: Gen<SIMD4<Double>>.unitVector) { vec in
             #expect(abs(length(vec) - 1) < 0.0001)
         }
     }
 
+    #if canImport(simd)
     @Test func testQuatF() async {
         await propertyCheck(input: Gen.simd_quatf) { quat in
             #expect(abs(quat.length - 1) < 0.0001)