libao2/ao_pcm.c

   1 /*
   2  * PCM audio output driver
   3  *
   4  * This file is part of MPlayer.
   5  *
   6  * MPlayer is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * MPlayer is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along
  17  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19  */
  20
  21 #include "config.h"
  22
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26
  27 #include "libavutil/common.h"
  28 #include "mpbswap.h"
  29 #include "subopt-helper.h"
  30 #include "libaf/af_format.h"
  31 #include "libaf/reorder_ch.h"
  32 #include "audio_out.h"
  33 #include "audio_out_internal.h"
  34 #include "mp_msg.h"
  35 #include "help_mp.h"
  36
  37 #ifdef __MINGW32__
  38 // for GetFileType to detect pipes
  39 #include <windows.h>
  40 #endif
  41
  42 static const ao_info_t info =
  43 {
  44     "RAW PCM/WAVE file writer audio output",
  45     "pcm",
  46     "Atmosfear",
  47     ""
  48 };
  49
  50 LIBAO_EXTERN(pcm)
  51
  52 extern int vo_pts;
  53
  54 static char *ao_outputfilename = NULL;
  55 static int ao_pcm_waveheader = 1;
  56 static int fast = 0;
  57
  58 #define WAV_ID_RIFF 0x46464952 /* "RIFF" */
  59 #define WAV_ID_WAVE 0x45564157 /* "WAVE" */
  60 #define WAV_ID_FMT  0x20746d66 /* "fmt " */
  61 #define WAV_ID_DATA 0x61746164 /* "data" */
  62 #define WAV_ID_PCM  0x0001
  63 #define WAV_ID_FLOAT_PCM  0x0003
  64
  65 struct WaveHeader
  66 {
  67     uint32_t riff;
  68     uint32_t file_length;
  69     uint32_t wave;
  70     uint32_t fmt;
  71     uint32_t fmt_length;
  72     uint16_t fmt_tag;
  73     uint16_t channels;
  74     uint32_t sample_rate;
  75     uint32_t bytes_per_second;
  76     uint16_t block_align;
  77     uint16_t bits;
  78     uint32_t data;
  79     uint32_t data_length;
  80 };
  81
  82 /* init with default values */
  83 static struct WaveHeader wavhdr;
  84 static uint64_t data_length;
  85
  86 static FILE *fp = NULL;
  87
  88 // to set/get/query special features/parameters
  89 static int control(int cmd,void *arg){
  90     return -1;
  91 }
  92
  93 // open & setup audio device
  94 // return: 1=success 0=fail
  95 static int init(int rate,int channels,int format,int flags){
  96     int bits;
  97     const opt_t subopts[] = {
  98         {"waveheader", OPT_ARG_BOOL, &ao_pcm_waveheader, NULL},
  99         {"file",       OPT_ARG_MSTRZ, &ao_outputfilename, NULL},
 100         {"fast",       OPT_ARG_BOOL, &fast, NULL},
 101         {NULL}
 102     };
 103     // set defaults
 104     ao_pcm_waveheader = 1;
 105
 106     if (subopt_parse(ao_subdevice, subopts) != 0) {
 107         return 0;
 108     }
 109     if (!ao_outputfilename){
 110         ao_outputfilename =
 111             strdup(ao_pcm_waveheader?"audiodump.wav":"audiodump.pcm");
 112     }
 113
 114     if (ao_pcm_waveheader)
 115     {
 116         // WAV files must have one of the following formats
 117
 118         switch(format){
 119         case AF_FORMAT_U8:
 120         case AF_FORMAT_S16_LE:
 121         case AF_FORMAT_S24_LE:
 122         case AF_FORMAT_S32_LE:
 123         case AF_FORMAT_FLOAT_LE:
 124         case AF_FORMAT_AC3_BE:
 125         case AF_FORMAT_AC3_LE:
 126              break;
 127         default:
 128             format = AF_FORMAT_S16_LE;
 129             break;
 130         }
 131     }
 132
 133     bits = af_fmt2bits(format);
 134
 135     ao_data.outburst = 65536;
 136     ao_data.buffersize= 2*65536;
 137     ao_data.channels=channels;
 138     ao_data.samplerate=rate;
 139     ao_data.format=format;
 140     ao_data.bps=channels*rate*(bits/8);
 141
 142     wavhdr.riff = le2me_32(WAV_ID_RIFF);
 143     wavhdr.wave = le2me_32(WAV_ID_WAVE);
 144     wavhdr.fmt = le2me_32(WAV_ID_FMT);
 145     wavhdr.fmt_length = le2me_32(16);
 146     wavhdr.fmt_tag = le2me_16(format == AF_FORMAT_FLOAT_LE ? WAV_ID_FLOAT_PCM : WAV_ID_PCM);
 147     wavhdr.channels = le2me_16(ao_data.channels);
 148     wavhdr.sample_rate = le2me_32(ao_data.samplerate);
 149     wavhdr.bytes_per_second = le2me_32(ao_data.bps);
 150     wavhdr.bits = le2me_16(bits);
 151     wavhdr.block_align = le2me_16(ao_data.channels * (bits / 8));
 152
 153     wavhdr.data = le2me_32(WAV_ID_DATA);
 154     wavhdr.data_length=le2me_32(0x7ffff000);
 155     wavhdr.file_length = wavhdr.data_length + sizeof(wavhdr) - 8;
 156
 157     mp_msg(MSGT_AO, MSGL_INFO, MSGTR_AO_PCM_FileInfo, ao_outputfilename,
 158            (ao_pcm_waveheader?"WAVE":"RAW PCM"), rate,
 159            (channels > 1) ? "Stereo" : "Mono", af_fmt2str_short(format));
 160     mp_msg(MSGT_AO, MSGL_INFO, MSGTR_AO_PCM_HintInfo);
 161
 162     fp = fopen(ao_outputfilename, "wb");
 163     if(fp) {
 164         if(ao_pcm_waveheader){ /* Reserve space for wave header */
 165             fwrite(&wavhdr,sizeof(wavhdr),1,fp);
 166         }
 167         return 1;
 168     }
 169     mp_msg(MSGT_AO, MSGL_ERR, MSGTR_AO_PCM_CantOpenOutputFile,
 170                ao_outputfilename);
 171     return 0;
 172 }
 173
 174 // close audio device
 175 static void uninit(int immed){
 176
 177     if(ao_pcm_waveheader){ /* Rewrite wave header */
 178         int broken_seek = 0;
 179 #ifdef __MINGW32__
 180         // Windows, in its usual idiocy "emulates" seeks on pipes so it always looks
 181         // like they work. So we have to detect them brute-force.
 182         broken_seek = GetFileType((HANDLE)_get_osfhandle(_fileno(fp))) != FILE_TYPE_DISK;
 183 #endif
 184         if (broken_seek || fseek(fp, 0, SEEK_SET) != 0)
 185             mp_msg(MSGT_AO, MSGL_ERR, "Could not seek to start, WAV size headers not updated!\n");
 186         else if (data_length > 0x7ffff000)
 187             mp_msg(MSGT_AO, MSGL_ERR, "File larger than allowed for WAV files, may play truncated!\n");
 188         else {
 189             wavhdr.file_length = data_length + sizeof(wavhdr) - 8;
 190             wavhdr.file_length = le2me_32(wavhdr.file_length);
 191             wavhdr.data_length = le2me_32(data_length);
 192             fwrite(&wavhdr,sizeof(wavhdr),1,fp);
 193         }
 194     }
 195     fclose(fp);
 196     if (ao_outputfilename)
 197         free(ao_outputfilename);
 198     ao_outputfilename = NULL;
 199 }
 200
 201 // stop playing and empty buffers (for seeking/pause)
 202 static void reset(void){
 203
 204 }
 205
 206 // stop playing, keep buffers (for pause)
 207 static void audio_pause(void)
 208 {
 209     // for now, just call reset();
 210     reset();
 211 }
 212
 213 // resume playing, after audio_pause()
 214 static void audio_resume(void)
 215 {
 216 }
 217
 218 // return: how many bytes can be played without blocking
 219 static int get_space(void){
 220
 221     if(vo_pts)
 222         return ao_data.pts < vo_pts + fast * 30000 ? ao_data.outburst : 0;
 223     return ao_data.outburst;
 224 }
 225
 226 // plays 'len' bytes of 'data'
 227 // it should round it down to outburst*n
 228 // return: number of bytes played
 229 static int play(void* data,int len,int flags){
 230
 231 // let libaf to do the conversion...
 232 #if 0
 233 //#if HAVE_BIGENDIAN
 234     if (ao_data.format == AFMT_S16_LE) {
 235       unsigned short *buffer = (unsigned short *) data;
 236       register int i;
 237       for(i = 0; i < len/2; ++i) {
 238         buffer[i] = le2me_16(buffer[i]);
 239       }
 240     }
 241 #endif
 242
 243     if (ao_data.channels == 5 || ao_data.channels == 6 || ao_data.channels == 8) {
 244         int frame_size = le2me_16(wavhdr.bits) / 8;
 245         len -= len % (frame_size * ao_data.channels);
 246         reorder_channel_nch(data, AF_CHANNEL_LAYOUT_MPLAYER_DEFAULT,
 247                             AF_CHANNEL_LAYOUT_WAVEEX_DEFAULT,
 248                             ao_data.channels,
 249                             len / frame_size, frame_size);
 250     }
 251
 252     //printf("PCM: Writing chunk!\n");
 253     fwrite(data,len,1,fp);
 254
 255     if(ao_pcm_waveheader)
 256         data_length += len;
 257
 258     return len;
 259 }
 260
 261 // return: delay in seconds between first and last sample in buffer
 262 static float get_delay(void){
 263
 264     return 0.0;
 265 }
 266
 267
 268
 269
 270
 271