mirror of https://github.com/mpv-player/mpv
425 lines
9.5 KiB
C
425 lines
9.5 KiB
C
#undef HAVE_DVB
|
|
#define PES_MAX_SIZE 2048
|
|
/*
|
|
* 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 <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.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_LIBAVCODEC
|
|
|
|
#include "../libavcodec/avcodec.h"
|
|
|
|
static unsigned char *picture_buf=NULL;
|
|
static unsigned char *outbuf=NULL;
|
|
static int outbuf_size = 100000;
|
|
|
|
static int s_pos_x,s_pos_y;
|
|
static int d_pos_x,d_pos_y;
|
|
|
|
static int osd_w,osd_h;
|
|
|
|
static AVPicture picture;
|
|
static AVCodec *codec=NULL;
|
|
static AVCodecContext codec_context;
|
|
extern int avcodec_inited;
|
|
|
|
#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
|
|
init(uint32_t s_width, uint32_t s_height, uint32_t width, uint32_t height, uint32_t fullscreen, char *title, uint32_t format)
|
|
{
|
|
#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_LIBAVCODEC
|
|
picture_buf=NULL;
|
|
if(format==IMGFMT_YV12){
|
|
int size;
|
|
|
|
if(!avcodec_inited){
|
|
avcodec_init();
|
|
avcodec_register_all();
|
|
avcodec_inited=1;
|
|
}
|
|
|
|
/* find the mpeg1 video encoder */
|
|
codec = avcodec_find_encoder(CODEC_ID_MPEG1VIDEO);
|
|
if (!codec) {
|
|
fprintf(stderr, "mpeg1 codec not found\n");
|
|
return -1;
|
|
}
|
|
memset(&codec_context,0,sizeof(codec_context));
|
|
codec_context.bit_rate=100000; // not used
|
|
codec_context.frame_rate=25*FRAME_RATE_BASE; // !!!!!
|
|
codec_context.gop_size=0; // I frames only
|
|
codec_context.flags=CODEC_FLAG_QSCALE;
|
|
codec_context.quality=1; // quality! 1..31 (1=best,slowest)
|
|
|
|
#if 0
|
|
codec_context.width=width;
|
|
codec_context.height=height;
|
|
#else
|
|
if(width<=352 && height<=288){
|
|
codec_context.width=352;
|
|
codec_context.height=288;
|
|
} else
|
|
if(width<=352 && height<=576){
|
|
codec_context.width=352;
|
|
codec_context.height=576;
|
|
} else
|
|
if(width<=480 && height<=576){
|
|
codec_context.width=480;
|
|
codec_context.height=576;
|
|
} else
|
|
if(width<=544 && height<=576){
|
|
codec_context.width=544;
|
|
codec_context.height=576;
|
|
} else {
|
|
codec_context.width=704;
|
|
codec_context.height=576;
|
|
}
|
|
#endif
|
|
|
|
osd_w=s_width;
|
|
d_pos_x=(codec_context.width-(int)s_width)/2;
|
|
if(d_pos_x<0){
|
|
s_pos_x=-d_pos_x;d_pos_x=0;
|
|
osd_w=codec_context.width;
|
|
} else s_pos_x=0;
|
|
|
|
osd_h=s_height;
|
|
d_pos_y=(codec_context.height-(int)s_height)/2;
|
|
if(d_pos_y<0){
|
|
s_pos_y=-d_pos_y;d_pos_y=0;
|
|
osd_h=codec_context.height;
|
|
} else s_pos_y=0;
|
|
|
|
printf("[vo] position mapping: %d;%d => %d;%d\n",s_pos_x,s_pos_y,d_pos_x,d_pos_y);
|
|
|
|
/* open it */
|
|
if (avcodec_open(&codec_context, codec) < 0) {
|
|
fprintf(stderr, "could not open codec\n");
|
|
return -1;
|
|
}
|
|
|
|
outbuf_size=10000+width*height; // must be enough!
|
|
outbuf = malloc(outbuf_size);
|
|
|
|
size = codec_context.width*codec_context.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
|
|
|
|
picture.data[0] = picture_buf;
|
|
picture.data[1] = picture.data[0] + size;
|
|
picture.data[2] = picture.data[1] + size / 4;
|
|
picture.linesize[0] = codec_context.width;
|
|
picture.linesize[1] = codec_context.width / 2;
|
|
picture.linesize[2] = codec_context.width / 2;
|
|
|
|
}
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
static const vo_info_t*
|
|
get_info(void)
|
|
{
|
|
return &vo_info;
|
|
}
|
|
|
|
#ifdef USE_LIBAVCODEC
|
|
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,
|
|
picture.data[0]+(x0+d_pos_x)+(y0+d_pos_y)*picture.linesize[0],picture.linesize[0]);
|
|
}
|
|
#endif
|
|
|
|
static void draw_osd(void)
|
|
{
|
|
#ifdef USE_LIBAVCODEC
|
|
if(picture_buf){ // YV12 only:
|
|
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];
|
|
|
|
static void send_pes_packet(unsigned char* data,int len,int id,int timestamp){
|
|
int x;
|
|
|
|
pes_header[0]=pes_header[1]=0;
|
|
pes_header[2]=id>>8; pes_header[3]=id&255;
|
|
|
|
while(1){
|
|
int payload_size=len+5; // data + PTS
|
|
if(6+payload_size>PES_MAX_SIZE) payload_size=PES_MAX_SIZE-6;
|
|
|
|
// construct PES header: (code from ffmpeg's libav)
|
|
// startcode:
|
|
// packetsize:
|
|
pes_header[4]=(payload_size)>>8;
|
|
pes_header[5]=(payload_size)&255;
|
|
// stuffing:
|
|
// 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;
|
|
|
|
payload_size-=5;
|
|
memcpy(&pes_header[6+5],data,payload_size);
|
|
my_write(pes_header,6+5+payload_size);
|
|
|
|
len-=payload_size; data+=payload_size;
|
|
if(len<=0) break;
|
|
}
|
|
|
|
// 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_LIBAVCODEC
|
|
if(picture_buf){ // YV12 only:
|
|
int out_size;
|
|
static int fno=0;
|
|
/* encode the image */
|
|
out_size = avcodec_encode_video(&codec_context, outbuf, outbuf_size, &picture);
|
|
send_pes_packet(outbuf,out_size,0x1E0,fno*(90000/25));++fno;
|
|
// 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_LIBAVCODEC
|
|
int y;
|
|
unsigned char* s;
|
|
unsigned char* d;
|
|
|
|
x0+=d_pos_x;
|
|
y0+=d_pos_y;
|
|
if(x0+w>picture.linesize[0]) w=picture.linesize[0]-x0; // !!
|
|
if(y0+h>codec_context.height) h=codec_context.height-y0;
|
|
|
|
// Y
|
|
s=srcimg[0]+s_pos_x+s_pos_y*stride[0];
|
|
d=picture.data[0]+x0+y0*picture.linesize[0];
|
|
for(y=0;y<h;y++){
|
|
memcpy(d,s,w);
|
|
s+=stride[0];
|
|
d+=picture.linesize[0];
|
|
}
|
|
|
|
w/=2;h/=2;x0/=2;y0/=2;
|
|
|
|
// U
|
|
s=srcimg[1]+(s_pos_x/2)+(s_pos_y/2)*stride[1];
|
|
d=picture.data[1]+x0+y0*picture.linesize[1];
|
|
for(y=0;y<h;y++){
|
|
memcpy(d,s,w);
|
|
s+=stride[1];
|
|
d+=picture.linesize[1];
|
|
}
|
|
|
|
// V
|
|
s=srcimg[2]+(s_pos_x/2)+(s_pos_y/2)*stride[2];
|
|
d=picture.data[2]+x0+y0*picture.linesize[2];
|
|
for(y=0;y<h;y++){
|
|
memcpy(d,s,w);
|
|
s+=stride[2];
|
|
d+=picture.linesize[2];
|
|
}
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
|
|
static uint32_t
|
|
query_format(uint32_t format)
|
|
{
|
|
if(format==IMGFMT_MPEGPES) return 1;
|
|
#ifdef USE_LIBAVCODEC
|
|
if(format==IMGFMT_YV12) return 1;
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
static void
|
|
uninit(void)
|
|
{
|
|
#ifdef USE_LIBAVCODEC
|
|
if(picture_buf){ // YV12 only:
|
|
free(outbuf);
|
|
free(picture_buf);
|
|
}
|
|
#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)
|
|
{
|
|
}
|
|
|