/******************************************************************************
 *
 * 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);
    }
}