AMRWB_E_SAMPLE.c revision 956c553ab0ce72f8074ad0fda2ffd66a0305700c 
1/*
2 ** Copyright 2003-2010, VisualOn, Inc.
3 **
4 ** Licensed under the Apache License, Version 2.0 (the "License");
5 ** you may not use this file except in compliance with the License.
6 ** You may obtain a copy of the License at
7 **
8 **     http://www.apache.org/licenses/LICENSE-2.0
9 **
10 ** Unless required by applicable law or agreed to in writing, software
11 ** distributed under the License is distributed on an "AS IS" BASIS,
12 ** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 ** See the License for the specific language governing permissions and
14 ** limitations under the License.
15 */
16
17
18#ifdef LINUX
19#include <dlfcn.h>
20#endif
21
22#include      <stdio.h>
23#include      <stdlib.h>
24#include      <time.h>
25#include      "voAMRWB.h"
26#include      "cmnMemory.h"
27
28#define VOAMRWB_RFC3267_HEADER_INFO "#!AMR-WB\n"
29
30#define  INPUT_SIZE   640
31#define  OUTPUT_SIZE  1024
32unsigned char  InputBuf[INPUT_SIZE];
33unsigned char  OutputBuf[OUTPUT_SIZE];
34
35void usage (void) {
36	printf ("AMR_WB Encoder HELP   Displays this text\n");
37	printf ("\n");
38	printf ("Usage:\n");
39	printf ("AMRWBEnc [options] Input_file output_file \n");
40	printf ("\n");
41	printf ("Options +M* +F* +DTX \n");
42	printf ("Support \n");
43	printf ("Options +M* for seting compression bitrate mode, default is 23.85kbps\n");
44	printf (" +M0 = 6.6kbps \n");
45	printf (" +M1 = 8.85kbps \n");
46	printf (" +M2 = 12.65kbps \n");
47	printf (" +M3 = 14.25kbps \n");
48	printf (" +M4 = 15.58kbps \n");
49	printf (" +M5 = 18.25kbps \n");
50	printf (" +M6 = 19.85kbps \n");
51	printf (" +M7 = 23.05kbps \n");
52	printf (" +M8 = 23.85kbps \n");
53	printf ("\n");
54	printf ("Options +F* for setting output frame Type, default is RFC3267 \n");
55	printf ("+F0 for AMR_WB Defualt bit extern short data frame type \n");
56	printf ("+F1 for AMR_WB_ITU bit extern short data frame type \n");
57	printf ("+F2 for RFC3267\n ");
58	printf ("\n");
59	printf ("Options +DTX enable DTX mode, default is disable.\n");
60	printf ("File names, input raw PCM data, and output is AMR_WB bit-stream file.\n");
61	printf ("\n");
62}
63
64int  GetNextBuf(FILE* inFile,unsigned char* dst,int size)
65{
66	int size2 = (int)fread(dst, sizeof(signed char), size,inFile);
67	return size2;
68}
69
70typedef int (VO_API * VOGETAUDIOENCAPI) (VO_AUDIO_CODECAPI * pEncHandle);
71
72int encode(
73		   int mode,
74		   short   allow_dtx,
75		   VOAMRWBFRAMETYPE frameType,
76		   const char* srcfile,
77		   const char* dstfile
78		   )
79{
80	int			ret = 0;
81	int         returnCode;
82	FILE		*fsrc = NULL;
83	FILE		*fdst = NULL;
84	int         framenum = 0;
85	int         eofFile = 0;
86	int         size1 = 0;
87	int         Relens;
88
89	VO_AUDIO_CODECAPI       AudioAPI;
90	VO_MEM_OPERATOR         moper;
91	VO_CODEC_INIT_USERDATA  useData;
92	VO_HANDLE               hCodec;
93	VO_CODECBUFFER          inData;
94	VO_CODECBUFFER          outData;
95	VO_AUDIO_OUTPUTINFO     outFormat;
96
97	unsigned char *inBuf = InputBuf;
98	unsigned char *outBuf = OutputBuf;
99
100
101#ifdef LINUX
102	void  *handle = NULL;
103	void  *pfunc;
104	VOGETAUDIOENCAPI pGetAPI;
105#endif
106
107	clock_t   start, finish;
108	double    duration = 0.0;
109
110	if ((fsrc = fopen (srcfile, "rb")) == NULL)
111	{
112		ret = -1;
113		goto safe_exit;
114	}
115
116	if ((fdst = fopen (dstfile, "wb")) == NULL)
117	{
118		ret = -1;
119		goto safe_exit;
120	}
121
122	moper.Alloc = cmnMemAlloc;
123	moper.Copy = cmnMemCopy;
124	moper.Free = cmnMemFree;
125	moper.Set = cmnMemSet;
126	moper.Check = cmnMemCheck;
127
128	useData.memflag = VO_IMF_USERMEMOPERATOR;
129	useData.memData = (VO_PTR)(&moper);
130
131#ifdef LINUX
132	handle = dlopen("/data/local/tmp/voAMRWBEnc.so", RTLD_NOW);
133	if(handle == 0)
134	{
135		printf("open dll error......");
136		return -1;
137	}
138
139	pfunc = dlsym(handle, "voGetAMRWBEncAPI");
140	if(pfunc == 0)
141	{
142		printf("open function error......");
143		return -1;
144	}
145
146	pGetAPI = (VOGETAUDIOENCAPI)pfunc;
147
148	returnCode  = pGetAPI(&AudioAPI);
149	if(returnCode)
150	{
151		printf("get APIs error......");
152		return -1;
153	}
154#else
155	ret = voGetAMRWBEncAPI(&AudioAPI);
156	if(ret)
157	{
158		ret = -1;
159		printf("get APIs error......");
160		goto safe_exit;
161	}
162#endif
163
164	//#######################################   Init Encoding Section   #########################################
165	ret = AudioAPI.Init(&hCodec, VO_AUDIO_CodingAMRWB, &useData);
166
167	if(ret)
168	{
169		ret = -1;
170		printf("APIs init error......");
171		goto safe_exit;
172	}
173
174	Relens = GetNextBuf(fsrc,InputBuf,INPUT_SIZE);
175	if(Relens!=INPUT_SIZE && !feof(fsrc))
176	{
177		ret = -1; //Invalid magic number
178		printf("get next buffer error......");
179		goto safe_exit;
180	}
181
182	//###################################### set encode Mode ##################################################
183	ret = AudioAPI.SetParam(hCodec, VO_PID_AMRWB_FRAMETYPE, &frameType);
184	ret = AudioAPI.SetParam(hCodec, VO_PID_AMRWB_MODE, &mode);
185	ret = AudioAPI.SetParam(hCodec, VO_PID_AMRWB_DTX, &allow_dtx);
186
187	if(frameType == VOAMRWB_RFC3267)
188	{
189		/* write RFC3267 Header info to indicate single channel AMR file storage format */
190		size1 = (int)strlen(VOAMRWB_RFC3267_HEADER_INFO);
191		memcpy(outBuf, VOAMRWB_RFC3267_HEADER_INFO, size1);
192		outBuf += size1;
193	}
194
195	//#######################################   Encoding Section   #########################################
196	printf(" \n ---------------- Running -------------------------\n ");
197
198	do{
199		inData.Buffer = (unsigned char *)inBuf;
200		inData.Length = Relens;
201		outData.Buffer = outBuf;
202
203		start = clock();
204
205		/* decode one amr block */
206		returnCode = AudioAPI.SetInputData(hCodec,&inData);
207
208		do {
209			returnCode = AudioAPI.GetOutputData(hCodec,&outData, &outFormat);
210			if(returnCode == 0)
211			{
212				framenum++;
213				printf(" Frames processed: %hd\r", framenum);
214				if(framenum == 1)
215				{
216					fwrite(OutputBuf, 1, outData.Length + size1, fdst);
217					fflush(fdst);
218				}
219				else
220				{
221					fwrite(outData.Buffer, 1, outData.Length, fdst);
222					fflush(fdst);
223				}
224			}
225			else if(returnCode == VO_ERR_LICENSE_ERROR)
226			{
227		        printf("Encoder time reach upper limit......");
228		        goto safe_exit;
229			}
230		} while(returnCode != VO_ERR_INPUT_BUFFER_SMALL);
231
232		finish = clock();
233		duration += finish - start;
234
235		if (!eofFile) {
236			Relens = GetNextBuf(fsrc, InputBuf, INPUT_SIZE);
237			inBuf = InputBuf;
238			if (feof(fsrc) && Relens == 0)
239				eofFile = 1;
240		}
241	} while (!eofFile && returnCode);
242	//#######################################   End Encoding Section   #########################################
243
244safe_exit:
245	returnCode = AudioAPI.Uninit(hCodec);
246
247	printf( "\n%2.5f seconds\n", (double)duration/CLOCKS_PER_SEC);
248
249	if (fsrc)
250		fclose(fsrc);
251	if (fdst)
252		fclose(fdst);
253
254#ifdef LINUX
255	dlclose(handle);
256#endif
257
258	return ret;
259}
260
261int main(int argc, char **argv)  // for gcc compiler;
262{
263	int     mode, r;
264	int     arg, filename=0;
265	char    *inFileName = NULL;
266	char    *outFileName = NULL;
267	short   allow_dtx;
268	VOAMRWBFRAMETYPE frameType;
269
270	printf("\n");
271	printf("************************Adaptive Multi-Rate Wide Band Encoder (AMR-WB)*******************************\n");
272	printf("***********************************DEFINITIONS:*******************************************************\n");
273	printf("AMR-WB encoder scheme is based on the principle of Algebraic Code Excited Linear Prediction algorithm\n");
274	printf("The AMR-WB encoder compression MONO liner PCM speech input data at 16kHz sampling rate\n");
275	printf("to one of nine data rate modes-6.60, 8.85, 12.65, 14.25, 15.85, 18.25, 19.25, 23.05 and 23.85kbps.\n");
276	printf("The encoder supports output format AMRWB ITU, AMRWB RFC3267.\n");
277	printf("\n");
278
279	/*Encoder Default setting */
280	mode = VOAMRWB_MD2385;
281	allow_dtx = 0;
282	frameType = VOAMRWB_RFC3267;
283
284	if(argc < 3){
285		usage();
286		return 0;
287	}else{
288		for (arg = 1; arg < argc; arg++) {
289			if (argv [arg] [0] == '+') {
290				if(argv[arg][1] == 'M')
291				{
292					switch(argv[arg][2])
293					{
294					case '0': mode = VOAMRWB_MD66;
295						break;
296					case '1': mode = VOAMRWB_MD885;
297						break;
298					case '2': mode = VOAMRWB_MD1265;
299						break;
300					case '3': mode = VOAMRWB_MD1425;
301						break;
302					case '4': mode = VOAMRWB_MD1585;
303						break;
304					case '5': mode = VOAMRWB_MD1825;
305						break;
306					case '6': mode = VOAMRWB_MD1985;
307						break;
308					case '7': mode = VOAMRWB_MD2305;
309						break;
310					case '8': mode = VOAMRWB_MD2385;
311						break;
312					default:
313						usage();
314						printf ("Invalid parameter '%s'.\n", argv [arg]);
315						break;
316					}
317				}else if(argv[arg][1] == 'F')
318				{
319					switch(argv[arg][2])
320					{
321					case '0': frameType = VOAMRWB_DEFAULT;
322						break;
323					case '1': frameType = VOAMRWB_ITU;
324						break;
325					case '2': frameType = VOAMRWB_RFC3267 ;
326						break;
327					default:
328						usage();
329						printf ("Invalid parameter '%s'.\n", argv [arg]);
330						break;
331
332
333					}
334				}else if(strcmp (argv[arg], "+DTX") == 0)
335				{
336					allow_dtx = 1;
337				}
338
339			} else {
340				switch (filename) {
341						case 0:
342							inFileName  = argv[arg];
343							break;
344						case 1:
345							outFileName = argv[arg];
346							break;
347						default:
348							usage ();
349							fprintf (stderr, "Invalid parameter '%s'.\n", argv [arg]);
350							return 0;
351				}
352				filename++;
353			}
354		}
355	}
356
357	r = encode(mode, allow_dtx, frameType, inFileName, outFileName);
358	if(r)
359	{
360		fprintf(stderr, "error: %d\n", r);
361	}
362	return r;
363}
364
365