Add packed data API.

savonet · Oct 21, 2024 · d8012b7 · d8012b7
1 parent 4c015e8
commit d8012b7
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/image.mli b/src/image.mli
@@ -314,6 +314,7 @@ module YUV420 : sig
   val v : t -> Data.t
   val uv_stride : t -> int
   val data : t -> Data.t * Data.t * Data.t
+  val packed_data : t -> Data.t
   val alpha : t -> Data.t option
   val set_alpha : t -> Data.t option -> unit
   val dimensions : t -> int * int

diff --git a/src/imageYUV420.ml b/src/imageYUV420.ml
@@ -115,6 +115,20 @@ let create ?(blank = false) ?y_stride ?uv_stride width height =
   if blank then blank_all img;
   img
 
+let packed_data img =
+  let y, u, v = data img in
+  let y_dim = Bigarray.Array1.dim y in
+  let u_dim = Bigarray.Array1.dim u in
+  let v_dim = Bigarray.Array1.dim v in
+  let data =
+    Bigarray.Array1.create Bigarray.int8_unsigned Bigarray.c_layout
+      (y_dim + u_dim + v_dim)
+  in
+  Bigarray.Array1.blit y (Bigarray.Array1.sub data 0 y_dim);
+  Bigarray.Array1.blit u (Bigarray.Array1.sub data y_dim u_dim);
+  Bigarray.Array1.blit v (Bigarray.Array1.sub data (y_dim + u_dim) v_dim);
+  data
+
 let has_alpha img = img.alpha <> None
 let remove_alpha img = img.alpha <- None