mpv/libao2/ao_mpegpes.c

318 lines
7.0 KiB
C

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <inttypes.h>
#include "config.h"
#ifdef HAVE_DVB_HEAD
#define HAVE_DVB 1
#endif
#ifdef HAVE_DVB
#include <sys/poll.h>
#include <sys/ioctl.h>
#endif
#include "audio_out.h"
#include "audio_out_internal.h"
#include "libaf/af_format.h"
#include "libmpdemux/mpeg_packetizer.h"
#include "subopt-helper.h"
#include "mp_msg.h"
#include "help_mp.h"
#ifdef HAVE_DVB
#ifndef HAVE_DVB_HEAD
#include <ost/audio.h>
audioMixer_t dvb_mixer={255,255};
#else
#include <linux/dvb/audio.h>
audio_mixer_t dvb_mixer={255,255};
#endif
#endif
#define true 1
#define false 0
extern int vo_mpegpes_fd;
int vo_mpegpes_fd2 = -1;
#include <errno.h>
static ao_info_t info =
{
#ifdef HAVE_DVB
"DVB audio output",
#else
"Mpeg-PES audio output",
#endif
"mpegpes",
"A'rpi",
""
};
LIBAO_EXTERN(mpegpes)
// to set/get/query special features/parameters
static int control(int cmd,void *arg){
#ifdef HAVE_DVB
switch(cmd){
case AOCONTROL_GET_VOLUME:
if(vo_mpegpes_fd2>=0){
((ao_control_vol_t*)(arg))->left=dvb_mixer.volume_left/2.56;
((ao_control_vol_t*)(arg))->right=dvb_mixer.volume_right/2.56;
return CONTROL_OK;
}
return CONTROL_ERROR;
case AOCONTROL_SET_VOLUME:
if(vo_mpegpes_fd2>=0){
dvb_mixer.volume_left=((ao_control_vol_t*)(arg))->left*2.56;
dvb_mixer.volume_right=((ao_control_vol_t*)(arg))->right*2.56;
if(dvb_mixer.volume_left>255) dvb_mixer.volume_left=255;
if(dvb_mixer.volume_right>255) dvb_mixer.volume_right=255;
// printf("Setting DVB volume: %d ; %d \n",dvb_mixer.volume_left,dvb_mixer.volume_right);
if ( (ioctl(vo_mpegpes_fd2,AUDIO_SET_MIXER, &dvb_mixer) < 0)){
mp_msg(MSGT_AO, MSGL_ERR, MSGTR_AO_MPEGPES_CantSetMixer,
strerror(errno));
return CONTROL_ERROR;
}
return CONTROL_OK;
}
return CONTROL_ERROR;
}
#endif
return CONTROL_UNKNOWN;
}
static int freq=0;
static int freq_id=0;
#ifdef HAVE_DVB
static int init_device(int card)
{
char ao_file[30];
#ifndef HAVE_DVB_HEAD
mp_msg(MSGT_VO,MSGL_INFO, "Opening /dev/ost/audio\n");
sprintf(ao_file, "/dev/ost/audio");
#else
mp_msg(MSGT_VO,MSGL_INFO, "Opening /dev/dvb/adapter%d/audio0\n", card);
sprintf(ao_file, "/dev/dvb/adapter%d/audio0", card);
#endif
if((vo_mpegpes_fd2 = open(ao_file,O_RDWR|O_NONBLOCK)) < 0)
{
perror("DVB AUDIO DEVICE: ");
return -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;
}
if((ioctl(vo_mpegpes_fd2,AUDIO_SET_AV_SYNC, true) < 0))
{
perror("DVB AUDIO SET AV SYNC: ");
return -1;
}
//FIXME: in vo_mpegpes audio was inited as MUTEd
if((ioctl(vo_mpegpes_fd2,AUDIO_SET_MUTE, false) < 0))
{
perror("DVB AUDIO SET MUTE: ");
return -1;
}
return vo_mpegpes_fd2;
}
#endif
static int preinit(const char *arg)
{
int card = 1;
char *ao_file = NULL;
opt_t subopts[] = {
{"card", OPT_ARG_INT, &card, NULL},
{"file", OPT_ARG_MSTRZ, &ao_file, NULL},
{NULL}
};
if(subopt_parse(ao_subdevice, subopts) != 0)
{
mp_msg(MSGT_VO, MSGL_ERR, "AO_MPEGPES, Unrecognized options\n");
return -1;
}
if((card < 1) || (card > 4))
{
mp_msg(MSGT_VO, MSGL_ERR, "DVB card number must be between 1 and 4\n");
return -1;
}
card--;
#ifdef HAVE_DVB
if(!ao_file)
return init_device(card);
#else
if(!ao_file)
return vo_mpegpes_fd; //video fd
#endif
vo_mpegpes_fd2=open(ao_file,O_WRONLY|O_CREAT,0666);
if(vo_mpegpes_fd2<0)
{
perror("ao_mpegpes");
return -1;
}
return vo_mpegpes_fd2;
}
static int my_ao_write(unsigned char* data,int len){
int orig_len = len;
#ifdef HAVE_DVB
#define NFD 1
struct pollfd pfd[NFD];
pfd[0].fd = vo_mpegpes_fd2;
pfd[0].events = POLLOUT;
while(len>0){
if(poll(pfd,NFD,1)){
if(pfd[0].revents & POLLOUT){
int ret=write(vo_mpegpes_fd2,data,len);
if(ret<=0){
perror("write");
usleep(0);
} else {
len-=ret;
data+=ret;
}
} else usleep(1000);
}
}
#else
if(vo_mpegpes_fd2<0) return 0; // no file
write(vo_mpegpes_fd2,data,len); // write to file
#endif
return orig_len;
}
// open & setup audio device
// return: 1=success 0=fail
static int init(int rate,int channels,int format,int flags){
if(preinit(NULL)<0) return 0;
ao_data.channels=2;
ao_data.outburst=2000;
switch(format){
case AF_FORMAT_S16_LE:
case AF_FORMAT_S16_BE:
case AF_FORMAT_MPEG2:
case AF_FORMAT_AC3:
ao_data.format=format;
break;
default:
ao_data.format=AF_FORMAT_S16_BE;
}
retry:
switch(rate){
case 48000: freq_id=0;break;
case 96000: freq_id=1;break;
case 44100: freq_id=2;break;
case 32000: freq_id=3;break;
default:
mp_msg(MSGT_AO, MSGL_ERR, MSGTR_AO_MPEGPES_UnsupSamplerate, rate);
#if 0
if(rate>48000) rate=96000; else
if(rate>44100) rate=48000; else
if(rate>32000) rate=44100; else
rate=32000;
goto retry;
#else
rate=48000; freq_id=0;
#endif
}
ao_data.bps=rate*2*2;
freq=ao_data.samplerate=rate;
return 1;
}
// close audio device
static void uninit(int immed){
}
// stop playing and empty buffers (for seeking/pause)
static void reset(void){
}
// stop playing, keep buffers (for pause)
static void audio_pause(void)
{
// for now, just call reset();
reset();
}
// resume playing, after audio_pause()
static void audio_resume(void)
{
}
void send_pes_packet(unsigned char* data,int len,int id,int timestamp);
void send_lpcm_packet(unsigned char* data,int len,int id,int timestamp,int freq_id);
extern int vo_pts;
// return: how many bytes can be played without blocking
static int get_space(void){
float x=(float)(vo_pts-ao_data.pts)/90000.0;
int y;
//FIXME: is it correct?
if(vo_mpegpes_fd < 0) return 32000; //not using -vo mpegpes
// printf("vo_pts: %5.3f ao_pts: %5.3f\n",vo_pts/90000.0,ao_data.pts/90000.0);
if(x<=0) return 0;
y=freq*4*x;y/=ao_data.outburst;y*=ao_data.outburst;
if(y>32000) y=32000;
// printf("diff: %5.3f -> %d \n",x,y);
return y;
}
// plays 'len' bytes of 'data'
// it should round it down to outburst*n
// return: number of bytes played
static int play(void* data,int len,int flags){
// printf("\nao_mpegpes: play(%d) freq=%d\n",len,freq_id);
if(ao_data.format==AF_FORMAT_MPEG2)
send_mpeg_pes_packet (data, len, 0x1C0, ao_data.pts, 1, my_ao_write);
else {
int i;
unsigned short *s=data;
// if(len>2000) len=2000;
// printf("ao_mpegpes: len=%d \n",len);
if(ao_data.format==AF_FORMAT_S16_LE || ao_data.format==AF_FORMAT_AC3)
for(i=0;i<len/2;i++) s[i]=(s[i]>>8)|(s[i]<<8); // le<->be
send_mpeg_lpcm_packet(data, len, 0xA0, ao_data.pts, freq_id, my_ao_write);
}
return len;
}
// return: delay in seconds between first and last sample in buffer
static float get_delay(void){
return 0.0;
}