mpv/libvo/vo_dxr3.c

655 lines
17 KiB
C

#define PES_MAX_SIZE 2048
/*
* vo_dxr3.c - DXR3/H+ video out
*
* Copyright (C) 2001 David Holm <dholm@iname.com>
*
* libav - MPEG-PS multiplexer, part of ffmpeg
* Copyright Gerard Lantau (see http://ffmpeg.sf.net)
*
*/
#include "fastmemcpy.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <linux/em8300.h>
#include <sys/ioctl.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <fcntl.h>
#include <stdio.h>
#include <time.h>
#include "config.h"
#include "video_out.h"
#include "video_out_internal.h"
#include "../postproc/rgb2rgb.h"
#ifdef HAVE_MMX
#include "mmx.h"
#endif
LIBVO_EXTERN (dxr3)
#ifdef USE_LIBAVCODEC
#ifdef USE_LIBAVCODEC_SO
#include <libffmpeg/avcodec.h>
#else
#include "../libavcodec/avcodec.h"
#endif
static AVPicture picture;
static AVCodec *codec=NULL;
static AVCodecContext codec_context;
extern int avcodec_inited;
#endif
static unsigned char *picture_buf=NULL;
static unsigned char *outbuf=NULL;
static unsigned char *spubuf=NULL;
static int outbuf_size = 0;
static int v_width,v_height;
static int s_width,s_height;
static int s_pos_x,s_pos_y;
static int d_pos_x,d_pos_y;
static int osd_w,osd_h;
static int img_format = 0;
static int palette[] = { 0x000000, 0x494949, 0xb5b5b5, 0xffffff };
static int fd_control = -1;
static int fd_video = -1;
static int fd_spu = -1;
static int ioval = 0;
static vo_info_t vo_info =
{
"DXR3/H+ video out",
"dxr3",
"David Holm <dholm@iname.com>",
""
};
static uint32_t
init(uint32_t scr_width, uint32_t scr_height, uint32_t width, uint32_t height, uint32_t fullscreen, char *title, uint32_t format)
{
int i;
char tmp;
fd_control = open( "/dev/em8300", O_WRONLY );
if( fd_control < 1 )
{
printf( "VO: [dxr3] Error opening /dev/em8300 for writing!\n" );
return -1;
}
fd_video = open( "/dev/em8300_mv", O_WRONLY );
if( fd_video < 0 )
{
printf( "VO: [dxr3] Error opening /dev/em8300_mv for writing!\n" );
return -1;
}
else printf( "VO: [dxr3] Opened /dev/em8300_mv\n" );
fd_spu = open( "/dev/em8300_sp", O_WRONLY );
if( fd_spu < 0 )
{
printf( "VO: [dxr3] Error opening /dev/em8300_sp for writing!\n" );
return -1;
}
/* Subpic code isn't working yet, don't set to ON
unless you are really sure what you are doing */
ioval = EM8300_SPUMODE_OFF;
if( ioctl( fd_control, EM8300_IOCTL_SET_SPUMODE, &ioval ) < 0 )
{
printf( "VO: [dxr3] Unable to set subpicture mode!\n" );
return -1;
}
if( ioctl( fd_spu, EM8300_IOCTL_SPU_SETPALETTE, palette ) < 0 )
{
printf( "VO: [dxr3] Unable to set subpicture palette!\n" );
return -1;
}
ioval = EM8300_PLAYMODE_PLAY;
if( ioctl( fd_control, EM8300_IOCTL_SET_PLAYMODE, &ioval ) < 0 )
printf( "VO: [dxr3] Unable to set playmode!\n" );
close( fd_control );
img_format = format;
v_width = width;
v_height = height;
spubuf = malloc(53220); /* 53220 bytes is the standardized max size of a subpic */
picture_buf=NULL;
if( format == IMGFMT_YV12 )
{
#ifdef USE_LIBAVCODEC
int size;
printf("VO: [dxr3] Format: YV12\n");
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)
{
printf( "VO: [dxr3] mpeg1 codec not found! Read DOCS/DXR3!\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;
codec_context.flags=CODEC_FLAG_QSCALE;
codec_context.quality=1;
codec_context.pix_fmt = PIX_FMT_YUV420P;
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;
}
s_width = codec_context.width;
s_height = codec_context.height;;
osd_w=scr_width;
d_pos_x=(codec_context.width-(int)scr_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=scr_height;
d_pos_y=(codec_context.height-(int)scr_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: [dxr3] 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)
{
printf( "VO: [dxr3] Could not open codec\n");
return -1;
}
outbuf_size=10000+width*height;
outbuf = malloc(outbuf_size);
size = codec_context.width*codec_context.height;
picture_buf = malloc((size * 3)/2); /* size for YUV 420 */
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;
return 0;
#endif
return -1;
}
else if(format==IMGFMT_BGR24)
{
#ifdef USE_LIBAVCODEC
int size = 0;
printf("VO: [dxr3] Format: BGR24\n");
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)
{
printf( "VO: [dxr3] mpeg1 codec not found! Read DOCS/DXR3!\n");
return -1;
}
outbuf_size=10000+width*height;
outbuf = malloc(outbuf_size);
memset(&codec_context,0,sizeof(codec_context));
codec_context.bit_rate=100000;
codec_context.frame_rate=25*FRAME_RATE_BASE;
codec_context.gop_size=0;
codec_context.flags=CODEC_FLAG_QSCALE;
codec_context.quality=1;
codec_context.pix_fmt = PIX_FMT_YUV420P;
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;
}
/* s_width = codec_context.width = width;
s_height = codec_context.height = height;
*/
osd_w=scr_width;
d_pos_x=(codec_context.width-(int)scr_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=scr_height;
d_pos_y=(codec_context.height-(int)scr_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: [dxr3] 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)
{
printf( "VO: [dxr3] could not open codec\n");
return -1;
}
size = 10000+codec_context.width*codec_context.height;
picture_buf = malloc((size * 3)/2);
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;
return 0;
#endif
return -1;
}
else if(format==IMGFMT_YUY2)
{
#ifdef USE_LIBAVCODEC
int size = 0;
printf("VO: [dxr3] Format: YUY2\n");
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)
{
printf( "VO: [dxr3] mpeg1 codec not found! Read DOCS/DXR3!\n");
return -1;
}
outbuf_size=10000+width*height;
outbuf = malloc(outbuf_size);
memset(&codec_context,0,sizeof(codec_context));
codec_context.bit_rate=100000;
codec_context.frame_rate=25*FRAME_RATE_BASE;
codec_context.gop_size=0;
codec_context.flags=CODEC_FLAG_QSCALE;
codec_context.quality=1;
codec_context.pix_fmt = PIX_FMT_YUV420P;
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;
}
s_width = codec_context.width;
s_height = codec_context.height;;
/* FOR DEBUGGING ONLY!! */
codec_context.width = width;
codec_context.height = height;
osd_w=scr_width;
d_pos_x=(codec_context.width-(int)scr_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=scr_height;
d_pos_y=(codec_context.height-(int)scr_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: [dxr3] 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)
{
printf(stderr, "VO: [dxr3] Could not open codec\n");
return -1;
}
size = 10000+codec_context.width*codec_context.height;
picture_buf = malloc((size * 3)/2);
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;
return 0;
#endif
return -1;
}
else if(format==IMGFMT_MPEGPES)
{
printf( "VO: [dxr3] Format: MPEG-PES (no conversion needed)\n" );
return 0;
}
printf( "VO: [dxr3] Format: Unsupported\n" );
return -1;
}
static const vo_info_t* get_info(void)
{
return &vo_info;
}
static void draw_alpha(int x0, int y0, int w, int h, unsigned char* src, unsigned char *srca, int srcstride)
{
}
static void draw_osd(void)
{
// vo_draw_text(osd_w,osd_h,draw_alpha);
}
static uint32_t draw_frame(uint8_t * src[])
{
int pts = 0;
if( img_format == IMGFMT_MPEGPES )
{
int data_left;
vo_mpegpes_t *p=(vo_mpegpes_t *)src[0];
unsigned char *data = p->data;
data_left = p->size;
pts = p->timestamp;
if( ioctl( fd_video, EM8300_IOCTL_VIDEO_SETPTS, &pts ) < 0 )
printf( "VO: [dxr3] Unable to set PTS in draw_frame\n" );
while( data_left )
data_left -= write( fd_video, &((unsigned char*)p->data)[p->size-data_left], data_left );
return 0;
}
#ifdef USE_LIBAVCODEC
else if( img_format == IMGFMT_YV12 )
{
printf("VO: [dxr3] ERROR: Uninplemented\n");
}
else if( img_format == IMGFMT_BGR24 )
{
int tmp_size, out_size;
int wrap, wrap3, x, y;
int r, g, b, R, G, B, h = v_height, w = v_width;
unsigned char *s, *Y, *U, *V;
if(d_pos_x+w>picture.linesize[0]) w=picture.linesize[0]-d_pos_x;
if(d_pos_y+h>codec_context.height) h=codec_context.height-d_pos_y;
Y = picture.data[0]+d_pos_x+(d_pos_y*picture.linesize[0]);
U = picture.data[1]+(d_pos_x/2)+((d_pos_y/2)*picture.linesize[1]);
V = picture.data[2]+(d_pos_x/2)+((d_pos_y/2)*picture.linesize[2]);
//BGR24->YUV420P from ffmpeg, see ffmpeg.sourceforge.net for terms of license
#define SCALEBITS 8
#define ONE_HALF (1 << (SCALEBITS - 1))
#define FIX(x) ((int) ((x) * (1L<<SCALEBITS) + 0.5))
wrap = s_width;
wrap3 = w * 3;
s = src[0]+s_pos_x+(s_pos_y*wrap3);
for( y = 0; y < h; y+=2 )
{
for( x = 0; x < w; x+=2 )
{
b = s[0];
g = s[1];
r = s[2];
R = r;
G = g;
B = b;
Y[0] = (FIX(0.29900) * r + FIX(0.58700) * g + FIX(0.11400) * b + ONE_HALF) >> SCALEBITS;
b = s[3];
g = s[4];
r = s[5];
R += r;
G += g;
B += b;
Y[1] = (FIX(0.29900) * r + FIX(0.58700) * g + FIX(0.11400) * b + ONE_HALF) >> SCALEBITS;
s += wrap3;
Y += wrap;
b = s[0];
g = s[1];
r = s[2];
R += r;
G += g;
B += b;
Y[0] = (FIX(0.29900) * r + FIX(0.58700) * g + FIX(0.11400) * b + ONE_HALF) >> SCALEBITS;
b = s[3];
g = s[4];
r = s[5];
R += r;
G += g;
B += b;
Y[1] = (FIX(0.29900) * r + FIX(0.58700) * g + FIX(0.11400) * b + ONE_HALF) >> SCALEBITS;
U[0] = ((- FIX(0.16874) * R - FIX(0.33126) * G - FIX(0.50000) * B + 4 * ONE_HALF - 1) >> (SCALEBITS + 2)) + 128;
V[0] = ((FIX(0.50000) * R - FIX(0.41869) * G - FIX(0.08131) * B + 4 * ONE_HALF - 1) >> (SCALEBITS + 2)) + 128;
U++;
V++;
s -= (wrap3-6);
Y -= (wrap-(3/2));
}
s += wrap3;
Y += wrap;
}
#undef SCALEBITS
#undef ONE_HALF
#undef FIX(x)
//End of ffmpeg code, see ffmpeg.sourceforge.net for terms of license
pts = vo_pts;
if( ioctl( fd_video, EM8300_IOCTL_VIDEO_SETPTS, &pts ) < 0 )
printf( "VO: [dxr3] Unable to set PTS in draw_frame\n" );
tmp_size = out_size = avcodec_encode_video(&codec_context, outbuf, outbuf_size, &picture);
while( out_size )
out_size -= write( fd_video, &outbuf[tmp_size-out_size], out_size );
return 0;
}
else if( img_format == IMGFMT_YUY2 )
{
int tmp_size, out_size;
int x, y, w = v_width, h = v_height;
for( y = 0; y < h; y++ )
{
for( x = 0; x < w; x++ )
{
}
}
pts = vo_pts;
if( ioctl( fd_video, EM8300_IOCTL_VIDEO_SETPTS, &pts ) < 0 )
printf( "VO: [dxr3] Unable to set PTS in draw_frame\n" );
tmp_size = out_size = avcodec_encode_video(&codec_context, outbuf, outbuf_size, &picture);
while( out_size )
out_size -= write( fd_video, &outbuf[tmp_size-out_size], out_size );
return 0;
}
#endif
printf( "VO: [dxr3] Error in draw_frame(...)\n" );
return -1;
}
static void flip_page (void)
{
}
static uint32_t draw_slice( uint8_t *srcimg[], int stride[], int w, int h, int x0, int y0 )
{
int y, pts;
unsigned char* s;
unsigned char* d;
int data_left;
vo_mpegpes_t *p = (vo_mpegpes_t *)srcimg[0];
if( img_format == IMGFMT_YV12 )
{
#ifdef USE_LIBAVCODEC
int out_size, tmp_size;
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];
}
pts = vo_pts;
if( ioctl( fd_video, EM8300_IOCTL_VIDEO_SETPTS, &pts ) < 0 )
printf( "VO: [dxr3] Unable to set PTS in draw_slice\n" );
tmp_size = out_size = avcodec_encode_video(&codec_context, outbuf, outbuf_size, &picture);
while( out_size )
out_size -= write( fd_video, &outbuf[tmp_size-out_size], out_size );
return 0;
#endif
printf( "VO: [dxr3] You need to install ffmpeg.so or libavcodec, read DOCS/DXR3\n" );
return -1;
}
return -1;
}
static uint32_t
query_format(uint32_t format)
{
if(format==IMGFMT_MPEGPES) return 1;
#ifdef USE_LIBAVCODEC
if(format==IMGFMT_YV12) return 1;
// if(format==IMGFMT_YUY2) return 0x1|0x4;
if(format==IMGFMT_BGR24) return 1;
#else
if(format==IMGFMT_YV12) {printf("VO: [dxr3] You need to compile with libavcodec or ffmpeg.so to play this file!\n" ); return 0;}
if(format==IMGFMT_YUY2) {printf("VO: [dxr3] You need to compile with libavcodec or ffmpeg.so to play this file!\n" ); return 0;}
if(format==IMGFMT_BGR24) {printf("VO: [dxr3] You need to compile with libavcodec or ffmpeg.so to play this file!\n" ); return 0;}
#endif
return 0;
}
static void
uninit(void)
{
printf( "VO: [dxr3] Uninitializing\n" );
free(outbuf);
free(picture_buf);
free(spubuf);
close(fd_video);
close(fd_spu);
}
static void check_events(void)
{
}