loader/dshow/DS_AudioDecoder.c

   1 /********************************************************
   2
   3          DirectShow audio decoder
   4          Copyright 2001 Eugene Kuznetsov  (divx@euro.ru)
   5
   6 *********************************************************/
   7 #include "config.h"
   8
   9 #include "libwin32.h"
  10
  11 #include "DS_Filter.h"
  12
  13 struct DS_AudioDecoder
  14 {
  15     WAVEFORMATEX in_fmt;
  16     AM_MEDIA_TYPE m_sOurType, m_sDestType;
  17     DS_Filter* m_pDS_Filter;
  18     char* m_sVhdr;
  19     char* m_sVhdr2;
  20 };
  21
  22 #include "DS_AudioDecoder.h"
  23 #ifdef WIN32_LOADER
  24 #include "loader/ldt_keeper.h"
  25 #endif
  26
  27 #include <string.h>
  28 #include <stdio.h>
  29 #include <stdlib.h>
  30
  31 typedef long STDCALL (*GETCLASS) (GUID*, GUID*, void**);
  32
  33 static SampleProcUserData sampleProcData;
  34
  35
  36 DS_AudioDecoder * DS_AudioDecoder_Open(char* dllname, GUID* guid, WAVEFORMATEX* wf)
  37 //DS_AudioDecoder * DS_AudioDecoder_Create(const CodecInfo * info, const WAVEFORMATEX* wf)
  38 {
  39     DS_AudioDecoder *this;
  40     int sz;
  41     WAVEFORMATEX* pWF;
  42
  43 #ifdef WIN32_LOADER
  44     Setup_LDT_Keeper();
  45     Setup_FS_Segment();
  46 #endif
  47
  48     this = malloc(sizeof(DS_AudioDecoder));
  49
  50     sz = 18 + wf->cbSize;
  51     this->m_sVhdr = malloc(sz);
  52     memcpy(this->m_sVhdr, wf, sz);
  53     this->m_sVhdr2 = malloc(18);
  54     memcpy(this->m_sVhdr2, this->m_sVhdr, 18);
  55
  56     pWF = (WAVEFORMATEX*)this->m_sVhdr2;
  57     pWF->wFormatTag = 1;
  58     pWF->wBitsPerSample = 16;
  59     pWF->nBlockAlign = pWF->nChannels * (pWF->wBitsPerSample + 7) / 8;
  60     pWF->cbSize = 0;
  61     pWF->nAvgBytesPerSec = pWF->nBlockAlign * pWF->nSamplesPerSec;
  62
  63     memcpy(&this->in_fmt,wf,sizeof(WAVEFORMATEX));
  64
  65     memset(&this->m_sOurType, 0, sizeof(this->m_sOurType));
  66     this->m_sOurType.majortype=MEDIATYPE_Audio;
  67     this->m_sOurType.subtype=MEDIASUBTYPE_PCM;
  68     this->m_sOurType.subtype.f1=wf->wFormatTag;
  69     this->m_sOurType.formattype=FORMAT_WaveFormatEx;
  70     this->m_sOurType.lSampleSize=wf->nBlockAlign;
  71     this->m_sOurType.bFixedSizeSamples=1;
  72     this->m_sOurType.bTemporalCompression=0;
  73     this->m_sOurType.pUnk=0;
  74     this->m_sOurType.cbFormat=sz;
  75     this->m_sOurType.pbFormat=this->m_sVhdr;
  76
  77     memset(&this->m_sDestType, 0, sizeof(this->m_sDestType));
  78     this->m_sDestType.majortype=MEDIATYPE_Audio;
  79     this->m_sDestType.subtype=MEDIASUBTYPE_PCM;
  80 //    this->m_sDestType.subtype.f1=pWF->wFormatTag;
  81     this->m_sDestType.formattype=FORMAT_WaveFormatEx;
  82     this->m_sDestType.bFixedSizeSamples=1;
  83     this->m_sDestType.bTemporalCompression=0;
  84     this->m_sDestType.lSampleSize=pWF->nBlockAlign;
  85     if (wf->wFormatTag == 0x130)
  86         // ACEL hack to prevent memory corruption
  87         // obviosly we are missing something here
  88         this->m_sDestType.lSampleSize *= 288;
  89     this->m_sDestType.pUnk=0;
  90     this->m_sDestType.cbFormat=18; //pWF->cbSize;
  91     this->m_sDestType.pbFormat=this->m_sVhdr2;
  92
  93 //print_wave_header(this->m_sVhdr,  MSGL_V);
  94 //print_wave_header(this->m_sVhdr2, MSGL_V);
  95
  96     /*try*/
  97     {
  98         this->m_pDS_Filter = DS_FilterCreate(dllname, guid, &this->m_sOurType, &this->m_sDestType,&sampleProcData);
  99         if( !this->m_pDS_Filter ) {
 100            free(this);
 101            return NULL;
 102         }
 103
 104         //Commit should be done before binary codec start
 105         this->m_pDS_Filter->m_pAll->vt->Commit(this->m_pDS_Filter->m_pAll);
 106
 107         this->m_pDS_Filter->Start(this->m_pDS_Filter);
 108
 109     }
 110     /*
 111     catch (FatalError& e)
 112     {
 113         e.PrintAll();
 114         delete[] m_sVhdr;
 115         delete[] m_sVhdr2;
 116         delete m_pDS_Filter;
 117         throw;
 118     }
 119     */
 120     return this;
 121 }
 122
 123 void DS_AudioDecoder_Destroy(DS_AudioDecoder *this)
 124 {
 125     free(this->m_sVhdr);
 126     free(this->m_sVhdr2);
 127     DS_Filter_Destroy(this->m_pDS_Filter);
 128     free(this);
 129 }
 130
 131 int DS_AudioDecoder_Convert(DS_AudioDecoder *this, const void* in_data, unsigned int in_size,
 132                              void* out_data, unsigned int out_size,
 133                              unsigned int* size_read, unsigned int* size_written)
 134 {
 135     unsigned int written = 0;
 136     unsigned int read = 0;
 137
 138     if (!in_data || !out_data)
 139         return -1;
 140
 141 #ifdef WIN32_LOADER
 142     Setup_FS_Segment();
 143 #endif
 144
 145     in_size -= in_size%this->in_fmt.nBlockAlign;
 146     while (in_size>0)
 147     {
 148         IMediaSample* sample=0;
 149         char* ptr;
 150         int result;
 151         this->m_pDS_Filter->m_pAll->vt->GetBuffer(this->m_pDS_Filter->m_pAll, &sample, 0, 0, 0);
 152         if (!sample)
 153         {
 154             Debug printf("DS_AudioDecoder::Convert() Error: null sample\n");
 155             break;
 156         }
 157         sample->vt->SetActualDataLength(sample, this->in_fmt.nBlockAlign);
 158         sample->vt->GetPointer(sample, (BYTE **)&ptr);
 159         memcpy(ptr, (const uint8_t*)in_data + read, this->in_fmt.nBlockAlign);
 160         sample->vt->SetSyncPoint(sample, 1);
 161         sample->vt->SetPreroll(sample, 0);
 162         result = this->m_pDS_Filter->m_pImp->vt->Receive(this->m_pDS_Filter->m_pImp, sample);
 163         if (result)
 164             Debug printf("DS_AudioDecoder::Convert() Error: putting data into input pin %x\n", result);
 165         if ((written + sampleProcData.frame_size) > out_size)
 166         {
 167             sample->vt->Release((IUnknown*)sample);
 168             break;
 169         }
 170         memcpy((uint8_t*)out_data + written, sampleProcData.frame_pointer, sampleProcData.frame_size);
 171         sample->vt->Release((IUnknown*)sample);
 172         read+=this->in_fmt.nBlockAlign;
 173         written+=sampleProcData.frame_size;
 174         break;
 175     }
 176     if (size_read)
 177         *size_read = read;
 178     if (size_written)
 179         *size_written = written;
 180     return 0;
 181 }
 182
 183 int DS_AudioDecoder_GetSrcSize(DS_AudioDecoder *this, int dest_size)
 184 {
 185     double efficiency =(double) this->in_fmt.nAvgBytesPerSec
 186         / (this->in_fmt.nSamplesPerSec*this->in_fmt.nBlockAlign);
 187     int frames = (int)(dest_size*efficiency);
 188
 189     if (frames < 1)
 190         frames = 1;
 191     return frames * this->in_fmt.nBlockAlign;
 192 }