Mercurial > mplayer.hg
view libvo/vo_mpegpes.c @ 5162:3cc3a36e3d0d
xp added
author | jaf |
---|---|
date | Sun, 17 Mar 2002 15:03:25 +0000 |
parents | 5823cde0dfdf |
children | b16820598c1b |
line wrap: on
line source
// Don't change for DVB card, it must be 2048 #define PES_MAX_SIZE 2048 // 100=best >=80 very good >=50 fast #define QUALITY 90 // #undef if you don't want mpeg1 transcoder (you'll be limited to mpeg1/2 playback only) #define USE_LIBFAME /* * Based on: * * test_av.c - Test program for new API * * Copyright (C) 2000 Ralph Metzler <ralph@convergence.de> * & Marcus Metzler <marcus@convergence.de> * for convergence integrated media GmbH * * libav - MPEG-PS multiplexer, part of ffmpeg * Copyright Gerard Lantau (see http://ffmpeg.sf.net) * */ #include <stdio.h> #include <stdlib.h> #include <string.h> #include <errno.h> #include <sys/types.h> #include <sys/stat.h> #include <fcntl.h> #include "config.h" #ifdef HAVE_DVB #include <sys/poll.h> #include <sys/ioctl.h> #include <stdio.h> #include <time.h> #include <unistd.h> #include <ost/dmx.h> #include <ost/frontend.h> #include <ost/sec.h> #include <ost/video.h> #include <ost/audio.h> #endif #include "config.h" #include "video_out.h" #include "video_out_internal.h" LIBVO_EXTERN (mpegpes) int vo_mpegpes_fd=-1; int vo_mpegpes_fd2=-1; #ifdef USE_LIBFAME #include "../libfame/fame.h" static unsigned char *picture_buf=NULL; static unsigned char *outbuf=NULL; static int outbuf_size = 1000000; static int s_pos_x,s_pos_y; static int d_pos_x,d_pos_y; static int osd_w,osd_h; static fame_parameters_t params; static fame_yuv_t yuv; static fame_context_t *ctx=NULL; #endif static vo_info_t vo_info = { #ifdef HAVE_DVB "Mpeg-PES to DVB card", #else "Mpeg-PES file", #endif "mpegpes", "A'rpi", "" }; static uint32_t config(uint32_t s_width, uint32_t s_height, uint32_t width, uint32_t height, uint32_t fullscreen, char *title, uint32_t format,const vo_tune_info_t *info) { #ifdef HAVE_DVB //|O_NONBLOCK if((vo_mpegpes_fd = open("/dev/ost/video",O_RDWR)) < 0){ perror("DVB VIDEO DEVICE: "); return -1; } if((vo_mpegpes_fd2 = open("/dev/ost/audio",O_RDWR|O_NONBLOCK)) < 0){ perror("DVB AUDIO DEVICE: "); return -1; } if ( (ioctl(vo_mpegpes_fd,VIDEO_SET_BLANK, false) < 0)){ perror("DVB VIDEO SET BLANK: "); return -1; } if ( (ioctl(vo_mpegpes_fd,VIDEO_SELECT_SOURCE, VIDEO_SOURCE_MEMORY) < 0)){ perror("DVB VIDEO SELECT SOURCE: "); return -1; } #if 1 if ( (ioctl(vo_mpegpes_fd2,AUDIO_SELECT_SOURCE, AUDIO_SOURCE_MEMORY) < 0)){ perror("DVB AUDIO SELECT SOURCE: "); return -1; } if ( (ioctl(vo_mpegpes_fd2,AUDIO_PLAY) < 0)){ perror("DVB AUDIO PLAY: "); return -1; } #else if ( (ioctl(vo_mpegpes_fd2,AUDIO_STOP,0) < 0)){ perror("DVB AUDIO STOP: "); return -1; } #endif if ( (ioctl(vo_mpegpes_fd,VIDEO_PLAY) < 0)){ perror("DVB VIDEO PLAY: "); return -1; } if ( (ioctl(vo_mpegpes_fd2,AUDIO_SET_AV_SYNC, false) < 0)){ perror("DVB AUDIO SET AV SYNC: "); return -1; } if ( (ioctl(vo_mpegpes_fd2,AUDIO_SET_MUTE, false) < 0)){ perror("DVB AUDIO SET MUTE: "); return -1; } #else vo_mpegpes_fd=open("grab.mpg",O_WRONLY|O_CREAT); if(vo_mpegpes_fd<0){ perror("vo_mpegpes"); return -1; } #endif #ifdef USE_LIBFAME picture_buf=NULL; if(format==IMGFMT_YV12){ int size; ctx=fame_open(); if(!ctx){ printf("FATAL: cannot open libFAME!\n"); return -1; } params.width=720; params.height=576; //params.coding="IPPPPP"; // seems to be buggy and eats more cpu params.coding="I"; params.quality=QUALITY; // 100=best >=80 very good >=50 fast params.bitrate=0; // 6*1000000 // bitrate (X bits/sec), 0=VBR params.slices_per_frame=1; params.frames_per_sequence=25; //0xffffffff; params.frame_rate_num=25; params.frame_rate_den=1; params.shape_quality=100; params.search_range=8; // for "IPPP" only params.verbose=0; params.profile=NULL; // TODO #if 0 params.width=width; params.height=height; #else if(width<=352 && height<=288){ params.width=352; params.height=288; } else if(width<=352 && height<=576){ params.width=352; params.height=576; } else if(width<=480 && height<=576){ params.width=480; params.height=576; } else if(width<=544 && height<=576){ params.width=544; params.height=576; } else { params.width=704; params.height=576; } #endif osd_w=s_width; d_pos_x=(params.width-(int)s_width)/2; if(d_pos_x<0){ s_pos_x=-d_pos_x;d_pos_x=0; osd_w=params.width; } else s_pos_x=0; d_pos_y=(params.height-(int)s_height)/2; if(d_pos_y<0){ s_pos_y=-d_pos_y;d_pos_y=0; osd_h=params.height; } else { s_pos_y=0; osd_h=s_height+d_pos_y; // if(d_pos_y) osd clear: s_height+d_pos_y .. params.height } printf("[vo] position mapping: %d;%d => %d;%d\n",s_pos_x,s_pos_y,d_pos_x,d_pos_y); /* open it */ outbuf_size=10000+width*height; // must be enough! outbuf = malloc(outbuf_size); fame_init(ctx,¶ms,outbuf,outbuf_size); size = params.width*params.height; picture_buf = malloc((size * 3) / 2); /* size for YUV 420 */ memset(picture_buf,0,size); // clear Y memset(picture_buf+size,128,size/2); // clear UV yuv.w=params.width; yuv.h=params.height; yuv.y=picture_buf; yuv.u=yuv.y+size; yuv.v=yuv.u+size/4; } #endif return 0; } static const vo_info_t* get_info(void) { return &vo_info; } #ifdef USE_LIBFAME static void draw_alpha(int x0,int y0, int w,int h, unsigned char* src, unsigned char *srca, int stride){ int x,y; vo_draw_alpha_yv12(w,h,src,srca,stride, yuv.y+(x0+d_pos_x)+(y0+d_pos_y)*yuv.w,yuv.w); } #endif static void draw_osd(void) { #ifdef USE_LIBFAME if(picture_buf){ // YV12 only: // if(d_pos_y) osd clear: s_height+d_pos_y .. params.height if(d_pos_y){ memset(yuv.y+osd_h*yuv.w,0,(params.height-osd_h)*yuv.w); } vo_draw_text(osd_w,osd_h,draw_alpha); } #endif } static void my_write(unsigned char* data,int len){ #ifdef HAVE_DVB #define NFD 2 struct pollfd pfd[NFD]; // printf("write %d bytes \n",len); pfd[0].fd = vo_mpegpes_fd; pfd[0].events = POLLOUT; pfd[1].fd = vo_mpegpes_fd2; pfd[1].events = POLLOUT; while(len>0){ if (poll(pfd,NFD,1)){ if (pfd[0].revents & POLLOUT){ int ret=write(vo_mpegpes_fd,data,len); // printf("ret=%d \n",ret); if(ret<=0){ perror("write"); usleep(0); } else { len-=ret; data+=ret; } } else usleep(1000); } } #else write(vo_mpegpes_fd,data,len); // write to file #endif } static unsigned char pes_header[PES_MAX_SIZE]; void send_pes_packet(unsigned char* data,int len,int id,int timestamp){ int ptslen=timestamp?5:1; // startcode: pes_header[0]=pes_header[1]=0; pes_header[2]=id>>8; pes_header[3]=id&255; while(len>0){ int payload_size=len; // data + PTS if(6+ptslen+payload_size>PES_MAX_SIZE) payload_size=PES_MAX_SIZE-(6+ptslen); // construct PES header: (code from ffmpeg's libav) // packetsize: pes_header[4]=(ptslen+payload_size)>>8; pes_header[5]=(ptslen+payload_size)&255; if(ptslen==5){ int x; // presentation time stamp: x=(0x02 << 4) | (((timestamp >> 30) & 0x07) << 1) | 1; pes_header[6]=x; x=((((timestamp >> 15) & 0x7fff) << 1) | 1); pes_header[7]=x>>8; pes_header[8]=x&255; x=((((timestamp) & 0x7fff) << 1) | 1); pes_header[9]=x>>8; pes_header[10]=x&255; } else { // stuffing and header bits: pes_header[6]=0x0f; } memcpy(&pes_header[6+ptslen],data,payload_size); my_write(pes_header,6+ptslen+payload_size); len-=payload_size; data+=payload_size; ptslen=1; // store PTS only once, at first packet! } // printf("PES: draw frame! pts=%d size=%d \n",timestamp,len); } void send_lpcm_packet(unsigned char* data,int len,int id,unsigned int timestamp,int freq_id){ int ptslen=timestamp?5:0; // startcode: pes_header[0]=pes_header[1]=0; pes_header[2]=1; pes_header[3]=0xBD; while(len>=4){ int payload_size; payload_size=PES_MAX_SIZE-6-20; // max possible data len if(payload_size>len) payload_size=len; payload_size&=(~3); // align! //if(6+payload_size>PES_MAX_SIZE) payload_size=PES_MAX_SIZE-6; // packetsize: pes_header[4]=(payload_size+3+ptslen+7)>>8; pes_header[5]=(payload_size+3+ptslen+7)&255; // stuffing: pes_header[6]=0x81; pes_header[7]=0x80; // hdrlen: pes_header[8]=ptslen; if(ptslen){ int x; // presentation time stamp: x=(0x02 << 4) | (((timestamp >> 30) & 0x07) << 1) | 1; pes_header[9]=x; x=((((timestamp >> 15) & 0x7fff) << 1) | 1); pes_header[10]=x>>8; pes_header[11]=x&255; x=((((timestamp) & 0x7fff) << 1) | 1); pes_header[12]=x>>8; pes_header[13]=x&255; } // ============ LPCM header: (7 bytes) ================= // Info by mocm@convergence.de // ID: pes_header[ptslen+9]=id; // number of frames: pes_header[ptslen+10]=0x07; // first acces unit pointer, i.e. start of audio frame: pes_header[ptslen+11]=0x00; pes_header[ptslen+12]=0x04; // audio emphasis on-off 1 bit // audio mute on-off 1 bit // reserved 1 bit // audio frame number 5 bit pes_header[ptslen+13]=0x0C; // quantization word length 2 bit // audio sampling frequency (48khz = 0, 96khz = 1) 2 bit // reserved 1 bit // number of audio channels - 1 (e.g. stereo = 1) 3 bit pes_header[ptslen+14]=1|(freq_id<<4); // dynamic range control (0x80 if off) pes_header[ptslen+15]=0x80; memcpy(&pes_header[6+3+ptslen+7],data,payload_size); my_write(pes_header,6+3+ptslen+7+payload_size); len-=payload_size; data+=payload_size; ptslen=0; // store PTS only once, at first packet! } // printf("PES: draw frame! pts=%d size=%d \n",timestamp,len); } static uint32_t draw_frame(uint8_t * src[]) { vo_mpegpes_t *p=(vo_mpegpes_t *)src[0]; unsigned char *data=p->data; // int tmp=-1; send_pes_packet(p->data,p->size,p->id,p->timestamp); // video data // send_pes_packet(&tmp,0,0x1C0,p->timestamp+30000); // fake audio data return 0; } static void flip_page (void) { #ifdef USE_LIBFAME if(picture_buf){ // YV12 only: int out_size; // static int fno=0; /* encode the image */ out_size = fame_encode_frame(ctx, &yuv, NULL); // send_pes_packet(outbuf,out_size,0x1E0,fno*(90000/25));++fno; send_pes_packet(outbuf,out_size,0x1E0,vo_pts); // printf("frame size: %d \n",out_size); } #endif } static uint32_t draw_slice(uint8_t *srcimg[], int stride[], int w,int h,int x0,int y0) { #ifdef USE_LIBFAME int y; unsigned char* s; unsigned char* d; x0+=d_pos_x; y0+=d_pos_y; if(x0+w>yuv.w) w=yuv.w-x0; // !! if(y0+h>params.height) h=params.height-y0; // Y s=srcimg[0]+s_pos_x+s_pos_y*stride[0]; d=yuv.y+x0+y0*yuv.w; for(y=0;y<h;y++){ memcpy(d,s,w); s+=stride[0]; d+=yuv.w; } w/=2;h/=2;x0/=2;y0/=2; // U s=srcimg[1]+(s_pos_x/2)+(s_pos_y/2)*stride[1]; d=yuv.u+x0+y0*(yuv.w>>1); for(y=0;y<h;y++){ memcpy(d,s,w); s+=stride[1]; d+=(yuv.w>>1); } // V s=srcimg[2]+(s_pos_x/2)+(s_pos_y/2)*stride[2]; d=yuv.v+x0+y0*(yuv.w>>1); for(y=0;y<h;y++){ memcpy(d,s,w); s+=stride[2]; d+=(yuv.w>>1); } #endif return 0; } static uint32_t query_format(uint32_t format) { if(format==IMGFMT_MPEGPES) return 1|256; #ifdef USE_LIBFAME if(format==IMGFMT_YV12) return 1|256; #endif return 0; } static void uninit(void) { #ifdef USE_LIBFAME if(picture_buf){ // YV12 only: free(outbuf); free(picture_buf); fame_close(ctx); ctx=NULL; } #endif if(vo_mpegpes_fd>=0){ close(vo_mpegpes_fd);vo_mpegpes_fd=-1;} if(vo_mpegpes_fd2>=0){ close(vo_mpegpes_fd2);vo_mpegpes_fd2=-1;} } static void check_events(void) { } static uint32_t preinit(const char *arg) { if(arg) { printf("vo_mpegpes: Unknown subdevice: %s\n",arg); return ENOSYS; } 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; }