aosp12/external/libavc/encoder/irc_frame_info_collector.c

/******************************************************************************
 *
 * 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 Includes                                                              */
/******************************************************************************/

/* User include files */
#include "irc_datatypes.h"
#include "irc_cntrl_param.h"
#include "irc_frame_info_collector.h"

void irc_init_frame_info(frame_info_t *frame_info)
{
    WORD32 i;

    for(i = 0; i < MAX_MB_TYPE; i++)
    {
        frame_info->mb_header_bits[i] = 0;
        frame_info->tot_mb_sad[i] = 0;
        frame_info->num_mbs[i] = 0;
        frame_info->qp_sum[i] = 0;
        frame_info->mb_texture_bits[i] = 0;
    }

    frame_info->other_header_bits = 0;
    frame_info->activity_sum = 0;
    frame_info->intra_mb_cost_sum = 0;
}

/******************************************************************************
 * GET Functions: Sending back collected information to the rate control module
 ******************************************************************************/
WORD32 irc_fi_get_total_header_bits(frame_info_t *frame_info)
{
    WORD32 total_header_bits = 0, i;

    for(i = 0; i < MAX_MB_TYPE; i++)
    {
        total_header_bits += frame_info->mb_header_bits[i];
    }
    total_header_bits += frame_info->other_header_bits;

    return (total_header_bits);
}

WORD32 irc_fi_get_total_texture_bits(frame_info_t *frame_info)
{
    WORD32 total_texture_bits = 0, i;

    for(i = 0; i < MAX_MB_TYPE; i++)
    {
        total_texture_bits += frame_info->mb_texture_bits[i];
    }

    return (total_texture_bits);
}

WORD32 irc_fi_get_total_frame_sad(frame_info_t *frame_info)
{
    WORD32 total_sad = 0, i;

    for(i = 0; i < MAX_MB_TYPE; i++)
    {
        total_sad += frame_info->tot_mb_sad[i];
    }

    return (total_sad);
}

WORD32 irc_fi_get_average_qp(frame_info_t *frame_info)
{
    WORD32 i, total_qp = 0, total_mbs = 0;

    for(i = 0; i < MAX_MB_TYPE; i++)
    {
        total_qp += frame_info->qp_sum[i];
        total_mbs += frame_info->num_mbs[i];
    }

    if(total_mbs)
    {
        return (total_qp / total_mbs);
    }
    else
    {
        return 0;
    }
}

WORD32 irc_fi_get_avg_mb_header(frame_info_t *frame_info, UWORD8 mb_type)
{
    if(frame_info->num_mbs[mb_type])
    {
        return (frame_info->mb_header_bits[mb_type]
                        / frame_info->num_mbs[mb_type]);
    }
    else
    {
        return 0;
    }
}

WORD32 irc_fi_get_total_mb_texture_bits(frame_info_t *frame_info,
                                        UWORD8 mb_type)
{
    return (frame_info->mb_texture_bits[mb_type]);
}

WORD32 irc_fi_get_total_mb_sad(frame_info_t *frame_info, UWORD8 mb_type)
{
    return (frame_info->tot_mb_sad[mb_type]);
}

WORD32 irc_fi_get_total_mb_qp(frame_info_t *frame_info, UWORD8 mb_type)
{
    if(frame_info->num_mbs[mb_type])
    {
        return (frame_info->qp_sum[mb_type]);
    }
    else
    {
        return 0;
    }
}

WORD32 irc_fi_get_total_mb(frame_info_t *frame_info, UWORD8 mb_type)
{
    return (frame_info->num_mbs[mb_type]);
}

WORD32 irc_fi_get_num_intra_mb(frame_info_t *frame_info)
{
    return (frame_info->num_mbs[MB_TYPE_INTRA]);
}

WORD32 irc_fi_get_avg_activity(frame_info_t *frame_info)
{
    WORD32 i;
    WORD32 i4_tot_mbs = 0;

    for(i = 0; i < MAX_MB_TYPE; i++)
    {
        i4_tot_mbs += frame_info->num_mbs[i];
    }

    if(i4_tot_mbs)
    {
        return (frame_info->activity_sum / i4_tot_mbs);
    }
    else
    {
        return 0;
    }
}

WORD32 irc_fi_get_total_intra_mb_cost(frame_info_t *frame_info)
{
    return (frame_info->intra_mb_cost_sum);
}