/*
* Initialize and register AMR codec factory to pjmedia endpoint.
*/
PJ_DEF(pj_status_t) pjmedia_codec_opencore_amr_init( pjmedia_endpt *endpt,
unsigned options)
{
pjmedia_codec_mgr *codec_mgr;
pj_str_t codec_name;
pj_status_t status;
if (amr_codec_factory.pool != NULL)
return PJ_SUCCESS;
/* Create AMR codec factory. */
amr_codec_factory.base.op = &amr_factory_op;
amr_codec_factory.base.factory_data = NULL;
amr_codec_factory.endpt = endpt;
#ifdef USE_AMRNB
amr_codec_factory.init[IDX_AMR_NB] = ((options & PJMEDIA_AMR_NO_NB) == 0);
#else
amr_codec_factory.init[IDX_AMR_NB] = PJ_FALSE;
#endif
#ifdef USE_AMRWB
amr_codec_factory.init[IDX_AMR_WB] = ((options & PJMEDIA_AMR_NO_WB) == 0);
#else
amr_codec_factory.init[IDX_AMR_WB] = PJ_FALSE;
#endif
amr_codec_factory.pool = pjmedia_endpt_create_pool(endpt, "amr", 1000,
1000);
if (!amr_codec_factory.pool)
return PJ_ENOMEM;
/* Get the codec manager. */
codec_mgr = pjmedia_endpt_get_codec_mgr(endpt);
if (!codec_mgr) {
status = PJ_EINVALIDOP;
goto on_error;
}
/* Register format match callback. */
pj_cstr(&codec_name, "AMR");
status = pjmedia_sdp_neg_register_fmt_match_cb(
&codec_name,
&pjmedia_codec_amr_match_sdp);
if (status != PJ_SUCCESS)
goto on_error;
/* Register codec factory to endpoint. */
status = pjmedia_codec_mgr_register_factory(codec_mgr,
&amr_codec_factory.base);
if (status != PJ_SUCCESS)
goto on_error;
/* Done. */
return PJ_SUCCESS;
on_error:
pj_pool_release(amr_codec_factory.pool);
amr_codec_factory.pool = NULL;
return status;
}
/*
* Initialize and register AMR-NB codec factory to pjmedia endpoint.
*/
PJ_DEF(pj_status_t) pjmedia_codec_opencore_amrnb_init( pjmedia_endpt *endpt )
{
pjmedia_codec_mgr *codec_mgr;
pj_status_t status;
pj_str_t codec_name;
if (amr_codec_factory.pool != NULL)
return PJ_SUCCESS;
/* Create AMR-NB codec factory. */
amr_codec_factory.base.op = &amr_factory_op;
amr_codec_factory.base.factory_data = NULL;
amr_codec_factory.endpt = endpt;
amr_codec_factory.pool = pjmedia_endpt_create_pool(endpt, "amrnb", 1000,
1000);
if (!amr_codec_factory.pool)
return PJ_ENOMEM;
/* Get the codec manager. */
codec_mgr = pjmedia_endpt_get_codec_mgr(endpt);
if (!codec_mgr) {
status = PJ_EINVALIDOP;
goto on_error;
}
/* Register format match callback. */
pj_cstr(&codec_name, "AMR");
status = pjmedia_sdp_neg_register_fmt_match_cb( &codec_name,
&pjmedia_codec_amr_match_sdp);
if (status != PJ_SUCCESS){
goto on_error;
}
/* Register codec factory to endpoint. */
status = pjmedia_codec_mgr_register_factory(codec_mgr,
&amr_codec_factory.base);
if (status != PJ_SUCCESS)
goto on_error;
/* Done. */
return PJ_SUCCESS;
on_error:
pj_pool_release(amr_codec_factory.pool);
amr_codec_factory.pool = NULL;
return status;
}
/*
* Initialize and register openh264 codec factory to pjmedia endpoint.
*/
PJ_DEF(pj_status_t) pjmedia_codec_openh264_vid_init(pjmedia_vid_codec_mgr *mgr,
pj_pool_factory *pf)
{
pj_pool_t *pool;
pj_status_t status;
openh264_codec_desc *desc = &codec_desc[0];
if (openh264_factory.pool != NULL) {
/* Already initialized. */
return PJ_SUCCESS;
}
if (!mgr)
mgr = pjmedia_vid_codec_mgr_instance();
PJ_ASSERT_RETURN(mgr, PJ_EINVAL);
/* Create openh264 codec factory. */
openh264_factory.base.op = &openh264_factory_op;
openh264_factory.base.factory_data = NULL;
openh264_factory.mgr = mgr;
openh264_factory.pf = pf;
pool = pj_pool_create(pf, "openh264 codec factory", 256, 256, NULL);
if (!pool)
return PJ_ENOMEM;
/* Create mutex. */
status = pj_mutex_create_simple(pool, "openh264 codec factory",
&openh264_factory.mutex);
if (status != PJ_SUCCESS)
goto on_error;
desc->info.dec_fmt_id[0] = PJMEDIA_FORMAT_I420;
desc->info.dec_fmt_id_cnt = 1;
desc->info.dir |= PJMEDIA_DIR_ENCODING;
desc->info.dir |= PJMEDIA_DIR_DECODING;
desc->enabled = PJ_TRUE;
if (desc->info.clock_rate == 0)
desc->info.clock_rate = 90000;
desc->info.packings |= PJMEDIA_VID_PACKING_WHOLE;
if (desc->packetize && desc->unpacketize)
desc->info.packings |= PJMEDIA_VID_PACKING_PACKETS;
/* Registering format match for SDP negotiation */
if (desc->sdp_fmt_match) {
status = pjmedia_sdp_neg_register_fmt_match_cb(
&desc->info.encoding_name,
desc->sdp_fmt_match);
pj_assert(status == PJ_SUCCESS);
}
/* Register codec factory to codec manager. */
status = pjmedia_vid_codec_mgr_register_factory(mgr,
&openh264_factory.base);
if (status != PJ_SUCCESS)
goto on_error;
openh264_factory.pool = pool;
/* Done. */
return PJ_SUCCESS;
on_error:
pj_pool_release(pool);
return status;
}
/*
* Initialize and register FFMPEG codec factory to pjmedia endpoint.
*/
PJ_DEF(pj_status_t) pjmedia_codec_ffmpeg_init(pjmedia_vid_codec_mgr *mgr,
pj_pool_factory *pf)
{
pj_pool_t *pool;
AVCodec *c;
pj_status_t status;
unsigned i;
if (ffmpeg_factory.pool != NULL) {
/* Already initialized. */
return PJ_SUCCESS;
}
if (!mgr) mgr = pjmedia_vid_codec_mgr_instance();
PJ_ASSERT_RETURN(mgr, PJ_EINVAL);
/* Create FFMPEG codec factory. */
ffmpeg_factory.base.op = &ffmpeg_factory_op;
ffmpeg_factory.base.factory_data = NULL;
ffmpeg_factory.mgr = mgr;
ffmpeg_factory.pf = pf;
pool = pj_pool_create(pf, "ffmpeg codec factory", 256, 256, NULL);
if (!pool)
return PJ_ENOMEM;
/* Create mutex. */
status = pj_mutex_create_simple(pool, "ffmpeg codec factory",
&ffmpeg_factory.mutex);
if (status != PJ_SUCCESS)
goto on_error;
avcodec_init();
avcodec_register_all();
av_log_set_level(AV_LOG_ERROR);
/* Enum FFMPEG codecs */
for (c=av_codec_next(NULL); c; c=av_codec_next(c)) {
ffmpeg_codec_desc *desc;
pjmedia_format_id fmt_id;
int codec_info_idx;
#if LIBAVCODEC_VERSION_MAJOR <= 52
# define AVMEDIA_TYPE_VIDEO CODEC_TYPE_VIDEO
#endif
if (c->type != AVMEDIA_TYPE_VIDEO)
continue;
/* Video encoder and decoder are usually implemented in separate
* AVCodec instances. While the codec attributes (e.g: raw formats,
* supported fps) are in the encoder.
*/
//PJ_LOG(3, (THIS_FILE, "%s", c->name));
status = CodecID_to_pjmedia_format_id(c->id, &fmt_id);
/* Skip if format ID is unknown */
if (status != PJ_SUCCESS)
continue;
codec_info_idx = find_codec_idx_by_fmt_id(fmt_id);
/* Skip if codec is unwanted by this wrapper (not listed in
* the codec info array)
*/
if (codec_info_idx < 0)
continue;
desc = &codec_desc[codec_info_idx];
/* Skip duplicated codec implementation */
if ((c->encode && (desc->info.dir & PJMEDIA_DIR_ENCODING)) ||
(c->decode && (desc->info.dir & PJMEDIA_DIR_DECODING)))
{
continue;
}
/* Get raw/decoded format ids in the encoder */
if (c->pix_fmts && c->encode) {
pjmedia_format_id raw_fmt[PJMEDIA_VID_CODEC_MAX_DEC_FMT_CNT];
unsigned raw_fmt_cnt = 0;
unsigned raw_fmt_cnt_should_be = 0;
const enum PixelFormat *p = c->pix_fmts;
for(;(p && *p != -1) &&
(raw_fmt_cnt < PJMEDIA_VID_CODEC_MAX_DEC_FMT_CNT);
++p)
{
pjmedia_format_id fmt_id;
raw_fmt_cnt_should_be++;
status = PixelFormat_to_pjmedia_format_id(*p, &fmt_id);
if (status != PJ_SUCCESS) {
PJ_LOG(6, (THIS_FILE, "Unrecognized ffmpeg pixel "
"format %d", *p));
continue;
}
//raw_fmt[raw_fmt_cnt++] = fmt_id;
/* Disable some formats due to H.264 error:
* x264 [error]: baseline profile doesn't support 4:4:4
*/
if (desc->info.pt != PJMEDIA_RTP_PT_H264 ||
fmt_id != PJMEDIA_FORMAT_RGB24)
{
raw_fmt[raw_fmt_cnt++] = fmt_id;
}
}
if (raw_fmt_cnt == 0) {
PJ_LOG(5, (THIS_FILE, "No recognized raw format "
"for codec [%s/%s], codec ignored",
c->name, c->long_name));
/* Skip this encoder */
continue;
}
if (raw_fmt_cnt < raw_fmt_cnt_should_be) {
PJ_LOG(6, (THIS_FILE, "Codec [%s/%s] have %d raw formats, "
"recognized only %d raw formats",
c->name, c->long_name,
raw_fmt_cnt_should_be, raw_fmt_cnt));
}
desc->info.dec_fmt_id_cnt = raw_fmt_cnt;
pj_memcpy(desc->info.dec_fmt_id, raw_fmt,
sizeof(raw_fmt[0])*raw_fmt_cnt);
}
/* Get supported framerates */
if (c->supported_framerates) {
const AVRational *fr = c->supported_framerates;
while ((fr->num != 0 || fr->den != 0) &&
desc->info.fps_cnt < PJMEDIA_VID_CODEC_MAX_FPS_CNT)
{
desc->info.fps[desc->info.fps_cnt].num = fr->num;
desc->info.fps[desc->info.fps_cnt].denum = fr->den;
++desc->info.fps_cnt;
++fr;
}
}
/* Get ffmpeg encoder instance */
if (c->encode && !desc->enc) {
desc->info.dir |= PJMEDIA_DIR_ENCODING;
desc->enc = c;
}
/* Get ffmpeg decoder instance */
if (c->decode && !desc->dec) {
desc->info.dir |= PJMEDIA_DIR_DECODING;
desc->dec = c;
}
/* Enable this codec when any ffmpeg codec instance are recognized
* and the supported raw formats info has been collected.
*/
if ((desc->dec || desc->enc) && desc->info.dec_fmt_id_cnt)
{
desc->enabled = PJ_TRUE;
}
/* Normalize default value of clock rate */
if (desc->info.clock_rate == 0)
desc->info.clock_rate = 90000;
/* Set supported packings */
desc->info.packings |= PJMEDIA_VID_PACKING_WHOLE;
if (desc->packetize && desc->unpacketize)
desc->info.packings |= PJMEDIA_VID_PACKING_PACKETS;
}
/* Review all codecs for applying base format, registering format match for
* SDP negotiation, etc.
*/
for (i = 0; i < PJ_ARRAY_SIZE(codec_desc); ++i) {
ffmpeg_codec_desc *desc = &codec_desc[i];
/* Init encoder/decoder description from base format */
if (desc->base_fmt_id && (!desc->dec || !desc->enc)) {
ffmpeg_codec_desc *base_desc = NULL;
int base_desc_idx;
pjmedia_dir copied_dir = PJMEDIA_DIR_NONE;
base_desc_idx = find_codec_idx_by_fmt_id(desc->base_fmt_id);
if (base_desc_idx != -1)
base_desc = &codec_desc[base_desc_idx];
if (!base_desc || !base_desc->enabled)
continue;
/* Copy description from base codec */
if (!desc->info.dec_fmt_id_cnt) {
desc->info.dec_fmt_id_cnt = base_desc->info.dec_fmt_id_cnt;
pj_memcpy(desc->info.dec_fmt_id, base_desc->info.dec_fmt_id,
sizeof(pjmedia_format_id)*desc->info.dec_fmt_id_cnt);
}
if (!desc->info.fps_cnt) {
desc->info.fps_cnt = base_desc->info.fps_cnt;
pj_memcpy(desc->info.fps, base_desc->info.fps,
sizeof(desc->info.fps[0])*desc->info.fps_cnt);
}
if (!desc->info.clock_rate) {
desc->info.clock_rate = base_desc->info.clock_rate;
}
if (!desc->dec && base_desc->dec) {
copied_dir |= PJMEDIA_DIR_DECODING;
desc->dec = base_desc->dec;
}
if (!desc->enc && base_desc->enc) {
copied_dir |= PJMEDIA_DIR_ENCODING;
desc->enc = base_desc->enc;
}
desc->info.dir |= copied_dir;
desc->enabled = (desc->info.dir != PJMEDIA_DIR_NONE);
/* Set supported packings */
desc->info.packings |= PJMEDIA_VID_PACKING_WHOLE;
if (desc->packetize && desc->unpacketize)
desc->info.packings |= PJMEDIA_VID_PACKING_PACKETS;
if (copied_dir != PJMEDIA_DIR_NONE) {
const char *dir_name[] = {NULL, "encoder", "decoder", "codec"};
PJ_LOG(5, (THIS_FILE, "The %.*s %s is using base codec (%.*s)",
desc->info.encoding_name.slen,
desc->info.encoding_name.ptr,
dir_name[copied_dir],
base_desc->info.encoding_name.slen,
base_desc->info.encoding_name.ptr));
}
}
/* Registering format match for SDP negotiation */
if (desc->sdp_fmt_match) {
status = pjmedia_sdp_neg_register_fmt_match_cb(
&desc->info.encoding_name,
desc->sdp_fmt_match);
pj_assert(status == PJ_SUCCESS);
}
/* Print warning about missing encoder/decoder */
if (!desc->enc) {
PJ_LOG(4, (THIS_FILE, "Cannot find %.*s encoder in ffmpeg library",
desc->info.encoding_name.slen,
desc->info.encoding_name.ptr));
}
if (!desc->dec) {
PJ_LOG(4, (THIS_FILE, "Cannot find %.*s decoder in ffmpeg library",
desc->info.encoding_name.slen,
desc->info.encoding_name.ptr));
}
}
/* Register codec factory to codec manager. */
status = pjmedia_vid_codec_mgr_register_factory(mgr,
&ffmpeg_factory.base);
if (status != PJ_SUCCESS)
goto on_error;
ffmpeg_factory.pool = pool;
/* Done. */
return PJ_SUCCESS;
on_error:
pj_pool_release(pool);
return status;
}
