interlacing support - Klaus Stengel <ks1@inter-ject.de>

git-svn-id: svn://svn.mplayerhq.hu/mplayer/trunk@6660 b3059339-0415-0410-9bf9-f77b7e298cf2

1 files changed, 287 insertions, 50 deletions

diff --git a/libvo/vo_yuv4mpeg.c b/libvo/vo_yuv4mpeg.c
index 33ba91783b..995716330e 100644
--- a/libvo/vo_yuv4mpeg.c
+++ b/libvo/vo_yuv4mpeg.c
@@ -8,6 +8,12 @@
  *
  * This is undoubtedly incomplete, inaccurate, or just plain wrong. :-)
  *
+ * 2002/06/19 Klaus Stengel <Klaus.Stengel@asamnet.de>
+ *            - added support for interlaced output
+ *              Activate by using '-vo yuv4mpeg:interlaced'
+ *              or '-vo yuv4mpeg:interlaced_bf' if your source has
+ *              bottom fields first
+ *            - added some additional checks to catch problems
  *
  * 2002/04/17 Juergen Hammelmann <juergen.hammelmann@gmx.de>
  *            - added support for output of subtitles
@@ -49,35 +55,85 @@ static uint8_t *image_y = NULL;
 static uint8_t *image_u = NULL;
 static uint8_t *image_v = NULL;
 
+static uint8_t *rgb_buffer = NULL;
+static uint8_t *rgb_line_buffer = NULL;
+
 static int using_format = 0;
 static FILE *yuv_out;
-int write_bytes;
+static int write_bytes;
+
+#define Y4M_ILACE_NONE         'p'  /* non-interlaced, progressive frame */
+#define Y4M_ILACE_TOP_FIRST    't'  /* interlaced, top-field first       */
+#define Y4M_ILACE_BOTTOM_FIRST 'b'  /* interlaced, bottom-field first    */
+
+/* Set progressive mode as default */
+static int config_interlace = Y4M_ILACE_NONE;
+#define Y4M_IS_INTERLACED (config_interlace != Y4M_ILACE_NONE)
 
 static uint32_t config(uint32_t width, uint32_t height, uint32_t d_width, 
        uint32_t d_height, uint32_t fullscreen, char *title, 
        uint32_t format, const vo_tune_info_t *tuneinfo)
 {
-    image_height = height;
-    image_width = width;
-	write_bytes = image_width * image_height * 3 / 2;
+	image_height = height;
+	image_width = width;
 	using_format = format;
-    image = malloc(write_bytes);
+
+	if (Y4M_IS_INTERLACED)
+	{
+		if (height % 4)
+		{
+			perror("yuv4mpeg: Interlaced mode requires image height to be divisable by 4");
+			return -1;
+		}
+		
+		rgb_line_buffer = malloc(image_width * 3);
+		if (!rgb_line_buffer)
+		{
+			perror("yuv4mpeg: Unable to allocate line buffer for interlaced mode");
+			return -1;
+		}
+		
+		if (using_format == IMGFMT_YV12)
+			printf("yuv4mpeg: WARNING: Input not RGB; Can't seperate chrominance by fields!\n");
+	}
+				
+	if (width % 2)
+	{
+		perror("yuv4mpeg: Image width must be divisable by 2");
+		return -1;
+	}	
+	
+	if(using_format != IMGFMT_YV12)
+	{
+		rgb_buffer = malloc(image_width * image_height * 3);
+		if (!rgb_buffer)
+		{
+			perror("yuv4mpeg: Not enough memory to allocate RGB framebuffer");
+			return -1;
+		}
+	}
+
+	write_bytes = image_width * image_height * 3 / 2;
+	image = malloc(write_bytes);
 
 	yuv_out = fopen("stream.yuv", "wb");
-	if (!yuv_out || image == NULL) 
+	if (!yuv_out || image == 0) 
 	{
-		perror("Can't get memory or file handle to stream.yuv");
+		perror("yuv4mpeg: Can't get memory or file handle to write stream.yuv");
 		return -1;
 	}
 	image_y = image;
 	image_u = image_y + image_width * image_height;
-	image_v = image_u + (image_width * image_height) / 4;
+	image_v = image_u + image_width * image_height / 4;
 	
 	// This isn't right.  
 	// But it should work as long as the file isn't interlaced
 	// or otherwise unusual (the "Ip A0:0" part).
-	fprintf(yuv_out, "YUV4MPEG2 W%d H%d F%ld:%ld Ip A0:0\n", 
-					image_width, image_height, (long)(vo_fps * 1000000.0), 1000000);
+
+	/* At least the interlacing is ok now */
+	fprintf(yuv_out, "YUV4MPEG2 W%d H%d F%ld:%ld I%c A0:0\n", 
+			image_width, image_height, (long)(vo_fps * 1000000.0), 
+			(long)1000000, config_interlace);
 
 	fflush(yuv_out);
 	return 0;
@@ -88,12 +144,43 @@ static const vo_info_t* get_info(void)
     return &vo_info;
 }
 
+/* Only use when h divisable by 2! */
+static void swap_fields(uint8_t *ptr, const int h, const int stride)
+{
+	int i;
+	
+	for (i=0; i<h; i +=2)
+	{
+		memcpy(rgb_line_buffer     , ptr + stride *  i   , stride);
+		memcpy(ptr + stride *  i   , ptr + stride * (i+1), stride);
+		memcpy(ptr + stride * (i+1), rgb_line_buffer     , stride);
+	}
+}
+
 static void draw_alpha(int x0, int y0, int w, int h, unsigned char *src,
                        unsigned char *srca, int stride) {
-    if(using_format == IMGFMT_YV12)
+	switch (using_format)
 	{
-	    vo_draw_alpha_yv12(w, h, src, srca, stride, 
-			       image+(y0*image_width+x0), image_width);
+    	case IMGFMT_YV12:
+	    	vo_draw_alpha_yv12(w, h, src, srca, stride, 
+				       image + y0 * image_width + x0, image_width);
+			break;
+		
+		case IMGFMT_BGR|24:
+		case IMGFMT_RGB|24:
+			if (config_interlace != Y4M_ILACE_BOTTOM_FIRST)
+				vo_draw_alpha_rgb24(w, h, src, srca, stride,
+						rgb_buffer + (y0 * image_width + x0) * 3, image_width * 3);
+			else
+			{
+				swap_fields (rgb_buffer, image_height, image_width * 3);
+
+				vo_draw_alpha_rgb24(w, h, src, srca, stride,
+						rgb_buffer + (y0 * image_width  + x0) * 3, image_width * 3);
+				
+				swap_fields (rgb_buffer, image_height, image_width * 3);
+			}
+			break;
 	}
 }
 
@@ -102,22 +189,117 @@ static void draw_osd(void)
     vo_draw_text(image_width, image_height, draw_alpha);
 }
 
+static void deinterleave_fields(uint8_t *ptr, const int stride,
+							  const int img_height)
+{
+	unsigned int i, j, k_start = 1, modv = img_height - 1;
+	unsigned char *line_state = malloc(modv);
+
+	for (i=0; i<modv; i++)
+		line_state[i] = 0;
+
+	line_state[0] = 1;
+	
+	while(k_start < modv)
+	{
+		i = j = k_start;
+		memcpy(rgb_line_buffer, ptr + stride * i, stride);
+
+		while (!line_state[j])
+		{
+			line_state[j] = 1;
+			i = j;
+			j = j * 2 % modv;
+			memcpy(ptr + stride * i, ptr + stride * j, stride);
+		}
+		memcpy(ptr + stride * i, rgb_line_buffer, stride);
+		
+		while(k_start < modv && line_state[k_start])
+			k_start++;
+	}
+	free(line_state);
+}
+
+static void vo_y4m_write(const void *ptr, const size_t num_bytes)
+{
+	if (fwrite(ptr, 1, num_bytes, yuv_out) != num_bytes)
+		perror("yuv4mpeg: Error writing image to output!");
+}
+
 static void flip_page (void)
 {
+	uint8_t *upper_y, *upper_u, *upper_v, *rgb_buffer_lower;
+	int rgb_stride, uv_stride, field_height;
+	unsigned int i, low_ofs;
+	
 	fprintf(yuv_out, "FRAME\n");
-	if(fwrite(image, 1, write_bytes, yuv_out) != write_bytes)
-		perror("Error writing image to output!");
-    return;
+	
+	if (using_format != IMGFMT_YV12)
+	{
+		rgb_stride = image_width * 3;
+		uv_stride = image_width / 2;
+		
+		if (Y4M_IS_INTERLACED)
+		{
+			field_height = image_height / 2;		
+
+			upper_y = image;
+			upper_u = upper_y + image_width * field_height;
+			upper_v = upper_u + image_width * field_height / 4;
+			low_ofs = image_width * field_height * 3 / 2;
+			rgb_buffer_lower = rgb_buffer + rgb_stride * field_height;
+		
+			deinterleave_fields(rgb_buffer, rgb_stride, image_height);
+
+			rgb24toyv12(rgb_buffer, upper_y, upper_u, upper_v,
+						 image_width, field_height,
+						 image_width, uv_stride, rgb_stride);
+			rgb24toyv12(rgb_buffer_lower,  upper_y + low_ofs,
+						 upper_u + low_ofs, upper_v + low_ofs,
+						 image_width, field_height,
+						 image_width, uv_stride, rgb_stride);
+		
+			/* Write Y plane */
+			for(i = 0; i < field_height; i++)
+			{
+				vo_y4m_write(upper_y + image_width * i,           image_width);
+				vo_y4m_write(upper_y + image_width * i + low_ofs, image_width);
+			}
+
+			/* Write U and V plane */
+			for(i = 0; i < field_height / 2; i++)
+			{
+				vo_y4m_write(upper_u + uv_stride * i,           uv_stride);
+				vo_y4m_write(upper_u + uv_stride * i + low_ofs, uv_stride);
+			}
+			for(i = 0; i < field_height / 2; i++)
+			{
+				vo_y4m_write(upper_v + uv_stride * i,           uv_stride);
+				vo_y4m_write(upper_v + uv_stride * i + low_ofs, uv_stride);
+			}
+			return; /* Image written; We have to stop here */
+		}
+
+		rgb24toyv12(rgb_buffer, image_y, image_u, image_v,
+					image_width, image_height,
+					image_width, uv_stride, rgb_stride);
+	}
+
+	/* Write progressive frame */
+	vo_y4m_write(image, write_bytes);
 }
 
 static uint32_t draw_slice(uint8_t *srcimg[], int stride[], int w,int h,int x,int y)
 {
-	if(using_format == IMGFMT_YV12)
+	int i;
+	uint8_t *dst, *src = srcimg[0];
+	
+	switch (using_format)
 	{
-		int i;
+		case IMGFMT_YV12:
+		
 		// copy Y:
-		uint8_t *dst = image_y + image_width * y + x;
-		uint8_t *src = srcimg[0];
+		dst = image_y + image_width * y + x;
 		for (i = 0; i < h; i++)
 		{
 			memcpy(dst, src, w);
@@ -141,11 +323,22 @@ static uint32_t draw_slice(uint8_t *srcimg[], int stride[], int w,int h,int x,in
 				dstv += imgstride;
 			}
 		}
+		break;
+		
+		case IMGFMT_BGR24:
+		case IMGFMT_RGB24:
+			dst = rgb_buffer + (image_width * y + x) * 3;
+			for (i = 0; i < h; i++)
+			{
+				memcpy(dst, src, w * 3);
+				src += stride[0];
+				dst += image_width * 3;
+			}
+			break;
 	}
 	return 0;
 }
 
-
 static uint32_t draw_frame(uint8_t * src[])
 {
 	switch(using_format)
@@ -153,27 +346,10 @@ static uint32_t draw_frame(uint8_t * src[])
 		case IMGFMT_YV12:
 			// gets done in draw_slice
 			break;
+
 		case IMGFMT_BGR|24:
-			{
-#ifdef GUESS_THIS_ISNT_NEEDED
-				int c;
-				uint8_t temp;
-				//switch BGR to RGB
-				for(c = 0; c < image_width * image_height; c++)
-				{
-					temp = src[0][c * 3];
-					src[0][c * 3] = src[0][c * 3 + 2];
-					src[0][c * 3 + 2] = temp;
-				}
-#endif
-			}
-			// intentional fall-through
 		case IMGFMT_RGB|24:
-			{
-				rgb24toyv12(src[0], image_y, image_u, image_v, 
-							image_width, image_height, 
-							image_width, image_width / 2, image_width * 3);
-			}
+			memcpy(rgb_buffer, src[0], image_width * image_height * 3);
 			break;
 	}
     return 0;
@@ -181,14 +357,35 @@ static uint32_t draw_frame(uint8_t * src[])
 
 static uint32_t query_format(uint32_t format)
 {
-    switch(format){
-    case IMGFMT_YV12:
-	return VFCAP_CSP_SUPPORTED|VFCAP_CSP_SUPPORTED_BY_HW|VFCAP_OSD;
-    case IMGFMT_BGR|24:
-    case IMGFMT_RGB|24:
-        return VFCAP_CSP_SUPPORTED;
-    }
-    return 0;
+    
+	if (Y4M_IS_INTERLACED)
+    {
+		/* When processing interlaced material we want to get the raw RGB
+         * data and do the YV12 conversion ourselves to have the chrominance
+         * information sampled correct. */
+		
+		switch(format)
+		{
+			case IMGFMT_YV12:
+				return VFCAP_CSP_SUPPORTED|VFCAP_OSD;
+			case IMGFMT_BGR|24:
+			case IMGFMT_RGB|24:
+				return VFCAP_CSP_SUPPORTED|VFCAP_CSP_SUPPORTED_BY_HW|VFCAP_OSD;
+		}
+	}
+	else
+	{
+
+		switch(format)
+		{
+			case IMGFMT_YV12:
+				return VFCAP_CSP_SUPPORTED|VFCAP_CSP_SUPPORTED_BY_HW|VFCAP_OSD;
+    		case IMGFMT_BGR|24:
+    		case IMGFMT_RGB|24:
+        		return VFCAP_CSP_SUPPORTED|VFCAP_OSD;
+    	}	
+	}
+	return 0;
 }
 
 static void uninit(void)
@@ -196,9 +393,18 @@ static void uninit(void)
     if(image)
 		free(image);
 	image = NULL;
+
 	if(yuv_out)
 		fclose(yuv_out);
 	yuv_out = NULL;
+	
+	if(rgb_buffer)
+		free(rgb_buffer);
+	rgb_buffer = NULL;
+
+	if(rgb_line_buffer)
+		free(rgb_line_buffer);
+	rgb_line_buffer = NULL;
 }
 
 
@@ -209,10 +415,41 @@ static void check_events(void)
 
 static uint32_t preinit(const char *arg)
 {
+    int arg_unrecognized = 0;
+
     if(arg) 
     {
-	printf("vo_yuv4mpeg: Unknown subdevice: %s\n",arg);
-	return ENOSYS;
+        /* configure output mode */
+	if (strcmp(arg, "interlaced"))
+            arg_unrecognized++;
+	else
+	    config_interlace = Y4M_ILACE_TOP_FIRST;
+
+        if (strcmp(arg, "interlaced_bf"))
+            arg_unrecognized++;
+        else
+            config_interlace = Y4M_ILACE_BOTTOM_FIRST;
+
+        /* If both tests failed the argument is invalid */
+        if (arg_unrecognized == 2)
+        {
+	        printf("vo_yuv4mpeg: Unknown subdevice: %s\n", arg);
+			return ENOSYS;
+		}
+    }
+
+    /* Inform user which output mode is used */
+    switch (config_interlace)
+    {
+        case Y4M_ILACE_TOP_FIRST:
+            printf("vo_yuv4mpeg: Interlaced output mode, top-field first\n");
+            break;
+        case Y4M_ILACE_BOTTOM_FIRST:
+            printf("vo_yuv4mpeg: Interlaced output mode, bottom-field first\n");
+            break;
+        default:
+            printf("vo_yuv4mpeg: Using (default) progressive frame mode\n");
+            break;
     }
     return 0;
 }