Add yuv420p10le pix_fmt

Author: tsnl

av/video/frame.pyx

@@ -21,7 +21,7 @@ supported_np_pix_fmts = {
     "gbrp14be", "gbrp14le", "gbrp16be", "gbrp16le", "gbrpf32be", "gbrpf32le", "gray",
     "gray16be", "gray16le", "gray8", "grayf32be", "grayf32le", "nv12", "pal8", "rgb24",
     "rgb48be", "rgb48le", "rgb8", "rgba", "rgba64be", "rgba64le", "yuv420p",
-    "yuv422p10le", "yuv444p", "yuv444p16be", "yuv444p16le", "yuva444p16be",
+    "yuv420p10le", "yuv422p10le", "yuv444p", "yuv444p16be", "yuv444p16le", "yuva444p16be",
     "yuva444p16le", "yuvj420p", "yuvj444p", "yuyv422",
 }
 
@@ -321,7 +321,7 @@ cdef class VideoFrame(Frame):
         import numpy as np
 
         # check size
-        if frame.format.name in {"yuv420p", "yuvj420p", "yuyv422", "yuv422p10le"}:
+        if frame.format.name in {"yuv420p", "yuvj420p", "yuyv422", "yuv420p10le", "yuv422p10le"}:
             assert frame.width % 2 == 0, "the width has to be even for this pixel format"
             assert frame.height % 2 == 0, "the height has to be even for this pixel format"
 
@@ -407,6 +407,16 @@ cdef class VideoFrame(Frame):
                 useful_array(frame.planes[1]),
                 useful_array(frame.planes[2]),
             ]).reshape(-1, frame.width)
+        if frame.format.name == "yuv420p10le":
+            # Read planes as uint16:
+            y = useful_array(frame.planes[0], 2, "uint16").reshape(frame.height, frame.width)
+            u = useful_array(frame.planes[1], 2, "uint16").reshape(frame.height // 2, frame.width // 2)
+            v = useful_array(frame.planes[2], 2, "uint16").reshape(frame.height // 2, frame.width // 2)
+            u_full = np.repeat(np.repeat(u, 2, axis=1), 2, axis=0)
+            v_full = np.repeat(np.repeat(u, 2, axis=1), 2, axis=0)
+            if channel_last:
+                return np.stack([y, u_full, v_full], axis=2)
+            return np.stack([y, u_full, v_full], axis=0)
         if frame.format.name == "yuv422p10le":
             # Read planes as uint16 at their original width
             y = useful_array(frame.planes[0], 2, "uint16").reshape(frame.height, frame.width)