view libmpcodecs/ad_pcm.c @ 21548:bf65ffcf0cdb

Set AVFMT_FLAG_GENPTS if -correct-pts is used. This should allow using -correct-pts (and thus filters which adjust pts or add frames) with dvd or other mpeg container files by specifying "-correct-pts -demuxer lavf -vc ffmpeg12". Might work with libmpeg2 decoder too but certainly not with internal demuxer. Using this flag isn't quite optimal as it can cause extra buffering of demuxed frames, but at least it's better than just failing until a more complex solution is implemented.

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include "config.h"
#include "ad_internal.h"
#include "libaf/af_format.h"

static ad_info_t info = 
{
	"Uncompressed PCM audio decoder",
	"pcm",
	"Nick Kurshev",
	"A'rpi",
	""
};

LIBAD_EXTERN(pcm)

static int init(sh_audio_t *sh_audio)
{
  WAVEFORMATEX *h=sh_audio->wf;
  sh_audio->i_bps=h->nAvgBytesPerSec;
  sh_audio->channels=h->nChannels;
  sh_audio->samplerate=h->nSamplesPerSec;
  sh_audio->samplesize=(h->wBitsPerSample+7)/8;
  sh_audio->sample_format=AF_FORMAT_S16_LE; // default
  switch(sh_audio->format){ /* hardware formats: */
    case 0x0:
    case 0x1: // Microsoft PCM
    case 0xfffe: // Extended
       switch (sh_audio->samplesize) {
         case 1: sh_audio->sample_format=AF_FORMAT_U8; break;
         case 2: sh_audio->sample_format=AF_FORMAT_S16_LE; break;
         case 3: sh_audio->sample_format=AF_FORMAT_S24_LE; break;
         case 4: sh_audio->sample_format=AF_FORMAT_S32_LE; break;
       }
       break;
    case 0x3: // IEEE float
       sh_audio->sample_format=AF_FORMAT_FLOAT_LE;
       break;
    case 0x6:  sh_audio->sample_format=AF_FORMAT_A_LAW;break;
    case 0x7:  sh_audio->sample_format=AF_FORMAT_MU_LAW;break;
    case 0x11: sh_audio->sample_format=AF_FORMAT_IMA_ADPCM;break;
    case 0x50: sh_audio->sample_format=AF_FORMAT_MPEG2;break;
/*    case 0x2000: sh_audio->sample_format=AFMT_AC3; */
    case 0x20776172: // 'raw '
       sh_audio->sample_format=AF_FORMAT_S16_BE;
       if(sh_audio->samplesize==1) sh_audio->sample_format=AF_FORMAT_U8;
       break;
    case 0x736F7774: // 'twos'
       sh_audio->sample_format=AF_FORMAT_S16_BE;
       // intended fall-through
    case 0x74776F73: // 'sowt'
       if(sh_audio->samplesize==1) sh_audio->sample_format=AF_FORMAT_S8;
       break;
    case 0x32336c66: // 'fl32', bigendian float32
       sh_audio->sample_format=AF_FORMAT_FLOAT_BE;
       sh_audio->samplesize=4;
       break;
    case 0x666c3332: // '23lf', little endian float32, MPlayer internal fourCC
       sh_audio->sample_format=AF_FORMAT_FLOAT_LE;
       sh_audio->samplesize=4;
       break;
/*    case 0x34366c66: // 'fl64', bigendian float64
       sh_audio->sample_format=AF_FORMAT_FLOAT_BE;
       sh_audio->samplesize=8;
       break;
    case 0x666c3634: // '46lf', little endian float64, MPlayer internal fourCC
       sh_audio->sample_format=AF_FORMAT_FLOAT_LE;
       sh_audio->samplesize=8;
       break;*/
    case 0x34326e69: // 'in24', bigendian int24
       sh_audio->sample_format=AF_FORMAT_S24_BE;
       sh_audio->samplesize=3;
       break;
    case 0x696e3234: // '42ni', little endian int24, MPlayer internal fourCC
       sh_audio->sample_format=AF_FORMAT_S24_LE;
       sh_audio->samplesize=3;
       break;
    case 0x32336e69: // 'in32', bigendian int32
       sh_audio->sample_format=AF_FORMAT_S32_BE;
       sh_audio->samplesize=4;
       break;
    case 0x696e3332: // '23ni', little endian int32, MPlayer internal fourCC
       sh_audio->sample_format=AF_FORMAT_S32_LE;
       sh_audio->samplesize=4;
       break;
    default: if(sh_audio->samplesize!=2) sh_audio->sample_format=AF_FORMAT_U8;
  }
  if (!sh_audio->samplesize) // this would cause MPlayer to hang later
    sh_audio->samplesize = 2;
  return 1;
}

static int preinit(sh_audio_t *sh)
{
  sh->audio_out_minsize=2048;
  return 1;
}

static void uninit(sh_audio_t *sh)
{
}

static int control(sh_audio_t *sh,int cmd,void* arg, ...)
{
  int skip;
    switch(cmd)
    {
      case ADCTRL_SKIP_FRAME:
	skip=sh->i_bps/16;
	skip=skip&(~3);
	demux_read_data(sh->ds,NULL,skip);
	return CONTROL_TRUE;
    }
  return CONTROL_UNKNOWN;
}

static int decode_audio(sh_audio_t *sh_audio,unsigned char *buf,int minlen,int maxlen)
{
  unsigned len = sh_audio->channels*sh_audio->samplesize;
  len = (minlen + len - 1) / len * len;
  if (len > maxlen)
      // if someone needs hundreds of channels adjust audio_out_minsize
      // based on channels in preinit()
      return -1;
  len=demux_read_data(sh_audio->ds,buf,len);
  return len;
}