blob: 68f846c593c1d9fbadc8b2f3c3ac9a2a8b500aaf [file] [log] [blame]
Nanang Izzuddin57b88572009-04-01 12:05:34 +00001/* $Id$ */
2/*
3 * Copyright (C) 2008-2009 Teluu Inc. (http://www.teluu.com)
4 * Copyright (C) 2003-2008 Benny Prijono <benny@prijono.org>
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 */
20#include <pjmedia-codec/g7221.h>
21#include <pjmedia/codec.h>
22#include <pjmedia/errno.h>
23#include <pjmedia/endpoint.h>
24#include <pjmedia/port.h>
25#include <pjmedia/silencedet.h>
26#include <pj/assert.h>
27#include <pj/log.h>
28#include <pj/math.h>
29#include <pj/pool.h>
30#include <pj/string.h>
31#include <pj/os.h>
32
33/*
34 * Only build this file if PJMEDIA_HAS_G7221_CODEC != 0
35 */
36#if defined(PJMEDIA_HAS_G7221_CODEC) && PJMEDIA_HAS_G7221_CODEC!=0
37
Nanang Izzuddin56e380a2009-04-15 14:45:41 +000038#include "../../../third_party/g7221/common/defs.h"
Nanang Izzuddin57b88572009-04-01 12:05:34 +000039
40#define THIS_FILE "g7221.c"
41
42/* Codec tag, it is the SDP encoding name and also MIME subtype name */
43#define CODEC_TAG "G7221"
44
45/* Sampling rates definition */
46#define WB_SAMPLE_RATE 16000
47#define UWB_SAMPLE_RATE 32000
48
49/* Maximum number of samples per frame. */
50#define MAX_SAMPLES_PER_FRAME (UWB_SAMPLE_RATE * 20 / 1000)
51
52/* Maximum number of codec params. */
53#define MAX_CODEC_MODES 8
54#define START_RSV_MODES_IDX 6
55
56
57/* Prototypes for G722.1 codec factory */
58static pj_status_t test_alloc( pjmedia_codec_factory *factory,
59 const pjmedia_codec_info *id );
60static pj_status_t default_attr( pjmedia_codec_factory *factory,
61 const pjmedia_codec_info *id,
62 pjmedia_codec_param *attr );
63static pj_status_t enum_codecs( pjmedia_codec_factory *factory,
64 unsigned *count,
65 pjmedia_codec_info codecs[]);
66static pj_status_t alloc_codec( pjmedia_codec_factory *factory,
67 const pjmedia_codec_info *id,
68 pjmedia_codec **p_codec);
69static pj_status_t dealloc_codec( pjmedia_codec_factory *factory,
70 pjmedia_codec *codec );
71
72/* Prototypes for G722.1 codec implementation. */
73static pj_status_t codec_init( pjmedia_codec *codec,
74 pj_pool_t *pool );
75static pj_status_t codec_open( pjmedia_codec *codec,
76 pjmedia_codec_param *attr );
77static pj_status_t codec_close( pjmedia_codec *codec );
78static pj_status_t codec_modify(pjmedia_codec *codec,
79 const pjmedia_codec_param *attr );
80static pj_status_t codec_parse( pjmedia_codec *codec,
81 void *pkt,
82 pj_size_t pkt_size,
83 const pj_timestamp *ts,
84 unsigned *frame_cnt,
85 pjmedia_frame frames[]);
86static pj_status_t codec_encode( pjmedia_codec *codec,
87 const struct pjmedia_frame *input,
88 unsigned output_buf_len,
89 struct pjmedia_frame *output);
90static pj_status_t codec_decode( pjmedia_codec *codec,
91 const struct pjmedia_frame *input,
92 unsigned output_buf_len,
93 struct pjmedia_frame *output);
94static pj_status_t codec_recover( pjmedia_codec *codec,
95 unsigned output_buf_len,
96 struct pjmedia_frame *output);
97
98/* Definition for G722.1 codec operations. */
99static pjmedia_codec_op codec_op =
100{
101 &codec_init,
102 &codec_open,
103 &codec_close,
104 &codec_modify,
105 &codec_parse,
106 &codec_encode,
107 &codec_decode,
108 &codec_recover
109};
110
111/* Definition for G722.1 codec factory operations. */
112static pjmedia_codec_factory_op codec_factory_op =
113{
114 &test_alloc,
115 &default_attr,
116 &enum_codecs,
117 &alloc_codec,
118 &dealloc_codec
119};
120
121
122/* Structure of G722.1 mode */
123typedef struct codec_mode
124{
125 pj_bool_t enabled; /* Is this mode enabled? */
126 pj_uint8_t pt; /* Payload type. */
127 unsigned sample_rate; /* Default sampling rate to be used.*/
128 unsigned bitrate; /* Bitrate. */
129 char bitrate_str[8]; /* Bitrate in string. */
130} codec_mode;
131
132
133/* G722.1 codec factory */
134static struct codec_factory {
135 pjmedia_codec_factory base; /**< Base class. */
136 pjmedia_endpt *endpt; /**< PJMEDIA endpoint instance. */
137 pj_pool_t *pool; /**< Codec factory pool. */
138 pj_mutex_t *mutex; /**< Codec factory mutex. */
139
140 unsigned mode_count; /**< Number of G722.1 modes. */
141 codec_mode modes[MAX_CODEC_MODES]; /**< The G722.1 modes. */
142 unsigned mode_rsv_start;/**< Start index of G722.1 non-
143 standard modes, currently
144 there can only be up to two
145 non-standard modes enabled
146 at the same time. */
147} codec_factory;
148
149/* G722.1 codec private data. */
150typedef struct codec_private {
151 pj_pool_t *pool; /**< Pool for each instance. */
152 pj_bool_t plc_enabled; /**< PLC enabled? */
153 pj_bool_t vad_enabled; /**< VAD enabled? */
154 pjmedia_silence_det *vad; /**< PJMEDIA VAD instance. */
155 pj_timestamp last_tx; /**< Timestamp of last transmit.*/
156
157 /* ITU ref implementation seems to leave the codec engine states to be
158 * managed by the application, so here we go.
159 */
160
161 /* Common engine state */
162 pj_uint16_t samples_per_frame; /**< Samples per frame. */
163 pj_uint16_t bitrate; /**< Coded stream bitrate. */
164 pj_uint16_t frame_size; /**< Coded frame size. */
165 pj_uint16_t frame_size_bits; /**< Coded frame size in bits. */
166 pj_uint16_t number_of_regions; /**< Number of regions. */
167
168 /* Encoder specific state */
169 Word16 *enc_old_frame;
170
171 /* Decoder specific state */
172 Word16 *dec_old_frame;
173 Rand_Obj dec_randobj;
174 Word16 dec_old_mag_shift;
175 Word16 *dec_old_mlt_coefs;
176} codec_private_t;
177
178/*
179 * Helper function for looking up mode based on payload type.
180 */
181static codec_mode* lookup_mode(unsigned pt)
182{
183 codec_mode* mode = NULL;
184 unsigned i;
185
186 for (i = 0; i < codec_factory.mode_count; ++i) {
187 mode = &codec_factory.modes[i];
188 if (mode->pt == pt)
189 break;
190 }
191
192 return mode;
193}
194
195/*
196 * Helper function to validate G722.1 mode. Valid modes are defined as:
197 * 1. sample rate must be 16kHz or 32kHz,
198 * 2. bitrate:
199 * - for sampling rate 16kHz: 16000 to 32000 bps, it must be a multiple
200 * of 400 (to keep RTP payload octed-aligned)
201 * - for sampling rate 32kHz: 24000 to 48000 bps, it must be a multiple
202 * of 400 (to keep RTP payload octed-aligned)
203 */
204static pj_bool_t validate_mode(unsigned sample_rate, unsigned bitrate)
205{
206 if (sample_rate == WB_SAMPLE_RATE) {
207 if (bitrate < 16000 || bitrate > 32000 ||
208 ((bitrate-16000) % 400 != 0))
209 {
210 return PJ_FALSE;
211 }
212 } else if (sample_rate == UWB_SAMPLE_RATE) {
213 if (bitrate < 24000 || bitrate > 48000 ||
214 ((bitrate-24000) % 400 != 0))
215 {
216 return PJ_FALSE;
217 }
218 } else {
219 return PJ_FALSE;
220 }
221
222 return PJ_TRUE;
223}
224
225/*
226 * Initialize and register G722.1 codec factory to pjmedia endpoint.
227 */
228PJ_DEF(pj_status_t) pjmedia_codec_g7221_init( pjmedia_endpt *endpt )
229{
230 pjmedia_codec_mgr *codec_mgr;
231 codec_mode *mode;
232 pj_status_t status;
233
234 if (codec_factory.pool != NULL) {
235 /* Already initialized. */
236 return PJ_SUCCESS;
237 }
238
239 /* Initialize codec modes, by default all standard bitrates are enabled */
240 mode = &codec_factory.modes[codec_factory.mode_count++];
241 mode->enabled = PJ_TRUE;
242 mode->pt = PJMEDIA_RTP_PT_G722_1_24;
243 mode->sample_rate = WB_SAMPLE_RATE;
244 mode->bitrate = 24000;
245 pj_utoa(mode->bitrate, mode->bitrate_str);
246
247 mode = &codec_factory.modes[codec_factory.mode_count++];
248 mode->enabled = PJ_TRUE;
249 mode->pt = PJMEDIA_RTP_PT_G722_1_32;
250 mode->sample_rate = WB_SAMPLE_RATE;
251 mode->bitrate = 32000;
252 pj_utoa(mode->bitrate, mode->bitrate_str);
253
254 mode = &codec_factory.modes[codec_factory.mode_count++];
255 mode->enabled = PJ_TRUE;
256 mode->pt = PJMEDIA_RTP_PT_G7221C_24;
257 mode->sample_rate = UWB_SAMPLE_RATE;
258 mode->bitrate = 24000;
259 pj_utoa(mode->bitrate, mode->bitrate_str);
260
261 mode = &codec_factory.modes[codec_factory.mode_count++];
262 mode->enabled = PJ_TRUE;
263 mode->pt = PJMEDIA_RTP_PT_G7221C_32;
264 mode->sample_rate = UWB_SAMPLE_RATE;
265 mode->bitrate = 32000;
266 pj_utoa(mode->bitrate, mode->bitrate_str);
267
268 mode = &codec_factory.modes[codec_factory.mode_count++];
269 mode->enabled = PJ_TRUE;
270 mode->pt = PJMEDIA_RTP_PT_G7221C_48;
271 mode->sample_rate = UWB_SAMPLE_RATE;
272 mode->bitrate = 48000;
273 pj_utoa(mode->bitrate, mode->bitrate_str);
274
275 /* Non-standard bitrates */
276
277 /* Bitrate 16kbps is non-standard but rather commonly used. */
278 mode = &codec_factory.modes[codec_factory.mode_count++];
279 mode->enabled = PJ_FALSE;
280 mode->pt = PJMEDIA_RTP_PT_G722_1_16;
281 mode->sample_rate = WB_SAMPLE_RATE;
282 mode->bitrate = 16000;
283 pj_utoa(mode->bitrate, mode->bitrate_str);
284
285 /* Reserved two modes for non-standard bitrates */
286 codec_factory.mode_rsv_start = codec_factory.mode_count;
287 mode = &codec_factory.modes[codec_factory.mode_count++];
288 mode->enabled = PJ_FALSE;
289 mode->pt = PJMEDIA_RTP_PT_G7221_RSV1;
290
291 mode = &codec_factory.modes[codec_factory.mode_count++];
292 mode->enabled = PJ_FALSE;
293 mode->pt = PJMEDIA_RTP_PT_G7221_RSV2;
294
295 pj_assert(codec_factory.mode_count <= MAX_CODEC_MODES);
296
297 /* Create G722.1 codec factory. */
298 codec_factory.base.op = &codec_factory_op;
299 codec_factory.base.factory_data = NULL;
300 codec_factory.endpt = endpt;
301
302 codec_factory.pool = pjmedia_endpt_create_pool(endpt, "G722.1 codec",
303 4000, 4000);
304 if (!codec_factory.pool)
305 return PJ_ENOMEM;
306
307 /* Create mutex. */
308 status = pj_mutex_create_simple(codec_factory.pool, "G722.1 codec",
309 &codec_factory.mutex);
310 if (status != PJ_SUCCESS)
311 goto on_error;
312
313 /* Get the codec manager. */
314 codec_mgr = pjmedia_endpt_get_codec_mgr(endpt);
315 if (!codec_mgr) {
316 status = PJ_EINVALIDOP;
317 goto on_error;
318 }
319
320 /* Register codec factory to endpoint. */
321 status = pjmedia_codec_mgr_register_factory(codec_mgr,
322 &codec_factory.base);
323 if (status != PJ_SUCCESS)
324 goto on_error;
325
326 /* Done. */
327 return PJ_SUCCESS;
328
329on_error:
330 if (codec_factory.mutex) {
331 pj_mutex_destroy(codec_factory.mutex);
332 codec_factory.mutex = NULL;
333 }
334
335 pj_pool_release(codec_factory.pool);
336 codec_factory.pool = NULL;
337 return status;
338}
339
340
341/**
342 * Enable and disable G722.1 modes, including non-standard modes.
343 */
344PJ_DEF(pj_status_t) pjmedia_codec_g7221_set_mode(unsigned sample_rate,
345 unsigned bitrate,
346 pj_bool_t enabled)
347{
348 unsigned i;
349
350 /* Validate mode */
351 if (!validate_mode(sample_rate, bitrate))
352 return PJMEDIA_CODEC_EINMODE;
353
354 /* Look up in factory modes table */
355 for (i = 0; i < codec_factory.mode_count; ++i) {
356 if (codec_factory.modes[i].sample_rate == sample_rate &&
357 codec_factory.modes[i].bitrate == bitrate)
358 {
359 codec_factory.modes[i].enabled = enabled;
360 return PJ_SUCCESS;
361 }
362 }
363
364 /* Mode not found in modes table, this may be a request to enable
365 * a non-standard G722.1 mode.
366 */
367
368 /* Non-standard mode need to be initialized first before user
369 * can disable it.
370 */
371 if (!enabled)
372 return PJ_ENOTFOUND;
373
374 /* Initialize a non-standard mode, look for available space. */
375 for (i = codec_factory.mode_rsv_start;
376 i < codec_factory.mode_count; ++i)
377 {
378 if (!codec_factory.modes[i].enabled)
379 {
380 codec_mode *mode = &codec_factory.modes[i];
381 mode->enabled = PJ_TRUE;
382 mode->sample_rate = sample_rate;
383 mode->bitrate = bitrate;
384 pj_utoa(mode->bitrate, mode->bitrate_str);
385
386 return PJ_SUCCESS;
387 }
388 }
389
390 /* No space for non-standard mode. */
391 return PJ_ETOOMANY;
392}
393
394
395/*
396 * Unregister G722.1 codec factory from pjmedia endpoint.
397 */
398PJ_DEF(pj_status_t) pjmedia_codec_g7221_deinit(void)
399{
400 pjmedia_codec_mgr *codec_mgr;
401 pj_status_t status;
402
403 if (codec_factory.pool == NULL) {
404 /* Already deinitialized */
405 return PJ_SUCCESS;
406 }
407
408 pj_mutex_lock(codec_factory.mutex);
409
410 /* Get the codec manager. */
411 codec_mgr = pjmedia_endpt_get_codec_mgr(codec_factory.endpt);
412 if (!codec_mgr) {
413 pj_pool_release(codec_factory.pool);
414 codec_factory.pool = NULL;
415 return PJ_EINVALIDOP;
416 }
417
418 /* Unregister G722.1 codec factory. */
419 status = pjmedia_codec_mgr_unregister_factory(codec_mgr,
420 &codec_factory.base);
421
422 /* Destroy mutex. */
423 pj_mutex_destroy(codec_factory.mutex);
424
425 /* Destroy pool. */
426 pj_pool_release(codec_factory.pool);
427 codec_factory.pool = NULL;
428
429 return status;
430}
431
432/*
433 * Check if factory can allocate the specified codec.
434 */
435static pj_status_t test_alloc( pjmedia_codec_factory *factory,
436 const pjmedia_codec_info *info )
437{
438 PJ_UNUSED_ARG(factory);
439
440 /* Type MUST be audio. */
441 if (info->type != PJMEDIA_TYPE_AUDIO)
442 return PJMEDIA_CODEC_EUNSUP;
443
444 /* Check encoding name. */
445 if (pj_stricmp2(&info->encoding_name, CODEC_TAG) != 0)
446 return PJMEDIA_CODEC_EUNSUP;
447
448 /* Check clock-rate */
449 if (info->clock_rate != WB_SAMPLE_RATE &&
450 info->clock_rate != UWB_SAMPLE_RATE)
451 {
452 return PJMEDIA_CODEC_EUNSUP;
453 }
454
455 return PJ_SUCCESS;
456}
457
458/*
459 * Generate default attribute.
460 */
461static pj_status_t default_attr ( pjmedia_codec_factory *factory,
462 const pjmedia_codec_info *id,
463 pjmedia_codec_param *attr )
464{
465 codec_mode *mode;
466
467 PJ_ASSERT_RETURN(factory==&codec_factory.base, PJ_EINVAL);
468
469 pj_bzero(attr, sizeof(pjmedia_codec_param));
470
471 mode = lookup_mode(id->pt);
472 if (mode == NULL || !mode->enabled)
473 return PJMEDIA_CODEC_EUNSUP;
474
475 attr->info.pt = (pj_uint8_t)id->pt;
476 attr->info.channel_cnt = 1;
477 attr->info.clock_rate = mode->sample_rate;
478 attr->info.max_bps = mode->bitrate;
479 attr->info.avg_bps = mode->bitrate;
480 attr->info.pcm_bits_per_sample = 16;
481 attr->info.frm_ptime = 20;
482
483 /* Default flags. */
484 attr->setting.plc = 1;
485 attr->setting.vad = 0;
486 attr->setting.frm_per_pkt = 1;
487
488 /* Default FMTP setting */
489 attr->setting.dec_fmtp.cnt = 1;
490 attr->setting.dec_fmtp.param[0].name = pj_str("bitrate");
491 attr->setting.dec_fmtp.param[0].val = pj_str(mode->bitrate_str);
492
493 return PJ_SUCCESS;
494}
495
496/*
497 * Enum codecs supported by this factory.
498 */
499static pj_status_t enum_codecs( pjmedia_codec_factory *factory,
500 unsigned *count,
501 pjmedia_codec_info codecs[])
502{
Nanang Izzuddinf484a1d2009-04-02 10:41:22 +0000503 unsigned i, max_cnt;
Nanang Izzuddin57b88572009-04-01 12:05:34 +0000504
505 PJ_ASSERT_RETURN(factory==&codec_factory.base, PJ_EINVAL);
506 PJ_ASSERT_RETURN(codecs && *count > 0, PJ_EINVAL);
507
Nanang Izzuddinf484a1d2009-04-02 10:41:22 +0000508 max_cnt = *count;
509 *count = 0;
Nanang Izzuddin57b88572009-04-01 12:05:34 +0000510
Nanang Izzuddinf484a1d2009-04-02 10:41:22 +0000511 for (i=0; (i < codec_factory.mode_count) && (*count < max_cnt); ++i)
512 {
Nanang Izzuddin57b88572009-04-01 12:05:34 +0000513 if (!codec_factory.modes[i].enabled)
514 continue;
515
Nanang Izzuddinf484a1d2009-04-02 10:41:22 +0000516 pj_bzero(&codecs[*count], sizeof(pjmedia_codec_info));
517 codecs[*count].encoding_name = pj_str((char*)CODEC_TAG);
518 codecs[*count].pt = codec_factory.modes[i].pt;
519 codecs[*count].type = PJMEDIA_TYPE_AUDIO;
520 codecs[*count].clock_rate = codec_factory.modes[i].sample_rate;
521 codecs[*count].channel_cnt = 1;
522
523 ++ *count;
Nanang Izzuddin57b88572009-04-01 12:05:34 +0000524 }
525
526 return PJ_SUCCESS;
527}
528
529/*
530 * Allocate a new codec instance.
531 */
532static pj_status_t alloc_codec( pjmedia_codec_factory *factory,
533 const pjmedia_codec_info *id,
534 pjmedia_codec **p_codec)
535{
536 codec_private_t *codec_data;
537 pjmedia_codec *codec;
538 pj_pool_t *pool;
539 pj_status_t status;
540
541 PJ_ASSERT_RETURN(factory && id && p_codec, PJ_EINVAL);
542 PJ_ASSERT_RETURN(factory == &codec_factory.base, PJ_EINVAL);
543
544 pj_mutex_lock(codec_factory.mutex);
545
546 /* Create pool for codec instance */
547 pool = pjmedia_endpt_create_pool(codec_factory.endpt, "G7221", 512, 512);
548 codec = PJ_POOL_ZALLOC_T(pool, pjmedia_codec);
549 codec->op = &codec_op;
550 codec->factory = factory;
551 codec->codec_data = PJ_POOL_ZALLOC_T(pool, codec_private_t);
552 codec_data = (codec_private_t*) codec->codec_data;
553 codec_data->pool = pool;
554
555 /* Create silence detector */
556 status = pjmedia_silence_det_create(pool, id->clock_rate,
557 id->clock_rate * 20 / 1000,
558 &codec_data->vad);
559 if (status != PJ_SUCCESS) {
560 pj_mutex_unlock(codec_factory.mutex);
561 return status;
562 }
563
564 pj_mutex_unlock(codec_factory.mutex);
565
566 *p_codec = codec;
567 return PJ_SUCCESS;
568}
569
570/*
571 * Free codec.
572 */
573static pj_status_t dealloc_codec( pjmedia_codec_factory *factory,
574 pjmedia_codec *codec )
575{
576 codec_private_t *codec_data;
577 pj_pool_t *pool;
578
579 PJ_ASSERT_RETURN(factory && codec, PJ_EINVAL);
580 PJ_ASSERT_RETURN(factory == &codec_factory.base, PJ_EINVAL);
581
582 /* Close codec, if it's not closed. */
583 codec_data = (codec_private_t*) codec->codec_data;
584 pool = codec_data->pool;
585 codec_close(codec);
586
587 /* Release codec pool */
588 pj_pool_release(pool);
589
590 return PJ_SUCCESS;
591}
592
593/*
594 * Init codec.
595 */
596static pj_status_t codec_init( pjmedia_codec *codec,
597 pj_pool_t *pool )
598{
599 PJ_UNUSED_ARG(codec);
600 PJ_UNUSED_ARG(pool);
601 return PJ_SUCCESS;
602}
603
604/*
605 * Open codec.
606 */
607static pj_status_t codec_open( pjmedia_codec *codec,
608 pjmedia_codec_param *attr )
609{
610 codec_private_t *codec_data = (codec_private_t*) codec->codec_data;
611 pj_pool_t *pool;
612 unsigned tmp;
613
614 /* Validation mode first! */
615 if (!validate_mode(attr->info.clock_rate, attr->info.avg_bps))
616 return PJMEDIA_CODEC_EINMODE;
617
618 pool = codec_data->pool;
619
620 /* Initialize common state */
621 codec_data->vad_enabled = (attr->setting.vad != 0);
622 codec_data->plc_enabled = (attr->setting.plc != 0);
623
624 codec_data->bitrate = (pj_uint16_t)attr->info.avg_bps;
625 codec_data->frame_size_bits = (pj_uint16_t)(attr->info.avg_bps*20/1000);
626 codec_data->frame_size = (pj_uint16_t)(codec_data->frame_size_bits>>3);
627 codec_data->samples_per_frame = (pj_uint16_t)
628 (attr->info.clock_rate*20/1000);
629 codec_data->number_of_regions = (pj_uint16_t)
630 (attr->info.clock_rate <= WB_SAMPLE_RATE?
631 NUMBER_OF_REGIONS:MAX_NUMBER_OF_REGIONS);
632
633 /* Initialize encoder state */
634 tmp = codec_data->samples_per_frame << 1;
635 codec_data->enc_old_frame = (Word16*)pj_pool_zalloc(pool, tmp);
636
637 /* Initialize decoder state */
638 tmp = codec_data->samples_per_frame;
639 codec_data->dec_old_frame = (Word16*)pj_pool_zalloc(pool, tmp);
640
641 tmp = codec_data->samples_per_frame << 1;
642 codec_data->dec_old_mlt_coefs = (Word16*)pj_pool_zalloc(pool, tmp);
643
644 codec_data->dec_randobj.seed0 = 1;
645 codec_data->dec_randobj.seed1 = 1;
646 codec_data->dec_randobj.seed2 = 1;
647 codec_data->dec_randobj.seed3 = 1;
648
649 return PJ_SUCCESS;
650}
651
652/*
653 * Close codec.
654 */
655static pj_status_t codec_close( pjmedia_codec *codec )
656{
657 PJ_UNUSED_ARG(codec);
658
659 return PJ_SUCCESS;
660}
661
662
663/*
664 * Modify codec settings.
665 */
666static pj_status_t codec_modify( pjmedia_codec *codec,
667 const pjmedia_codec_param *attr )
668{
669 codec_private_t *codec_data = (codec_private_t*) codec->codec_data;
670
671 codec_data->vad_enabled = (attr->setting.vad != 0);
672 codec_data->plc_enabled = (attr->setting.plc != 0);
673
674 return PJ_SUCCESS;
675}
676
677/*
678 * Get frames in the packet.
679 */
680static pj_status_t codec_parse( pjmedia_codec *codec,
681 void *pkt,
682 pj_size_t pkt_size,
683 const pj_timestamp *ts,
684 unsigned *frame_cnt,
685 pjmedia_frame frames[])
686{
687 codec_private_t *codec_data = (codec_private_t*) codec->codec_data;
688 unsigned count = 0;
689
690 PJ_ASSERT_RETURN(frame_cnt, PJ_EINVAL);
691
692 /* Parse based on fixed frame size. */
693 while (pkt_size >= codec_data->frame_size && count < *frame_cnt) {
694 frames[count].type = PJMEDIA_FRAME_TYPE_AUDIO;
695 frames[count].buf = pkt;
696 frames[count].size = codec_data->frame_size;
697 frames[count].timestamp.u64 = ts->u64 +
698 count * codec_data->samples_per_frame;
699
700 pkt = (pj_uint8_t*)pkt + codec_data->frame_size;
701 pkt_size -= codec_data->frame_size;
702
703 ++count;
704 }
705
706 pj_assert(pkt_size == 0);
707 *frame_cnt = count;
708
709 return PJ_SUCCESS;
710}
711
712/*
713 * Encode frames.
714 */
715static pj_status_t codec_encode( pjmedia_codec *codec,
716 const struct pjmedia_frame *input,
717 unsigned output_buf_len,
718 struct pjmedia_frame *output)
719{
720 codec_private_t *codec_data = (codec_private_t*) codec->codec_data;
721 Word16 mlt_coefs[MAX_SAMPLES_PER_FRAME];
722 Word16 mag_shift;
723
724 /* Check frame in & out size */
725 PJ_ASSERT_RETURN((pj_uint16_t)input->size ==
726 (codec_data->samples_per_frame<<1),
727 PJMEDIA_CODEC_EPCMTOOSHORT);
728 PJ_ASSERT_RETURN(output_buf_len >= codec_data->frame_size,
729 PJMEDIA_CODEC_EFRMTOOSHORT);
730
731 /* Apply silence detection if VAD is enabled */
732 if (codec_data->vad_enabled) {
733 pj_bool_t is_silence;
734 pj_int32_t silence_duration;
735
736 pj_assert(codec_data->vad);
737
738 silence_duration = pj_timestamp_diff32(&codec_data->last_tx,
739 &input->timestamp);
740
741 is_silence = pjmedia_silence_det_detect(codec_data->vad,
742 (const pj_int16_t*) input->buf,
743 (input->size >> 1),
744 NULL);
745 if (is_silence &&
746 PJMEDIA_CODEC_MAX_SILENCE_PERIOD != -1 &&
747 silence_duration < (PJMEDIA_CODEC_MAX_SILENCE_PERIOD *
748 (int)codec_data->samples_per_frame / 20))
749 {
750 output->type = PJMEDIA_FRAME_TYPE_NONE;
751 output->buf = NULL;
752 output->size = 0;
753 output->timestamp = input->timestamp;
754 return PJ_SUCCESS;
755 } else {
756 codec_data->last_tx = input->timestamp;
757 }
758 }
759
760 /* Convert input samples to rmlt coefs */
761 mag_shift = samples_to_rmlt_coefs((Word16*)input->buf,
762 codec_data->enc_old_frame,
763 mlt_coefs,
764 codec_data->samples_per_frame);
765
Nanang Izzuddin134989a2009-04-06 11:45:25 +0000766 /* Encode the mlt coefs. Note that encoder output stream is 16 bit array,
767 * so we need to take care about endianness.
768 */
Nanang Izzuddin57b88572009-04-01 12:05:34 +0000769 encoder(codec_data->frame_size_bits,
770 codec_data->number_of_regions,
771 mlt_coefs,
772 mag_shift,
773 output->buf);
774
Nanang Izzuddin134989a2009-04-06 11:45:25 +0000775#if defined(PJ_IS_LITTLE_ENDIAN) && PJ_IS_LITTLE_ENDIAN!=0
776 {
777 pj_uint16_t *p, *p_end;
778
779 p = (pj_uint16_t*)output->buf;
780 p_end = p + codec_data->frame_size/2;
781 while (p < p_end) {
782 *p = pj_htons(*p);
783 ++p;
784 }
785 }
786#endif
787
Nanang Izzuddin57b88572009-04-01 12:05:34 +0000788 output->type = PJMEDIA_FRAME_TYPE_AUDIO;
789 output->size = codec_data->frame_size;
790
791 return PJ_SUCCESS;
792}
793
794/*
795 * Decode frame.
796 */
797static pj_status_t codec_decode( pjmedia_codec *codec,
798 const struct pjmedia_frame *input,
799 unsigned output_buf_len,
800 struct pjmedia_frame *output)
801{
802 codec_private_t *codec_data = (codec_private_t*) codec->codec_data;
803 Word16 mlt_coefs[MAX_SAMPLES_PER_FRAME];
804 Word16 mag_shift;
805 Bit_Obj bitobj;
806 Word16 frame_error_flag = 0;
807
808 /* Check frame out length size */
809 PJ_ASSERT_RETURN(output_buf_len >=
810 (unsigned)(codec_data->samples_per_frame<<1),
811 PJMEDIA_CODEC_EPCMTOOSHORT);
812
813 /* If input is NULL, perform PLC by settting frame_error_flag to 1 */
814 if (input) {
815 /* Check frame in length size */
816 PJ_ASSERT_RETURN((pj_uint16_t)input->size == codec_data->frame_size,
817 PJMEDIA_CODEC_EFRMINLEN);
Nanang Izzuddin134989a2009-04-06 11:45:25 +0000818
819 /* Decoder requires input of 16-bits array, so we need to take care
820 * about endianness.
821 */
822#if defined(PJ_IS_LITTLE_ENDIAN) && PJ_IS_LITTLE_ENDIAN!=0
823 {
824 pj_uint16_t *p, *p_end;
825
826 p = (pj_uint16_t*)input->buf;
827 p_end = p + codec_data->frame_size/2;
828 while (p < p_end) {
829 *p = pj_ntohs(*p);
830 ++p;
831 }
832 }
833#endif
834
Nanang Izzuddin57b88572009-04-01 12:05:34 +0000835 bitobj.code_word_ptr = (Word16*)input->buf;
836 bitobj.current_word = *bitobj.code_word_ptr;
837 bitobj.code_bit_count = 0;
838 bitobj.number_of_bits_left = codec_data->frame_size_bits;
839 } else {
840 pj_bzero(&bitobj, sizeof(bitobj));
841 frame_error_flag = 1;
842 }
843
844 /* Process the input frame to get mlt coefs */
845 decoder(&bitobj,
846 &codec_data->dec_randobj,
847 codec_data->number_of_regions,
848 mlt_coefs,
849 &mag_shift,
850 &codec_data->dec_old_mag_shift,
851 codec_data->dec_old_mlt_coefs,
852 frame_error_flag);
853
854 /* Convert the mlt_coefs to PCM samples */
855 rmlt_coefs_to_samples(mlt_coefs,
856 codec_data->dec_old_frame,
857 (Word16*)output->buf,
858 codec_data->samples_per_frame,
859 mag_shift);
860
861 output->type = PJMEDIA_FRAME_TYPE_AUDIO;
862 output->size = codec_data->samples_per_frame << 1;
863
864 return PJ_SUCCESS;
865}
866
867/*
868 * Recover lost frame.
869 */
870static pj_status_t codec_recover( pjmedia_codec *codec,
871 unsigned output_buf_len,
872 struct pjmedia_frame *output)
873{
874 codec_private_t *codec_data = (codec_private_t*) codec->codec_data;
875
876 /* Use native PLC when PLC is enabled. */
877 if (codec_data->plc_enabled)
878 return codec_decode(codec, NULL, output_buf_len, output);
879
880 /* Otherwise just return zero-fill frame. */
881 output->type = PJMEDIA_FRAME_TYPE_AUDIO;
882 output->size = codec_data->samples_per_frame << 1;
883
884 pjmedia_zero_samples((pj_int16_t*)output->buf,
885 codec_data->samples_per_frame);
886
887 return PJ_SUCCESS;
888}
889
890#endif /* PJMEDIA_HAS_G7221_CODEC */