/*
 * Copyright 2015 Rockchip Electronics 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 "h264e_sps"

#include "mpp_common.h"

#include "mpp_bitwrite.h"
#include "h264e_debug.h"
#include "h264e_pps.h"

static void write_scaling_list(MppWriteCtx *bit, RK_S32 mode)
{
    switch (mode) {
    case 0 : {
        // flat scaling matrix
        /* scaling_list_present_flag */
        mpp_writer_put_bits(bit, 0, 1);
    } break;
    case 1 : {
        /* scaling_list_present_flag */
        mpp_writer_put_bits(bit, 1, 1);
        /* delta_scale */
        mpp_writer_put_se(bit, -8);
    } break;
    default : {
        mpp_err_f("unsupport scaling list mode %d\n", mode);
    } break;
    }
}

MPP_RET h264e_pps_update(H264ePps *pps, MppEncCfgSet *cfg)
{
    MppEncH264Cfg *codec = &cfg->codec.h264;

    pps->pps_id = 0;
    pps->sps_id = 0;

    pps->entropy_coding_mode = codec->entropy_coding_mode;
    pps->bottom_field_pic_order_in_frame_present = 0;
    pps->num_slice_groups = 1;

    pps->num_ref_idx_l0_default_active = 1;
    pps->num_ref_idx_l1_default_active = 1;

    pps->weighted_pred = 0;
    pps->weighted_bipred_idc = 0;

    pps->pic_init_qp = 26;
    pps->pic_init_qs = pps->pic_init_qp;

    pps->chroma_qp_index_offset = codec->chroma_cb_qp_offset;
    pps->second_chroma_qp_index_offset = codec->chroma_cb_qp_offset;
    pps->deblocking_filter_control = 1;
    pps->constrained_intra_pred = codec->constrained_intra_pred_mode;
    pps->redundant_pic_cnt = 0;

    // if (more_rbsp_data())
    pps->transform_8x8_mode = codec->transform8x8_mode;
    mpp_assert(codec->scaling_list_mode == 0 || codec->scaling_list_mode == 1);
    pps->pic_scaling_matrix_present = codec->scaling_list_mode;
    if (codec->scaling_list_mode) {
        /* NOTE: H.264 current encoder do NOT split detail matrix case */
        pps->use_default_scaling_matrix[H264_INTRA_4x4_Y] = 1;
        pps->use_default_scaling_matrix[H264_INTRA_4x4_U] = 1;
        pps->use_default_scaling_matrix[H264_INTRA_4x4_V] = 1;
        pps->use_default_scaling_matrix[H264_INTER_4x4_Y] = 1;
        pps->use_default_scaling_matrix[H264_INTER_4x4_U] = 1;
        pps->use_default_scaling_matrix[H264_INTER_4x4_V] = 1;
        pps->use_default_scaling_matrix[H264_INTRA_8x8_Y] = 1;
        pps->use_default_scaling_matrix[H264_INTER_8x8_Y] = 1;
    }

    if (codec->profile < H264_PROFILE_HIGH) {
        pps->second_chroma_qp_index_offset_present = 0;
        if (pps->transform_8x8_mode) {
            pps->transform_8x8_mode = 0;
            mpp_log_f("warning: for profile %d transform_8x8_mode should be 0\n",
                      codec->profile);
        }
    } else {
        pps->second_chroma_qp_index_offset_present = 1;
        pps->second_chroma_qp_index_offset = codec->chroma_cr_qp_offset;
    }

    if (codec->profile == H264_PROFILE_BASELINE && pps->entropy_coding_mode) {
        mpp_log_f("warning: for baseline profile entropy_coding_mode should be 0\n");
        pps->entropy_coding_mode = 0;
    }

    return MPP_OK;
}

RK_S32 h264e_pps_to_packet(H264ePps *pps, MppPacket packet, RK_S32 *offset, RK_S32 *len)
{
    void *pos = mpp_packet_get_pos(packet);
    void *data = mpp_packet_get_data(packet);
    size_t size = mpp_packet_get_size(packet);
    size_t length = mpp_packet_get_length(packet);
    void *p = pos + length;
    RK_S32 buf_size = (data + size) - (pos + length);
    MppWriteCtx bit_ctx;
    MppWriteCtx *bit = &bit_ctx;
    RK_S32 pps_size = 0;

    mpp_writer_init(bit, p, buf_size);

    /* start_code_prefix 00 00 00 01 */
    mpp_writer_put_raw_bits(bit, 0, 24);
    mpp_writer_put_raw_bits(bit, 1, 8);
    /* forbidden_zero_bit */
    mpp_writer_put_raw_bits(bit, 0, 1);
    /* nal_ref_idc */
    mpp_writer_put_raw_bits(bit, H264_NALU_PRIORITY_HIGHEST, 2);
    /* nal_unit_type */
    mpp_writer_put_raw_bits(bit, H264_NALU_TYPE_PPS, 5);

    /* pic_parameter_set_id */
    mpp_writer_put_ue(bit, pps->pps_id);
    /* seq_parameter_set_id */
    mpp_writer_put_ue(bit, pps->sps_id);
    /* entropy_coding_mode_flag */
    mpp_writer_put_bits(bit, pps->entropy_coding_mode, 1);
    /* bottom_field_pic_order_in_frame_present */
    mpp_writer_put_bits(bit, pps->bottom_field_pic_order_in_frame_present, 1);
    /* num_slice_groups_minus1 */
    mpp_writer_put_ue(bit, pps->num_slice_groups - 1);
    /* num_ref_idx_l0_active_minus1 */
    mpp_writer_put_ue(bit, pps->num_ref_idx_l0_default_active - 1);
    /* num_ref_idx_l1_active_minus1 */
    mpp_writer_put_ue(bit, pps->num_ref_idx_l1_default_active - 1);
    /* weighted_pred_flag */
    mpp_writer_put_bits(bit, pps->weighted_pred, 1);
    /* weighted_bipred_idc */
    mpp_writer_put_bits(bit, pps->weighted_bipred_idc, 2);
    /* pic_init_qp_minus26 */
    mpp_writer_put_se(bit, pps->pic_init_qp - 26);
    /* pic_init_qs_minus26 */
    mpp_writer_put_se(bit, pps->pic_init_qs - 26);
    /* chroma_qp_index_offset */
    mpp_writer_put_se(bit, pps->chroma_qp_index_offset);
    /* deblocking_filter_control_present_flag */
    mpp_writer_put_bits(bit, pps->deblocking_filter_control, 1);
    /* constrained_intra_pred_flag */
    mpp_writer_put_bits(bit, pps->constrained_intra_pred, 1);
    /* redundant_pic_cnt_present_flag */
    mpp_writer_put_bits(bit, pps->redundant_pic_cnt, 1);

    if (pps->transform_8x8_mode ||
        pps->second_chroma_qp_index_offset_present ||
        pps->pic_scaling_matrix_present) {
        /* transform_8x8_mode_flag */
        mpp_writer_put_bits(bit, pps->transform_8x8_mode, 1);

        /* pic_scaling_matrix_present_flag */
        mpp_writer_put_bits(bit, pps->pic_scaling_matrix_present, 1);
        if (pps->pic_scaling_matrix_present) {
            /* Only support default scaling list */
            /* pic_scaling_list_present_flag[i] */
            RK_S32 count = pps->transform_8x8_mode ? 8 : 6;
            RK_S32 i;

            for (i = 0; i < count; i++)
                write_scaling_list(bit, pps->use_default_scaling_matrix[i]);
        }

        /* second_chroma_qp_index_offset */
        mpp_writer_put_se(bit, pps->second_chroma_qp_index_offset);
    }

    mpp_writer_trailing(bit);

    pps_size = mpp_writer_bytes(bit);
    if (len)
        *len = pps_size;
    if (offset)
        *offset = length;

    mpp_packet_set_length(packet, length + pps_size);

    return pps_size;
}

MPP_RET h264e_pps_dump(H264ePps *pps)
{
    (void) pps;
    return MPP_OK;
}