学步园

一.LINUX DVR源码
#include <stdio.h><br /> #include <stdlib.h><br /> #include <stdint.h><br /> #include <errno.h><br /> #include <fcntl.h><br /> #include <signal.h><br /> #include <sys/ioctl.h><br /> #include <sys/mman.h><br /> #include <linux/videodev.h><br /> #include "xvid.h"<br /> #define FRAMERATE_INCR 1001<br /> #define SMALL_EPS (1e-10)<br /> #define VIDEO "/dev/video"<br /> #define INDEX 0<br /> #define IOCTL(fd, req, addr ) /<br /> ((-1==ioctl(fd,req,addr))?(perror(#req),close(fd),exit(EXIT_FAILURE)):0)</p> <p>static float ARG_FRAMERATE = 25.00f;<br /> static int ARG_QUALITY = 4;<br /> static int ARG_MAXKEYINTERVAL = 250;<br /> static char * ARG_OUTPUTFILE= "out.m4u";<br /> static FILE * fp = NULL;<br /> static int done = 0;<br /> static const int motion_presets[] = {<br /> /* quality 0 */<br /> 0,</p> <p> /* quality 1 */<br /> XVID_ME_ADVANCEDDIAMOND16,</p> <p> /* quality 2 */<br /> XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16,</p> <p> /* quality 3 */<br /> XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16 |<br /> XVID_ME_ADVANCEDDIAMOND8 | XVID_ME_HALFPELREFINE8,</p> <p> /* quality 4 */<br /> XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16 |<br /> XVID_ME_ADVANCEDDIAMOND8 | XVID_ME_HALFPELREFINE8 |<br /> XVID_ME_CHROMA_PVOP | XVID_ME_CHROMA_BVOP,</p> <p> /* quality 5 */<br /> XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16 |<br /> XVID_ME_ADVANCEDDIAMOND8 | XVID_ME_HALFPELREFINE8 |<br /> XVID_ME_CHROMA_PVOP | XVID_ME_CHROMA_BVOP,</p> <p> /* quality 6 */<br /> XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16 | XVID_ME_EXTSEARCH16 |<br /> XVID_ME_ADVANCEDDIAMOND8 | XVID_ME_HALFPELREFINE8 | XVID_ME_EXTSEARCH8 |<br /> XVID_ME_CHROMA_PVOP | XVID_ME_CHROMA_BVOP,</p> <p>};<br /> static const int vop_presets[] = {<br /> /* quality 0 */<br /> 0,</p> <p> /* quality 1 */<br /> 0,</p> <p> /* quality 2 */<br /> XVID_VOP_HALFPEL,</p> <p> /* quality 3 */<br /> XVID_VOP_HALFPEL | XVID_VOP_INTER4V,</p> <p> /* quality 4 */<br /> XVID_VOP_HALFPEL | XVID_VOP_INTER4V,</p> <p> /* quality 5 */<br /> XVID_VOP_HALFPEL | XVID_VOP_INTER4V |<br /> XVID_VOP_TRELLISQUANT,</p> <p> /* quality 6 */<br /> XVID_VOP_HALFPEL | XVID_VOP_INTER4V |<br /> XVID_VOP_TRELLISQUANT | XVID_VOP_HQACPRED,</p> <p>};</p> <p>static int XDIM=352,YDIM=288;<br /> static use_assembler = 0;<br /> static void *enc_handle = NULL;</p> <p>static int enc_init(int use_assembler);<br /> static int enc_main(unsigned char *image,<br /> unsigned char *bitstream,<br /> int *key,<br /> int *stats_type,<br /> int *stats_quant,<br /> int *stats_length,<br /> int stats[3]);<br /> static int enc_stop();<br /> //signal act method defination<br /> static void clean(int signum)<br /> {<br /> done=1;<br /> }<br /> int<br /> main(int argc,char *argv[])<br /> {<br /> //register SIGINT action method ,to save the m4u file<br /> signal(SIGINT,clean);<br /> int fd;<br /> struct v4l2_capability cap;<br /> struct v4l2_format format;<br /> struct v4l2_requestbuffers reqbuf;<br /> struct v4l2_buffer buffer;<br /> struct v4l2_input input;<br /> int index=0,i;<br /> for (i=1; i< argc; i++) {</p> <p> if (strcmp("-asm", argv[i]) == 0 ) {<br /> use_assembler = 1;<br /> }else if(strcmp("-f", argv[i]) == 0 && i < argc -1){<br /> i++;<br /> ARG_OUTPUTFILE = argv[i];<br /> }else if(strcmp("-i", argv[i]) == 0 && i < argc -1){<br /> i++;<br /> index = atoi(argv[i]);<br /> } else if(strcmp("-w", argv[i]) == 0 && i < argc -1){<br /> i++;<br /> XDIM = atoi(argv[i]);<br /> } else if(strcmp("-h", argv[i]) == 0 && i < argc -1){<br /> i++;<br /> YDIM = atoi(argv[i]);<br /> }<br /> }</p> <p> //init capture card</p> <p> fd = open(VIDEO,O_RDWR);<br /> if(fd<0)<br /> {<br /> perror("Can't open /dev/video device");<br /> exit(errno);<br /> }<br /> //if card can't support video capture and by means of steamio,exit </p> <p> IOCTL(fd, VIDIOC_QUERYCAP, &cap);<br /> if ( !(cap.capabilities & V4L2_CAP_VIDEO_CAPTURE)<br /> &&!(cap.capabilities & V4L2_CAP_STREAMING))<br /> {<br /> fprintf(stderr, "Couldn't use the card /n");<br /> exit(3);<br /> } </p> <p> //query input and select the desired input</p> <p> IOCTL(fd, VIDIOC_G_INPUT, &index);<br /> input.index = index;<br /> IOCTL(fd, VIDIOC_ENUMINPUT, &input);<br /> printf ("Current input: Index %i,Name %s/n", index,input.name);<br /> for(i=0;;i++)<br /> {<br /> if(i!=index)<br /> {<br /> input.index = i;<br /> if(-1==ioctl(fd, VIDIOC_ENUMINPUT, &input))<br /> break;<br /> else<br /> printf ("Other input: Index %i,Name %s/n",i,input.name);<br /> }<br /> }<br /> IOCTL(fd, VIDIOC_S_INPUT, &index);</p> <p> //get current video format,set the desired format</p> <p> format.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;<br /> IOCTL(fd,VIDIOC_G_FMT,&format);<br /> format.fmt.pix.width = XDIM;<br /> format.fmt.pix.height = YDIM;<br /> format.fmt.pix.pixelformat = V4L2_PIX_FMT_YVU420;<br /> format.fmt.pix.field = V4L2_FIELD_ANY;//V4L2_FIELD_INTERLACED</p> <p> IOCTL(fd,VIDIOC_S_FMT,&format);</p> <p> XDIM = format.fmt.pix.width;<br /> printf("Gotten width %d/n",XDIM);<br /> YDIM = format.fmt.pix.height;<br /> printf("Gotten height %d/n",YDIM);</p> <p> //let driver get the buffers</p> <p> reqbuf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;<br /> reqbuf.memory = V4L2_MEMORY_MMAP;<br /> reqbuf.count = 4;<br /> IOCTL(fd,VIDIOC_REQBUFS,&reqbuf );<br /> if (reqbuf.count < 4)<br /> {<br /> printf ("Not enough buffer memory for driver/n");<br /> exit (5);<br /> }</p> <p>//mmap the driver's buffer to application address</p> <p> //create relative buffers struct to be used by mmap</p> <p> struct<br /> {<br /> void * start;<br /> size_t length;<br /> } buffers[reqbuf.count];</p> <p> //mmap the driver's kernel buffer into application</p> <p> for (i=0;i<reqbuf.count;i++)<br /> {<br /> buffer.type = reqbuf.type;<br /> buffer.index =i ;</p> <p> //query the status of one buffer ,mmap the driver buffer</p> <p> IOCTL(fd,VIDIOC_QUERYBUF,&buffer);<br /> buffers[i].length = buffer.length;<br /> buffers[i].start = mmap (NULL,buffer.length,<br /> PROT_READ | PROT_WRITE,<br /> MAP_SHARED,<br /> fd,buffer.m.offset);<br /> if (buffers[i].start == MAP_FAILED)<br /> {<br /> close(fd);<br /> perror ("mmap");<br /> exit(errno);<br /> }<br /> }<br /> //begin video capture</p> <p> IOCTL(fd,VIDIOC_STREAMON,&reqbuf.type); </p> <p> //enqueue all driver buffers<br /> for (i=0;i<reqbuf.count;i++)<br /> {<br /> buffer.type = reqbuf.type;<br /> buffer.index = i;<br /> IOCTL(fd,VIDIOC_QUERYBUF, &buffer);<br /> IOCTL(fd,VIDIOC_QBUF,&buffer);<br /> }<br /> //init xvid</p> <p> int result;<br /> result = enc_init(1);<br /> if(result!=0)<br /> {<br /> fprintf(stderr, "Encore INIT problem, return value %d/n", result);<br /> goto clean_all;<br /> }</p> <p> //get mem of mpg4 frame</p> <p> uint8_t *mp4_buffer = NULL;<br /> mp4_buffer = (unsigned char *) malloc(XDIM*YDIM);<br /> if (mp4_buffer==NULL)<br /> {<br /> fprintf(stderr,"malloc error");<br /> goto clean_all;<br /> }<br /> int key;<br /> int stats_type;<br /> int stats_quant;<br /> int stats_length;<br /> int sse[3]; </p> <p> //create store mp4 file</p> <p> fp= fopen(ARG_OUTPUTFILE, "rb");<br /> if (fp== NULL) {<br /> perror("Error opening output file.");<br /> exit(-1);<br /> } </p> <p>//main loop ,frame capture,compressed </p> <p> i = 0;<br /> int outbytes,m4v_size;<br /> while(!done)<br /> {<br /> //dequeue one frame</p> <p> buffer.type = reqbuf.type;<br /> buffer.index = i;<br /> IOCTL(fd, VIDIOC_QUERYBUF, &buffer);<br /> IOCTL(fd, VIDIOC_DQBUF, &buffer);<br /> /*debug info<br /> printf("current frame's unix time seconds :%d/n",buffer.timestamp.tv_sec);<br /> printf("current frame's unix time mcicoseconds :%d/n",buffer.timestamp.tv_usec);<br /> */</p> <p> //compress a frame</p> <p> m4v_size = enc_main((uint8_t *)buffers[i].start,<br /> mp4_buffer+16,<br /> &key, &stats_type,&stats_quant, &stats_length, sse);</p> <p> //store into output file<br /> outbytes = fwrite(mp4_buffer, 1, m4v_size, fp);<br /> if(outbytes != m4v_size)<br /> {<br /> fprintf(stderr,"Error writing the m4u file/n");<br /> exit(7);<br /> }</p> <p> //enqueue one frame</p> <p> buffer.type = reqbuf.type;<br /> buffer.index = i;<br /> IOCTL(fd, VIDIOC_QUERYBUF, &buffer );<br /> IOCTL(fd, VIDIOC_QBUF, &buffer );<br /> i++;<br /> if( i >= reqbuf.count )<br /> i = 0;<br /> }<br /> clean_all:<br /> fclose(fp);<br /> enc_stop();<br /> //stop capture<br /> IOCTL(fd, VIDIOC_STREAMOFF, &reqbuf.type );<br /> //unmmap the apllication buffer<br /> for (i=0;i<reqbuf.count;i++)<br /> munmap (buffers[i].start,buffers[i].length);<br /> //release the dynamic mem<br /> close(fd);<br /> return 0;<br /> }</p> <p>int<br /> enc_init(int use_assembler)<br /> {<br /> int xerr;<br /> //xvid_plugin_cbr_t cbr;<br /> xvid_plugin_single_t single;<br /> xvid_plugin_2pass1_t rc2pass1;<br /> xvid_plugin_2pass2_t rc2pass2;<br /> //xvid_plugin_fixed_t rcfixed;<br /> xvid_enc_plugin_t plugins[7];<br /> xvid_gbl_init_t xvid_gbl_init;<br /> xvid_enc_create_t xvid_enc_create;</p> <p> /*------------------------------------------------------------------------<br /> * XviD core initialization<br /> *----------------------------------------------------------------------*/</p> <p> /* Set version -- version checking will done by xvidcore */<br /> memset(&xvid_gbl_init, 0, sizeof(xvid_gbl_init));<br /> xvid_gbl_init.version = XVID_VERSION;<br /> xvid_gbl_init.debug = 0;</p> <p> /* Do we have to enable ASM optimizations ? */<br /> if (use_assembler) {<br /> xvid_gbl_init.cpu_flags = 0;<br /> }</p> <p> /* Initialize XviD core -- Should be done once per __process__ */<br /> xvid_global(NULL, XVID_GBL_INIT, &xvid_gbl_init, NULL);</p> <p> /*------------------------------------------------------------------------<br /> * XviD encoder initialization<br /> *----------------------------------------------------------------------*/</p> <p> /* Version again */<br /> memset(&xvid_enc_create, 0, sizeof(xvid_enc_create));<br /> xvid_enc_create.version = XVID_VERSION;</p> <p> /* Width and Height of input frames */<br /> xvid_enc_create.width = XDIM;<br /> xvid_enc_create.height = YDIM;<br /> xvid_enc_create.profile = XVID_PROFILE_S_L3;</p> <p> /* init plugins */<br /> xvid_enc_create.zones = NULL;<br /> xvid_enc_create.num_zones = 0;</p> <p> xvid_enc_create.plugins = NULL;<br /> xvid_enc_create.num_plugins = 0;</p> <p> /* No fancy thread tests */<br /> xvid_enc_create.num_threads = 0;</p> <p> /* Frame rate - Do some quick float fps = fincr/fbase hack */<br /> if ((ARG_FRAMERATE - (int) ARG_FRAMERATE) < SMALL_EPS) {<br /> xvid_enc_create.fincr = 1;<br /> xvid_enc_create.fbase = (int) ARG_FRAMERATE;<br /> } else {<br /> xvid_enc_create.fincr = FRAMERATE_INCR;<br /> xvid_enc_create.fbase = (int) (FRAMERATE_INCR * ARG_FRAMERATE);<br /> }</p> <p> /* Maximum key frame interval */<br /> if (ARG_MAXKEYINTERVAL > 0) {<br /> xvid_enc_create.max_key_interval = ARG_MAXKEYINTERVAL;<br /> }else {<br /> xvid_enc_create.max_key_interval = (int) ARG_FRAMERATE *10;<br /> }</p> <p> /* Bframes settings */<br /> xvid_enc_create.max_bframes = 0;<br /> xvid_enc_create.bquant_ratio = 150;<br /> xvid_enc_create.bquant_offset = 100;</p> <p> /* Dropping ratio frame -- we don't need that */<br /> xvid_enc_create.frame_drop_ratio = 0;</p> <p> /* Global encoder options */<br /> xvid_enc_create.global = 0;</p> <p> /* I use a small value here, since will not encode whole movies, but short clips */<br /> xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);</p> <p> /* Retrieve the encoder instance from the structure */<br /> enc_handle = xvid_enc_create.handle;</p> <p> return (xerr);<br /> }</p> <p>int<br /> enc_main(unsigned char *image,<br /> unsigned char *bitstream,<br /> int *key,<br /> int *stats_type,<br /> int *stats_quant,<br /> int *stats_length,<br /> int sse[3])<br /> {<br /> int ret;</p> <p> xvid_enc_frame_t xvid_enc_frame;<br /> xvid_enc_stats_t xvid_enc_stats;</p> <p> /* Version for the frame and the stats */<br /> memset(&xvid_enc_frame, 0, sizeof(xvid_enc_frame));<br /> xvid_enc_frame.version = XVID_VERSION;</p> <p> memset(&xvid_enc_stats, 0, sizeof(xvid_enc_stats));<br /> xvid_enc_stats.version = XVID_VERSION;</p> <p> /* Bind output buffer */<br /> xvid_enc_frame.bitstream = bitstream;<br /> xvid_enc_frame.length = -1;</p> <p> /* Initialize input image fields */<br /> if (image) {<br /> xvid_enc_frame.input.plane[0] = image;<br /> xvid_enc_frame.input.csp = XVID_CSP_I420;<br /> xvid_enc_frame.input.stride[0] = XDIM;<br /> } else {<br /> xvid_enc_frame.input.csp = XVID_CSP_NULL;<br /> }</p> <p> /* Set up core's general features */<br /> xvid_enc_frame.vol_flags = 0;</p> <p> /* Set up core's general features */<br /> xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY];</p> <p> /* Frame type -- let core decide for us */<br /> xvid_enc_frame.type = XVID_TYPE_AUTO;</p> <p> /* Force the right quantizer -- It is internally managed by RC plugins */<br /> xvid_enc_frame.quant = 3;</p> <p> /* Set up motion estimation flags */<br /> xvid_enc_frame.motion = motion_presets[ARG_QUALITY];</p> <p> /* We don't use special matrices */<br /> xvid_enc_frame.quant_intra_matrix = NULL;<br /> xvid_enc_frame.quant_inter_matrix = NULL;</p> <p> /* Encode the frame */<br /> ret = xvid_encore(enc_handle, XVID_ENC_ENCODE, &xvid_enc_frame,<br /> &xvid_enc_stats);</p> <p> *key = (xvid_enc_frame.out_flags & XVID_KEYFRAME);<br /> *stats_type = xvid_enc_stats.type;<br /> *stats_quant = xvid_enc_stats.quant;<br /> *stats_length = xvid_enc_stats.length;<br /> sse[0] = xvid_enc_stats.sse_y;<br /> sse[1] = xvid_enc_stats.sse_u;<br /> sse[2] = xvid_enc_stats.sse_v;<br /> return (ret);<br /> }</p> <p>int<br /> enc_stop()<br /> {<br /> int xerr;</p> <p> /* Destroy the encoder instance */<br /> xerr = xvid_encore(enc_handle, XVID_ENC_DESTROY, NULL, NULL);</p> <p> return (xerr);<br /> }</p> <p>

下面是基于SDL的跨平台MPEG4播放器源代码,支持硬件YUV加速
#include <stdint.h><br /> #include <stdio.h><br /> #include <stdlib.h><br /> #include <string.h><br /> #include <time.h><br /> #include <SDL/SDL.h><br /> #include "xvid.h"</p> <p>#define USE_PNM 1<br /> #define USE_TGA 0</p> <p>static int SHXDIM = 0;//display size<br /> static int SHYDIM = 0;<br /> static int XDIM = 352;//real size<br /> static int YDIM = 288;<br /> static int ARG_SAVEDECOUTPUT = 0;<br /> static int ARG_SAVEMPEGSTREAM = 0;<br /> static char *ARG_INPUTFILE = NULL;<br /> static int CSP = XVID_CSP_I420;<br /> static int BPP = 1;<br /> static int FORMAT = USE_PNM;<br /> static char filepath[256] = "./";<br /> static void *dec_handle = NULL;<br /> #define BUFFER_SIZE (2*1024*1024)<br /> static const int display_buffer_bytes = 0;<br /> static SDL_Surface *screen;<br /> static SDL_Overlay *overlay;<br /> static SDL_Rect rect;<br /> static double msecond();<br /> static int dec_init(int use_assembler, int debug_level);<br /> static int dec_main(unsigned char *istream,<br /> unsigned char *ostream,<br /> int istream_size,<br /> xvid_dec_stats_t *xvid_dec_stats);<br /> static int dec_stop();<br /> static void usage();<br /> static int write_image(char *prefix, unsigned char *image);<br /> static int write_pnm(char *filename, unsigned char *image);<br /> static int write_tga(char *filename, unsigned char *image);<br /> const char * type2str(int type)<br /> {<br /> if (type==XVID_TYPE_IVOP)<br /> return "I";<br /> if (type==XVID_TYPE_PVOP)<br /> return "P";<br /> if (type==XVID_TYPE_BVOP)<br /> return "B";<br /> return "S";<br /> }</p> <p>static void init_SDL()<br /> {<br /> if (SDL_Init (SDL_INIT_VIDEO) < 0)<br /> {<br /> fprintf (stderr, "Couldn't initialize SDL: %s/n", SDL_GetError());<br /> exit (1);<br /> }<br /> atexit (SDL_Quit);<br /> screen = SDL_SetVideoMode (SHXDIM, SHYDIM, 0, SDL_HWSURFACE<br /> | SDL_DOUBLEBUF<br /> | SDL_ANYFORMAT<br /> | SDL_RESIZABLE);<br /> if (screen == NULL)<br /> {<br /> fprintf(stderr, "Couldn't set video mode: %s/n", SDL_GetError());<br /> exit(2);<br /> }<br /> if (0 == screen->flags & SDL_HWSURFACE)<br /> {<br /> fprintf(stderr,"Can't get hardware surface/n");<br /> exit(3);<br /> }<br /> SDL_WM_SetCaption ("SDL MultiMedia Application", NULL);<br /> overlay = SDL_CreateYUVOverlay(XDIM, YDIM, SDL_YV12_OVERLAY, screen);<br /> if (!overlay)<br /> {<br /> fprintf(stderr, "Couldn't create overlay: %s/n", SDL_GetError());<br /> exit(4);<br /> }<br /> //show the overlay status<br /> printf("Created %dx%dx%d %s %s overlay/n",overlay->w,overlay->h,overlay->planes,<br /> overlay->hw_overlay?"hardware":"software",<br /> overlay->format==SDL_YV12_OVERLAY?"YV12":<br /> overlay->format==SDL_IYUV_OVERLAY?"IYUV":<br /> overlay->format==SDL_YUY2_OVERLAY?"YUY2":<br /> overlay->format==SDL_UYVY_OVERLAY?"UYVY":<br /> overlay->format==SDL_YVYU_OVERLAY?"YVYU":<br /> "Unknown");<br /> rect.x=0;<br /> rect.y=0;<br /> rect.w=SHXDIM;<br /> rect.h=SHYDIM;<br /> }</p> <p>int main(int argc, char *argv[])<br /> {<br /> SDL_Event event;<br /> uint32_t lastftick;<br /> unsigned char *mp4_buffer = NULL;<br /> unsigned char *mp4_ptr = NULL;<br /> unsigned char *out_buffer = NULL;<br /> int useful_bytes;<br /> xvid_dec_stats_t xvid_dec_stats;</p> <p> double totaldectime;</p> <p> long totalsize;<br /> int status;<br /> int fps = 25;<br /> int fpsdelay;<br /> int paused=0;<br /> int resized=0;<br /> int use_assembler = 1;<br /> int debug_level = 0;</p> <p> char filename[256];</p> <p> FILE *in_file;<br /> int filenr;<br /> int i;</p> <p> for (i=1; i< argc; i++) {</p> <p> if (strcmp("-asm", argv[i]) == 0 ) {<br /> use_assembler = 1;<br /> } else if (strcmp("-debug", argv[i]) == 0 && i < argc - 1 ) {<br /> i++;<br /> if (sscanf(argv[i], "0x%x", &debug_level) != 1) {<br /> debug_level = atoi(argv[i]);<br /> }<br /> } else if (strcmp("-d", argv[i]) == 0) {<br /> ARG_SAVEDECOUTPUT = 1;<br /> } else if (strcmp("-i", argv[i]) == 0 && i < argc - 1 ) {<br /> i++;<br /> ARG_INPUTFILE = argv[i];<br /> } else if (strcmp("-c", argv[i]) == 0 && i < argc - 1 ) {<br /> i++;<br /> if (strcmp(argv[i], "rgb16") == 0) {<br /> CSP = XVID_CSP_RGB555;<br /> BPP = 2;<br /> } else if (strcmp(argv[i], "rgb24") == 0) {<br /> CSP = XVID_CSP_BGR;<br /> BPP = 3;<br /> } else if (strcmp(argv[i], "rgb32") == 0) {<br /> CSP = XVID_CSP_BGRA;<br /> BPP = 4;<br /> } else if (strcmp(argv[i], "yv12") == 0) {<br /> CSP = XVID_CSP_YV12;<br /> BPP = 1;<br /> } else {<br /> CSP = XVID_CSP_I420;<br /> BPP = 1;<br /> }<br /> } else if (strcmp("-f", argv[i]) == 0 && i < argc -1) {<br /> i++;<br /> if (strcmp(argv[i], "tga") == 0) {<br /> FORMAT = USE_TGA;<br /> } else {<br /> FORMAT = USE_PNM;<br /> }<br /> } else if (strcmp("-help", argv[i]) == 0) {<br /> usage();<br /> return(0);<br /> } else if(strcmp("-w", argv[i]) == 0 && i < argc -1){<br /> i++;<br /> SHXDIM = atoi(argv[i]);<br /> } else if(strcmp("-h", argv[i]) == 0 && i < argc -1){<br /> i++;<br /> SHYDIM = atoi(argv[i]);<br /> } else if(strcmp("-fps", argv[i]) == 0 && i < argc -1){<br /> i++;<br /> fps = atoi(argv[i]);<br /> }<br /> else {<br /> usage();<br /> exit(-1);<br /> }<br /> }</p> <p> if (ARG_INPUTFILE==NULL) {<br /> fprintf(stderr, "Warning: MSVC build does not read EOF correctly from stdin. Use the -i switch./n/n");<br /> }<br /> in_file = fopen(ARG_INPUTFILE, "rb");<br /> if (in_file == NULL) {<br /> fprintf(stderr, "Error opening input file %s/n", ARG_INPUTFILE);<br /> return(-1);<br /> }</p> <p> /* PNM/PGM format can't handle 16/32 bit data */<br /> if (BPP != 1 && BPP != 3 && FORMAT == USE_PNM) {<br /> FORMAT = USE_TGA;<br /> }</p> <p> /* Memory for encoded mp4 stream */<br /> mp4_buffer = (unsigned char *) malloc(BUFFER_SIZE);<br /> if (!mp4_buffer)<br /> goto free_all_memory;<br /> out_buffer = (unsigned char *)malloc(XDIM*YDIM*3/2);<br /> if (!out_buffer)<br /> goto free_all_memory;<br /> /******************************************************************************<br /> * INIT SDL<br /> *******************************************************************************/<br /> init_SDL(); </p> <p>/*****************************************************************************<br /> * XviD PART Start<br /> ****************************************************************************/</p> <p> status = dec_init(use_assembler, debug_level);<br /> if (status) {<br /> fprintf(stderr,<br /> "Decore INIT problem, return value %d/n", status);<br /> goto release_all;<br /> } </p> <p>/*****************************************************************************<br /> * Main loop<br /> ****************************************************************************/</p> <p> /* Fill the buffer ,create 2M buffer*/<br /> useful_bytes = fread(mp4_buffer, 1, BUFFER_SIZE, in_file);</p> <p> totaldectime = 0;<br /> totalsize = 0;<br /> filenr = 0;<br /> mp4_ptr = mp4_buffer;<br /> uint8_t *outy,*outu,*outv,*op[3];<br /> int y;<br /> /* set the start frame */<br /> i=0;<br /> fpsdelay=1000/fps;</p> <p> lastftick=SDL_GetTicks();<br /> do<br /> {<br /> int used_bytes = 0;<br /> double dectime;<br /> while (SDL_PollEvent(&event))<br /> {<br /> switch (event.type)<br /> {<br /> case SDL_VIDEORESIZE:<br /> screen=SDL_SetVideoMode(event.resize.w, event.resize.h, 0, SDL_RESIZABLE | SDL_SWSURFACE);<br /> rect.w=event.resize.w;<br /> rect.h=event.resize.h;<br /> if (paused)<br /> {<br /> resized=1;<br /> }<br /> break;<br /> case SDL_KEYDOWN:<br /> if (event.key.keysym.sym == SDLK_SPACE)<br /> {<br /> paused=!paused;<br /> break;<br /> }<br /> if (event.key.keysym.sym != SDLK_ESCAPE)<br /> {<br /> goto release_all;<br /> }<br /> case SDL_QUIT:<br /> goto release_all;<br /> }<br /> }<br /> /*<br /> * If the buffer is half empty or there are no more bytes in it<br /> * then fill it.<br /> */<br /> if (mp4_ptr > mp4_buffer + BUFFER_SIZE/2)<br /> {<br /> int already_in_buffer = (mp4_buffer + BUFFER_SIZE - mp4_ptr);<br /> /* Move data if needed */<br /> if (already_in_buffer > 0)<br /> memcpy(mp4_buffer, mp4_ptr, already_in_buffer);<br /> /* Update mp4_ptr */<br /> mp4_ptr = mp4_buffer;<br /> /* read new data */<br /> if(feof(in_file))<br /> break;<br /> useful_bytes += fread(mp4_buffer + already_in_buffer,<br /> 1, BUFFER_SIZE - already_in_buffer,<br /> in_file);</p> <p> }<br /> if ((!paused)||(resized))<br /> {<br /> if (((SDL_GetTicks()-lastftick)>fpsdelay)||(resized))<br /> {<br /> lastftick=SDL_GetTicks();<br /> /* This loop is needed to handle VOL/NVOP reading */<br /> do{<br /> /* Decode frame */<br /> dectime = msecond();<br /> used_bytes = dec_main(mp4_ptr, out_buffer, useful_bytes, &xvid_dec_stats);<br /> dectime = msecond() - dectime;<br /> if(xvid_dec_stats.type==XVID_TYPE_VOL<br /> && (xvid_dec_stats.data.vol.width != XDIM<br /> ||xvid_dec_stats.data.vol.height != YDIM))<br /> {<br /> //reallocate bigger out frame<br /> free(out_buffer);<br /> XDIM = xvid_dec_stats.data.vol.width;<br /> YDIM = xvid_dec_stats.data.vol.height;<br /> out_buffer = (unsigned char *) malloc(XDIM*YDIM*3/2);<br /> if (!out_buffer)<br /> goto free_all_memory;</p> <p> //reallocate bigger yuv overlay</p> <p> SDL_FreeYUVOverlay(overlay);<br /> overlay = SDL_CreateYUVOverlay(XDIM, YDIM, SDL_YV12_OVERLAY, screen);<br /> if (!overlay)<br /> {<br /> fprintf(stderr, "Couldn't create overlay: %s/n", SDL_GetError());<br /> exit(4);<br /> } </p> <p> }<br /> /* Update buffer pointers */<br /> if(used_bytes > 0) {<br /> mp4_ptr += used_bytes;<br /> useful_bytes -= used_bytes;<br /> /* Total size */<br /> totalsize += used_bytes;<br /> }<br /> }while (xvid_dec_stats.type <= 0 && useful_bytes > 0);<br /> /* Check if there is a negative number of useful bytes left in buffer<br /> * This means we went too far */<br /> if(useful_bytes < 0)<br /> break;<br /> /* Updated data - Count only usefull decode time */<br /> totaldectime += dectime;<br /> //display the decoded frame<br /> SDL_LockSurface(screen);<br /> SDL_LockYUVOverlay(overlay);<br /> outy = out_buffer;<br /> outu = out_buffer+XDIM*YDIM;<br /> outv = out_buffer+XDIM*YDIM*5/4;<br /> for(y=0;y<screen->h && y<overlay->h;y++)<br /> {<br /> op[0]=overlay->pixels[0]+overlay->pitches[0]*y;<br /> op[1]=overlay->pixels[1]+overlay->pitches[1]*(y/2);<br /> op[2]=overlay->pixels[2]+overlay->pitches[2]*(y/2);<br /> memcpy(op[0],outy+y*XDIM,XDIM);<br /> if(y%2 == 0)<br /> {<br /> memcpy(op[1],outu+XDIM/2*y/2,XDIM/2);<br /> memcpy(op[2],outv+XDIM/2*y/2,XDIM/2);<br /> }<br /> }<br /> SDL_UnlockYUVOverlay(overlay);<br /> SDL_UnlockSurface(screen);<br /> SDL_DisplayYUVOverlay(overlay, &rect);<br /> /* Save output frame if required */<br /> if (ARG_SAVEDECOUTPUT) {<br /> sprintf(filename, "%sdec%05d", filepath, filenr);<br /> if(write_image(filename, out_buffer)) {<br /> fprintf(stderr,<br /> "Error writing decoded frame %s/n",<br /> filename);<br /> }<br /> }<br /> filenr++;<br /> if (resized)<br /> resized = 0;<br /> }<br /> }<br /> SDL_Delay(10);<br /> } while (useful_bytes>0 || !feof(in_file));</p> <p> useful_bytes = 0; /* Empty buffer */</p> <p>/*****************************************************************************<br /> * Flush decoder buffers<br /> ****************************************************************************/</p> <p> do {</p> <p> /* Fake vars */<br /> int used_bytes;<br /> double dectime;</p> <p> do {<br /> dectime = msecond();<br /> used_bytes = dec_main(NULL, out_buffer, -1, &xvid_dec_stats);<br /> dectime = msecond() - dectime;<br /> } while(used_bytes>=0 && xvid_dec_stats.type <= 0);</p> <p> if (used_bytes < 0) { /* XVID_ERR_END */<br /> break;<br /> }</p> <p> /* Updated data - Count only usefull decode time */<br /> totaldectime += dectime;</p> <p> /* Prints some decoding stats */<br /> if (!display_buffer_bytes) {<br /> printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d/n",<br /> filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);<br /> }</p> <p> /* Save output frame if required */<br /> if (ARG_SAVEDECOUTPUT) {<br /> sprintf(filename, "%sdec%05d", filepath, filenr);<br /> if(write_image(filename, out_buffer)) {<br /> fprintf(stderr,<br /> "Error writing decoded frame %s/n",<br /> filename);<br /> }<br /> }</p> <p> filenr++;</p> <p> }while(1);</p> <p>/*****************************************************************************<br /> * Calculate totals and averages for output, print results<br /> ****************************************************************************/</p> <p> if (filenr>0) {<br /> totalsize /= filenr;<br /> totaldectime /= filenr;<br /> printf("Avg: dectime(ms) =%7.2f, fps =%7.2f, length(bytes) =%7d/n",<br /> totaldectime, 1000/totaldectime, (int)totalsize);<br /> }else{<br /> printf("Nothing was decoded!/n");<br /> }</p> <p>/*****************************************************************************<br /> * XviD PART Stop<br /> ****************************************************************************/</p> <p> release_all:<br /> SDL_FreeYUVOverlay(overlay);<br /> if (dec_handle) {<br /> status = dec_stop();<br /> if (status)<br /> fprintf(stderr, "decore RELEASE problem return value %d/n", status);<br /> }</p> <p> free_all_memory:<br /> free(out_buffer);<br /> free(mp4_buffer);<br /> return 0;<br /> }</p> <p>/*****************************************************************************<br /> * Usage function<br /> ****************************************************************************/</p> <p>static void usage()<br /> {</p> <p> fprintf(stderr, "Usage : xvid_decraw [OPTIONS]/n");<br /> fprintf(stderr, "Options :/n");<br /> fprintf(stderr, " -asm : use assembly optimizations (default=disabled)/n");<br /> fprintf(stderr, " -i string : input filename (default=stdin)/n");<br /> fprintf(stderr, " -d : save decoder output/n");<br /> fprintf(stderr, " -f format : choose output file format (tga, pnm, pgm)/n");<br /> fprintf(stderr, " -w width : init window width/n");<br /> fprintf(stderr, " -h height : init window height/n");<br /> fprintf(stderr, " -help : This help message/n");<br /> fprintf(stderr, " (* means default)/n");</p> <p>}</p> <p>/* return the current time in milli seconds */<br /> static double<br /> msecond()<br /> {<br /> clock_t clk;<br /> clk = clock();<br /> return(clk * 1000 / CLOCKS_PER_SEC);<br /> }</p> <p>static int write_image(char *prefix, unsigned char *image)<br /> {<br /> char filename[1024];<br /> char *ext;<br /> int ret;</p> <p> if (FORMAT == USE_PNM && BPP == 1) {<br /> ext = "pgm";<br /> } else if (FORMAT == USE_PNM && BPP == 3) {<br /> ext = "pnm";<br /> } else if (FORMAT == USE_TGA) {<br /> ext = "tga";<br /> } else {<br /> fprintf(stderr, "Bug: should not reach this path code -- please report to xvid-devel@xvid.org with command line options used");<br /> exit(-1);<br /> }</p> <p> sprintf(filename, "%s.%s", prefix, ext);</p> <p> if (FORMAT == USE_PNM) {<br /> ret = write_pnm(filename, image);<br /> } else {<br /> ret = write_tga(filename, image);<br /> }</p> <p> return(ret);<br /> }</p> <p>static int write_tga(char *filename, unsigned char *image)<br /> {<br /> FILE * f;<br /> char hdr[18];</p> <p> f = fopen(filename, "wb");<br /> if ( f == NULL) {<br /> return -1;<br /> }</p> <p> hdr[0] = 0; /* ID length */<br /> hdr[1] = 0; /* Color map type */<br /> hdr[2] = (BPP>1)?2:3; /* Uncompressed true color (2) or greymap (3) */<br /> hdr[3] = 0; /* Color map specification (not used) */<br /> hdr[4] = 0; /* Color map specification (not used) */<br /> hdr[5] = 0; /* Color map specification (not used) */<br /> hdr[6] = 0; /* Color map specification (not used) */<br /> hdr[7] = 0; /* Color map specification (not used) */<br /> hdr[8] = 0; /* LSB X origin */<br /> hdr[9] = 0; /* MSB X origin */<br /> hdr[10] = 0; /* LSB Y origin */<br /> hdr[11] = 0; /* MSB Y origin */<br /> hdr[12] = (XDIM>>0)&0xff; /* LSB Width */<br /> hdr[13] = (XDIM>>8)&0xff; /* MSB Width */<br /> if (BPP > 1) {<br /> hdr[14] = (YDIM>>0)&0xff; /* LSB Height */<br /> hdr[15] = (YDIM>>8)&0xff; /* MSB Height */<br /> } else {<br /> hdr[14] = ((YDIM*3)>>1)&0xff; /* LSB Height */<br /> hdr[15] = ((YDIM*3)>>9)&0xff; /* MSB Height */<br /> }<br /> hdr[16] = BPP*8;<br /> hdr[17] = 0x00 | (1<<5) /* Up to down */ | (0<<4); /* Image descriptor */</p> <p> /* Write header */<br /> fwrite(hdr, 1, sizeof(hdr), f);</p> <p> /* write first plane */<br /> fwrite(image, 1, XDIM*YDIM*BPP, f);</p> <p> /* Write Y and V planes for YUV formats */<br /> if (BPP == 1) {<br /> int i;</p> <p> /* Write the two chrominance planes */<br /> for (i=0; i<YDIM/2; i++) {<br /> fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);<br /> fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);<br /> }<br /> }</p> <p> /* Close the file */<br /> fclose(f);</p> <p> return(0);<br /> }</p> <p>static int write_pnm(char *filename, unsigned char *image)<br /> {<br /> FILE * f;</p> <p> f = fopen(filename, "wb");<br /> if ( f == NULL) {<br /> return -1;<br /> }</p> <p> if (BPP == 1) {<br /> int i;<br /> fprintf(f, "P5/n#xvid/n%i %i/n255/n", XDIM, YDIM*3/2);</p> <p> fwrite(image, 1, XDIM*YDIM, f);</p> <p> for (i=0; i<YDIM/2;i++) {<br /> fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);<br /> fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);<br /> }<br /> } else if (BPP == 3) {<br /> int i;<br /> fprintf(f, "P6/n#xvid/n%i %i/n255/n", XDIM, YDIM);<br /> for (i=0; i<XDIM*YDIM*3; i+=3) {<br /> fputc(image[i+2], f);<br /> fputc(image[i+1], f);<br /> fputc(image[i+0], f);<br /> }<br /> }</p> <p> fclose(f);</p> <p> return 0;<br /> }</p> <p>/*****************************************************************************<br /> * Routines for decoding: init decoder, use, and stop decoder<br /> ****************************************************************************/</p> <p>/* init decoder before first run */<br /> static int<br /> dec_init(int use_assembler, int debug_level)<br /> {<br /> int ret;</p> <p> xvid_gbl_init_t xvid_gbl_init;<br /> xvid_dec_create_t xvid_dec_create;</p> <p> /* Reset the structure with zeros */<br /> memset(&xvid_gbl_init, 0, sizeof(xvid_gbl_init_t));<br /> memset(&xvid_dec_create, 0, sizeof(xvid_dec_create_t));</p> <p> /*------------------------------------------------------------------------<br /> * XviD core initialization<br /> *----------------------------------------------------------------------*/</p> <p> /* Version */<br /> xvid_gbl_init.version = XVID_VERSION;</p> <p> /* Assembly setting */<br /> if(use_assembler)<br /> xvid_gbl_init.cpu_flags = 0;<br /> else<br /> xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;</p> <p> xvid_gbl_init.debug = debug_level;</p> <p> xvid_global(NULL, 0, &xvid_gbl_init, NULL);</p> <p> /*------------------------------------------------------------------------<br /> * XviD encoder initialization<br /> *----------------------------------------------------------------------*/</p> <p> /* Version */<br /> xvid_dec_create.version = XVID_VERSION;</p> <p> /*<br /> * Image dimensions -- set to 0, xvidcore will resize when ever it is<br /> * needed<br /> */<br /> xvid_dec_create.width = 0;<br /> xvid_dec_create.height = 0;</p> <p> ret = xvid_decore(NULL, XVID_DEC_CREATE, &xvid_dec_create, NULL);</p> <p> dec_handle = xvid_dec_create.handle;</p> <p> return(ret);<br /> }</p> <p>/* decode one frame */<br /> static int<br /> dec_main(unsigned char *istream,<br /> unsigned char *ostream,<br /> int istream_size,<br /> xvid_dec_stats_t *xvid_dec_stats)<br /> {</p> <p> int ret;</p> <p> xvid_dec_frame_t xvid_dec_frame;</p> <p> /* Reset all structures */<br /> memset(&xvid_dec_frame, 0, sizeof(xvid_dec_frame_t));<br /> memset(xvid_dec_stats, 0, sizeof(xvid_dec_stats_t));</p> <p> /* Set version */<br /> xvid_dec_frame.version = XVID_VERSION;<br /> xvid_dec_stats->version = XVID_VERSION;</p> <p> /* No general flags to set */<br /> xvid_dec_frame.general = 0;</p> <p> /* Input stream */<br /> xvid_dec_frame.bitstream = istream;<br /> xvid_dec_frame.length = istream_size;</p> <p> /* Output frame structure */<br /> xvid_dec_frame.output.plane[0] = ostream;<br /> xvid_dec_frame.output.stride[0] = XDIM*BPP;<br /> xvid_dec_frame.output.csp = CSP;</p> <p> ret = xvid_decore(dec_handle, XVID_DEC_DECODE, &xvid_dec_frame, xvid_dec_stats);</p> <p> return(ret);<br /> }</p> <p>/* close decoder to release resources */<br /> static int<br /> dec_stop()<br /> {<br /> int ret;</p> <p> ret = xvid_decore(dec_handle, XVID_DEC_DESTROY, NULL, NULL);</p> <p> return(ret);<br /> }