libmpdemux/demux_mpc.c

   1 /*
   2  * demuxer for Musepack v7 bitstream
   3  * copyright (c) 2005 Reimar Doeffinger <Reimar.Doeffinger@stud.uni-karlsruhe.de>
   4  *
   5  * This code may be be relicensed under the terms of the GNU LGPL when it
   6  * becomes part of the FFmpeg project (ffmpeg.org)
   7  *
   8  * This file is part of MPlayer.
   9  *
  10  * MPlayer is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * MPlayer is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License along
  21  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  22  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  23  */
  24
  25 #include "config.h"
  26
  27 #include <stdlib.h>
  28 #include <stdio.h>
  29 #include <string.h>
  30 #include "mp_msg.h"
  31 #include "libavutil/common.h"
  32 #include "mpbswap.h"
  33 #include "stream/stream.h"
  34 #include "demuxer.h"
  35 #include "stheader.h"
  36
  37
  38 #define HDR_SIZE (6 * 4)
  39
  40 typedef struct da_priv {
  41   float last_pts;
  42   float pts_per_packet;
  43   uint32_t dword;
  44   int pos;
  45   float length;
  46 } da_priv_t;
  47
  48 static uint32_t get_bits(da_priv_t* priv, stream_t* s, int bits) {
  49   uint32_t out = priv->dword;
  50   uint32_t mask = (1 << bits) - 1;
  51   priv->pos += bits;
  52   if (priv->pos < 32) {
  53     out >>= (32 - priv->pos);
  54   }
  55   else {
  56     stream_read(s, (void *)&priv->dword, 4);
  57     priv->dword = le2me_32(priv->dword);
  58     priv->pos -= 32;
  59     if (priv->pos) {
  60       out <<= priv->pos;
  61       out |= priv->dword >> (32 - priv->pos);
  62     }
  63   }
  64   return out & mask;
  65 }
  66
  67 static int demux_mpc_check(demuxer_t* demuxer) {
  68   stream_t *s = demuxer->stream;
  69   uint8_t hdr[HDR_SIZE];
  70   int i;
  71
  72   if (stream_read(s, hdr, HDR_SIZE) != HDR_SIZE)
  73     return 0;
  74   for (i = 0; i < 30000 && !s->eof; i++) {
  75     if (hdr[0] == 'M' && hdr[1] == 'P' && hdr[2] == '+')
  76       break;
  77     memmove(hdr, &hdr[1], HDR_SIZE - 1);
  78     stream_read(s, &hdr[HDR_SIZE - 1], 1);
  79   }
  80
  81   if (hdr[0] != 'M' || hdr[1] != 'P' || hdr[2] != '+')
  82     return 0;
  83   demuxer->movi_start = stream_tell(s) - HDR_SIZE;
  84   demuxer->movi_end = s->end_pos;
  85   demuxer->priv = malloc(HDR_SIZE);
  86   memcpy(demuxer->priv, hdr, HDR_SIZE);
  87   return DEMUXER_TYPE_MPC;
  88 }
  89
  90 static demuxer_t *demux_mpc_open(demuxer_t* demuxer) {
  91   float seconds = 0;
  92   stream_t *s = demuxer->stream;
  93   sh_audio_t* sh_audio;
  94   da_priv_t* priv = demuxer->priv;
  95
  96   sh_audio = new_sh_audio(demuxer,0);
  97
  98   {
  99     char *wf = calloc(1, sizeof(WAVEFORMATEX) + HDR_SIZE);
 100     char *header = &wf[sizeof(WAVEFORMATEX)];
 101     const int freqs[4] = {44100, 48000, 37800, 32000};
 102     int frames;
 103     sh_audio->format = mmioFOURCC('M', 'P', 'C', ' ');
 104     memcpy(header, priv, HDR_SIZE);
 105     free(priv);
 106     frames = header[4] | header[5] << 8 | header[6] << 16 | header[7] << 24;
 107     sh_audio->wf = (WAVEFORMATEX *)wf;
 108     sh_audio->wf->wFormatTag = sh_audio->format;
 109     sh_audio->wf->nChannels = 2;
 110     sh_audio->wf->nSamplesPerSec = freqs[header[10] & 3];
 111     sh_audio->wf->nBlockAlign = 32 * 36;
 112     sh_audio->wf->wBitsPerSample = 16;
 113     seconds = 1152 * frames / (float)sh_audio->wf->nSamplesPerSec;
 114     if (demuxer->movi_end > demuxer->movi_start && seconds > 1)
 115       sh_audio->wf->nAvgBytesPerSec = (demuxer->movi_end - demuxer->movi_start) / seconds;
 116     else
 117       sh_audio->wf->nAvgBytesPerSec = 32 * 1024; // dummy to make mencoder not hang
 118     sh_audio->wf->cbSize = HDR_SIZE;
 119     demuxer->movi_start = stream_tell(s);
 120     demuxer->movi_end = s->end_pos;
 121   }
 122
 123   priv = malloc(sizeof(da_priv_t));
 124   priv->last_pts = -1;
 125   priv->pts_per_packet = (32 * 36) / (float)sh_audio->wf->nSamplesPerSec;
 126   priv->length = seconds;
 127   priv->dword = 0;
 128   priv->pos = 32; // empty bit buffer
 129   get_bits(priv, s, 8); // discard first 8 bits
 130   demuxer->priv = priv;
 131   demuxer->audio->id = 0;
 132   demuxer->audio->sh = sh_audio;
 133   sh_audio->ds = demuxer->audio;
 134   sh_audio->samplerate = sh_audio->wf->nSamplesPerSec;
 135   sh_audio->i_bps = sh_audio->wf->nAvgBytesPerSec;
 136   sh_audio->audio.dwSampleSize = 0;
 137   sh_audio->audio.dwScale = 32 * 36;
 138   sh_audio->audio.dwRate = sh_audio->samplerate;
 139
 140   return demuxer;
 141 }
 142
 143 static int demux_mpc_fill_buffer(demuxer_t *demux, demux_stream_t *ds) {
 144   int l;
 145   int bit_len;
 146   demux_packet_t* dp;
 147   sh_audio_t* sh_audio = ds->sh;
 148   da_priv_t* priv = demux->priv;
 149   stream_t* s = demux->stream;
 150   sh_audio = ds->sh;
 151
 152   if (s->eof)
 153     return 0;
 154
 155   bit_len = get_bits(priv, s, 20);
 156   dp = new_demux_packet((bit_len + 7) / 8);
 157   for (l = 0; l < (bit_len / 8); l++)
 158     dp->buffer[l] = get_bits(priv, s, 8);
 159   bit_len %= 8;
 160   if (bit_len)
 161     dp->buffer[l] = get_bits(priv, s, bit_len) << (8 - bit_len);
 162   if (priv->last_pts < 0)
 163     priv->last_pts = 0;
 164   else
 165     priv->last_pts += priv->pts_per_packet;
 166   dp->pts = priv->last_pts;
 167   ds_add_packet(ds, dp);
 168   return 1;
 169 }
 170
 171 static void demux_mpc_seek(demuxer_t *demuxer,float rel_seek_secs,float audio_delay,int flags){
 172   sh_audio_t* sh_audio = demuxer->audio->sh;
 173   da_priv_t* priv = demuxer->priv;
 174   stream_t* s = demuxer->stream;
 175   float target = rel_seek_secs;
 176   if (flags & SEEK_FACTOR)
 177     target *= priv->length;
 178   if (!(flags & SEEK_ABSOLUTE))
 179     target += priv->last_pts;
 180   if (target < priv->last_pts) {
 181     stream_seek(s, demuxer->movi_start);
 182     priv->pos = 32; // empty bit buffer
 183     get_bits(priv, s, 8); // discard first 8 bits
 184     priv->last_pts = 0;
 185   }
 186   while (target > priv->last_pts) {
 187     int bit_len = get_bits(priv, s, 20);
 188     if (bit_len > 32) {
 189       stream_skip(s, bit_len / 32 * 4 - 4);
 190       get_bits(priv, s, 32); // make sure dword is reloaded
 191     }
 192     get_bits(priv, s, bit_len % 32);
 193     priv->last_pts += priv->pts_per_packet;
 194     if (s->eof) break;
 195   }
 196   if (!sh_audio) return;
 197 }
 198
 199 static void demux_close_mpc(demuxer_t* demuxer) {
 200   da_priv_t* priv = demuxer->priv;
 201
 202   free(priv);
 203 }
 204
 205 static int demux_mpc_control(demuxer_t *demuxer,int cmd, void *arg){
 206   da_priv_t* priv = demuxer->priv;
 207   switch (cmd) {
 208     case DEMUXER_CTRL_GET_TIME_LENGTH:
 209       if (priv->length < 1) return DEMUXER_CTRL_DONTKNOW;
 210       *((double *)arg) = priv->length;
 211       return DEMUXER_CTRL_OK;
 212     case DEMUXER_CTRL_GET_PERCENT_POS:
 213       if (priv->length < 1) return DEMUXER_CTRL_DONTKNOW;
 214       *((int *)arg) = priv->last_pts * 100 / priv->length;
 215       return DEMUXER_CTRL_OK;
 216   }
 217   return DEMUXER_CTRL_NOTIMPL;
 218 }
 219
 220
 221 const demuxer_desc_t demuxer_desc_mpc = {
 222   "Musepack demuxer",
 223   "mpc",
 224   "MPC",
 225   "Reza Jelveh, Reimar Doeffinger",
 226   "supports v7 bitstream only",
 227   DEMUXER_TYPE_MPC,
 228   0, // unsafe autodetect
 229   demux_mpc_check,
 230   demux_mpc_fill_buffer,
 231   demux_mpc_open,
 232   demux_close_mpc,
 233   demux_mpc_seek,
 234   demux_mpc_control
 235 };