From bd843d3f9118bb0e07b71c63e864cd77e383d00c Mon Sep 17 00:00:00 2001
From: Dione Julek <dnjulek@hotmail.com>
Date: Mon, 19 Aug 2024 01:52:43 -0300
Subject: [PATCH] add `getStride2` and `getDimensions2`

---
 src/zigapi.zig | 28 ++++++++++++++++++----------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/src/zigapi.zig b/src/zigapi.zig
index 933ae2d..a963e17 100644
--- a/src/zigapi.zig
+++ b/src/zigapi.zig
@@ -99,14 +99,6 @@ pub const Frame = struct {
         return .{ self.getWidth(plane), self.getHeight(plane), self.getStride(plane) };
     }
 
-    pub fn getDimensions2(self: Self, plane: usize) struct { width: u32, height: u32, stride: u32 } {
-        return .{
-            .width = self.getWidth(plane),
-            .height = self.getHeight(plane),
-            .stride = self.getStride(plane),
-        };
-    }
-
     pub fn getReadSlice(self: Self, plane: usize) []const u8 {
         const ptr = self.vsapi.?.getReadPtr.?(self.frame, @intCast(plane));
         const len = self.getHeight(plane) * self.getStride(plane);
@@ -119,17 +111,33 @@ pub const Frame = struct {
         return ptr[0..len];
     }
 
+    pub fn getStride2(self: Self, comptime T: type, plane: usize) u32 {
+        return @intCast(self.vsapi.?.getStride.?(self.frame, @intCast(plane)) >> (@sizeOf(T) >> 1));
+    }
+
+    pub fn getDimensions2(self: Self, comptime T: type, plane: usize) struct { u32, u32, u32 } {
+        return .{ self.getWidth(plane), self.getHeight(plane), self.getStride2(T, plane) };
+    }
+
     pub fn getReadSlice2(self: Self, comptime T: type, plane: usize) []const T {
         const ptr = self.vsapi.?.getReadPtr.?(self.frame, @intCast(plane));
-        const len = self.getHeight(plane) * self.getStride(plane);
+        const len = self.getHeight(plane) * self.getStride2(T, plane);
         return @as([*]const T, @ptrCast(@alignCast(ptr)))[0..len];
     }
 
     pub fn getWriteSlice2(self: Self, comptime T: type, plane: usize) []T {
         const ptr = self.vsapi.?.getWritePtr.?(@constCast(self.frame), @intCast(plane));
-        const len = self.getHeight(plane) * self.getStride(plane);
+        const len = self.getHeight(plane) * self.getStride2(T, plane);
         return @as([*]T, @ptrCast(@alignCast(ptr)))[0..len];
     }
+
+    pub fn getDimensions3(self: Self, plane: usize) struct { width: u32, height: u32, stride: u32 } {
+        return .{
+            .width = self.getWidth(plane),
+            .height = self.getHeight(plane),
+            .stride = self.getStride(plane),
+        };
+    }
 };
 
 pub const Map = struct {