8325
|
1 /********************************************************
|
|
2
|
|
3 DirectShow audio decoder
|
|
4 Copyright 2001 Eugene Kuznetsov (divx@euro.ru)
|
|
5
|
|
6 *********************************************************/
|
|
7
|
|
8 #ifndef NOAVIFILE_HEADERS
|
|
9 #include "audiodecoder.h"
|
|
10 #include "except.h"
|
|
11 #else
|
|
12 #include "libwin32.h"
|
|
13 #endif
|
|
14
|
|
15 #include "DMO_Filter.h"
|
|
16 #include "DMO_AudioDecoder.h"
|
|
17
|
|
18 struct _DMO_AudioDecoder
|
|
19 {
|
|
20 DMO_MEDIA_TYPE m_sOurType, m_sDestType;
|
|
21 DMO_Filter* m_pDMO_Filter;
|
|
22 char* m_sVhdr;
|
|
23 char* m_sVhdr2;
|
|
24 int m_iFlushed;
|
|
25 };
|
|
26
|
|
27 #include "DMO_AudioDecoder.h"
|
|
28
|
|
29 #include <string.h>
|
|
30 #include <stdio.h>
|
|
31 #include <stdlib.h>
|
|
32
|
|
33 #define __MODULE__ "DirectShow audio decoder"
|
|
34
|
|
35 typedef long STDCALL (*GETCLASS) (GUID*, GUID*, void**);
|
|
36
|
|
37 DMO_AudioDecoder * DMO_AudioDecoder_Open(char* dllname, GUID* guid, WAVEFORMATEX* wf,int out_channels)
|
|
38 //DMO_AudioDecoder * DMO_AudioDecoder_Create(const CodecInfo * info, const WAVEFORMATEX* wf)
|
|
39 {
|
|
40 DMO_AudioDecoder *this;
|
|
41 int sz;
|
|
42 WAVEFORMATEX* pWF;
|
|
43
|
|
44 Setup_LDT_Keeper();
|
|
45 Setup_FS_Segment();
|
|
46
|
|
47 this = malloc(sizeof(DMO_AudioDecoder));
|
|
48
|
|
49 this->m_iFlushed=1;
|
|
50
|
|
51 sz = 18 + wf->cbSize;
|
|
52 this->m_sVhdr = malloc(sz);
|
|
53 memcpy(this->m_sVhdr, wf, sz);
|
|
54 this->m_sVhdr2 = malloc(18);
|
|
55 memcpy(this->m_sVhdr2, this->m_sVhdr, 18);
|
|
56
|
|
57 pWF = (WAVEFORMATEX*)this->m_sVhdr2;
|
|
58 pWF->wFormatTag = 1;
|
|
59 pWF->wBitsPerSample = 16;
|
|
60 pWF->nChannels = out_channels;
|
|
61 pWF->nBlockAlign = 2*pWF->nChannels; //pWF->nChannels * (pWF->wBitsPerSample + 7) / 8;
|
|
62 pWF->nAvgBytesPerSec = pWF->nBlockAlign * pWF->nSamplesPerSec;
|
|
63 pWF->cbSize = 0;
|
|
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.cbFormat=sz;
|
|
74 this->m_sOurType.pbFormat=this->m_sVhdr;
|
|
75
|
|
76 memset(&this->m_sDestType, 0, sizeof(this->m_sDestType));
|
|
77 this->m_sDestType.majortype=MEDIATYPE_Audio;
|
|
78 this->m_sDestType.subtype=MEDIASUBTYPE_PCM;
|
|
79 this->m_sDestType.formattype=FORMAT_WaveFormatEx;
|
|
80 this->m_sDestType.bFixedSizeSamples=1;
|
|
81 this->m_sDestType.bTemporalCompression=0;
|
|
82 this->m_sDestType.lSampleSize=pWF->nBlockAlign;
|
|
83 this->m_sDestType.cbFormat=18; //pWF->cbSize;
|
|
84 this->m_sDestType.pbFormat=this->m_sVhdr2;
|
|
85
|
|
86 print_wave_header(this->m_sVhdr);
|
|
87 print_wave_header(this->m_sVhdr2);
|
|
88
|
|
89 this->m_pDMO_Filter = DMO_FilterCreate(dllname, guid, &this->m_sOurType, &this->m_sDestType);
|
|
90 if( !this->m_pDMO_Filter ) {
|
|
91 free(this);
|
|
92 return NULL;
|
|
93 }
|
|
94
|
|
95 return this;
|
|
96 }
|
|
97
|
|
98 void DMO_AudioDecoder_Destroy(DMO_AudioDecoder *this)
|
|
99 {
|
|
100 free(this->m_sVhdr);
|
|
101 free(this->m_sVhdr2);
|
|
102 DMO_Filter_Destroy(this->m_pDMO_Filter);
|
|
103 free(this);
|
|
104 }
|
|
105
|
|
106 int DMO_AudioDecoder_Convert(DMO_AudioDecoder *this, const void* in_data, unsigned int in_size,
|
|
107 void* out_data, unsigned int out_size,
|
|
108 unsigned int* size_read, unsigned int* size_written)
|
|
109 {
|
|
110 DMO_OUTPUT_DATA_BUFFER db;
|
|
111 CMediaBuffer* bufferin;
|
|
112 unsigned int written = 0;
|
|
113 unsigned int read = 0;
|
|
114 int r = 0;
|
|
115
|
|
116 if (!in_data || !out_data)
|
|
117 return -1;
|
|
118
|
|
119 Setup_FS_Segment();
|
|
120
|
|
121 //m_pDMO_Filter->m_pMedia->vt->Lock(m_pDMO_Filter->m_pMedia, 1);
|
|
122 bufferin = CMediaBufferCreate(in_size, (void*)in_data, in_size, 1);
|
|
123 r = this->m_pDMO_Filter->m_pMedia->vt->ProcessInput(this->m_pDMO_Filter->m_pMedia, 0,
|
|
124 (IMediaBuffer*)bufferin,
|
|
125 (this->m_iFlushed) ? DMO_INPUT_DATA_BUFFERF_SYNCPOINT : 0,
|
|
126 0, 0);
|
|
127 if (r == 0){
|
|
128 ((IMediaBuffer*)bufferin)->vt->GetBufferAndLength((IMediaBuffer*)bufferin, 0, &read);
|
|
129 this->m_iFlushed = 0;
|
|
130 }
|
|
131
|
|
132 ((IMediaBuffer*)bufferin)->vt->Release((IUnknown*)bufferin);
|
|
133
|
|
134 //printf("RESULTA: %d 0x%x %ld %d %d\n", r, r, read, m_iFlushed, out_size);
|
|
135 if (r == 0 || (unsigned)r == DMO_E_NOTACCEPTING){
|
|
136 unsigned long status = 0;
|
|
137 /* something for process */
|
|
138 db.rtTimestamp = 0;
|
|
139 db.rtTimelength = 0;
|
|
140 db.dwStatus = 0;
|
|
141 db.pBuffer = (IMediaBuffer*) CMediaBufferCreate(out_size, out_data, 0, 0);
|
|
142 //printf("OUTSIZE %d\n", out_size);
|
|
143 r = this->m_pDMO_Filter->m_pMedia->vt->ProcessOutput(this->m_pDMO_Filter->m_pMedia,
|
|
144 0, 1, &db, &status);
|
|
145
|
|
146 ((IMediaBuffer*)db.pBuffer)->vt->GetBufferAndLength((IMediaBuffer*)db.pBuffer, 0, &written);
|
|
147 ((IMediaBuffer*)db.pBuffer)->vt->Release((IUnknown*)db.pBuffer);
|
|
148
|
|
149 //printf("RESULTB: %d 0x%x %ld\n", r, r, written);
|
|
150 //printf("Converted %d -> %d\n", in_size, out_size);
|
|
151 }
|
|
152 else if (in_size > 0)
|
|
153 printf("ProcessInputError r:0x%x=%d\n", r, r);
|
|
154
|
|
155 if (size_read)
|
|
156 *size_read = read;
|
|
157 if (size_written)
|
|
158 *size_written = written;
|
|
159 return r;
|
|
160 }
|
|
161
|
|
162 int DMO_AudioDecoder_GetSrcSize(DMO_AudioDecoder *this, int dest_size)
|
|
163 {
|
|
164 // unsigned long inputs, outputs;
|
|
165 // Setup_FS_Segment();
|
|
166 // this->m_pDMO_Filter->m_pMedia->vt->GetOutputSizeInfo(this->m_pDMO_Filter->m_pMedia, 0, &inputs, &outputs);
|
|
167 return ((WAVEFORMATEX*)this->m_sVhdr)->nBlockAlign*4;
|
|
168 }
|