Merge pull request #42 from sergeypdev/main

Add mut variants for vector fields
kooparse · Feb 16, 2024 · 01785b0 · 01785b0
2 parents 27013ef + 232ff76
commit 01785b0
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/generic_vector.zig b/src/generic_vector.zig
@@ -63,6 +63,10 @@ pub fn GenericVector(comptime dimensions: comptime_int, comptime T: type) type {
                     return self.data[2];
                 }
 
+                pub fn zMut(self: *Self) *T {
+                    return &self.data[2];
+                }
+
                 /// Shorthand for (0, 0, 1).
                 pub fn forward() Self {
                     return new(0, 0, 1);
@@ -112,6 +116,14 @@ pub fn GenericVector(comptime dimensions: comptime_int, comptime T: type) type {
                 pub fn w(self: Self) T {
                     return self.data[3];
                 }
+
+                pub fn zMut(self: *Self) *T {
+                    return &self.data[2];
+                }
+
+                pub fn wMut(self: *Self) *T {
+                    return &self.data[3];
+                }
             },
             else => unreachable,
         };
@@ -124,6 +136,14 @@ pub fn GenericVector(comptime dimensions: comptime_int, comptime T: type) type {
             return self.data[1];
         }
 
+        pub fn xMut(self: *Self) *T {
+            return &self.data[0];
+        }
+
+        pub fn yMut(self: *Self) *T {
+            return &self.data[1];
+        }
+
         /// Set all components to the same given value.
         pub fn set(val: T) Self {
             const result: Data = @splat(val);