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

#ifndef _FRAME_INFO_COLLECTOR_H_
#define _FRAME_INFO_COLLECTOR_H_

typedef struct
{
    /* Number of MBs in each type */
    WORD32 num_mbs[MAX_MB_TYPE];

    /* Sum of all MB SADs of each MB type  */
    WORD32 tot_mb_sad[MAX_MB_TYPE];

    /* Sum of QPs for each mb type */
    WORD32 qp_sum[MAX_MB_TYPE];

    /* Header bits consumed other than MB headers */
    WORD32 other_header_bits;

    /* Header bits consumed for each type of MBs */
    WORD32 mb_header_bits[MAX_MB_TYPE];

    /* Texture bits consumed for each type of MBs */
    WORD32 mb_texture_bits[MAX_MB_TYPE];

    /* Sum of all MB activity */
    WORD32 activity_sum;

    /* Sum of all the Intra MB cost values for the entire frame */
    WORD32 intra_mb_cost_sum;

} frame_info_t;

void irc_init_frame_info(frame_info_t *frame_info);

/*
 * Update functions: Collecting information from encoder
 */
#define FI_UPDATE_OTHER_HEADER_BITS(frame_info,header_bits)\
    {(frame_info)->other_header_bits += (header_bits);}

#define FI_UPDATE_MB_HEADER(frame_info,header_bits,mb_type)\
    {(frame_info)->mb_header_bits[(mb_type)] += (header_bits);}

#define FI_UPDATE_MB_TEXTURE(frame_info,texture_bits,mb_type)\
    {(frame_info)->mb_texture_bits[(mb_type)] += (texture_bits);}

#define FI_UPDATE_MB_SAD(frame_info,mb_sad,mb_type)\
    {(frame_info)->tot_mb_sad[(mb_type)] += (mb_sad);}

#define FI_UPDATE_MB_QP(frame_info,qp,mb_type)\
    {(frame_info)->qp_sum[(mb_type)] += (qp);(frame_info)->num_mbs[(mb_type)]++;}

#define FI_UPDATE_ACTIVITY(frame_info,mb_activity)\
    {(frame_info)->activity_sum += (mb_activity);}

#define FI_UPDATE_INTRA_MB_COST(frame_info,intra_mb_cost)\
    {(frame_info)->intra_mb_cost_sum += (intra_mb_cost);}

/*
 * GET Functions: Sending back collected information to the rate control module
 */

/* Frame Level Model Information */
WORD32 irc_fi_get_total_header_bits(frame_info_t *frame_info);

WORD32 irc_fi_get_total_texture_bits(frame_info_t *frame_info);

WORD32 irc_fi_get_average_qp(frame_info_t *frame_info);

WORD32 irc_fi_get_total_frame_sad(frame_info_t *frame_info);

WORD32 irc_fi_get_avg_activity(frame_info_t *frame_info);

/* Number of Intra MBs for Scene Change Detection */
WORD32 irc_fi_get_num_intra_mb(frame_info_t *frame_info);

/* MB Level Model Information */
WORD32 irc_fi_get_avg_mb_header(frame_info_t *frame_info, UWORD8 mb_type);

WORD32 irc_fi_get_total_mb_texture_bits(frame_info_t *frame_info,
                                        UWORD8 mb_type);

WORD32 irc_fi_get_total_mb_sad(frame_info_t *frame_info, UWORD8 mb_type);

WORD32 irc_fi_get_total_mb_qp(frame_info_t *frame_info, UWORD8 mb_type);

WORD32 irc_fi_get_total_mb(frame_info_t *frame_info, UWORD8 mb_type);

WORD32 irc_fi_get_total_intra_mb_cost(frame_info_t *frame_info);
#endif