changeset 1582:754781df8033 libavformat

RFC 1 stage: Improve maintainability of the cursoir painting code - Turned the mouse cursor painting code into a macro for more maintainability - Dropped mouse cursor painting in 8bit mode - Removed log when dropping frames - Free the shared memory segment on close (patch by Edouard Gomez)

--- a/x11grab.c	Wed Dec 13 08:46:31 2006 +0000
+++ b/x11grab.c	Wed Dec 13 08:47:14 2006 +0000
@@ -188,12 +188,12 @@
 #if 0
 		GetColorInfo (image, &c_info);
 		if ( c_info.alpha_mask == 0xFF000000 && image->green_mask == 0xFF00 ) {
-			// byte order is relevant here, not endianness
-			// endianness is handled by avcodec, but atm no such thing
-			// as having ABGR, instead of ARGB in a word. Since we
-			// need this for Solaris/SPARC, but need to do the conversion
-			// for every frame we do it outside of this loop, cf. below
-			// this matches both ARGB32 and ABGR32
+			/* byte order is relevant here, not endianness
+			 * endianness is handled by avcodec, but atm no such thing
+			 * as having ABGR, instead of ARGB in a word. Since we
+			 * need this for Solaris/SPARC, but need to do the conversion
+			 * for every frame we do it outside of this loop, cf. below
+			 * this matches both ARGB32 and ABGR32 */
 			input_pixfmt = PIX_FMT_ARGB32;
 		}  else {
 			av_log(s1, AV_LOG_ERROR,"image depth %i not supported ... aborting\n", image->bits_per_pixel);
@@ -236,13 +236,13 @@
 	return AVERROR_IO;  
 }
 
-static const uint16_t mousePointerBlack[] =
+static uint16_t mousePointerBlack[] =
 {
 	0, 49152, 40960, 36864, 34816, 33792, 33280, 33024, 32896, 32832,
 	33728, 37376, 43264, 51456, 1152, 1152, 576, 576, 448, 0
 };
 
-static const uint16_t mousePointerWhite[] =
+static uint16_t mousePointerWhite[] =
 {
 	0, 0, 16384, 24576, 28672, 30720, 31744, 32256, 32512, 32640, 31744,
 	27648, 17920, 1536, 768, 768, 384, 384, 0, 0
@@ -266,6 +266,36 @@
 	}
 }
 
+#define DRAW_CURSOR_TEMPLATE(type_t) \
+do { \
+type_t *cursor; \
+int width_cursor; \
+uint16_t bm_b, bm_w, mask; \
+\
+for (line = 0; line < min(20, (y_off + height) - *y); line++ ) { \
+	if (s->mouse_wanted == 1) { \
+		bm_b = mousePointerBlack[line]; \
+		bm_w = mousePointerWhite[line]; \
+	} else { \
+		bm_b = mousePointerWhite[line]; \
+		bm_w = mousePointerBlack[line]; \
+	} \
+	mask = ( 0x0001 << 15 ); \
+\
+	for (cursor = (type_t*) im_data, width_cursor = 0; \
+	     ((width_cursor + *x) < (width + x_off) && width_cursor < 16); \
+	     cursor++, width_cursor++) { \
+		if ( ( bm_b & mask ) > 0 ) { \
+			*cursor &= ((~ image->red_mask) & (~ image->green_mask) & (~image->blue_mask )); \
+		} else if ( ( bm_w & mask ) > 0 ) { \
+			*cursor |= (image->red_mask | image->green_mask | image->blue_mask ); \
+		} \
+		mask >>= 1; \
+	} \
+	im_data += image->bytes_per_line; \
+} \
+} while (0)
+
 static void
 paintMousePointer(AVFormatContext *s1, X11Grab *s, int *x, int *y, XImage *image)
 {
@@ -283,135 +313,15 @@
 		im_data += (image->bits_per_pixel / 8 * (*x - x_off)); // shift to right pixel
     
 		switch(image->bits_per_pixel) {
-		case 32: 
-		{
-			uint32_t *cursor;
-			int width_cursor;
-			uint16_t bm_b, bm_w, mask;
-      
-			for (line = 0; line < min(20, (y_off + height) - *y); line++ ) {
-				if (s->mouse_wanted == 1) {
-					bm_b = mousePointerBlack[line];
-					bm_w = mousePointerWhite[line];
-				} else {
-					bm_b = mousePointerWhite[line];
-					bm_w = mousePointerBlack[line];
-				}
-				mask = (0x0001 << 15);
-	  
-				for (cursor = (uint32_t*) im_data, width_cursor = 0; 
-				     ((width_cursor + *x) < (width + x_off) && width_cursor < 16);
-				     cursor++, width_cursor++) {
-					// Boolean pointer_b_bit, pointer_w_bit;	  
-					// pointer_b_bit = ( ( bm_b & mask ) > 0 );
-					// pointer_w_bit = ( ( bm_w & mask ) > 0 );
-					// printf("%i ", pointer_b_bit, pointer_w_bit );
-					if (( bm_b & mask) > 0 ) {
-						*cursor &= (  (~image->red_mask)
-							    & (~image->green_mask)
-							    & (~image->blue_mask));
-					} else if (( bm_w & mask) > 0 ) {
-						*cursor |= (  image->red_mask
-							    | image->green_mask
-							    | image->blue_mask );
-					}
-					mask >>= 1;
-				}
-				//						printf("\n");
-				im_data += image->bytes_per_line;
-			}
-		}
-		break;
-#if 0
-		case 24: // not sure this can occur at all ..........
-			av_log(s1, AV_LOG_ERROR, "input image bits_per_pixel %i not implemented with mouse pointer capture ... aborting!\n",
-				image->bits_per_pixel);
-			av_log(s1, AV_LOG_ERROR, "Please file a bug at http://www.sourceforge.net/projects/xvidcap/\n");
-			exit(1);
+		case 32:
+			DRAW_CURSOR_TEMPLATE(uint32_t);
 			break;
 		case 16:
-		{
-			uint16_t *cursor;
-			int width;
-			uint16_t bm_b, bm_w, mask;
-                
-			for (line = 0; line < 16; line++) {
-				if (mjob->mouseWanted == 1) {
-					bm_b = mousePointerBlack[line];
-					bm_w = mousePointerWhite[line];
-				} else {
-					bm_b = mousePointerWhite[line];
-					bm_w = mousePointerBlack[line];
-				}
-				mask = (0x0001 << 15);
-                    
-				for (cursor = (uint16_t*) im_data, width = 0;
-				     ((width + *x) < (mjob->area->width + mjob->area->x)&&width < 6);
-				     cursor++, width++) {
-					// Boolean pointer_b_bit, pointer_w_bit;
-					// pointer_b_bit = ( ( bm_b & mask ) > 0 );
-					// pointer_w_bit = ( ( bm_w & mask ) > 0 );
-					// printf("%i ", pointer_b_bit, pointer_w_bit );
-					if (( bm_b & mask ) > 0 ) {
-						*cursor &= (  (~image->red_mask)
-							    & (~image->green_mask)
-							    & (~image->blue_mask));
-					} else if (( bm_w & mask ) > 0 ) {
-						*cursor |= (  image->red_mask
-							    | image->green_mask
-							    | image->blue_mask );
-					}
-					mask >>= 1;
-				}
-				// printf("\n");
-                    
-				im_data += image->bytes_per_line;
-			}
-		}
-			break;
-		case 8:
-		{
-			uint8_t *cursor;
-			int width;
-			uint16_t bm_b, bm_w, mask;
-                
-			for (line = 0; line < 16; line++ ) {
-				if (mjob->mouseWanted == 1) {
-					bm_b = mousePointerBlack[line];
-					bm_w = mousePointerWhite[line];
-				} else {
-					bm_b = mousePointerWhite[line];
-					bm_w = mousePointerBlack[line];
-				}
-				mask = ( 0x0001 << 15 );
-                    
-				for (cursor = im_data, width = 0;
-				     ((width + *x) < (mjob->area->width + mjob->area->x)&&width < 6);
-				     cursor++, width++) {
-					// Boolean pointer_b_bit, pointer_w_bit;
-					// pointer_b_bit = ( ( bm_b & mask ) > 0 );
-					// pointer_w_bit = ( ( bm_w & mask ) > 0 );
-					// printf("%i ", pointer_b_bit, pointer_w_bit );
-					if ((bm_b & mask) > 0 ) {
-						*cursor = 0;
-					} else if (( bm_w & mask) > 0 ) {
-						*cursor = 1;
-					}
-					mask >>= 1;
-                        
-				}
-				// printf("\n");
-                    
-				im_data += image->bytes_per_line;
-			}
-		}
+			DRAW_CURSOR_TEMPLATE(uint16_t);
 			break;
 		default:
-			av_log(s1, AV_LOG_ERROR, "input image bits_per_pixel %i not supported with mouse pointer capture ... aborting!\n",
-			       image->bits_per_pixel);
-			exit(1);
-                
-#endif        
+			/* XXX: How do we deal with 24bit and 8bit modes ? */
+			break;
 		}
 	}
 }
@@ -478,7 +388,6 @@
 		delay = s->time_frame  * s->frame_rate_base / s->frame_rate - curtime;
 		if (delay <= 0) {
 			if (delay < int64_t_C(-1000000) * s->frame_rate_base / s->frame_rate) {
-				av_log(s1, AV_LOG_DEBUG, "grabbing is %d frames late (dropping)\n", (int) -(delay / 16666));
 				s->time_frame += int64_t_C(1000000);
 			}
 			break;
@@ -510,14 +419,23 @@
 		paintMousePointer(s1, s, &pointer_x, &pointer_y, image);
 	}
 
-#warning FIXME - avoid memcpy
-	memcpy(pkt->data, image->data, s->frame_size); 
+
+	/* XXX: avoid memcpy */
+	memcpy(pkt->data, image->data, s->frame_size);
 	return s->frame_size;
 }
 
 static int x11grab_read_close(AVFormatContext *s1)
 {
 	X11Grab *x11grab = s1->priv_data;
+
+	/* Free shared mem if necessary */
+	if (x11grab->use_shm) {
+		shmdt(x11grab->shminfo.shmaddr);
+		shmctl(x11grab->shminfo.shmid, IPC_RMID, NULL);
+	}
+
+	/* Free X11 display */
 	XCloseDisplay(x11grab->dpy);
 	return 0;
 }
@@ -525,7 +443,7 @@
 AVInputFormat x11_grab_device_demuxer =
 {
 	"x11grab",
-	"X11 frame graber",
+	"X11grab",
 	sizeof(X11Grab),
 	NULL,
 	x11grab_read_header,