From 3d9a7065280e64797c5a68c794a04d60fb3f767c Mon Sep 17 00:00:00 2001 From: dec05eba Date: Thu, 7 Mar 2024 22:53:04 +0100 Subject: Set frame size to video codec context size, nicer rgb to nv12 --- study/create_matrix.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) (limited to 'study') diff --git a/study/create_matrix.py b/study/create_matrix.py index 96bfcd4..1599a12 100755 --- a/study/create_matrix.py +++ b/study/create_matrix.py @@ -9,6 +9,12 @@ def usage(): print(" create_matrix.py 0.2126 0.7152 0.0722 limited") exit(1) +def a(v): + if v >= 0: + return " %f" % v + else: + return "%f" % v + def main(argv): if len(argv) != 5: usage() @@ -34,9 +40,9 @@ def main(argv): ] # Transform from row major to column major for glsl - print("%f, %f, %f, %f" % (matrix[0][0] * transform_range, matrix[1][0] * transform_range, matrix[2][0] * transform_range, 0.0)) - print("%f, %f, %f, %f" % (matrix[0][1] * transform_range, matrix[1][1] * transform_range, matrix[2][1] * transform_range, 0.0)) - print("%f, %f, %f, %f" % (matrix[0][2] * transform_range, matrix[1][2] * transform_range, matrix[2][2] * transform_range, 0.0)) - print("%f, %f, %f, %f" % (matrix[3][0] + luma_offset, matrix[3][1], matrix[3][2], 1.0)) + print("const mat4 RGBtoYUV = mat4(%f, %s, %s, %f," % (matrix[0][0] * transform_range, a(matrix[1][0] * transform_range), a(matrix[2][0] * transform_range), 0.0)) + print(" %f, %s, %s, %f," % (matrix[0][1] * transform_range, a(matrix[1][1] * transform_range), a(matrix[2][1] * transform_range), 0.0)) + print(" %f, %s, %s, %f," % (matrix[0][2] * transform_range, a(matrix[1][2] * transform_range), a(matrix[2][2] * transform_range), 0.0)) + print(" %f, %s, %s, %f);" % (matrix[3][0] + luma_offset, a(matrix[3][1]), a(matrix[3][2]), 1.0)) main(sys.argv) -- cgit v1.2.3