aboutsummaryrefslogtreecommitdiff
path: root/examples/svcenc/recon.c
diff options
context:
space:
mode:
Diffstat (limited to 'examples/svcenc/recon.c')
-rw-r--r--examples/svcenc/recon.c215
1 files changed, 215 insertions, 0 deletions
diff --git a/examples/svcenc/recon.c b/examples/svcenc/recon.c
new file mode 100644
index 0000000..c43d39e
--- /dev/null
+++ b/examples/svcenc/recon.c
@@ -0,0 +1,215 @@
+/******************************************************************************
+ *
+ * Copyright (C) 2022 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 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 "isvce.h"
+#include "app.h"
+
+/*****************************************************************************/
+/* Constant Macros */
+/*****************************************************************************/
+
+/*****************************************************************************/
+/* Macros */
+/*****************************************************************************/
+
+/*****************************************************************************/
+/* Function Declarations */
+/*****************************************************************************/
+
+IV_STATUS_T write_recon(FILE *fp, iv_raw_buf_t *ps_raw_buf)
+{
+ WORD32 bytes;
+ WORD32 wd, ht;
+ UWORD8 *pu1_buf;
+ WORD32 i;
+ WORD32 comp;
+ WORD32 num_comp;
+
+ num_comp = 2;
+ if(IV_YUV_420P == ps_raw_buf->e_color_fmt) num_comp = 3;
+
+ for(comp = 0; comp < num_comp; comp++)
+ {
+ wd = ps_raw_buf->au4_wd[comp];
+ ht = ps_raw_buf->au4_ht[comp];
+ pu1_buf = ps_raw_buf->apv_bufs[comp];
+ for(i = 0; i < ht; i++)
+ {
+ bytes = (WORD32) fwrite(pu1_buf, sizeof(UWORD8), wd, fp);
+ if(bytes != wd)
+ {
+ return (IV_FAIL);
+ }
+ pu1_buf += wd;
+ }
+ }
+
+ fflush(fp);
+ return IV_SUCCESS;
+}
+
+void allocate_recon(app_ctxt_t *ps_app_ctxt)
+{
+ WORD32 num_bufs;
+ WORD32 pic_size;
+ WORD32 luma_size;
+ WORD32 chroma_size;
+ WORD32 i;
+ UWORD8 *pu1_buf;
+
+ num_bufs = DEFAULT_NUM_RECON_BUFS;
+
+ assert(ps_app_ctxt->s_get_buf_info_op.u4_rec_comp_cnt == 3);
+
+ /* Size of buffer for YUV420/420SP */
+ luma_size = ps_app_ctxt->s_get_buf_info_op.au4_min_rec_buf_size[0];
+ chroma_size = ps_app_ctxt->s_get_buf_info_op.au4_min_rec_buf_size[1] +
+ ps_app_ctxt->s_get_buf_info_op.au4_min_rec_buf_size[2];
+ pic_size = luma_size + chroma_size;
+
+ for(i = 0; i < num_bufs; i++)
+ {
+ pu1_buf = (UWORD8 *) isvca_aligned_malloc(16, pic_size);
+
+ if(NULL == pu1_buf)
+ {
+ CHAR ac_error[STRLENGTH];
+ snprintf(ac_error, sizeof(ac_error) - 1,
+ "Allocation failed for recon buffer of size %d\n", pic_size);
+ codec_exit(ac_error);
+ }
+
+ ps_app_ctxt->as_recon_buf[i].pu1_buf = pu1_buf;
+ ps_app_ctxt->as_recon_buf[i].u4_buf_size = pic_size;
+ ps_app_ctxt->as_recon_buf[i].u4_is_free = 1;
+ }
+
+ if(ps_app_ctxt->u4_psnr_enable)
+ {
+ pu1_buf = (UWORD8 *) isvca_aligned_malloc(16, pic_size);
+
+ if(NULL == pu1_buf)
+ {
+ CHAR ac_error[STRLENGTH];
+ snprintf(ac_error, sizeof(ac_error) - 1,
+ "Allocation failed for recon buffer of size %d\n", pic_size);
+ codec_exit(ac_error);
+ }
+
+ ps_app_ctxt->pu1_psnr_buf = pu1_buf;
+ ps_app_ctxt->u4_psnr_buf_size = pic_size;
+ }
+}
+
+void free_recon(app_ctxt_t *ps_app_ctxt)
+{
+ WORD32 num_bufs;
+ WORD32 i;
+
+ num_bufs = DEFAULT_NUM_RECON_BUFS;
+
+ for(i = 0; i < num_bufs; i++)
+ {
+ isvca_aligned_free(ps_app_ctxt->as_recon_buf[i].pu1_buf);
+ }
+
+ if(ps_app_ctxt->u4_psnr_enable)
+ {
+ isvca_aligned_free(ps_app_ctxt->pu1_psnr_buf);
+ }
+}
+
+void init_raw_buf_descr(app_ctxt_t *ps_app_ctxt, iv_raw_buf_t *ps_raw_buf, UWORD8 *pu1_buf,
+ IV_COLOR_FORMAT_T e_color_fmt)
+{
+ WORD32 luma_size;
+ WORD32 au4_chroma_sizes[2];
+
+ assert(IV_YUV_420P == e_color_fmt);
+
+ luma_size = ps_app_ctxt->s_get_buf_info_op.au4_min_rec_buf_size[0];
+ au4_chroma_sizes[0] = ps_app_ctxt->s_get_buf_info_op.au4_min_rec_buf_size[1];
+ au4_chroma_sizes[1] = ps_app_ctxt->s_get_buf_info_op.au4_min_rec_buf_size[2];
+
+ ps_raw_buf->apv_bufs[0] = pu1_buf;
+ pu1_buf += luma_size;
+
+ ps_raw_buf->apv_bufs[1] = pu1_buf;
+ pu1_buf += au4_chroma_sizes[0];
+
+ ps_raw_buf->apv_bufs[2] = NULL;
+ if(IV_YUV_420P == e_color_fmt)
+ {
+ ps_raw_buf->apv_bufs[2] = pu1_buf;
+ }
+
+ ps_raw_buf->e_color_fmt = e_color_fmt;
+ ps_raw_buf->au4_wd[0] = ps_app_ctxt->u4_enc_wd;
+ ps_raw_buf->au4_ht[0] = ps_app_ctxt->u4_enc_ht;
+ ps_raw_buf->au4_strd[0] = ps_app_ctxt->u4_enc_wd;
+
+ /* Initialize for 420SP */
+ {
+ ps_raw_buf->au4_wd[1] = ps_app_ctxt->u4_enc_wd;
+ ps_raw_buf->au4_wd[2] = 0;
+
+ ps_raw_buf->au4_ht[1] = ps_app_ctxt->u4_enc_ht / 2;
+ ps_raw_buf->au4_ht[2] = 0;
+
+ ps_raw_buf->au4_strd[1] = ps_app_ctxt->u4_enc_wd;
+ ps_raw_buf->au4_strd[2] = 0;
+ }
+
+ if(IV_YUV_420P == e_color_fmt)
+ {
+ ps_raw_buf->au4_wd[1] = ps_app_ctxt->u4_enc_wd / 2;
+ ps_raw_buf->au4_wd[2] = ps_app_ctxt->u4_enc_wd / 2;
+
+ ps_raw_buf->au4_ht[1] = ps_app_ctxt->u4_enc_ht / 2;
+ ps_raw_buf->au4_ht[2] = ps_app_ctxt->u4_enc_ht / 2;
+
+ ps_raw_buf->au4_strd[1] = ps_app_ctxt->u4_enc_wd / 2;
+ ps_raw_buf->au4_strd[2] = ps_app_ctxt->u4_enc_wd / 2;
+ }
+ /* If stride is not initialized, then use width as stride */
+ if(0 == ps_raw_buf->au4_strd[0])
+ {
+ ps_raw_buf->au4_strd[0] = ps_raw_buf->au4_wd[0];
+ ps_raw_buf->au4_strd[1] = ps_raw_buf->au4_wd[1];
+ ps_raw_buf->au4_strd[2] = ps_raw_buf->au4_wd[2];
+ }
+
+ ps_raw_buf->u4_size = sizeof(iv_raw_buf_t);
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-05-25 22:13:29 +0000