aboutsummaryrefslogtreecommitdiff
path: root/examples/avcenc/input.c
diff options
context:
space:
mode:
Diffstat (limited to 'examples/avcenc/input.c')
-rw-r--r--examples/avcenc/input.c346
1 files changed, 346 insertions, 0 deletions
diff --git a/examples/avcenc/input.c b/examples/avcenc/input.c
new file mode 100644
index 0000000..80a637d
--- /dev/null
+++ b/examples/avcenc/input.c
@@ -0,0 +1,346 @@
+/******************************************************************************
+ *
+ * Copyright (C) 2015 The Android Open Source Project
+ *
+ * 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.
+ *
+ *****************************************************************************
+ * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
+*/
+
+/**
+*******************************************************************************
+* @file
+* input.c
+*
+* @brief
+* Contains functions necessary for managing input buffers
+*
+* @author
+* ittiam
+*
+* @remarks
+* none
+*
+*******************************************************************************
+*/
+
+/*****************************************************************************/
+/* File Includes */
+/*****************************************************************************/
+
+/* System include files */
+#include <stdlib.h>
+#include <stdio.h>
+#include <assert.h>
+#include <string.h>
+
+/* User include files */
+#include "ih264_typedefs.h"
+#include "iv2.h"
+#include "ive2.h"
+#include "ih264e.h"
+#include "app.h"
+
+/*****************************************************************************/
+/* Constant Macros */
+/*****************************************************************************/
+
+
+/*****************************************************************************/
+/* Macros */
+/*****************************************************************************/
+
+
+/*****************************************************************************/
+/* Function Definitions */
+/*****************************************************************************/
+
+/**
+**************************************************************************
+* @brief read pic level metadata from a file
+**************************************************************************
+*/
+IV_STATUS_T read_pic_info(app_ctxt_t *ps_app_ctxt, void *pv_pic_info)
+{
+ IV_STATUS_T ret = IV_SUCCESS;
+ WORD32 size, bytes;
+
+ switch(ps_app_ctxt->u4_pic_info_type)
+ {
+ case 1:
+ size = sizeof(ih264e_pic_info1_t);
+ ps_app_ctxt->u4_pic_info_size = sizeof(ih264e_pic_info1_t);
+ break;
+ case 2:
+ size = sizeof(ih264e_pic_info2_t);
+ ps_app_ctxt->u4_pic_info_size = sizeof(ih264e_pic_info2_t);
+ break;
+ default:
+ size = 0;
+ break;
+ }
+
+ bytes = fread(pv_pic_info, 1, size, ps_app_ctxt->fp_pic_info);
+ if(bytes != size)
+ ret = IV_FAIL;
+
+ return ret;
+}
+
+/**
+**************************************************************************
+* @brief read mb level metadata from a file
+**************************************************************************
+*/
+IV_STATUS_T read_mb_info(app_ctxt_t *ps_app_ctxt, void *pv_mb_info)
+{
+ IV_STATUS_T ret = IV_SUCCESS;
+ WORD32 num_mbs;
+ WORD32 size;
+ WORD32 bytes;
+
+ num_mbs = ALIGN16(ps_app_ctxt->u4_wd) * ALIGN16(ps_app_ctxt->u4_ht);
+ num_mbs /= 256;
+
+ switch(ps_app_ctxt->u4_mb_info_type)
+ {
+ case 1:
+ size = sizeof(ih264e_mb_info1_t) * num_mbs;
+ ps_app_ctxt->u4_mb_info_size = sizeof(ih264e_mb_info1_t);
+ break;
+ case 2:
+ size = sizeof(ih264e_mb_info2_t) * num_mbs;
+ ps_app_ctxt->u4_mb_info_size = sizeof(ih264e_mb_info2_t);
+ break;
+ case 3:
+ size = sizeof(ih264e_mb_info3_t) * num_mbs;
+ ps_app_ctxt->u4_mb_info_size = sizeof(ih264e_mb_info3_t);
+ break;
+ case 4:
+ size = sizeof(ih264e_mb_info4_t) * num_mbs;
+ ps_app_ctxt->u4_mb_info_size = sizeof(ih264e_mb_info4_t);
+ break;
+ default:
+ size = 0;
+ break;
+ }
+
+ bytes = fread(pv_mb_info, 1, size, ps_app_ctxt->fp_mb_info);
+ if(bytes != size)
+ ret = IV_FAIL;
+
+ return ret;
+}
+
+/**
+**************************************************************************
+* @brief read input from a file
+**************************************************************************
+*/
+IV_STATUS_T read_input(FILE *fp, iv_raw_buf_t *ps_raw_buf)
+{
+ WORD32 i;
+
+ if(IV_YUV_422ILE == ps_raw_buf->e_color_fmt)
+ {
+ WORD32 wd = ps_raw_buf->au4_wd[0];
+ WORD32 ht = ps_raw_buf->au4_ht[0];
+ WORD32 strd = ps_raw_buf->au4_strd[0];
+ UWORD8 *pu1_buf = ps_raw_buf->apv_bufs[0];
+
+ for(i = 0; i < ht; i++)
+ {
+ WORD32 bytes = fread(pu1_buf, sizeof(UWORD8), wd, fp);
+
+ if(bytes != wd)
+ {
+ return (IV_FAIL);
+ }
+ pu1_buf += strd;
+ }
+ }
+ else
+ {
+ WORD32 num_comp = 2;
+ WORD32 comp_idx;
+
+ if(IV_YUV_420P == ps_raw_buf->e_color_fmt)
+ num_comp = 3;
+
+ for(comp_idx = 0; comp_idx < num_comp; comp_idx++)
+ {
+ WORD32 wd = ps_raw_buf->au4_wd[comp_idx];
+ WORD32 ht = ps_raw_buf->au4_ht[comp_idx];
+ WORD32 strd = ps_raw_buf->au4_strd[comp_idx];
+ UWORD8 *pu1_buf = ps_raw_buf->apv_bufs[comp_idx];
+
+ for(i = 0; i < ht; i++)
+ {
+ WORD32 bytes = fread(pu1_buf, sizeof(UWORD8), wd, fp);
+
+ if(bytes != wd)
+ {
+ return (IV_FAIL);
+ }
+ pu1_buf += strd;
+ }
+ }
+ }
+ return IV_SUCCESS;
+}
+
+/**
+**************************************************************************
+* @brief write input to a file
+**************************************************************************
+*/
+IV_STATUS_T dump_input(FILE *fp, iv_raw_buf_t *ps_raw_buf)
+{
+ WORD32 i;
+
+ if(IV_YUV_422ILE == ps_raw_buf->e_color_fmt)
+ {
+ WORD32 wd = ps_raw_buf->au4_wd[0];
+ WORD32 ht = ps_raw_buf->au4_ht[0];
+ WORD32 strd = ps_raw_buf->au4_strd[0];
+ UWORD8 *pu1_buf = ps_raw_buf->apv_bufs[0];
+
+ for(i = 0; i < ht; i++)
+ {
+ WORD32 bytes = fwrite(pu1_buf, sizeof(UWORD8), wd, fp);
+
+ if(bytes != wd)
+ {
+ return (IV_FAIL);
+ }
+ pu1_buf += strd;
+ }
+ }
+ else
+ {
+ WORD32 num_comp = 2;
+ WORD32 comp_idx;
+
+ if(IV_YUV_420P == ps_raw_buf->e_color_fmt)
+ num_comp = 3;
+
+ for(comp_idx = 0; comp_idx < num_comp; comp_idx++)
+ {
+ WORD32 wd = ps_raw_buf->au4_wd[comp_idx];
+ WORD32 ht = ps_raw_buf->au4_ht[comp_idx];
+ WORD32 strd = ps_raw_buf->au4_strd[comp_idx];
+ UWORD8 *pu1_buf = ps_raw_buf->apv_bufs[comp_idx];
+
+ for(i = 0; i < ht; i++)
+ {
+ WORD32 bytes = fwrite(pu1_buf, sizeof(UWORD8), wd, fp);
+
+ if(bytes != wd)
+ {
+ return (IV_FAIL);
+ }
+ pu1_buf += strd;
+ }
+ }
+ }
+ return IV_SUCCESS;
+}
+
+/**
+**************************************************************************
+* @brief allocate input buffers
+**************************************************************************
+*/
+void allocate_input(app_ctxt_t *ps_app_ctxt)
+{
+ WORD32 num_bufs;
+ WORD32 luma_size = ps_app_ctxt->u4_wd * ps_app_ctxt->u4_ht;
+ WORD32 chroma_size = luma_size >> 1;
+ WORD32 pic_size = luma_size + chroma_size * 2;
+ WORD32 num_mbs;
+ WORD32 i;
+ ih264e_ctl_getbufinfo_op_t *ps_get_buf_info_op = &ps_app_ctxt->s_get_buf_info_op;
+
+ num_bufs = MAX(DEFAULT_NUM_INPUT_BUFS,
+ ps_get_buf_info_op->s_ive_op.u4_min_inp_bufs);
+ num_bufs = MIN(DEFAULT_MAX_INPUT_BUFS, num_bufs);
+
+ num_mbs = ALIGN16(ps_app_ctxt->u4_max_wd) * ALIGN16(ps_app_ctxt->u4_max_ht);
+ num_mbs /= 256;
+
+ /* Memset the input buffer array to set is_free to 0 */
+ memset(ps_app_ctxt->as_input_buf, 0,
+ sizeof(input_buf_t) * DEFAULT_MAX_INPUT_BUFS);
+
+ for(i = 0; i < num_bufs; i++)
+ {
+ UWORD8 *pu1_buf = (UWORD8 *)ih264a_aligned_malloc(16, pic_size);
+ if(NULL == pu1_buf)
+ {
+ CHAR ac_error[STRLENGTH];
+ sprintf(ac_error, "Allocation failed for input buffer of size %d\n",
+ pic_size);
+ codec_exit(ac_error);
+ }
+ ps_app_ctxt->as_input_buf[i].pu1_buf = pu1_buf;
+
+ pu1_buf = (UWORD8 *)ih264a_aligned_malloc(
+ 16, num_mbs * sizeof(ih264e_mb_info_t));
+ if(NULL == pu1_buf)
+ {
+ CHAR ac_error[STRLENGTH];
+ sprintf(ac_error,
+ "Allocation failed for mb info buffer of size %d\n",
+ (WORD32)(num_mbs * sizeof(ih264e_mb_info_t)));
+ codec_exit(ac_error);
+ }
+ ps_app_ctxt->as_input_buf[i].pv_mb_info = pu1_buf;
+ pu1_buf = (UWORD8 *)ih264a_aligned_malloc(16,
+ sizeof(ih264e_pic_info2_t));
+ if(NULL == pu1_buf)
+ {
+ CHAR ac_error[STRLENGTH];
+ sprintf(ac_error,
+ "Allocation failed for pic info buffer of size %d\n",
+ (WORD32)sizeof(ih264e_pic_info2_t));
+ codec_exit(ac_error);
+ }
+ ps_app_ctxt->as_input_buf[i].pv_pic_info = pu1_buf;
+ ps_app_ctxt->as_input_buf[i].u4_buf_size = pic_size;
+ ps_app_ctxt->as_input_buf[i].u4_is_free = 1;
+ }
+}
+
+/**
+**************************************************************************
+* @brief free input buffers
+**************************************************************************
+*/
+void free_input(app_ctxt_t *ps_app_ctxt)
+{
+ WORD32 num_bufs;
+ WORD32 i;
+
+ num_bufs = MAX(DEFAULT_NUM_INPUT_BUFS,
+ ps_app_ctxt->s_get_buf_info_op.s_ive_op.u4_min_inp_bufs);
+ num_bufs = MIN(DEFAULT_MAX_INPUT_BUFS, num_bufs);
+
+ for(i = 0; i < num_bufs; i++)
+ {
+ ih264a_aligned_free(ps_app_ctxt->as_input_buf[i].pu1_buf);
+ ih264a_aligned_free(ps_app_ctxt->as_input_buf[i].pv_mb_info);
+ ih264a_aligned_free(ps_app_ctxt->as_input_buf[i].pv_pic_info);
+ }
+}
+
generated by cgit v1.2.3 (git 2.25.1) at 2024-05-25 17:22:59 +0000