/*
|
* Copyright 2010 Rockchip Electronics S.LSI Co. LTD
|
*
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
*/
|
|
#define MODULE_TAG "enc_impl"
|
|
#include <string.h>
|
|
#include "mpp_mem.h"
|
#include "mpp_debug.h"
|
#include "mpp_common.h"
|
#include "mpp_packet_impl.h"
|
|
#include "h264e_api_v2.h"
|
#include "jpege_api_v2.h"
|
#include "h265e_api.h"
|
#include "vp8e_api_v2.h"
|
#include "enc_impl.h"
|
|
/*
|
* all encoder controller static register here
|
*/
|
static const EncImplApi *enc_apis[] = {
|
#if HAVE_H264E
|
&api_h264e,
|
#endif
|
#if HAVE_H265E
|
&api_h265e,
|
#endif
|
#if HAVE_JPEGE
|
&api_jpege,
|
#endif
|
#if HAVE_VP8E
|
&api_vp8e,
|
#endif
|
};
|
|
typedef struct EncImplCtx_t {
|
EncImplCfg cfg;
|
const EncImplApi *api;
|
void *ctx;
|
} EncImplCtx;
|
|
MPP_RET enc_impl_init(EncImpl *impl, EncImplCfg *cfg)
|
{
|
if (NULL == impl || NULL == cfg) {
|
mpp_err_f("found NULL input controller %p config %p\n", impl, cfg);
|
return MPP_ERR_NULL_PTR;
|
}
|
|
*impl = NULL;
|
|
RK_U32 i;
|
const EncImplApi **apis = enc_apis;
|
RK_U32 api_cnt = MPP_ARRAY_ELEMS(enc_apis);
|
|
for (i = 0; i < api_cnt; i++) {
|
const EncImplApi *api = apis[i];
|
|
if (cfg->coding == api->coding) {
|
EncImplCtx *p = mpp_calloc(EncImplCtx, 1);
|
void *ctx = mpp_calloc_size(void, api->ctx_size);
|
|
if (NULL == ctx || NULL == p) {
|
mpp_err_f("failed to alloc encoder context\n");
|
mpp_free(p);
|
mpp_free(ctx);
|
return MPP_ERR_MALLOC;
|
}
|
|
MPP_RET ret = api->init(ctx, cfg);
|
if (MPP_OK != ret) {
|
mpp_err_f("failed to init controller\n");
|
mpp_free(p);
|
mpp_free(ctx);
|
return ret;
|
}
|
|
p->api = api;
|
p->ctx = ctx;
|
memcpy(&p->cfg, cfg, sizeof(p->cfg));
|
*impl = p;
|
return MPP_OK;
|
}
|
}
|
|
return MPP_NOK;
|
}
|
|
MPP_RET enc_impl_deinit(EncImpl impl)
|
{
|
if (NULL == impl) {
|
mpp_err_f("found NULL input\n");
|
return MPP_ERR_NULL_PTR;
|
}
|
|
EncImplCtx *p = (EncImplCtx *)impl;
|
if (p->api->deinit)
|
p->api->deinit(p->ctx);
|
|
mpp_free(p->ctx);
|
mpp_free(p);
|
return MPP_OK;
|
}
|
|
MPP_RET enc_impl_proc_cfg(EncImpl impl, MpiCmd cmd, void *para)
|
{
|
if (NULL == impl) {
|
mpp_err_f("found NULL input\n");
|
return MPP_ERR_NULL_PTR;
|
}
|
|
MPP_RET ret = MPP_OK;
|
EncImplCtx *p = (EncImplCtx *)impl;
|
if (p->api->proc_cfg)
|
ret = p->api->proc_cfg(p->ctx, cmd, para);
|
|
return ret;
|
}
|
|
MPP_RET enc_impl_gen_hdr(EncImpl impl, MppPacket pkt)
|
{
|
if (NULL == impl) {
|
mpp_err_f("found NULL input\n");
|
return MPP_ERR_NULL_PTR;
|
}
|
|
MPP_RET ret = MPP_OK;
|
EncImplCtx *p = (EncImplCtx *)impl;
|
|
if (p->api->gen_hdr) {
|
if (pkt)
|
mpp_packet_reset_segment(pkt);
|
|
ret = p->api->gen_hdr(p->ctx, pkt);
|
}
|
|
return ret;
|
}
|
|
MPP_RET enc_impl_start(EncImpl impl, HalEncTask *task)
|
{
|
if (NULL == impl) {
|
mpp_err_f("found NULL input\n");
|
return MPP_ERR_NULL_PTR;
|
}
|
|
MPP_RET ret = MPP_OK;
|
EncImplCtx *p = (EncImplCtx *)impl;
|
if (p->api->start)
|
ret = p->api->start(p->ctx, task);
|
|
return ret;
|
}
|
|
MPP_RET enc_impl_proc_dpb(EncImpl impl, HalEncTask *task)
|
{
|
if (NULL == impl) {
|
mpp_err_f("found NULL input\n");
|
return MPP_ERR_NULL_PTR;
|
}
|
|
MPP_RET ret = MPP_OK;
|
EncImplCtx *p = (EncImplCtx *)impl;
|
if (p->api->proc_dpb)
|
ret = p->api->proc_dpb(p->ctx, task);
|
|
return ret;
|
}
|
|
MPP_RET enc_impl_proc_hal(EncImpl impl, HalEncTask *task)
|
{
|
if (NULL == impl || NULL == task) {
|
mpp_err_f("found NULL input\n");
|
return MPP_ERR_NULL_PTR;
|
}
|
|
MPP_RET ret = MPP_OK;
|
EncImplCtx *p = (EncImplCtx *)impl;
|
if (p->api->proc_hal)
|
ret = p->api->proc_hal(p->ctx, task);
|
|
return ret;
|
}
|
|
MPP_RET enc_impl_add_prefix(EncImpl impl, MppPacket pkt, RK_S32 *length,
|
RK_U8 uuid[16], const void *data, RK_S32 size)
|
{
|
if (NULL == pkt || NULL == data) {
|
mpp_err_f("found NULL input\n");
|
return MPP_ERR_NULL_PTR;
|
}
|
|
MPP_RET ret = MPP_OK;
|
EncImplCtx *p = (EncImplCtx *)impl;
|
|
if (NULL == p->api->add_prefix)
|
return ret;
|
|
if (p->api->add_prefix)
|
ret = p->api->add_prefix(pkt, length, uuid, data, size);
|
|
return ret;
|
}
|
|
MPP_RET enc_impl_sw_enc(EncImpl impl, HalEncTask *task)
|
{
|
if (NULL == impl || NULL == task) {
|
mpp_err_f("found NULL input\n");
|
return MPP_ERR_NULL_PTR;
|
}
|
|
MPP_RET ret = MPP_OK;
|
EncImplCtx *p = (EncImplCtx *)impl;
|
if (p->api->sw_enc)
|
ret = p->api->sw_enc(p->ctx, task);
|
|
return ret;
|
}
|