summaryrefslogtreecommitdiff
path: root/src/gallium/frontends/omx/tizonia/h264eprc.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/gallium/frontends/omx/tizonia/h264eprc.c')
-rw-r--r--src/gallium/frontends/omx/tizonia/h264eprc.c698
1 files changed, 698 insertions, 0 deletions
diff --git a/src/gallium/frontends/omx/tizonia/h264eprc.c b/src/gallium/frontends/omx/tizonia/h264eprc.c
new file mode 100644
index 00000000000..b52e489f21c
--- /dev/null
+++ b/src/gallium/frontends/omx/tizonia/h264eprc.c
@@ -0,0 +1,698 @@
+/**************************************************************************
+ *
+ * Copyright 2013 Advanced Micro Devices, Inc.
+ * All Rights Reserved.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sub license, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the
+ * next paragraph) shall be included in all copies or substantial portions
+ * of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+ * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
+ * IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR
+ * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+ * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+ * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ *
+ **************************************************************************/
+
+#include <tizplatform.h>
+#include <tizkernel.h>
+#include <tizutils.h>
+
+#include "pipe/p_screen.h"
+#include "pipe/p_video_codec.h"
+#include "util/u_memory.h"
+
+#include "entrypoint.h"
+#include "h264e.h"
+#include "h264eprc.h"
+#include "vid_omx_common.h"
+#include "vid_enc_common.h"
+
+static OMX_ERRORTYPE init_port_structs(vid_enc_PrivateType * priv) {
+ const void * p_krn = NULL;
+
+ assert(priv);
+
+ /* Initialisation */
+ TIZ_INIT_OMX_PORT_STRUCT(priv->in_port_def_,
+ OMX_VID_ENC_AVC_INPUT_PORT_INDEX);
+ TIZ_INIT_OMX_PORT_STRUCT(priv->out_port_def_,
+ OMX_VID_ENC_AVC_OUTPUT_PORT_INDEX);
+ TIZ_INIT_OMX_PORT_STRUCT(priv->bitrate,
+ OMX_VID_ENC_AVC_OUTPUT_PORT_INDEX);
+ TIZ_INIT_OMX_PORT_STRUCT(priv->quant,
+ OMX_VID_ENC_AVC_OUTPUT_PORT_INDEX);
+ TIZ_INIT_OMX_PORT_STRUCT(priv->profile_level,
+ OMX_VID_ENC_AVC_OUTPUT_PORT_INDEX);
+
+ /* Get values */
+ p_krn = tiz_get_krn(handleOf(priv));
+
+ tiz_check_omx(
+ tiz_api_GetParameter(p_krn, handleOf(priv),
+ OMX_IndexParamPortDefinition, &(priv->in_port_def_)));
+ tiz_check_omx(
+ tiz_api_GetParameter(p_krn, handleOf(priv),
+ OMX_IndexParamPortDefinition, &(priv->out_port_def_)));
+ tiz_check_omx(
+ tiz_api_GetParameter(p_krn, handleOf(priv),
+ OMX_IndexParamVideoBitrate, &(priv->bitrate)));
+ tiz_check_omx(
+ tiz_api_GetParameter(p_krn, handleOf(priv),
+ OMX_IndexParamVideoQuantization, &(priv->quant)));
+ tiz_check_omx(
+ tiz_api_GetParameter(p_krn, handleOf(priv),
+ OMX_IndexParamVideoProfileLevelCurrent, &(priv->profile_level)));
+
+ return OMX_ErrorNone;
+}
+
+static OMX_BUFFERHEADERTYPE * get_input_buffer(vid_enc_PrivateType * priv) {
+ assert(priv);
+
+ if (priv->in_port_disabled_) {
+ return NULL;
+ }
+
+ assert(!priv->p_inhdr_); /* encode_frame expects new buffers every time */
+
+ tiz_krn_claim_buffer(tiz_get_krn(handleOf(priv)),
+ OMX_VID_ENC_AVC_INPUT_PORT_INDEX, 0,
+ &priv->p_inhdr_);
+ return priv->p_inhdr_;
+}
+
+static OMX_BUFFERHEADERTYPE * get_output_buffer(vid_enc_PrivateType * priv) {
+ assert(priv);
+
+ if (priv->out_port_disabled_) {
+ return NULL;
+ }
+
+ if (!priv->p_outhdr_) {
+ tiz_krn_claim_buffer(tiz_get_krn(handleOf(priv)),
+ OMX_VID_ENC_AVC_OUTPUT_PORT_INDEX, 0,
+ &priv->p_outhdr_);
+ }
+ return priv->p_outhdr_;
+}
+
+static OMX_ERRORTYPE h264e_buffer_emptied(vid_enc_PrivateType * priv, OMX_BUFFERHEADERTYPE * p_hdr)
+{
+ OMX_ERRORTYPE r = OMX_ErrorNone;
+
+ assert(priv);
+ assert(priv->p_inhdr_ == p_hdr);
+
+ if (!priv->out_port_disabled_) {
+ p_hdr->nOffset = 0;
+
+ if ((p_hdr->nFlags & OMX_BUFFERFLAG_EOS) != 0) {
+ priv->eos_ = true;
+ }
+
+ r = tiz_krn_release_buffer(tiz_get_krn(handleOf(priv)), 0, p_hdr);
+ priv->p_inhdr_ = NULL;
+ }
+
+ return r;
+}
+
+static OMX_ERRORTYPE h264e_buffer_filled(vid_enc_PrivateType * priv, OMX_BUFFERHEADERTYPE * p_hdr)
+{
+ OMX_ERRORTYPE r = OMX_ErrorNone;
+
+ assert(priv);
+ assert(priv->p_outhdr_ == p_hdr);
+ assert(p_hdr);
+
+ if (!priv->in_port_disabled_) {
+ p_hdr->nOffset = 0;
+
+ if (priv->eos_) {
+ /* EOS has been received and all the input data has been consumed
+ * already, so its time to propagate the EOS flag */
+ priv->p_outhdr_->nFlags |= OMX_BUFFERFLAG_EOS;
+ }
+
+ r = tiz_krn_release_buffer(tiz_get_krn(handleOf(priv)),
+ OMX_VID_ENC_AVC_OUTPUT_PORT_INDEX,
+ p_hdr);
+ priv->p_outhdr_ = NULL;
+ }
+
+ return r;
+}
+
+
+static void release_input_header(vid_enc_PrivateType * priv) {
+ assert(!priv->in_port_disabled_);
+ if (priv->p_inhdr_) {
+ (void) tiz_krn_release_buffer(tiz_get_krn(handleOf(priv)),
+ OMX_VID_ENC_AVC_INPUT_PORT_INDEX,
+ priv->p_inhdr_);
+ }
+ priv->p_inhdr_ = NULL;
+}
+
+static void release_output_header(vid_enc_PrivateType * priv) {
+ if (priv->p_outhdr_) {
+ assert(!priv->out_port_disabled_);
+ (void) tiz_krn_release_buffer(tiz_get_krn(handleOf(priv)),
+ OMX_VID_ENC_AVC_OUTPUT_PORT_INDEX,
+ priv->p_outhdr_);
+ priv->p_outhdr_ = NULL;
+ }
+}
+
+static OMX_ERRORTYPE h264e_release_all_headers(vid_enc_PrivateType * priv)
+{
+ assert(priv);
+
+ release_input_header(priv);
+ release_output_header(priv);
+
+ return OMX_ErrorNone;
+}
+
+static void reset_stream_parameters(vid_enc_PrivateType * priv)
+{
+ assert(priv);
+ init_port_structs(priv);
+ priv->p_inhdr_ = 0;
+ priv->p_outhdr_ = 0;
+ priv->eos_ = false;
+}
+
+/* Replacement for bellagio's omx_base_filter_BufferMgmtFunction */
+static OMX_ERRORTYPE h264e_manage_buffers(vid_enc_PrivateType* priv) {
+ OMX_BUFFERHEADERTYPE * in_buf = priv->p_inhdr_;
+ OMX_BUFFERHEADERTYPE * out_buf = priv->p_outhdr_;
+ OMX_ERRORTYPE r = OMX_ErrorNone;
+
+ if (in_buf->nFilledLen > 0) {
+ vid_enc_BufferEncoded_common(priv, in_buf, out_buf);
+ } else {
+ in_buf->nFilledLen = 0;
+ }
+
+ out_buf->nTimeStamp = in_buf->nTimeStamp;
+
+ /* Release input buffer if possible */
+ if (in_buf->nFilledLen == 0) {
+ r = h264e_buffer_emptied(priv, in_buf);
+ }
+
+ /* Realase output buffer if filled or eos */
+ if ((out_buf->nFilledLen != 0) || priv->eos_) {
+ r = h264e_buffer_filled(priv, out_buf);
+ }
+
+ return r;
+}
+
+static struct encode_task *enc_NeedTask(vid_enc_PrivateType * priv)
+{
+ OMX_VIDEO_PORTDEFINITIONTYPE *def = &priv->in_port_def_.format.video;
+
+ return enc_NeedTask_common(priv, def);
+}
+
+static void enc_ScaleInput(vid_enc_PrivateType * priv, struct pipe_video_buffer **vbuf, unsigned *size)
+{
+ OMX_VIDEO_PORTDEFINITIONTYPE *def = &priv->in_port_def_.format.video;
+ enc_ScaleInput_common(priv, def, vbuf, size);
+}
+
+static void enc_HandleTask(vid_enc_PrivateType * priv, struct encode_task *task,
+ enum pipe_h264_enc_picture_type picture_type)
+{
+ unsigned size = priv->out_port_def_.nBufferSize;
+ struct pipe_video_buffer *vbuf = task->buf;
+ struct pipe_h264_enc_picture_desc picture = {};
+
+ /* -------------- scale input image --------- */
+ enc_ScaleInput(priv, &vbuf, &size);
+ priv->s_pipe->flush(priv->s_pipe, NULL, 0);
+
+ /* -------------- allocate output buffer --------- */
+ task->bitstream = pipe_buffer_create(priv->s_pipe->screen,
+ PIPE_BIND_VERTEX_BUFFER,
+ PIPE_USAGE_STAGING, /* map for read */
+ size);
+
+ picture.picture_type = picture_type;
+ picture.pic_order_cnt = task->pic_order_cnt;
+ if (priv->restricted_b_frames && picture_type == PIPE_H264_ENC_PICTURE_TYPE_B)
+ picture.not_referenced = true;
+ enc_ControlPicture_common(priv, &picture);
+
+ /* -------------- encode frame --------- */
+ priv->codec->begin_frame(priv->codec, vbuf, &picture.base);
+ priv->codec->encode_bitstream(priv->codec, vbuf, task->bitstream, &task->feedback);
+ priv->codec->end_frame(priv->codec, vbuf, &picture.base);
+}
+
+static void enc_ClearBframes(vid_enc_PrivateType * priv, struct input_buf_private *inp)
+{
+ struct encode_task *task;
+
+ if (list_is_empty(&priv->b_frames))
+ return;
+
+ task = LIST_ENTRY(struct encode_task, priv->b_frames.prev, list);
+ list_del(&task->list);
+
+ /* promote last from to P frame */
+ priv->ref_idx_l0 = priv->ref_idx_l1;
+ enc_HandleTask(priv, task, PIPE_H264_ENC_PICTURE_TYPE_P);
+ list_addtail(&task->list, &inp->tasks);
+ priv->ref_idx_l1 = priv->frame_num++;
+
+ /* handle B frames */
+ LIST_FOR_EACH_ENTRY(task, &priv->b_frames, list) {
+ enc_HandleTask(priv, task, PIPE_H264_ENC_PICTURE_TYPE_B);
+ if (!priv->restricted_b_frames)
+ priv->ref_idx_l0 = priv->frame_num;
+ priv->frame_num++;
+ }
+
+ enc_MoveTasks(&priv->b_frames, &inp->tasks);
+}
+
+static OMX_ERRORTYPE enc_LoadImage(vid_enc_PrivateType * priv, OMX_BUFFERHEADERTYPE *buf,
+ struct pipe_video_buffer *vbuf)
+{
+ OMX_VIDEO_PORTDEFINITIONTYPE *def = &priv->in_port_def_.format.video;
+ return enc_LoadImage_common(priv, def, buf, vbuf);
+}
+
+static OMX_ERRORTYPE encode_frame(vid_enc_PrivateType * priv, OMX_BUFFERHEADERTYPE * in_buf)
+{
+ struct input_buf_private *inp = in_buf->pInputPortPrivate;
+ enum pipe_h264_enc_picture_type picture_type;
+ struct encode_task *task;
+ unsigned stacked_num = 0;
+ OMX_ERRORTYPE err;
+
+ enc_MoveTasks(&inp->tasks, &priv->free_tasks);
+ task = enc_NeedTask(priv);
+ if (!task)
+ return OMX_ErrorInsufficientResources;
+
+ /* EOS */
+ if (in_buf->nFilledLen == 0) {
+ if (in_buf->nFlags & OMX_BUFFERFLAG_EOS) {
+ in_buf->nFilledLen = in_buf->nAllocLen;
+ enc_ClearBframes(priv, inp);
+ enc_MoveTasks(&priv->stacked_tasks, &inp->tasks);
+ priv->codec->flush(priv->codec);
+ }
+ return h264e_manage_buffers(priv);
+ }
+
+ if (in_buf->pOutputPortPrivate) {
+ struct pipe_video_buffer *vbuf = in_buf->pOutputPortPrivate;
+ in_buf->pOutputPortPrivate = task->buf;
+ task->buf = vbuf;
+ } else {
+ /* ------- load input image into video buffer ---- */
+ err = enc_LoadImage(priv, in_buf, task->buf);
+ if (err != OMX_ErrorNone) {
+ FREE(task);
+ return err;
+ }
+ }
+
+ /* -------------- determine picture type --------- */
+ if (!(priv->pic_order_cnt % OMX_VID_ENC_IDR_PERIOD_DEFAULT) ||
+ priv->force_pic_type.IntraRefreshVOP) {
+ enc_ClearBframes(priv, inp);
+ picture_type = PIPE_H264_ENC_PICTURE_TYPE_IDR;
+ priv->force_pic_type.IntraRefreshVOP = OMX_FALSE;
+ priv->frame_num = 0;
+ } else if (priv->codec->profile == PIPE_VIDEO_PROFILE_MPEG4_AVC_BASELINE ||
+ !(priv->pic_order_cnt % OMX_VID_ENC_P_PERIOD_DEFAULT) ||
+ (in_buf->nFlags & OMX_BUFFERFLAG_EOS)) {
+ picture_type = PIPE_H264_ENC_PICTURE_TYPE_P;
+ } else {
+ picture_type = PIPE_H264_ENC_PICTURE_TYPE_B;
+ }
+
+ task->pic_order_cnt = priv->pic_order_cnt++;
+
+ if (picture_type == PIPE_H264_ENC_PICTURE_TYPE_B) {
+ /* put frame at the tail of the queue */
+ list_addtail(&task->list, &priv->b_frames);
+ } else {
+ /* handle I or P frame */
+ priv->ref_idx_l0 = priv->ref_idx_l1;
+ enc_HandleTask(priv, task, picture_type);
+ list_addtail(&task->list, &priv->stacked_tasks);
+ LIST_FOR_EACH_ENTRY(task, &priv->stacked_tasks, list) {
+ ++stacked_num;
+ }
+ if (stacked_num == priv->stacked_frames_num) {
+ struct encode_task *t;
+ t = LIST_ENTRY(struct encode_task, priv->stacked_tasks.next, list);
+ list_del(&t->list);
+ list_addtail(&t->list, &inp->tasks);
+ }
+ priv->ref_idx_l1 = priv->frame_num++;
+
+ /* handle B frames */
+ LIST_FOR_EACH_ENTRY(task, &priv->b_frames, list) {
+ enc_HandleTask(priv, task, PIPE_H264_ENC_PICTURE_TYPE_B);
+ if (!priv->restricted_b_frames)
+ priv->ref_idx_l0 = priv->frame_num;
+ priv->frame_num++;
+ }
+
+ enc_MoveTasks(&priv->b_frames, &inp->tasks);
+ }
+
+ if (list_is_empty(&inp->tasks)) {
+ return h264e_buffer_emptied(priv, in_buf);
+ } else {
+ return h264e_manage_buffers(priv);
+ }
+}
+
+static OMX_ERRORTYPE h264e_prc_create_encoder(void *ap_obj)
+{
+ vid_enc_PrivateType *priv = ap_obj;
+ struct pipe_screen *screen;
+
+ priv->screen = omx_get_screen();
+ if (!priv->screen)
+ return OMX_ErrorInsufficientResources;
+
+ screen = priv->screen->pscreen;
+ if (!screen->get_video_param(screen, PIPE_VIDEO_PROFILE_MPEG4_AVC_HIGH,
+ PIPE_VIDEO_ENTRYPOINT_ENCODE, PIPE_VIDEO_CAP_SUPPORTED))
+ return OMX_ErrorBadParameter;
+
+ priv->s_pipe = pipe_create_multimedia_context(screen);
+ if (!priv->s_pipe)
+ return OMX_ErrorInsufficientResources;
+
+ enc_InitCompute_common(priv);
+
+ if (!vl_compositor_init(&priv->compositor, priv->s_pipe)) {
+ priv->s_pipe->destroy(priv->s_pipe);
+ priv->s_pipe = NULL;
+ return OMX_ErrorInsufficientResources;
+ }
+
+ if (!vl_compositor_init_state(&priv->cstate, priv->s_pipe)) {
+ vl_compositor_cleanup(&priv->compositor);
+ priv->s_pipe->destroy(priv->s_pipe);
+ priv->s_pipe = NULL;
+ return OMX_ErrorInsufficientResources;
+ }
+
+ priv->t_pipe = pipe_create_multimedia_context(screen);
+ if (!priv->t_pipe)
+ return OMX_ErrorInsufficientResources;
+
+ list_inithead(&priv->free_tasks);
+ list_inithead(&priv->used_tasks);
+ list_inithead(&priv->b_frames);
+ list_inithead(&priv->stacked_tasks);
+
+ return OMX_ErrorNone;
+}
+
+static void h264e_prc_destroy_encoder(void *ap_obj)
+{
+ vid_enc_PrivateType *priv = ap_obj;
+ int i;
+
+ assert (priv);
+
+ enc_ReleaseTasks(&priv->free_tasks);
+ enc_ReleaseTasks(&priv->used_tasks);
+ enc_ReleaseTasks(&priv->b_frames);
+ enc_ReleaseTasks(&priv->stacked_tasks);
+
+ for (i = 0; i < OMX_VID_ENC_NUM_SCALING_BUFFERS; ++i)
+ if (priv->scale_buffer[i])
+ priv->scale_buffer[i]->destroy(priv->scale_buffer[i]);
+
+ if (priv->s_pipe) {
+ vl_compositor_cleanup_state(&priv->cstate);
+ vl_compositor_cleanup(&priv->compositor);
+ enc_ReleaseCompute_common(priv);
+ priv->s_pipe->destroy(priv->s_pipe);
+ }
+
+ if (priv->t_pipe)
+ priv->t_pipe->destroy(priv->t_pipe);
+
+ if (priv->screen)
+ omx_put_screen();
+}
+
+/*
+ * h264eprc
+ */
+
+static void * h264e_prc_ctor(void *ap_obj, va_list * app)
+{
+ vid_enc_PrivateType *priv = super_ctor(typeOf(ap_obj, "h264eprc"), ap_obj, app);
+ assert (priv);
+
+ if (h264e_prc_create_encoder(ap_obj) != OMX_ErrorNone)
+ return priv;
+
+ priv->p_inhdr_ = 0;
+ priv->p_outhdr_ = 0;
+ priv->profile_level.eProfile = OMX_VIDEO_AVCProfileBaseline;
+ priv->profile_level.eLevel = OMX_VIDEO_AVCLevel51;
+ priv->force_pic_type.IntraRefreshVOP = OMX_FALSE;
+ priv->frame_num = 0;
+ priv->pic_order_cnt = 0;
+ priv->restricted_b_frames = debug_get_bool_option("OMX_USE_RESTRICTED_B_FRAMES", FALSE);
+ priv->scale.xWidth = OMX_VID_ENC_SCALING_WIDTH_DEFAULT;
+ priv->scale.xHeight = OMX_VID_ENC_SCALING_WIDTH_DEFAULT;
+ priv->in_port_disabled_ = false;
+ priv->out_port_disabled_ = false;
+ reset_stream_parameters(priv);
+
+ return priv;
+}
+
+static void * h264e_prc_dtor(void *ap_obj)
+{
+ h264e_prc_destroy_encoder(ap_obj);
+ return super_dtor(typeOf(ap_obj, "h264eprc"), ap_obj);
+}
+
+static OMX_ERRORTYPE h264e_prc_allocate_resources(void *ap_obj, OMX_U32 a_pid)
+{
+ vid_enc_PrivateType *priv = ap_obj;
+
+ assert(priv);
+ if (!priv->screen)
+ return OMX_ErrorInsufficientResources;
+
+ return OMX_ErrorNone;
+}
+
+static OMX_ERRORTYPE h264e_prc_deallocate_resources(void *ap_obj)
+{
+ return OMX_ErrorNone;
+}
+
+static OMX_ERRORTYPE h264e_prc_prepare_to_transfer(void *ap_obj, OMX_U32 a_pid)
+{
+ vid_enc_PrivateType *priv = ap_obj;
+
+ assert(priv);
+
+ init_port_structs(priv);
+
+ priv->eos_ = false;
+
+ struct pipe_video_codec templat = {};
+
+ templat.profile = enc_TranslateOMXProfileToPipe(priv->profile_level.eProfile);
+ templat.level = enc_TranslateOMXLevelToPipe(priv->profile_level.eLevel);
+ templat.entrypoint = PIPE_VIDEO_ENTRYPOINT_ENCODE;
+ templat.chroma_format = PIPE_VIDEO_CHROMA_FORMAT_420;
+ templat.width = priv->scale_buffer[priv->current_scale_buffer] ?
+ priv->scale.xWidth : priv->in_port_def_.format.video.nFrameWidth;
+ templat.height = priv->scale_buffer[priv->current_scale_buffer] ?
+ priv->scale.xHeight : priv->in_port_def_.format.video.nFrameHeight;
+
+ if (templat.profile == PIPE_VIDEO_PROFILE_MPEG4_AVC_BASELINE) {
+ struct pipe_screen *screen = priv->screen->pscreen;
+ templat.max_references = 1;
+ priv->stacked_frames_num =
+ screen->get_video_param(screen,
+ PIPE_VIDEO_PROFILE_MPEG4_AVC_HIGH,
+ PIPE_VIDEO_ENTRYPOINT_ENCODE,
+ PIPE_VIDEO_CAP_STACKED_FRAMES);
+ } else {
+ templat.max_references = OMX_VID_ENC_P_PERIOD_DEFAULT;
+ priv->stacked_frames_num = 1;
+ }
+ priv->codec = priv->s_pipe->create_video_codec(priv->s_pipe, &templat);
+
+ return OMX_ErrorNone;
+}
+
+static OMX_ERRORTYPE h264e_prc_transfer_and_process(void *ap_obj, OMX_U32 a_pid)
+{
+ return OMX_ErrorNone;
+}
+
+static OMX_ERRORTYPE h264e_prc_stop_and_return(void *ap_obj)
+{
+ vid_enc_PrivateType *priv = (vid_enc_PrivateType *) ap_obj;
+ return h264e_release_all_headers(priv);
+}
+
+static OMX_ERRORTYPE h264e_prc_buffers_ready(const void *ap_obj)
+{
+ vid_enc_PrivateType *priv = (vid_enc_PrivateType *) ap_obj;
+ OMX_BUFFERHEADERTYPE *in_buf = NULL;
+ OMX_BUFFERHEADERTYPE *out_buf = NULL;
+ OMX_ERRORTYPE r = OMX_ErrorNone;
+
+ assert(priv);
+
+ /* Don't get input buffer if output buffer not found */
+ while (!priv->eos_ && (out_buf = get_output_buffer(priv)) && (in_buf = get_input_buffer(priv))) {
+ if (!priv->out_port_disabled_) {
+ r = encode_frame(priv, in_buf);
+ }
+ }
+
+ return r;
+}
+
+static OMX_ERRORTYPE h264e_prc_port_flush(const void *ap_obj, OMX_U32 a_pid)
+{
+ vid_enc_PrivateType *priv = (vid_enc_PrivateType *) ap_obj;
+ if (OMX_ALL == a_pid || OMX_VID_ENC_AVC_INPUT_PORT_INDEX == a_pid) {
+ release_input_header(priv);
+ reset_stream_parameters(priv);
+ }
+ if (OMX_ALL == a_pid || OMX_VID_ENC_AVC_OUTPUT_PORT_INDEX == a_pid) {
+ release_output_header(priv);
+ }
+ return OMX_ErrorNone;
+}
+
+static OMX_ERRORTYPE h264e_prc_port_disable(const void *ap_obj, OMX_U32 a_pid)
+{
+ vid_enc_PrivateType *priv = (vid_enc_PrivateType *) ap_obj;
+ assert(priv);
+ if (OMX_ALL == a_pid || OMX_VID_ENC_AVC_INPUT_PORT_INDEX == a_pid) {
+ /* Release all buffers */
+ h264e_release_all_headers(priv);
+ reset_stream_parameters(priv);
+ priv->in_port_disabled_ = true;
+ }
+ if (OMX_ALL == a_pid || OMX_VID_ENC_AVC_OUTPUT_PORT_INDEX == a_pid) {
+ release_output_header(priv);
+ priv->out_port_disabled_ = true;
+ }
+ return OMX_ErrorNone;
+}
+
+static OMX_ERRORTYPE h264e_prc_port_enable(const void *ap_obj, OMX_U32 a_pid)
+{
+ vid_enc_PrivateType * priv = (vid_enc_PrivateType *) ap_obj;
+ assert(priv);
+ if (OMX_ALL == a_pid || OMX_VID_ENC_AVC_INPUT_PORT_INDEX == a_pid) {
+ if (priv->in_port_disabled_) {
+ reset_stream_parameters(priv);
+ priv->in_port_disabled_ = false;
+ }
+ }
+ if (OMX_ALL == a_pid || OMX_VID_ENC_AVC_OUTPUT_PORT_INDEX == a_pid) {
+ priv->out_port_disabled_ = false;
+ }
+ return OMX_ErrorNone;
+}
+
+/*
+ * h264e_prc_class
+ */
+
+static void * h264e_prc_class_ctor(void *ap_obj, va_list * app)
+{
+ /* NOTE: Class methods might be added in the future. None for now. */
+ return super_ctor(typeOf(ap_obj, "h264eprc_class"), ap_obj, app);
+}
+
+/*
+ * initialization
+ */
+
+void * h264e_prc_class_init(void * ap_tos, void * ap_hdl)
+{
+ void * tizprc = tiz_get_type(ap_hdl, "tizprc");
+ void * h264eprc_class = factory_new
+ /* TIZ_CLASS_COMMENT: class type, class name, parent, size */
+ (classOf(tizprc), "h264eprc_class", classOf(tizprc),
+ sizeof(h264e_prc_class_t),
+ /* TIZ_CLASS_COMMENT: */
+ ap_tos, ap_hdl,
+ /* TIZ_CLASS_COMMENT: class constructor */
+ ctor, h264e_prc_class_ctor,
+ /* TIZ_CLASS_COMMENT: stop value*/
+ 0);
+ return h264eprc_class;
+}
+
+void * h264e_prc_init(void * ap_tos, void * ap_hdl)
+{
+ void * tizprc = tiz_get_type(ap_hdl, "tizprc");
+ void * h264eprc_class = tiz_get_type(ap_hdl, "h264eprc_class");
+ TIZ_LOG_CLASS (h264eprc_class);
+ void * h264eprc = factory_new
+ /* TIZ_CLASS_COMMENT: class type, class name, parent, size */
+ (h264eprc_class, "h264eprc", tizprc, sizeof(vid_enc_PrivateType),
+ /* TIZ_CLASS_COMMENT: */
+ ap_tos, ap_hdl,
+ /* TIZ_CLASS_COMMENT: class constructor */
+ ctor, h264e_prc_ctor,
+ /* TIZ_CLASS_COMMENT: class destructor */
+ dtor, h264e_prc_dtor,
+ /* TIZ_CLASS_COMMENT: */
+ tiz_srv_allocate_resources, h264e_prc_allocate_resources,
+ /* TIZ_CLASS_COMMENT: */
+ tiz_srv_deallocate_resources, h264e_prc_deallocate_resources,
+ /* TIZ_CLASS_COMMENT: */
+ tiz_srv_prepare_to_transfer, h264e_prc_prepare_to_transfer,
+ /* TIZ_CLASS_COMMENT: */
+ tiz_srv_transfer_and_process, h264e_prc_transfer_and_process,
+ /* TIZ_CLASS_COMMENT: */
+ tiz_srv_stop_and_return, h264e_prc_stop_and_return,
+ /* TIZ_CLASS_COMMENT: */
+ tiz_prc_buffers_ready, h264e_prc_buffers_ready,
+ /* TIZ_CLASS_COMMENT: */
+ tiz_prc_port_flush, h264e_prc_port_flush,
+ /* TIZ_CLASS_COMMENT: */
+ tiz_prc_port_disable, h264e_prc_port_disable,
+ /* TIZ_CLASS_COMMENT: */
+ tiz_prc_port_enable, h264e_prc_port_enable,
+ /* TIZ_CLASS_COMMENT: stop value*/
+ 0);
+
+ return h264eprc;
+}