mpv/stream/stream_dvdnav.c

666 lines
19 KiB
C

#include "config.h"
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include "mp_msg.h"
#include "osdep/timer.h"
#include "input/input.h"
#include "stream.h"
#include "libmpdemux/demuxer.h"
#include "stream_dvdnav.h"
#include "libvo/video_out.h"
#include "libavutil/common.h"
#include "spudec.h"
#include "m_option.h"
#include "m_struct.h"
#include "help_mp.h"
extern char *dvd_device;
extern int dvd_chapter;
extern int dvd_last_chapter;
extern int dvd_angle;
extern char *audio_lang, *dvdsub_lang;
static struct stream_priv_s {
int track;
char* device;
} stream_priv_dflts = {
0,
NULL
};
#define ST_OFF(f) M_ST_OFF(struct stream_priv_s,f)
/// URL definition
static m_option_t stream_opts_fields[] = {
{"filename", ST_OFF(device), CONF_TYPE_STRING, 0, 0, 0, NULL },
{"hostname", ST_OFF(track), CONF_TYPE_INT, 0, 0, 0, NULL},
{ NULL, NULL, 0, 0, 0, 0, NULL }
};
static struct m_struct_st stream_opts = {
"dvd",
sizeof(struct stream_priv_s),
&stream_priv_dflts,
stream_opts_fields
};
int dvd_nav_still=0; /* are we on a still picture? */
static int seek(stream_t *s, off_t newpos);
static dvdnav_priv_t * new_dvdnav_stream(char * filename) {
char * title_str;
dvdnav_priv_t *priv;
if (!filename)
return NULL;
if (!(priv=calloc(1,sizeof(dvdnav_priv_t))))
return NULL;
if (!(priv->filename=strdup(filename))) {
free(priv);
return NULL;
}
if(dvdnav_open(&(priv->dvdnav),priv->filename)!=DVDNAV_STATUS_OK)
{
free(priv->filename);
free(priv);
return NULL;
}
if (!priv->dvdnav) {
free(priv);
return NULL;
}
if(1) //from vlc: if not used dvdnav from cvs will fail
{
int len, event;
char buf[2048];
dvdnav_get_next_block(priv->dvdnav,buf,&event,&len);
dvdnav_sector_search(priv->dvdnav, 0, SEEK_SET);
}
/* turn off dvdnav caching */
dvdnav_set_readahead_flag(priv->dvdnav, 0);
if(dvdnav_set_PGC_positioning_flag(priv->dvdnav, 1) != DVDNAV_STATUS_OK)
mp_msg(MSGT_OPEN,MSGL_ERR,"stream_dvdnav, failed to set PGC positioning\n");
#if 1
/* report the title?! */
if (dvdnav_get_title_string(priv->dvdnav,&title_str)==DVDNAV_STATUS_OK) {
mp_msg(MSGT_IDENTIFY, MSGL_INFO,"Title: '%s'\n",title_str);
}
#endif
//dvdnav_event_clear(priv);
return priv;
}
static void dvdnav_get_highlight (dvdnav_priv_t *priv, int display_mode) {
pci_t *pnavpci = NULL;
dvdnav_highlight_event_t *hlev = &(priv->hlev);
int btnum;
if (!priv || !priv->dvdnav)
return;
pnavpci = dvdnav_get_current_nav_pci (priv->dvdnav);
if (!pnavpci)
return;
dvdnav_get_current_highlight (priv->dvdnav, &(hlev->buttonN));
hlev->display = display_mode; /* show */
if (hlev->buttonN > 0 && pnavpci->hli.hl_gi.btn_ns > 0 && hlev->display) {
for (btnum = 0; btnum < pnavpci->hli.hl_gi.btn_ns; btnum++) {
btni_t *btni = &(pnavpci->hli.btnit[btnum]);
if (hlev->buttonN == btnum + 1) {
hlev->sx = FFMIN (btni->x_start, btni->x_end);
hlev->ex = FFMAX (btni->x_start, btni->x_end);
hlev->sy = FFMIN (btni->y_start, btni->y_end);
hlev->ey = FFMAX (btni->y_start, btni->y_end);
hlev->palette = (btni->btn_coln == 0) ? 0 :
pnavpci->hli.btn_colit.btn_coli[btni->btn_coln - 1][0];
break;
}
}
} else { /* hide button or no button */
hlev->sx = hlev->ex = 0;
hlev->sy = hlev->ey = 0;
hlev->palette = hlev->buttonN = 0;
}
}
static int dvdnav_stream_read(dvdnav_priv_t * priv, unsigned char *buf, int *len) {
int event = DVDNAV_NOP;
if (!len) return -1;
*len=-1;
if (!priv) return -1;
if (!buf) return -1;
if (dvd_nav_still) {
mp_msg(MSGT_OPEN,MSGL_V, "%s: got a stream_read while I should be asleep!\n",__FUNCTION__);
*len=0;
return -1;
}
if (dvdnav_get_next_block(priv->dvdnav,buf,&event,len)!=DVDNAV_STATUS_OK) {
mp_msg(MSGT_OPEN,MSGL_V, "Error getting next block from DVD %d (%s)\n",event, dvdnav_err_to_string(priv->dvdnav) );
*len=-1;
}
else if (event!=DVDNAV_BLOCK_OK) {
// need to handle certain events internally (like skipping stills)
switch (event) {
case DVDNAV_NAV_PACKET:
return event;
case DVDNAV_STILL_FRAME: {
dvdnav_still_skip(priv->dvdnav); // don't let dvdnav stall on this image
break;
}
case DVDNAV_HIGHLIGHT: {
dvdnav_get_highlight (priv, 1);
break;
}
case DVDNAV_CELL_CHANGE: {
dvdnav_cell_change_event_t *ev = (dvdnav_cell_change_event_t*)buf;
if(ev->pgc_length)
priv->duration = ev->pgc_length/90;
break;
}
case DVDNAV_SPU_CLUT_CHANGE: {
memcpy(priv->spu_clut, buf, 16*sizeof(unsigned int));
priv->spu_set = 1;
break;
}
case DVDNAV_WAIT:
dvdnav_wait_skip(priv->dvdnav);
break;
}
*len=0;
}
return event;
}
static void update_title_len(stream_t *stream) {
dvdnav_priv_t *priv = stream->priv;
dvdnav_status_t status;
uint32_t pos = 0, len = 0;
status = dvdnav_get_position(priv->dvdnav, &pos, &len);
if(status == DVDNAV_STATUS_OK && len) {
stream->end_pos = (off_t) len * 2048;
stream->seek = seek;
} else {
stream->seek = NULL;
stream->end_pos = 0;
}
}
static int seek(stream_t *s, off_t newpos) {
uint32_t sector = 0;
dvdnav_priv_t *priv = s->priv;
if(s->end_pos && newpos > s->end_pos)
newpos = s->end_pos;
sector = newpos / 2048ULL;
if(dvdnav_sector_search(priv->dvdnav, (uint64_t) sector, SEEK_SET) != DVDNAV_STATUS_OK)
goto fail;
s->pos = newpos;
return 1;
fail:
mp_msg(MSGT_STREAM,MSGL_INFO,"dvdnav_stream, seeking to %"PRIu64" failed: %s\n", newpos, dvdnav_err_to_string(priv->dvdnav));
return 1;
}
static void stream_dvdnav_close(stream_t *s) {
dvdnav_priv_t *priv = s->priv;
dvdnav_close(priv->dvdnav);
priv->dvdnav = NULL;
free(priv);
}
static int fill_buffer(stream_t *s, char *but, int len)
{
int event;
dvdnav_priv_t* priv=s->priv;
len=0;
if(!s->end_pos)
update_title_len(s);
while(!len) /* grab all event until DVDNAV_BLOCK_OK (len=2048), DVDNAV_STOP or DVDNAV_STILL_FRAME */
{
event=dvdnav_stream_read(priv, s->buffer, &len);
if(event==-1 || len==-1)
{
mp_msg(MSGT_CPLAYER,MSGL_ERR, "DVDNAV stream read error!\n");
return 0;
}
switch (event) {
case DVDNAV_STOP:
case DVDNAV_BLOCK_OK:
case DVDNAV_NAV_PACKET:
return len;
case DVDNAV_VTS_CHANGE: {
int tit = 0, part = 0;
dvdnav_vts_change_event_t *vts_event = (dvdnav_vts_change_event_t *)s->buffer;
mp_msg(MSGT_CPLAYER,MSGL_INFO, "DVDNAV, switched to title: %d\r\n", vts_event->new_vtsN);
s->end_pos = 0;
update_title_len(s);
if(dvdnav_current_title_info(priv->dvdnav, &tit, &part) == DVDNAV_STATUS_OK) {
mp_msg(MSGT_CPLAYER,MSGL_V, "\r\nDVDNAV, NEW TITLE %d\r\n", tit);
dvdnav_get_highlight (priv, 0);
if(priv->title > 0 && tit != priv->title)
return 0;
}
break;
}
case DVDNAV_CELL_CHANGE: {
if(priv->title > 0 && dvd_last_chapter > 0) {
int tit=0, part=0;
if(dvdnav_current_title_info(priv->dvdnav, &tit, &part) == DVDNAV_STATUS_OK && part > dvd_last_chapter)
return 0;
}
}
break;
}
}
mp_msg(MSGT_STREAM,MSGL_DBG2,"DVDNAV fill_buffer len: %d\n",len);
return len;
}
static int control(stream_t *stream, int cmd, void* arg) {
dvdnav_priv_t* priv=stream->priv;
int tit, part;
switch(cmd)
{
case STREAM_CTRL_SEEK_TO_CHAPTER:
{
int chap = *((unsigned int *)arg)+1;
if(chap < 1 || dvdnav_current_title_info(priv->dvdnav, &tit, &part) != DVDNAV_STATUS_OK)
break;
if(dvdnav_part_play(priv->dvdnav, tit, chap) != DVDNAV_STATUS_OK)
break;
return 1;
}
case STREAM_CTRL_GET_NUM_CHAPTERS:
{
if(dvdnav_current_title_info(priv->dvdnav, &tit, &part) != DVDNAV_STATUS_OK)
break;
if(dvdnav_get_number_of_parts(priv->dvdnav, tit, &part) != DVDNAV_STATUS_OK)
break;
if(!part)
break;
*((unsigned int *)arg) = part;
return 1;
}
case STREAM_CTRL_GET_CURRENT_CHAPTER:
{
if(dvdnav_current_title_info(priv->dvdnav, &tit, &part) != DVDNAV_STATUS_OK)
break;
*((unsigned int *)arg) = part - 1;
return 1;
}
case STREAM_CTRL_GET_TIME_LENGTH:
{
if(priv->duration)
{
*((double *)arg) = (double)priv->duration / 1000.0;
return 1;
}
break;
}
case STREAM_CTRL_GET_ASPECT_RATIO:
{
uint8_t ar = dvdnav_get_video_aspect(priv->dvdnav);
*((double *)arg) = !ar ? 4.0/3.0 : 16.0/9.0;
return 1;
}
#ifdef MP_DVDNAV
case STREAM_CTRL_GET_CURRENT_TIME:
{
double tm;
tm = dvdnav_get_current_time(priv->dvdnav)/90000.0f;
if(tm != -1)
{
*((double *)arg) = tm;
return 1;
}
break;
}
case STREAM_CTRL_SEEK_TO_TIME:
{
uint64_t tm = (uint64_t) (*((double*)arg) * 90000);
if(dvdnav_time_search(priv->dvdnav, tm) == DVDNAV_STATUS_OK)
return 1;
break;
}
#endif
}
return STREAM_UNSUPPORTED;
}
static int open_s(stream_t *stream,int mode, void* opts, int* file_format) {
struct stream_priv_s* p = (struct stream_priv_s*)opts;
char *filename;
dvdnav_priv_t *priv;
if(p->device) filename = p->device;
else if(dvd_device) filename= dvd_device;
else filename = DEFAULT_DVD_DEVICE;
if(!(priv=new_dvdnav_stream(filename))) {
mp_msg(MSGT_OPEN,MSGL_ERR,MSGTR_CantOpenDVD,filename);
return STREAM_UNSUPPORTED;
}
if(p->track > 0) {
if(dvd_chapter > 0 && dvd_last_chapter > 0 && dvd_chapter > dvd_last_chapter) {
mp_msg(MSGT_OPEN,MSGL_FATAL,"dvdnav_stream, invalid chapter range: %d > %d\n", dvd_chapter, dvd_last_chapter);
return STREAM_UNSUPPORTED;
}
priv->title = p->track;
if(dvdnav_title_play(priv->dvdnav, p->track) != DVDNAV_STATUS_OK) {
mp_msg(MSGT_OPEN,MSGL_FATAL,"dvdnav_stream, couldn't select title %d, error '%s'\n", p->track, dvdnav_err_to_string(priv->dvdnav));
return STREAM_UNSUPPORTED;
}
if(dvd_chapter > 0)
dvdnav_part_play(priv->dvdnav, p->track, dvd_chapter);
} else if(p->track == -1)
dvdnav_menu_call(priv->dvdnav, DVD_MENU_Root);
else {
mp_msg(MSGT_OPEN,MSGL_INFO,"dvdnav_stream, you didn't specify a track number (as in dvdnav://1), playing whole disc\n");
dvdnav_menu_call(priv->dvdnav, DVD_MENU_Title);
}
if(dvd_angle > 1)
dvdnav_angle_change(priv->dvdnav, dvd_angle);
stream->sector_size = 2048;
stream->flags = STREAM_READ | STREAM_SEEK;
stream->fill_buffer = fill_buffer;
stream->seek = seek;
stream->control = control;
stream->close = stream_dvdnav_close;
stream->type = STREAMTYPE_DVDNAV;
stream->priv=(void*)priv;
*file_format = DEMUXER_TYPE_MPEG_PS;
update_title_len(stream);
if(!stream->pos)
mp_msg(MSGT_OPEN,MSGL_ERR, "INIT ERROR: couldn't get init pos %s\r\n", dvdnav_err_to_string(priv->dvdnav));
mp_msg(MSGT_OPEN,MSGL_INFO, "Remember to disable MPlayer's cache when playing dvdnav:// streams (adding -nocache to your command line)\r\n");
return STREAM_OK;
}
int mp_dvdnav_handle_input(stream_t *stream, int cmd, int *button) {
dvdnav_priv_t * priv=(dvdnav_priv_t*)stream->priv;
dvdnav_t *nav = priv->dvdnav;
dvdnav_status_t status=DVDNAV_STATUS_ERR;
pci_t *pci = dvdnav_get_current_nav_pci(nav);
int reset = 0;
if(cmd != MP_CMD_DVDNAV_SELECT && !pci)
return 0;
switch(cmd) {
case MP_CMD_DVDNAV_UP:
status = dvdnav_upper_button_select(nav, pci);
break;
case MP_CMD_DVDNAV_DOWN:
status = dvdnav_lower_button_select(nav, pci);
break;
case MP_CMD_DVDNAV_LEFT:
status = dvdnav_left_button_select(nav, pci);
break;
case MP_CMD_DVDNAV_RIGHT:
status = dvdnav_right_button_select(nav, pci);
break;
case MP_CMD_DVDNAV_MENU:
status = dvdnav_menu_call(nav,DVD_MENU_Root);
reset = 1;
break;
case MP_CMD_DVDNAV_PREVMENU: {
int title=0, part=0;
dvdnav_current_title_info(nav, &title, &part);
if(title) {
if(dvdnav_menu_call(nav, DVD_MENU_Part) == DVDNAV_STATUS_OK
|| dvdnav_menu_call(nav, DVD_MENU_Title) == DVDNAV_STATUS_OK) {
reset = 1;
break;
}
}
if(dvdnav_menu_call(nav, DVD_MENU_Root) == DVDNAV_STATUS_OK)
reset = 1;
}
break;
case MP_CMD_DVDNAV_SELECT:
status = dvdnav_button_activate(nav, pci);
if(status == DVDNAV_STATUS_OK) reset = 1;
break;
case MP_CMD_DVDNAV_MOUSECLICK:
/*
this is a workaround: in theory the simple dvdnav_lower_button_select()+dvdnav_button_activate()
should be enough (and generally it is), but there are cases when the calls to dvdnav_lower_button_select()
and friends fail! Hence we have to call dvdnav_mouse_activate(priv->mousex, priv->mousey) with
the coodinates saved by mp_dvdnav_update_mouse_pos().
This last call always works well
*/
status = dvdnav_mouse_activate(nav, pci, priv->mousex, priv->mousey);
if(status == DVDNAV_STATUS_OK) reset = 1;
break;
default:
mp_msg(MSGT_CPLAYER, MSGL_V, "Unknown DVDNAV cmd %d\n", cmd);
break;
}
if(status == DVDNAV_STATUS_OK)
dvdnav_get_current_highlight(nav, button);
return reset;
}
void mp_dvdnav_update_mouse_pos(stream_t *stream, int32_t x, int32_t y, int* button) {
dvdnav_priv_t * priv=(dvdnav_priv_t*)stream->priv;
dvdnav_t *nav = priv->dvdnav;
dvdnav_status_t status;
pci_t *pci = dvdnav_get_current_nav_pci(nav);
if(!pci) return;
status = dvdnav_mouse_select(nav, pci, x, y);
if(status == DVDNAV_STATUS_OK) dvdnav_get_current_highlight(nav, button);
else *button = -1;
priv->mousex = x;
priv->mousey = y;
}
/**
* \brief dvdnav_aid_from_lang() returns the audio id corresponding to the language code 'lang'
* \param stream: - stream pointer
* \param lang: 2-characters language code[s], eventually separated by spaces of commas
* \return -1 on error, current subtitle id if successful
*/
int dvdnav_aid_from_lang(stream_t *stream, unsigned char *language) {
dvdnav_priv_t * priv=(dvdnav_priv_t*)stream->priv;
int k;
uint8_t format, lg;
uint16_t lang, lcode;;
#ifdef DVDNAV_FORMAT_AC3
//this macro is defined only in libdvdnav-cvs
while(language && strlen(language)>=2) {
lcode = (language[0] << 8) | (language[1]);
for(k=0; k<32; k++) {
lg = dvdnav_get_audio_logical_stream(priv->dvdnav, k);
if(lg == 0xff) continue;
lang = dvdnav_audio_stream_to_lang(priv->dvdnav, lg);
if(lang != 0xFFFF && lang == lcode) {
format = dvdnav_audio_stream_format(priv->dvdnav, lg);
switch(format) {
case DVDNAV_FORMAT_AC3:
return k+128;
case DVDNAV_FORMAT_DTS:
return k+136;
case DVDNAV_FORMAT_LPCM:
return k+160;
case DVDNAV_FORMAT_MPEGAUDIO:
return k;
default:
return -1;
}
}
}
language += 2;
while(language[0]==',' || language[0]==' ') ++language;
}
#endif
return -1;
}
/**
* \brief dvdnav_lang_from_aid() assigns to buf the language corresponding to audio id 'aid'
* \param stream: - stream pointer
* \param sid: physical subtitle id
* \param buf: buffer to contain the 2-chars language string
* \return 0 on error, 1 if successful
*/
int dvdnav_lang_from_aid(stream_t *stream, int aid, unsigned char *buf) {
uint8_t lg;
uint16_t lang;
dvdnav_priv_t * priv=(dvdnav_priv_t*)stream->priv;
if(aid < 0)
return 0;
lg = dvdnav_get_audio_logical_stream(priv->dvdnav, aid & 0x7);
if(lg == 0xff) return 0;
lang = dvdnav_audio_stream_to_lang(priv->dvdnav, lg);
if(lang == 0xffff) return 0;
buf[0] = lang >> 8;
buf[1] = lang & 0xFF;
buf[2] = 0;
return 1;
}
/**
* \brief dvdnav_sid_from_lang() returns the subtitle id corresponding to the language code 'lang'
* \param stream: - stream pointer
* \param lang: 2-characters language code[s], eventually separated by spaces of commas
* \return -1 on error, current subtitle id if successful
*/
int dvdnav_sid_from_lang(stream_t *stream, unsigned char *language) {
dvdnav_priv_t * priv=(dvdnav_priv_t*)stream->priv;
uint8_t lg, k;
uint16_t lang, lcode;
while(language && strlen(language)>=2) {
lcode = (language[0] << 8) | (language[1]);
for(k=0; k<32; k++) {
lg = dvdnav_get_spu_logical_stream(priv->dvdnav, k);
if(lg == 0xff) continue;
lang = dvdnav_spu_stream_to_lang(priv->dvdnav, lg);
if(lang != 0xFFFF && lang == lcode) {
return k;
}
}
language += 2;
while(language[0]==',' || language[0]==' ') ++language;
}
return -1;
}
/**
* \brief dvdnav_lang_from_sid() assigns to buf the language corresponding to subtitle id 'sid'
* \param stream: - stream pointer
* \param sid: physical subtitle id
* \param buf: buffer to contain the 2-chars language string
* \return 0 on error, 1 if successful
*/
int dvdnav_lang_from_sid(stream_t *stream, int sid, unsigned char *buf) {
uint8_t lg;
uint16_t lang;
dvdnav_priv_t *priv=(dvdnav_priv_t*)stream->priv;
if(sid < 0) return 0;
lg = dvdnav_get_spu_logical_stream(priv->dvdnav, sid);
lang = dvdnav_spu_stream_to_lang(priv->dvdnav, lg);
if(lang == 0xffff) return 0;
buf[0] = lang >> 8;
buf[1] = lang & 0xFF;
buf[2] = 0;
return 1;
}
/**
* \brief dvdnav_number_of_subs() returns the count of available subtitles
* \param stream: - stream pointer
* \return 0 on error, something meaningful otherwise
*/
int dvdnav_number_of_subs(stream_t *stream) {
dvdnav_priv_t * priv=(dvdnav_priv_t*)stream->priv;
uint8_t lg, k, n=0;
for(k=0; k<32; k++) {
lg = dvdnav_get_spu_logical_stream(priv->dvdnav, k);
if(lg == 0xff) continue;
n++;
}
return n;
}
/**
* \brief mp_dvdnav_get_spu_clut() returns the spu clut
* \param stream: - stream pointer
* \return spu clut pointer
*/
unsigned int *mp_dvdnav_get_spu_clut(stream_t *stream) {
dvdnav_priv_t *priv=(dvdnav_priv_t*)stream->priv;
if(!priv->spu_set) return NULL;
return priv->spu_clut;
}
/**
* \brief mp_dvdnav_get_highlight() get dvdnav highlight struct
* \param stream: - stream pointer
* \param hl : - highlight struct pointer
*/
void mp_dvdnav_get_highlight (stream_t *stream, nav_highlight_t *hl) {
dvdnav_priv_t *priv = (dvdnav_priv_t *) stream->priv;
dvdnav_highlight_event_t hlev = priv->hlev;
hl->sx = hlev.sx;
hl->sy = hlev.sy;
hl->ex = hlev.ex;
hl->ey = hlev.ey;
}
stream_info_t stream_info_dvdnav = {
"DVDNAV stream",
"null",
"",
"",
open_s,
{ "dvdnav", NULL },
&stream_opts,
1 // Urls are an option string
};