mpv/libvo/vo_mpegpes.c

543 lines
12 KiB
C

// 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,&params,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;
}