X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=src%2Fmodules%2Fvid.stab%2Fcommon.c;fp=src%2Fmodules%2Fvid.stab%2Fcommon.c;h=13acc67f0ac06e79bfc0c7293ef21b44838cda78;hb=868437163988b81b721567ed571ff37d0c64d717;hp=0000000000000000000000000000000000000000;hpb=09923c8a08da705269f3b6c105fcbedd4d09ae67;p=mlt

diff --git a/src/modules/vid.stab/common.c b/src/modules/vid.stab/common.c
new file mode 100644
index 00000000..13acc67f
--- /dev/null
+++ b/src/modules/vid.stab/common.c
@@ -0,0 +1,156 @@
+/*
+ * common.c
+ * Copyright (C) 2014 Brian Matherly <pez4brian@yahoo.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ */
+#include <framework/mlt.h>
+#include <vid.stab/libvidstab.h>
+#include "common.h"
+
+mlt_image_format validate_format( mlt_image_format format )
+{
+	switch( format )
+	{
+	case mlt_image_rgb24a:
+	case mlt_image_rgb24:
+		return mlt_image_rgb24;
+	case mlt_image_yuv420p:
+		return mlt_image_yuv420p;
+	default:
+	case mlt_image_none:
+	case mlt_image_yuv422:
+	case mlt_image_opengl:
+	case mlt_image_glsl:
+	case mlt_image_glsl_texture:
+		return mlt_image_yuv422;
+	}
+}
+
+VSPixelFormat mltimage_to_vsimage( mlt_image_format mlt_format, int width, int height, uint8_t* mlt_img, uint8_t** vs_img )
+{
+	switch( mlt_format )
+	{
+	case mlt_image_rgb24:
+		// Convert RGB24 to YUV444 because it is the only planar
+		// format with comparable bit depth.
+		{
+			*vs_img = mlt_pool_alloc( width * height * 3 );
+			int y, u, v, r, g, b;
+			int total = width * height + 1;
+			uint8_t* yp = *vs_img;
+			uint8_t* up = yp + ( width * height );
+			uint8_t* vp = up + ( width * height );
+
+			while( --total )
+			{
+				r = *mlt_img++;
+				g = *mlt_img++;
+				b = *mlt_img++;
+				RGB2YUV_601_SCALED(r, g, b, y, u, v);
+				*yp++ = y;
+				*up++ = u;
+				*vp++ = v;
+			}
+
+			return PF_YUV444P;
+		}
+	case mlt_image_yuv420p:
+		// This format maps with no conversion
+		{
+			*vs_img = mlt_img;
+			return PF_YUV420P;
+		}
+	case mlt_image_yuv422:
+		// Convert packed to planar
+		{
+			*vs_img = mlt_pool_alloc( width * height * 2 );
+			uint8_t* yp = *vs_img;
+			uint8_t* up = yp + ( width * height );
+			uint8_t* vp = up + ( width * height / 2 );
+			int total = ( width * height / 2 ) + 1;
+
+			while( --total )
+			{
+				*yp++ = *mlt_img++;
+				*up++ = *mlt_img++;
+				*yp++ = *mlt_img++;
+				*vp++ = *mlt_img++;
+			}
+
+			return PF_YUV422P;
+		}
+	default:
+		return PF_NONE;
+	}
+}
+
+void vsimage_to_mltimage( uint8_t* vs_img, uint8_t* mlt_img, mlt_image_format mlt_format, int width, int height )
+{
+	switch( mlt_format )
+	{
+	case mlt_image_rgb24:
+		// Convert YUV444 to RGB24.
+		{
+			int y, u, v, r, g, b;
+			int total = width * height + 1;
+			uint8_t* yp = vs_img;
+			uint8_t* up = yp + ( width * height );
+			uint8_t* vp = up + ( width * height );
+
+			while( --total )
+			{
+				y = *yp++;
+				u = *up++;
+				v = *vp++;
+				YUV2RGB_601_SCALED( y, u, v, r, g, b );
+				*mlt_img++ = r;
+				*mlt_img++ = g;
+				*mlt_img++ = b;
+			}
+		}
+		break;
+	case mlt_image_yuv420p:
+		// This format was never converted
+		break;
+	case mlt_image_yuv422:
+		// Convert planar to packed
+		{
+			uint8_t* yp = vs_img;
+			uint8_t* up = yp + ( width * height );
+			uint8_t* vp = up + ( width * height / 2 );
+			int total = ( width * height / 2 ) + 1;
+
+			while( --total )
+			{
+				*mlt_img++ = *yp++;
+				*mlt_img++ = *up++;
+				*mlt_img++ = *yp++;
+				*mlt_img++ = *vp++;
+			}
+		}
+		break;
+	default:
+		return PF_NONE;
+	}
+}
+
+void free_vsimage( uint8_t* vs_img, VSPixelFormat format )
+{
+	if( format != PF_YUV420P )
+	{
+		mlt_pool_release( vs_img );
+	}
+}