From c45966f5970d20f507aad6c2bdd271a1076128a5 Mon Sep 17 00:00:00 2001
From: Doncho Gunchev <dgunchev@gmail.com>
Date: Sat, 15 Jun 2019 16:18:54 +0300
Subject: [PATCH] Add yuv420p_to_png.py and yuv422_to_png.py

From https://shrex999.wordpress.com/2013/07/31/yuv-to-rgb-python-imaging-library/
---
 video/yuv420p_to_png.py | 61 +++++++++++++++++++++++++++++++++++++++++
 video/yuv422_to_png.py  | 58 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 119 insertions(+)
 create mode 100755 video/yuv420p_to_png.py
 create mode 100755 video/yuv422_to_png.py

diff --git a/video/yuv420p_to_png.py b/video/yuv420p_to_png.py
new file mode 100755
index 0000000..43f1fc7
--- /dev/null
+++ b/video/yuv420p_to_png.py
@@ -0,0 +1,61 @@
+# Author : sherin.s@gmail.com (Sherin Sasidhan)
+# Date   : 05-Oct-2010
+ 
+import Image
+import sys
+from struct import *
+import array
+ 
+if len(sys.argv) != 4:
+        print "***** Usage syntax Error!!!! *****\n"
+        print "Usage:"
+        print "python <script> <.yuv file yuv420p> <width&gt> <height> "
+        sys.exit(1) # exit
+else:
+        pass
+ 
+image_name = sys.argv[1]
+width = int(sys.argv[2])
+height = int(sys.argv[3])
+ 
+y = array.array('B')
+u = array.array('B')
+v = array.array('B')
+ 
+f_y = open(image_name, "rb")
+f_uv = open(image_name, "rb")
+f_uv.seek(width*height, 1)
+ 
+image_out = Image.new("RGB", (width, height))
+pix = image_out.load()
+ 
+print "width=", width, "height=", height
+ 
+for i in range(0, height/2):
+    for j in range(0, width/2):
+        u.append(ord(f_uv.read(1)));
+ 
+for i in range(0, height/2):
+    for j in range(0, width/2):
+        v.append(ord(f_uv.read(1)));
+for i in range(0,height):
+    for j in range(0, width):
+        y.append(ord(f_y.read(1)));
+        #print "i=", i, "j=", j , (i*width), ((i*width) +j)
+        #pix[j, i] = y[(i*width) +j], y[(i*width) +j], y[(i*width) +j]
+        Y_val = y[(i*width)+j]
+        U_val = u[((i/2)*(width/2))+(j/2)]
+        V_val = v[((i/2)*(width/2))+(j/2)]
+        B = 1.164 * (Y_val-16) + 2.018 * (U_val - 128)
+        G = 1.164 * (Y_val-16) - 0.813 * (V_val - 128) - 0.391 * (U_val - 128)
+        R = 1.164 * (Y_val-16) + 1.596*(V_val - 128)
+        pix[j, i] = int(R), int(G), int(B)
+ 
+######################################################
+# B = 1.164(Y - 16)                   + 2.018(U - 128)
+# G = 1.164(Y - 16) - 0.813(V - 128) - 0.391(U - 128)
+# R = 1.164(Y - 16) + 1.596(V - 128)
+######################################################
+ 
+#image_out.save("out.bmp")
+image_out.show()
diff --git a/video/yuv422_to_png.py b/video/yuv422_to_png.py
new file mode 100755
index 0000000..abc5e0e
--- /dev/null
+++ b/video/yuv422_to_png.py
@@ -0,0 +1,58 @@
+# Author : sherin.s@gmail.com (Sherin Sasidhan)
+# Date   : 05-Oct-2010
+ 
+import Image
+import sys
+from struct import *
+import array
+ 
+if len(sys.argv) != 4:
+        print "***** Usage syntax Error!!!! *****\n"
+        print "Usage:"
+        print "python <script> <.yuv file uyvy422> <width&gt> <height> "
+        sys.exit(1) # exit
+else:
+        pass
+ 
+image_name = sys.argv[1]
+width = int(sys.argv[2])
+height = int(sys.argv[3])
+ 
+y = array.array('B')
+u = array.array('B')
+v = array.array('B')
+ 
+f_uyvy = open(image_name, "rb")
+f_uv = open(image_name, "rb")
+f_uv.seek(width*height, 1)
+ 
+image_out = Image.new("RGB", (width, height))
+pix = image_out.load()
+ 
+print "width=", width, "height=", height
+ 
+for i in range(0,height):
+    for j in range(0, width/2):
+        u  = ord(f_uyvy.read(1));
+        y1 = ord(f_uyvy.read(1));
+        v  = ord(f_uyvy.read(1));
+        y2 = ord(f_uyvy.read(1));
+ 
+        B = 1.164 * (y1-16) + 2.018 * (u - 128)
+        G = 1.164 * (y1-16) - 0.813 * (v - 128) - 0.391 * (u - 128)
+        R = 1.164 * (y1-16) + 1.596*(v - 128)
+        pix[j*2, i] = int(R), int(G), int(B)
+ 
+        B = 1.164 * (y2-16) + 2.018 * (u - 128)
+        G = 1.164 * (y2-16) - 0.813 * (v - 128) - 0.391 * (u - 128)
+        R = 1.164 * (y2-16) + 1.596*(v - 128)
+        pix[j*2+1, i] = int(R), int(G), int(B)
+ 
+######################################################
+# B = 1.164(Y - 16)                   + 2.018(U - 128)
+# G = 1.164(Y - 16) - 0.813(V - 128) - 0.391(U - 128)
+# R = 1.164(Y - 16) + 1.596(V - 128)
+######################################################
+ 
+#image_out.save("out.bmp")
+image_out.show()