libmpdemux/demux_audio.c

   1
   2 #include "config.h"
   3 #include "../mp_msg.h"
   4
   5 #include <stdlib.h>
   6 #include <stdio.h>
   7 #include "stream.h"
   8 #include "demuxer.h"
   9 #include "stheader.h"
  10 #include "genres.h"
  11 #include "mp3_hdr.h"
  12
  13 #include <string.h>
  14 #ifdef MP_DEBUG
  15 #include <assert.h>
  16 #endif
  17
  18 #define MP3 1
  19 #define WAV 2
  20 #define fLaC 3
  21
  22
  23 #define HDR_SIZE 4
  24
  25 typedef struct da_priv {
  26   int frmt;
  27   float last_pts;
  28 } da_priv_t;
  29
  30 extern void free_sh_audio(sh_audio_t* sh);
  31 extern void resync_audio_stream(sh_audio_t *sh_audio);
  32 extern void print_wave_header(WAVEFORMATEX *h);
  33
  34 int hr_mp3_seek = 0;
  35
  36 int demux_audio_open(demuxer_t* demuxer) {
  37   stream_t *s;
  38   sh_audio_t* sh_audio;
  39   uint8_t hdr[HDR_SIZE];
  40   int st_pos = 0,frmt = 0, n = 0, pos = 0, step, mp3_freq,mp3_chans;
  41   da_priv_t* priv;
  42 #ifdef MP_DEBUG
  43   assert(demuxer != NULL);
  44   assert(demuxer->stream != NULL);
  45 #endif
  46
  47   s = demuxer->stream;
  48
  49   while(n < 5 && ! s->eof) {
  50     st_pos = stream_tell(s);
  51     step = 1;
  52     if(pos < HDR_SIZE) {
  53       stream_read(s,&hdr[pos],HDR_SIZE-pos);
  54       pos = HDR_SIZE;
  55     }
  56
  57     if( hdr[0] == 'R' && hdr[1] == 'I' && hdr[2] == 'F' && hdr[3] == 'F' ) {
  58       stream_skip(s,4);
  59       if(s->eof)
  60         break;
  61       stream_read(s,hdr,4);
  62       if(s->eof)
  63         break;
  64       if(hdr[0] != 'W' || hdr[1] != 'A' || hdr[2] != 'V'  || hdr[3] != 'E' )
  65         stream_skip(s,-8);
  66       else
  67       // We found wav header. Now we can have 'fmt ' or a mp3 header
  68       // empty the buffer
  69         step = 4;
  70     } else if( hdr[0] == 'I' && hdr[1] == 'D' && hdr[2] == '3' && (hdr[3] >= 2)) {
  71       int len;
  72       stream_skip(s,2);
  73       stream_read(s,hdr,4);
  74       len = (hdr[0]<<21) | (hdr[1]<<14) | (hdr[2]<<7) | hdr[3];
  75       stream_skip(s,len);
  76       step = 4;
  77     } else if( hdr[0] == 'f' && hdr[1] == 'm' && hdr[2] == 't' && hdr[3] == ' ' ) {
  78       frmt = WAV;
  79       break;
  80     } else if((n = mp_get_mp3_header(hdr,&mp3_chans,&mp3_freq)) > 0) {
  81       frmt = MP3;
  82       break;
  83     } else if( hdr[0] == 'f' && hdr[1] == 'L' && hdr[2] == 'a' && hdr[3] == 'C' ) {
  84       frmt = fLaC;
  85       stream_skip(s,-4);
  86       break;
  87     }
  88     // Add here some other audio format detection
  89     if(step < HDR_SIZE)
  90       memmove(hdr,&hdr[step],HDR_SIZE-step);
  91     pos -= step;
  92   }
  93
  94   if(!frmt)
  95     return 0;
  96
  97   sh_audio = new_sh_audio(demuxer,0);
  98
  99   switch(frmt) {
 100   case MP3:
 101     sh_audio->format = 0x55;
 102     demuxer->movi_start = st_pos-HDR_SIZE+n;
 103     sh_audio->audio.dwSampleSize= 0;
 104     sh_audio->audio.dwScale = 1152;
 105     sh_audio->audio.dwRate = mp3_freq;
 106     sh_audio->wf = malloc(sizeof(WAVEFORMATEX));
 107     sh_audio->wf->wFormatTag = sh_audio->format;
 108     sh_audio->wf->nChannels = mp3_chans;
 109     sh_audio->wf->nSamplesPerSec = mp3_freq;
 110     sh_audio->wf->nBlockAlign = 1152;
 111     sh_audio->wf->wBitsPerSample = 16;
 112     sh_audio->wf->cbSize = 0;
 113     for(n = 0; n < 5 ; n++) {
 114       pos = mp_decode_mp3_header(hdr);
 115       if(pos < 0)
 116         return 0;
 117       stream_skip(s,pos-4);
 118       if(s->eof)
 119         return 0;
 120       stream_read(s,hdr,4);
 121       if(s->eof)
 122         return 0;
 123     }
 124     if(s->end_pos) {
 125       char tag[4];
 126       stream_seek(s,s->end_pos-128);
 127       stream_read(s,tag,3);
 128       tag[3] = '\0';
 129       if(strcmp(tag,"TAG"))
 130         demuxer->movi_end = s->end_pos;
 131       else {
 132         char buf[31];
 133         uint8_t g;
 134         demuxer->movi_end = stream_tell(s)-3;
 135         stream_read(s,buf,30);
 136         buf[30] = '\0';
 137         demux_info_add(demuxer,"Title",buf);
 138         stream_read(s,buf,30);
 139         buf[30] = '\0';
 140         demux_info_add(demuxer,"Artist",buf);
 141         stream_read(s,buf,30);
 142         buf[30] = '\0';
 143         demux_info_add(demuxer,"Album",buf);
 144         stream_read(s,buf,4);
 145         buf[4] = '\0';
 146         demux_info_add(demuxer,"Year",buf);
 147         stream_read(s,buf,30);
 148         buf[30] = '\0';
 149         demux_info_add(demuxer,"Comment",buf);
 150         if(buf[28] == 0 && buf[29] != 0) {
 151           uint8_t trk = (uint8_t)buf[29];
 152           sprintf(buf,"%d",trk);
 153           demux_info_add(demuxer,"Track",buf);
 154         }
 155         g = stream_read_char(s);
 156         demux_info_add(demuxer,"Genre",genres[g]);
 157       }
 158     }
 159     break;
 160   case WAV: {
 161     unsigned int chunk_type;
 162     unsigned int chunk_size;
 163     WAVEFORMATEX* w;
 164     int l;
 165     l = stream_read_dword_le(s);
 166     if(l < 16) {
 167       mp_msg(MSGT_DEMUX,MSGL_ERR,"[demux_audio] Bad wav header length: too short (%d)!!!\n",l);
 168       free_sh_audio(sh_audio);
 169       return 0;
 170     }
 171     sh_audio->wf = w = (WAVEFORMATEX*)malloc(l);
 172     w->wFormatTag = sh_audio->format = stream_read_word_le(s);
 173     w->nChannels = sh_audio->channels = stream_read_word_le(s);
 174     w->nSamplesPerSec = sh_audio->samplerate = stream_read_dword_le(s);
 175     w->nAvgBytesPerSec = stream_read_dword_le(s);
 176     w->nBlockAlign = stream_read_word_le(s);
 177     w->wBitsPerSample = sh_audio->samplesize = stream_read_word_le(s);
 178     w->cbSize = 0;
 179     l -= 16;
 180     if (l > 0) {
 181     w->cbSize = stream_read_word_le(s);
 182     l -= 2;
 183      if (w->cbSize > 0) {
 184       if (l < w->cbSize) {
 185         mp_msg(MSGT_DEMUX,MSGL_ERR,"[demux_audio] truncated extradata (%d < %d)\n",
 186         l,w->cbSize);
 187         stream_read(s,(char*)((char*)(w)+sizeof(WAVEFORMATEX)),l);
 188         l = 0;
 189       } else {
 190         stream_read(s,(char*)((char*)(w)+sizeof(WAVEFORMATEX)),w->cbSize);
 191         l -= w->cbSize;
 192       }
 193      }
 194     }
 195
 196     if(verbose>0) print_wave_header(w);
 197     if(l)
 198       stream_skip(s,l);
 199     do
 200     {
 201       chunk_type = stream_read_fourcc(demuxer->stream);
 202       chunk_size = stream_read_dword_le(demuxer->stream);
 203       if (chunk_type != mmioFOURCC('d', 'a', 't', 'a'))
 204         stream_skip(demuxer->stream, chunk_size);
 205     } while (chunk_type != mmioFOURCC('d', 'a', 't', 'a'));
 206     demuxer->movi_start = stream_tell(s);
 207     demuxer->movi_end = s->end_pos;
 208 //    printf("wav: %X .. %X\n",(int)demuxer->movi_start,(int)demuxer->movi_end);
 209   } break;
 210   case fLaC:
 211             sh_audio->format = mmioFOURCC('f', 'L', 'a', 'C');
 212             demuxer->movi_start = stream_tell(s);
 213             demuxer->movi_end = s->end_pos;
 214             break;
 215   }
 216
 217   priv = (da_priv_t*)malloc(sizeof(da_priv_t));
 218   priv->frmt = frmt;
 219   priv->last_pts = -1;
 220   demuxer->priv = priv;
 221   demuxer->audio->id = 0;
 222   demuxer->audio->sh = sh_audio;
 223   sh_audio->ds = demuxer->audio;
 224   sh_audio->samplerate = sh_audio->audio.dwRate;
 225
 226   if(stream_tell(s) != demuxer->movi_start)
 227     stream_seek(s,demuxer->movi_start);
 228
 229   mp_msg(MSGT_DEMUX,MSGL_V,"demux_audio: audio data 0x%X - 0x%X  \n",demuxer->movi_start,demuxer->movi_end);
 230
 231   return 1;
 232 }
 233
 234
 235 int demux_audio_fill_buffer(demux_stream_t *ds) {
 236   sh_audio_t* sh_audio;
 237   demuxer_t* demux;
 238   da_priv_t* priv;
 239   stream_t* s;
 240 #ifdef MP_DEBUG
 241   assert(ds != NULL);
 242   assert(ds->sh != NULL);
 243   assert(ds->demuxer != NULL);
 244 #endif
 245   sh_audio = ds->sh;
 246   demux = ds->demuxer;
 247   priv = demux->priv;
 248   s = demux->stream;
 249
 250   if(s->eof || (demux->movi_end && stream_tell(s) >= demux->movi_end) )
 251     return 0;
 252
 253   switch(priv->frmt) {
 254   case MP3 :
 255     while(! s->eof || (demux->movi_end && stream_tell(s) >= demux->movi_end) ) {
 256       uint8_t hdr[4];
 257       int len;
 258       stream_read(s,hdr,4);
 259       len = mp_decode_mp3_header(hdr);
 260       if(len < 0) {
 261         stream_skip(s,-3);
 262       } else {
 263         demux_packet_t* dp;
 264         if(s->eof  || (demux->movi_end && stream_tell(s) >= demux->movi_end) )
 265           return 0;
 266         dp = new_demux_packet(len);
 267         memcpy(dp->buffer,hdr,4);
 268         stream_read(s,dp->buffer + 4,len-4);
 269         priv->last_pts = priv->last_pts < 0 ? 0 : priv->last_pts + 1152/(float)sh_audio->samplerate;
 270         ds->pts = priv->last_pts - (ds_tell_pts(demux->audio)-sh_audio->a_in_buffer_len)/(float)sh_audio->i_bps;
 271         ds_add_packet(ds,dp);
 272         return 1;
 273       }
 274     } break;
 275   case WAV : {
 276     int l = sh_audio->wf->nAvgBytesPerSec;
 277     demux_packet_t*  dp = new_demux_packet(l);
 278     l = stream_read(s,dp->buffer,l);
 279     resize_demux_packet(dp, l);
 280     priv->last_pts = priv->last_pts < 0 ? 0 : priv->last_pts + l/(float)sh_audio->i_bps;
 281     ds->pts = priv->last_pts - (ds_tell_pts(demux->audio)-sh_audio->a_in_buffer_len)/(float)sh_audio->i_bps;
 282     ds_add_packet(ds,dp);
 283     return 1;
 284   }
 285   case fLaC: {
 286     int l = 65535;
 287     demux_packet_t*  dp = new_demux_packet(l);
 288     l = stream_read(s,dp->buffer,l);
 289     resize_demux_packet(dp, l);
 290     priv->last_pts = priv->last_pts < 0 ? 0 : priv->last_pts + l/(float)sh_audio->i_bps;
 291     ds->pts = priv->last_pts - (ds_tell_pts(demux->audio)-sh_audio->a_in_buffer_len)/(float)sh_audio->i_bps;
 292     ds_add_packet(ds,dp);
 293     return 1;
 294   }
 295   default:
 296     printf("Audio demuxer : unknown format %d\n",priv->frmt);
 297   }
 298
 299
 300   return 0;
 301 }
 302
 303 static void high_res_mp3_seek(demuxer_t *demuxer,float time) {
 304   uint8_t hdr[4];
 305   int len,nf;
 306   da_priv_t* priv = demuxer->priv;
 307   sh_audio_t* sh = (sh_audio_t*)demuxer->audio->sh;
 308
 309   nf = time*sh->samplerate/1152;
 310   while(nf > 0) {
 311     stream_read(demuxer->stream,hdr,4);
 312     len = mp_decode_mp3_header(hdr);
 313     if(len < 0) {
 314       stream_skip(demuxer->stream,-3);
 315       continue;
 316     }
 317     stream_skip(demuxer->stream,len-4);
 318     priv->last_pts += 1152/(float)sh->samplerate;
 319     nf--;
 320   }
 321 }
 322
 323 void demux_audio_seek(demuxer_t *demuxer,float rel_seek_secs,int flags){
 324   sh_audio_t* sh_audio;
 325   stream_t* s;
 326   int base,pos;
 327   float len;
 328   da_priv_t* priv;
 329
 330   if(!(sh_audio = demuxer->audio->sh))
 331     return;
 332   s = demuxer->stream;
 333   priv = demuxer->priv;
 334
 335   if(priv->frmt == MP3 && hr_mp3_seek && !(flags & 2)) {
 336     len = (flags & 1) ? rel_seek_secs - priv->last_pts : rel_seek_secs;
 337     if(len < 0) {
 338       stream_seek(s,demuxer->movi_start);
 339       len = priv->last_pts + len;
 340       priv->last_pts = 0;
 341     }
 342     if(len > 0)
 343       high_res_mp3_seek(demuxer,len);
 344     sh_audio->delay = priv->last_pts -  (ds_tell_pts(demuxer->audio)-sh_audio->a_in_buffer_len)/(float)sh_audio->i_bps;
 345     resync_audio_stream(sh_audio);
 346     return;
 347   }
 348
 349   base = flags&1 ? demuxer->movi_start : stream_tell(s);
 350   if(flags&2)
 351     pos = base + ((demuxer->movi_end - demuxer->movi_start)*rel_seek_secs);
 352   else
 353     pos = base + (rel_seek_secs*sh_audio->i_bps);
 354
 355   if(demuxer->movi_end && pos >= demuxer->movi_end) {
 356      pos = demuxer->movi_end;
 357     //sh_audio->delay = (stream_tell(s) - demuxer->movi_start)/(float)sh_audio->i_bps;
 358     //return;
 359   } else if(pos < demuxer->movi_start)
 360     pos = demuxer->movi_start;
 361
 362   priv->last_pts = (pos-demuxer->movi_start)/(float)sh_audio->i_bps;
 363   sh_audio->delay = priv->last_pts - (ds_tell_pts(demuxer->audio)-sh_audio->a_in_buffer_len)/(float)sh_audio->i_bps;
 364
 365   switch(priv->frmt) {
 366   case WAV:
 367     pos -= (pos % (sh_audio->channels * sh_audio->samplesize) );
 368     // We need to decrease the pts by one step to make it the "last one"
 369     priv->last_pts -= sh_audio->wf->nAvgBytesPerSec/(float)sh_audio->i_bps;
 370     break;
 371   }
 372
 373   stream_seek(s,pos);
 374
 375   resync_audio_stream(sh_audio);
 376
 377 }
 378
 379 void demux_close_audio(demuxer_t* demuxer) {
 380   da_priv_t* priv = demuxer->priv;
 381
 382   if(!priv)
 383     return;
 384   free(priv);
 385 }
 386
 387 int demux_audio_control(demuxer_t *demuxer,int cmd, void *arg){
 388     sh_audio_t *sh_audio=demuxer->audio->sh;
 389     int audio_length = demuxer->movi_end / sh_audio->i_bps;
 390     da_priv_t* priv = demuxer->priv;
 391
 392     switch(cmd) {
 393         case DEMUXER_CTRL_GET_TIME_LENGTH:
 394             if (audio_length<=0) return DEMUXER_CTRL_DONTKNOW;
 395             *((unsigned long *)arg)=(unsigned long)audio_length;
 396             return DEMUXER_CTRL_GUESS;
 397
 398         case DEMUXER_CTRL_GET_PERCENT_POS:
 399             if (audio_length<=0)
 400                 return DEMUXER_CTRL_DONTKNOW;
 401             *((int *)arg)=(int)( (priv->last_pts*100)  / audio_length);
 402             return DEMUXER_CTRL_OK;
 403
 404         default:
 405             return DEMUXER_CTRL_NOTIMPL;
 406     }
 407 }