Mercurial > mplayer.hg
view libvo/vo_yuv4mpeg.c @ 4624:080882dddb2e
fixedsize video samples fixed (dvntsc-qt.mov)
author | arpi |
---|---|
date | Sun, 10 Feb 2002 00:45:56 +0000 |
parents | e406d918e361 |
children | f737f948fa0b |
line wrap: on
line source
/* * vo_yuv4mpeg.c, yuv4mpeg (mjpegtools) interface * * Thrown together by * Robert Kesterson <robertk@robertk.com> * Based on the pgm output plugin, the rgb2rgb postproc filter, divxdec, * and probably others. * * This is undoubtedly incomplete, inaccurate, or just plain wrong. :-) * */ #include <stdio.h> #include <stdlib.h> #include <string.h> #include "config.h" #include "video_out.h" #include "video_out_internal.h" #include <unistd.h> #include <stdlib.h> #include <stdio.h> #include <fcntl.h> #include <string.h> #include <sys/ioctl.h> #include <sys/mman.h> #include <signal.h> #include <sys/wait.h> #ifndef __FreeBSD__ #include <linux/videodev.h> #endif LIBVO_EXTERN (yuv4mpeg) static vo_info_t vo_info = { "yuv4mpeg output for mjpegtools (to \"stream.yuv\")", "yuv4mpeg", "Robert Kesterson <robertk@robertk.com>", "" }; static int image_width; static int image_height; static uint8_t *image = NULL; static uint8_t *image_y = NULL; static uint8_t *image_u = NULL; static uint8_t *image_v = NULL; static int using_format = 0; static FILE *yuv_out; int write_bytes; /** * * height should be a multiple of 2 and width should be a multiple of 2 * chrominance data is only taken from every secound line others are ignored */ #define RGB2YUV_SHIFT 8 #define BY ((int)( 0.098*(1<<RGB2YUV_SHIFT)+0.5)) #define BV ((int)(-0.071*(1<<RGB2YUV_SHIFT)+0.5)) #define BU ((int)( 0.439*(1<<RGB2YUV_SHIFT)+0.5)) #define GY ((int)( 0.504*(1<<RGB2YUV_SHIFT)+0.5)) #define GV ((int)(-0.368*(1<<RGB2YUV_SHIFT)+0.5)) #define GU ((int)(-0.291*(1<<RGB2YUV_SHIFT)+0.5)) #define RY ((int)( 0.257*(1<<RGB2YUV_SHIFT)+0.5)) #define RV ((int)( 0.439*(1<<RGB2YUV_SHIFT)+0.5)) #define RU ((int)(-0.148*(1<<RGB2YUV_SHIFT)+0.5)) static inline void rgb24toyv12(const uint8_t *src, uint8_t *ydst, uint8_t *udst, uint8_t *vdst, unsigned int width, unsigned int height, unsigned int lumStride, unsigned int chromStride, unsigned int srcStride) { int y; const int chromWidth= width>>1; for(y=0; y<height; y+=2) { int i; for(i=0; i<chromWidth; i++) { unsigned int b= src[6*i+0]; unsigned int g= src[6*i+1]; unsigned int r= src[6*i+2]; unsigned int Y = ((RY*r + GY*g + BY*b)>>RGB2YUV_SHIFT) + 16; unsigned int V = ((RV*r + GV*g + BV*b)>>RGB2YUV_SHIFT) + 128; unsigned int U = ((RU*r + GU*g + BU*b)>>RGB2YUV_SHIFT) + 128; udst[i] = U; vdst[i] = V; ydst[2*i] = Y; b= src[6*i+3]; g= src[6*i+4]; r= src[6*i+5]; Y = ((RY*r + GY*g + BY*b)>>RGB2YUV_SHIFT) + 16; ydst[2*i+1] = Y; } ydst += lumStride; src += srcStride; for(i=0; i<chromWidth; i++) { unsigned int b= src[6*i+0]; unsigned int g= src[6*i+1]; unsigned int r= src[6*i+2]; unsigned int Y = ((RY*r + GY*g + BY*b)>>RGB2YUV_SHIFT) + 16; ydst[2*i] = Y; b= src[6*i+3]; g= src[6*i+4]; r= src[6*i+5]; Y = ((RY*r + GY*g + BY*b)>>RGB2YUV_SHIFT) + 16; ydst[2*i+1] = Y; } udst += chromStride; vdst += chromStride; ydst += lumStride; src += srcStride; } } 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; using_format = format; image = malloc(write_bytes); yuv_out = fopen("stream.yuv", "wb"); if (!yuv_out || image == NULL) { perror("Can't get memory or file handle to stream.yuv"); return -1; } image_y = image; image_u = image_y + image_width * image_height; 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); fflush(yuv_out); return 0; } static const vo_info_t* get_info(void) { return &vo_info; } static void draw_osd(void) { } static void flip_page (void) { fprintf(yuv_out, "FRAME\n"); if(fwrite(image, 1, write_bytes, yuv_out) != write_bytes) perror("Error writing image to output!"); return; } 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; // copy Y: uint8_t *dst = image_y + image_width * y + x; uint8_t *src = srcimg[0]; for (i = 0; i < h; i++) { memcpy(dst, src, w); src += stride[0]; dst += image_width; } { // copy U + V: uint8_t *src1 = srcimg[1]; uint8_t *src2 = srcimg[2]; uint8_t *dstu = image_u + image_width * (y / 2) + (x / 2); uint8_t *dstv = image_v + image_width * (y / 2) + (x / 2); for (i = 0; i < h / 2; i++) { memcpy(dstu, src1 , w / 2); memcpy(dstv, src2, w / 2); src1 += stride[1]; src2 += stride[2]; dstu += image_width / 2; dstv += image_width / 2; } } } return 0; } static uint32_t draw_frame(uint8_t * src[]) { switch(using_format) { 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); // RGB2YUV(image_width, image_height, src[0], image_y, image_u, image_v, 1); } break; } return 0; } static uint32_t query_format(uint32_t format) { switch(format){ case IMGFMT_YV12: case IMGFMT_BGR|24: case IMGFMT_RGB|24: return 1; } return 0; } static void uninit(void) { if(image) free(image); image = NULL; if(yuv_out) fclose(yuv_out); yuv_out = NULL; } static void check_events(void) { } static uint32_t preinit(const char *arg) { return 0; } static uint32_t control(uint32_t request, void *data, ...) { switch (request) { case VOCTRL_QUERY_FORMAT: return query_format(*((uint32_t*)data)); } return VO_NOTIMPL; }