The-Homebrew-Cloud/Cemu
2
0
Fork 0
mirror of https://github.com/cemu-project/Cemu.git synced 2024-11-25 18:46:55 +01:00
Code Issues Packages Projects Releases Wiki Activity
0afa2d75aa
Cemu/dependencies/ih264d/common/ih264_dpb_mgr.c
Exzap d60742f52b Add all the files
2022-08-22 22:21:23 +02:00

1177 lines
36 KiB
C
Raw Blame History

/******************************************************************************
*
* 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
* ih264_dpb_mgr.c
*
* @brief
* Function definitions used for decoded picture buffer management
*
* @author
* Srinivas T
*
* @par List of Functions:
* - ih264_dpb_mgr_init()
* - ih264_dpb_mgr_sort_short_term_fields_by_frame_num()
* - ih264_dpb_mgr_sort_short_term_fields_by_poc_l0()
* - ih264_dpb_mgr_sort_short_term_fields_by_poc_l1()
* - ih264_dpb_mgr_sort_long_term_fields_by_frame_idx()
* - ih264_dpb_mgr_alternate_ref_fields()
* - ih264_dpb_mgr_insert_ref_field()
* - ih264_dpb_mgr_insert_ref_frame()
* - ih264_dpb_mgr_count_ref_frames()
* - ih264_dpb_mgr_delete_ref_frame()
* - ih264_dpb_mgr_delete_long_ref_fields_max_frame_idx()
* - ih264_dpb_mgr_delete_short_ref_frame()
* - ih264_dpb_mgr_delete_all_ref_frames()
* - ih264_dpb_mgr_reset()
* - ih264_dpb_mgr_release_pics()
*
* @remarks
* None
*
*******************************************************************************
*/
#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
#include "ih264_typedefs.h"
#include "ih264_defs.h"
#include "ih264_macros.h"
#include "ih264_error.h"
#include "ih264_structs.h"
#include "ih264_buf_mgr.h"
#include "ih264_dpb_mgr.h"
#include "ih264_debug.h"
/**
*******************************************************************************
*
* @brief
* DPB manager initializer
*
* @par Description:
* Initialises the DPB manager structure
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
void ih264_dpb_mgr_init(dpb_mgr_t *ps_dpb_mgr)
{
UWORD32 i;
dpb_info_t *ps_dpb_info = ps_dpb_mgr->as_dpb_info;
for(i = 0; i < MAX_DPB_BUFS; i++)
{
ps_dpb_info[i].ps_prev_dpb = NULL;
ps_dpb_info[i].ps_pic_buf = NULL;
ps_dpb_mgr->as_top_field_pics[i].i4_used_as_ref = INVALID;
ps_dpb_mgr->as_bottom_field_pics[i].i4_used_as_ref = INVALID;
ps_dpb_mgr->as_top_field_pics[i].i1_field_type = INVALID;
ps_dpb_mgr->as_bottom_field_pics[i].i1_field_type = INVALID;
ps_dpb_mgr->as_top_field_pics[i].i4_long_term_frame_idx = -1;
ps_dpb_mgr->as_bottom_field_pics[i].i4_long_term_frame_idx = -1;
}
ps_dpb_mgr->u1_num_short_term_ref_bufs = 0;
ps_dpb_mgr->u1_num_long_term_ref_bufs = 0;
ps_dpb_mgr->ps_dpb_short_term_head = NULL;
ps_dpb_mgr->ps_dpb_long_term_head = NULL;
}
/**
*******************************************************************************
*
* @brief
* Function to sort sort term pics by frame_num.
*
* @par Description:
* Sorts short term fields by frame_num. For 2 fields having same frame_num,
* orders them based on requested first field type.
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @param[in] curr_frame_num
* frame_num of the current pic
*
* @param[in] first_field_type
* For complementary fields, required first field
*
* @param[in] max_frame_num
* Maximum frame_num allowed
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
WORD32 ih264_dpb_mgr_sort_short_term_fields_by_frame_num(dpb_mgr_t *ps_dpb_mgr,
WORD32 curr_frame_num,
WORD32 first_field_type,
WORD32 max_frame_num)
{
dpb_info_t *ps_dpb_node1 = ps_dpb_mgr->ps_dpb_short_term_head;
dpb_info_t *ps_dpb_node2;
WORD32 frame_num_node1;
WORD32 frame_num_node2;
pic_buf_t *ps_pic_buf;
if(ps_dpb_node1 == NULL)
return -1;
for (; ps_dpb_node1 != NULL; ps_dpb_node1 = ps_dpb_node1->ps_prev_dpb)
{
for (ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb; ps_dpb_node2 != NULL; ps_dpb_node2 = ps_dpb_node2->ps_prev_dpb)
{
frame_num_node1 = ps_dpb_node1->ps_pic_buf->i4_frame_num;
frame_num_node2 = ps_dpb_node2->ps_pic_buf->i4_frame_num;
if(frame_num_node1 > curr_frame_num)
frame_num_node1 = frame_num_node1 - max_frame_num;
if(frame_num_node2 > curr_frame_num)
frame_num_node2 = frame_num_node2 - max_frame_num;
if(frame_num_node1 < frame_num_node2)
{
ps_pic_buf = ps_dpb_node1->ps_pic_buf;
ps_dpb_node1->ps_pic_buf = ps_dpb_node2->ps_pic_buf;
ps_dpb_node2->ps_pic_buf = ps_pic_buf;
}
}
}
/**
* For frames and complementary field pairs,
* ensure first_field_type appears first in the list
*/
ps_dpb_node1 = ps_dpb_mgr->ps_dpb_short_term_head;
ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb;
while(ps_dpb_node2 != NULL)
{
pic_buf_t *ps_pic_node1 = ps_dpb_node1->ps_pic_buf;
pic_buf_t *ps_pic_node2 = ps_dpb_node2->ps_pic_buf;
frame_num_node1 = ps_pic_node1->i4_frame_num;
frame_num_node2 = ps_pic_node2->i4_frame_num;
if(frame_num_node1 == frame_num_node2)
{
ASSERT(ps_pic_node1->i1_field_type != ps_pic_node2->i1_field_type);
if(ps_pic_node1->i1_field_type != first_field_type)
{
ps_dpb_node1->ps_pic_buf = ps_pic_node2;
ps_dpb_node2->ps_pic_buf = ps_pic_node1;
}
}
ps_dpb_node1 = ps_dpb_node2;
ps_dpb_node2 = ps_dpb_node2->ps_prev_dpb;
}
return 0;
}
/**
*******************************************************************************
*
* @brief
* Function to sort sort term pics by poc for list 0.
*
* @par Description:
* Orders all the pocs less than current poc in the descending order.
* Then orders all the pocs greater than current poc in the ascending order.
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @param[in] curr_poc
* Poc of the current pic
*
* @param[in] first_field_type
* For complementary fields, required first field
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
WORD32 ih264_dpb_mgr_sort_short_term_fields_by_poc_l0(dpb_mgr_t *ps_dpb_mgr,
WORD32 curr_poc,
WORD32 first_field_type)
{
dpb_info_t *ps_dpb_node1 = ps_dpb_mgr->ps_dpb_short_term_head;
dpb_info_t *ps_dpb_node2;
WORD32 poc_node1;
WORD32 poc_node2;
WORD32 frame_num_node1;
WORD32 frame_num_node2;
pic_buf_t *ps_pic_buf;
if(ps_dpb_node1 == NULL)
return -1;
/**
* Sort the fields by poc.
* All POCs less than current poc are first placed in the descending order.
* Then all POCs greater than current poc are placed in the ascending order.
*/
for (; ps_dpb_node1 != NULL; ps_dpb_node1 = ps_dpb_node1->ps_prev_dpb)
{
for (ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb; ps_dpb_node2 != NULL; ps_dpb_node2 = ps_dpb_node2->ps_prev_dpb)
{
poc_node1 = ps_dpb_node1->ps_pic_buf->i4_abs_poc;
poc_node2 = ps_dpb_node2->ps_pic_buf->i4_abs_poc;
ASSERT(poc_node1 != curr_poc);
ASSERT(poc_node2 != curr_poc);
if(((poc_node1 < curr_poc) && (poc_node2 > curr_poc)) ||
((poc_node1 < curr_poc) && (poc_node2 < curr_poc) && (poc_node1 > poc_node2)) ||
((poc_node1 > curr_poc) && (poc_node2 > curr_poc) && (poc_node1 < poc_node2)))
continue;
ps_pic_buf = ps_dpb_node1->ps_pic_buf;
ps_dpb_node1->ps_pic_buf = ps_dpb_node2->ps_pic_buf;
ps_dpb_node2->ps_pic_buf = ps_pic_buf;
}
}
ps_dpb_node1 = ps_dpb_mgr->ps_dpb_short_term_head;
ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb;
while(ps_dpb_node2 != NULL)
{
pic_buf_t *ps_pic_node1 = ps_dpb_node1->ps_pic_buf;
pic_buf_t *ps_pic_node2 = ps_dpb_node2->ps_pic_buf;
frame_num_node1 = ps_pic_node1->i4_frame_num;
frame_num_node2 = ps_pic_node2->i4_frame_num;
if(frame_num_node1 == frame_num_node2)
{
ASSERT(ps_pic_node1->i1_field_type != ps_pic_node2->i1_field_type);
if(ps_pic_node1->i1_field_type != first_field_type)
{
ps_dpb_node1->ps_pic_buf = ps_pic_node2;
ps_dpb_node2->ps_pic_buf = ps_pic_node1;
}
}
ps_dpb_node1 = ps_dpb_node2;
ps_dpb_node2 = ps_dpb_node2->ps_prev_dpb;
}
return 0;
}
/**
*******************************************************************************
*
* @brief
* Function to sort sort term pics by poc for list 1.
*
* @par Description:
* Orders all the pocs greater than current poc in the ascending order.
* Then rrders all the pocs less than current poc in the descending order.
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @param[in] curr_poc
* Poc of the current pic
*
* @param[in] first_field_type
* For complementary fields, required first field
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
WORD32 ih264_dpb_mgr_sort_short_term_fields_by_poc_l1(dpb_mgr_t *ps_dpb_mgr,
WORD32 curr_poc,
WORD32 first_field_type)
{
dpb_info_t *ps_dpb_node1 = ps_dpb_mgr->ps_dpb_short_term_head;
dpb_info_t *ps_dpb_node2;
WORD32 poc_node1;
WORD32 poc_node2;
WORD32 frame_num_node1;
WORD32 frame_num_node2;
pic_buf_t *ps_pic_buf;
if(ps_dpb_node1 == NULL)
return -1;
/**
* Sort the fields by poc.
* All POCs greater than current poc are first placed in the ascending order.
* Then all POCs less than current poc are placed in the decending order.
*/
for (; ps_dpb_node1 != NULL; ps_dpb_node1 = ps_dpb_node1->ps_prev_dpb)
{
for (ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb; ps_dpb_node2 != NULL; ps_dpb_node2 = ps_dpb_node2->ps_prev_dpb)
{
poc_node1 = ps_dpb_node1->ps_pic_buf->i4_abs_poc;
poc_node2 = ps_dpb_node2->ps_pic_buf->i4_abs_poc;
ASSERT(poc_node1 != curr_poc);
ASSERT(poc_node2 != curr_poc);
if(((poc_node1 > curr_poc) && (poc_node2 < curr_poc)) ||
((poc_node1 < curr_poc) && (poc_node2 < curr_poc) && (poc_node1 > poc_node2)) ||
((poc_node1 > curr_poc) && (poc_node2 > curr_poc) && (poc_node1 < poc_node2)))
continue;
ps_pic_buf = ps_dpb_node1->ps_pic_buf;
ps_dpb_node1->ps_pic_buf = ps_dpb_node2->ps_pic_buf;
ps_dpb_node2->ps_pic_buf = ps_pic_buf;
}
}
ps_dpb_node1 = ps_dpb_mgr->ps_dpb_short_term_head;
ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb;
while(ps_dpb_node2 != NULL)
{
pic_buf_t *ps_pic_node1 = ps_dpb_node1->ps_pic_buf;
pic_buf_t *ps_pic_node2 = ps_dpb_node2->ps_pic_buf;
frame_num_node1 = ps_pic_node1->i4_frame_num;
frame_num_node2 = ps_pic_node2->i4_frame_num;
if(frame_num_node1 == frame_num_node2)
{
ASSERT(ps_pic_node1->i1_field_type != ps_pic_node2->i1_field_type);
if(ps_pic_node1->i1_field_type != first_field_type)
{
ps_dpb_node1->ps_pic_buf = ps_pic_node2;
ps_dpb_node2->ps_pic_buf = ps_pic_node1;
}
}
ps_dpb_node1 = ps_dpb_node2;
ps_dpb_node2 = ps_dpb_node2->ps_prev_dpb;
}
return 0;
}
/**
*******************************************************************************
*
* @brief
* Function to sort long term pics by long term frame idx.
*
* @par Description:
* Sorts long term fields by long term frame idx. For 2 fields
* having same frame_num, orders them based on requested first field type.
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @param[in] first_field_type
* For complementary fields, required first field
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
WORD32 ih264_dpb_mgr_sort_long_term_fields_by_frame_idx(dpb_mgr_t *ps_dpb_mgr,
WORD32 first_field_type)
{
dpb_info_t *ps_dpb_node1 = ps_dpb_mgr->ps_dpb_long_term_head;
dpb_info_t *ps_dpb_node2;
WORD32 frame_idx_node1;
WORD32 frame_idx_node2;
pic_buf_t *ps_pic_buf;
if(ps_dpb_node1 == NULL)
return -1;
/* Sort the fields by frame idx */
for (; ps_dpb_node1 != NULL; ps_dpb_node1 = ps_dpb_node1->ps_prev_dpb)
{
for (ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb; ps_dpb_node2 != NULL; ps_dpb_node2 = ps_dpb_node2->ps_prev_dpb)
{
frame_idx_node1 = ps_dpb_node1->ps_pic_buf->i4_long_term_frame_idx;
frame_idx_node2 = ps_dpb_node2->ps_pic_buf->i4_long_term_frame_idx;
if(frame_idx_node1 > frame_idx_node2)
{
ps_pic_buf = ps_dpb_node1->ps_pic_buf;
ps_dpb_node1->ps_pic_buf = ps_dpb_node2->ps_pic_buf;
ps_dpb_node2->ps_pic_buf = ps_pic_buf;
}
}
}
/**
* For frames and complementary field pairs,
* ensure first_field_type appears first in the list
*/
ps_dpb_node1 = ps_dpb_mgr->ps_dpb_long_term_head;
ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb;
while(ps_dpb_node2 != NULL)
{
pic_buf_t *ps_pic_node1 = ps_dpb_node1->ps_pic_buf;
pic_buf_t *ps_pic_node2 = ps_dpb_node2->ps_pic_buf;
frame_idx_node1 = ps_pic_node1->i4_long_term_frame_idx;
frame_idx_node2 = ps_pic_node2->i4_long_term_frame_idx;
if(frame_idx_node1 == frame_idx_node2)
{
ASSERT(ps_pic_node1->i1_field_type != ps_pic_node2->i1_field_type);
if(ps_pic_node1->i1_field_type != first_field_type)
{
ps_dpb_node1->ps_pic_buf = ps_pic_node2;
ps_dpb_node2->ps_pic_buf = ps_pic_node1;
}
}
ps_dpb_node1 = ps_dpb_node2;
ps_dpb_node2 = ps_dpb_node2->ps_prev_dpb;
}
return 0;
}
/**
*******************************************************************************
*
* @brief
* Function to alternate fields.
*
* @par Description:
* In the ordered list of fields, alternate fields starting with
* first_field_type
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @param[in] reference_type
* This is used to select between short-term and long-term linked list.
*
* @param[in] first_field_type
* For complementary fields, required first field
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
WORD32 ih264_dpb_mgr_alternate_ref_fields(dpb_mgr_t *ps_dpb_mgr,
WORD32 reference_type,
WORD32 first_field_type)
{
dpb_info_t s_dpb_head;
dpb_info_t *ps_dpb_head;
dpb_info_t *ps_dpb_node1;
dpb_info_t *ps_dpb_node2;
dpb_info_t *ps_dpb_node3;
dpb_info_t *ps_dpb_node4;
WORD32 expected_field;
expected_field = first_field_type;
ps_dpb_head = &s_dpb_head;
ps_dpb_head->ps_prev_dpb = (reference_type == SHORT_TERM_REF) ?
ps_dpb_mgr->ps_dpb_short_term_head:
ps_dpb_mgr->ps_dpb_long_term_head;
ps_dpb_node1 = ps_dpb_head;
ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb;
while(ps_dpb_node2 != NULL)
{
pic_buf_t *ps_pic_node2 = ps_dpb_node2->ps_pic_buf;
if(ps_pic_node2->i1_field_type != expected_field)
{
/*
* If it is not expected field, loop over the node till
* the expected field.
*/
ps_dpb_node3 = ps_dpb_node2;
ps_dpb_node4 = ps_dpb_node2->ps_prev_dpb;
while((ps_dpb_node4 != NULL) &&
(ps_dpb_node4->ps_pic_buf->i1_field_type != expected_field))
{
ps_dpb_node3 = ps_dpb_node4;
ps_dpb_node4 = ps_dpb_node4->ps_prev_dpb;
}
if(ps_dpb_node4 != NULL)
{
ps_dpb_node1->ps_prev_dpb = ps_dpb_node4;
ps_dpb_node3->ps_prev_dpb = ps_dpb_node4->ps_prev_dpb;
ps_dpb_node4->ps_prev_dpb = ps_dpb_node2;
}
else
{
/* node4 null means we have reached the end */
break;
}
}
ps_dpb_node1 = ps_dpb_node1->ps_prev_dpb;
ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb;
expected_field = (ps_dpb_node1->ps_pic_buf->i1_field_type == TOP_FIELD)?
BOTTOM_FIELD:TOP_FIELD;
}
if(reference_type == SHORT_TERM_REF)
{
ps_dpb_mgr->ps_dpb_short_term_head = ps_dpb_head->ps_prev_dpb;
}
else
{
ps_dpb_mgr->ps_dpb_long_term_head = ps_dpb_head->ps_prev_dpb;
}
return 0;
}
/**
*******************************************************************************
*
* @brief
* Add a ref field to short-term or long-term linked list.
*
* @par Description:
* This function adds a ref field to either short-term or long-term linked
* list. It picks up memory for the link from the array of dpb_info in
* dpb_mgr. The field is added to the beginning of the linked list and the
* head is set the the field.
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @param[in] ps_pic_buf
* Pic buf structure for the field being added.
*
* @param[in] reference_type
* This is used to select between short-term and long-term linked list.
*
* @param[in] frame_num
* frame_num for the field.
*
* @param[in] long_term_frame_idx
* If the ref being added is long-term, long_term_frame_idx of the field.
* Otherwise invalid.
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
WORD32 ih264_dpb_mgr_insert_ref_field(dpb_mgr_t *ps_dpb_mgr,
pic_buf_t *ps_pic_buf,
WORD32 reference_type,
UWORD32 frame_num,
WORD32 long_term_frame_idx)
{
WORD32 i;
dpb_info_t *ps_dpb_info;
dpb_info_t *ps_dpb_head;
ps_dpb_info = ps_dpb_mgr->as_dpb_info;
/* Return error if buffer is already present in the DPB */
for(i = 0; i < MAX_DPB_BUFS; i++)
{
if( (ps_dpb_info[i].ps_pic_buf == ps_pic_buf)
&& (ps_dpb_info[i].ps_pic_buf->i4_used_as_ref == reference_type) )
{
return (-1);
}
}
/* Find an unused DPB location */
for(i = 0; i < MAX_DPB_BUFS; i++)
{
if(NULL == ps_dpb_info[i].ps_pic_buf)
{
break;
}
}
if(i == MAX_DPB_BUFS)
{
return (-1);
}
ps_dpb_head = (reference_type == SHORT_TERM_REF)
?ps_dpb_mgr->ps_dpb_short_term_head
:ps_dpb_mgr->ps_dpb_long_term_head;
if(reference_type == SHORT_TERM_REF)
long_term_frame_idx = -1;
/* Create DPB info */
ps_dpb_info[i].ps_pic_buf = ps_pic_buf;
ps_dpb_info[i].ps_prev_dpb = ps_dpb_head;
ps_dpb_info[i].ps_pic_buf->i4_used_as_ref = reference_type;
ps_dpb_info[i].ps_pic_buf->i4_frame_num = frame_num;
ps_dpb_info[i].ps_pic_buf->i4_long_term_frame_idx = long_term_frame_idx;
/* update the head node of linked list to point to the current picture */
if(reference_type == SHORT_TERM_REF)
{
ps_dpb_mgr->ps_dpb_short_term_head = ps_dpb_info + i;
/* Increment Short term buffer count */
ps_dpb_mgr->u1_num_short_term_ref_bufs++;
}
else
{
ps_dpb_mgr->ps_dpb_long_term_head = ps_dpb_info + i;
/* Increment Long term buffer count */
ps_dpb_mgr->u1_num_long_term_ref_bufs++;
}
return 0;
}
/**
*******************************************************************************
*
* @brief
* Add a ref frame to short-term or long-term linked list.
*
* @par Description:
* This function adds a ref frame to either short-term or long-term linked
* list. Internally it calls add ref field twice to add top and bottom field.
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @param[in] ps_pic_buf
* Pic buf structure for the field being added.
*
* @param[in] reference_type
* This is used to select between short-term and long-term linked list.
*
* @param[in] frame_num
* frame_num for the field.
*
* @param[in] long_term_frame_idx
* If the ref being added is long-term, long_term_frame_idx of the field.
* Otherwise invalid.
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
WORD32 ih264_dpb_mgr_insert_ref_frame(dpb_mgr_t *ps_dpb_mgr,
pic_buf_t *ps_pic_buf,
WORD32 reference_type,
UWORD32 frame_num,
WORD32 long_term_frame_idx)
{
WORD32 buf_id;
pic_buf_t *ps_pic_top;
pic_buf_t *ps_pic_bottom;
WORD32 ret;
/*
* For a frame, since the ps_pic_buf passed to this function is that of top field
* obtain bottom field using buf_id.
*/
ps_pic_top = ps_pic_buf;
buf_id = ps_pic_top->i4_buf_id;
ps_pic_bottom = &ps_dpb_mgr->as_bottom_field_pics[buf_id];
/* Insert top field */
ret = ih264_dpb_mgr_insert_ref_field(ps_dpb_mgr,
ps_pic_top,
reference_type,
frame_num,
long_term_frame_idx);
if(ret != 0)
return ret;
/* Insert bottom field */
ret = ih264_dpb_mgr_insert_ref_field(ps_dpb_mgr,
ps_pic_bottom,
reference_type,
frame_num,
long_term_frame_idx);
if(ret != 0)
return ret;
return ret;
}
/**
*******************************************************************************
*
* @brief
* Returns the number of ref frames in both the linked list.
*
* @par Description:
* Returns the count of number of frames, number of complementary field pairs
* and number of unpaired fields.
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @param[in] curr_frame_num
* frame_num for the field.
*
* @param[in] max_frame_num
* Maximum frame_num allowed
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
WORD32 ih264_dpb_mgr_count_ref_frames(dpb_mgr_t *ps_dpb_mgr,
WORD32 curr_frame_num,
WORD32 max_frame_num)
{
WORD32 numShortTerm = 0;
WORD32 numLongTerm = 0;
dpb_info_t *ps_dpb_node;
WORD32 frame_num;
WORD32 prev_frame_num;
/*
* Compute the number of short-term frames/complementary field pairs/
* unpaired fields
*/
if(ps_dpb_mgr->ps_dpb_short_term_head != NULL)
{
/* Sort the short-term list by frame_num */
ih264_dpb_mgr_sort_short_term_fields_by_frame_num(ps_dpb_mgr,
curr_frame_num,
TOP_FIELD,
max_frame_num);
ps_dpb_node = ps_dpb_mgr->ps_dpb_short_term_head;
if(ps_dpb_node != NULL)
{
numShortTerm++;
prev_frame_num = ps_dpb_node->ps_pic_buf->i4_frame_num;
ps_dpb_node = ps_dpb_node->ps_prev_dpb;
}
while(ps_dpb_node != NULL)
{
frame_num = ps_dpb_node->ps_pic_buf->i4_frame_num;
if(frame_num != prev_frame_num)
numShortTerm++;
prev_frame_num = ps_dpb_node->ps_pic_buf->i4_frame_num;
ps_dpb_node = ps_dpb_node->ps_prev_dpb;
}
}
/*
* Compute the number of long-term frames/complementary field pairs/
* unpaired fields
*/
if(ps_dpb_mgr->ps_dpb_long_term_head != NULL)
{
ih264_dpb_mgr_sort_long_term_fields_by_frame_idx(ps_dpb_mgr,
TOP_FIELD);
ps_dpb_node = ps_dpb_mgr->ps_dpb_long_term_head;
if(ps_dpb_node != NULL)
{
numLongTerm++;
prev_frame_num = ps_dpb_node->ps_pic_buf->i4_frame_num;
ps_dpb_node = ps_dpb_node->ps_prev_dpb;
}
while(ps_dpb_node != NULL)
{
frame_num = ps_dpb_node->ps_pic_buf->i4_frame_num;
if(frame_num != prev_frame_num)
numLongTerm++;
prev_frame_num = ps_dpb_node->ps_pic_buf->i4_frame_num;
ps_dpb_node = ps_dpb_node->ps_prev_dpb;
}
}
return (numShortTerm + numLongTerm);
}
/**
*******************************************************************************
*
* @brief
* Deletes the ref frame at the end of the linked list.
*
* @par Description:
* Deletes the ref frame at the end of the linked list. For unpaired fields,
* it deletes just the last node. For frame or complementary field pair, it
* deletes the last two nodes.
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @param[in] reference_type
* This is used to select between short-term and long-term linked list.
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
WORD32 ih264_dpb_mgr_delete_ref_frame(dpb_mgr_t *ps_dpb_mgr,
WORD32 reference_type)
{
dpb_info_t *ps_dpb_node1;
dpb_info_t *ps_dpb_node2;
dpb_info_t *ps_dpb_node3;
/*
* Assumption: The nodes sorted for frame num.
*/
/* Select bw short-term and long-term list. */
ps_dpb_node1 = (reference_type == SHORT_TERM_REF)
?ps_dpb_mgr->ps_dpb_short_term_head
:ps_dpb_mgr->ps_dpb_long_term_head;
/* If null, no entries in the list. Hence return. */
if(ps_dpb_node1 == NULL)
return 0;
/* If only one node in the list, set as unsed for refer and return. */
if(ps_dpb_node1->ps_prev_dpb == NULL)
{
/* Set the picture as unused for reference */
ps_dpb_node1->ps_pic_buf->i4_used_as_ref = UNUSED_FOR_REF;
ps_dpb_node1->ps_pic_buf = NULL;
if(reference_type == SHORT_TERM_REF)
{
ps_dpb_mgr->ps_dpb_short_term_head = NULL;
/* Increment Short term buffer count */
ps_dpb_mgr->u1_num_short_term_ref_bufs = 0;
}
else
{
ps_dpb_mgr->ps_dpb_long_term_head = NULL;
/* Increment Long term buffer count */
ps_dpb_mgr->u1_num_long_term_ref_bufs = 0;
}
return 0;
}
/**
* If there are only 2 nodes in the list, set second node as unused for reference.
* If the frame_num of second node and first node is same, set first node also as
* unused for reference and set the corresponding head to NULL.
*/
ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb;
if(ps_dpb_node2->ps_prev_dpb == NULL)
{
/* Set the picture as unused for reference */
if(ps_dpb_node2->ps_pic_buf->i4_frame_num == ps_dpb_node1->ps_pic_buf->i4_frame_num)
{
/* Set the picture as unused for reference */
ps_dpb_node1->ps_pic_buf->i4_used_as_ref = UNUSED_FOR_REF;
ps_dpb_node1->ps_pic_buf = NULL;
if(reference_type == SHORT_TERM_REF)
{
ps_dpb_mgr->ps_dpb_short_term_head = NULL;
/* Increment Short term buffer count */
ps_dpb_mgr->u1_num_short_term_ref_bufs = 0;
}
else
{
ps_dpb_mgr->ps_dpb_long_term_head = NULL;
/* Increment Long term buffer count */
ps_dpb_mgr->u1_num_long_term_ref_bufs = 0;
}
}
ps_dpb_node2->ps_pic_buf->i4_used_as_ref = UNUSED_FOR_REF;
ps_dpb_node2->ps_pic_buf = NULL;
ps_dpb_node1->ps_prev_dpb = NULL;
return 0;
}
/*
* If there are more than 2 nodes, run a loop to get the last 3 nodes.
*/
ps_dpb_node3 = ps_dpb_node2->ps_prev_dpb;
while(ps_dpb_node3->ps_prev_dpb != NULL)
{
ps_dpb_node1 = ps_dpb_node2;
ps_dpb_node2 = ps_dpb_node3;
ps_dpb_node3 = ps_dpb_node3->ps_prev_dpb;
}
/*
* If node 2 and node 3 frame_nums are same, set node 2 also as unsed for
* reference and del reference from node1.
*/
if(ps_dpb_node2->ps_pic_buf->i4_frame_num == ps_dpb_node3->ps_pic_buf->i4_frame_num)
{
ps_dpb_node2->ps_pic_buf->i4_used_as_ref = UNUSED_FOR_REF;
ps_dpb_node2->ps_pic_buf = NULL;
ps_dpb_node1->ps_prev_dpb = NULL;
}
/* Set the third node as unused for reference */
ps_dpb_node3->ps_pic_buf->i4_used_as_ref = UNUSED_FOR_REF;
ps_dpb_node3->ps_pic_buf = NULL;
ps_dpb_node2->ps_prev_dpb = NULL;
return 0;
}
/**
*******************************************************************************
*
* @brief
* Delete long-term ref fields above max frame idx.
*
* @par Description:
* Deletes all the long-term ref fields having idx greater than max_frame_idx
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @param[in] max_frame_idx
* Max long-term frame idx allowed.
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
WORD32 ih264_dpb_mgr_delete_long_ref_fields_max_frame_idx(dpb_mgr_t *ps_dpb_mgr,
WORD32 max_frame_idx)
{
dpb_info_t *ps_dpb_node1;
dpb_info_t *ps_dpb_node2;
/*
* Loop until there is node which isn't to be deleted is encountered.
*/
while(ps_dpb_mgr->ps_dpb_long_term_head != NULL)
{
if(ps_dpb_mgr->ps_dpb_long_term_head->ps_pic_buf->i4_long_term_frame_idx
<= max_frame_idx)
{
break;
}
ps_dpb_mgr->ps_dpb_long_term_head->ps_pic_buf->i4_used_as_ref = UNUSED_FOR_REF;
ps_dpb_mgr->ps_dpb_long_term_head->ps_pic_buf = NULL;
ps_dpb_mgr->ps_dpb_long_term_head = ps_dpb_mgr->ps_dpb_long_term_head->ps_prev_dpb;
}
ps_dpb_node1 = ps_dpb_mgr->ps_dpb_long_term_head;
if(ps_dpb_node1 == NULL)
return 0;
/*
* With the node that isn't to be deleted as head, loop until the end.
*/
ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb;
while(ps_dpb_node2 != NULL)
{
if(ps_dpb_node2->ps_pic_buf->i4_long_term_frame_idx > max_frame_idx)
{
ps_dpb_node2->ps_pic_buf->i4_used_as_ref = UNUSED_FOR_REF;
ps_dpb_node2->ps_pic_buf = NULL;
ps_dpb_node1->ps_prev_dpb = ps_dpb_node2->ps_prev_dpb;
}
ps_dpb_node1 = ps_dpb_node1->ps_prev_dpb;
if(ps_dpb_node1 == NULL)
break;
ps_dpb_node2 = ps_dpb_node1->ps_prev_dpb;
}
return 0;
}
/**
*******************************************************************************
*
* @brief
* Deletes the short-term with least frame_num
*
* @par Description:
* Deletes the short-term with least frame_num. It sorts the function the
* short-term linked list by frame-num and the function that deletes the last
* frame in the linked list.
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @param[in] curr_frame_num
* frame_num of the current pic
*
* @param[in] max_frame_num
* Maximum frame_num allowed
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
WORD32 ih264_dpb_mgr_delete_short_ref_frame(dpb_mgr_t *ps_dpb_mgr,
WORD32 curr_frame_num,
WORD32 max_frame_num)
{
WORD32 ret;
/* Sort the short-term list by frame_num */
ret = ih264_dpb_mgr_sort_short_term_fields_by_frame_num(ps_dpb_mgr,
curr_frame_num,
TOP_FIELD,
max_frame_num);
/* Delete the last reference frame or field */
ret = ih264_dpb_mgr_delete_ref_frame(ps_dpb_mgr,SHORT_TERM_REF);
if(ret != 0)
{
ASSERT(0);
}
return ret;
}
/**
*******************************************************************************
*
* @brief
* Deletes all the ref frames.
*
* @par Description:
* Deletes all of the ref frames/fields in the short-term and long-term linked
* list.
*
* @param[in] ps_dpb_mgr
* Pointer to the DPB manager structure
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
WORD32 ih264_dpb_mgr_delete_all_ref_frames(dpb_mgr_t *ps_dpb_mgr)
{
/* Loop over short-term linked list. */
while(ps_dpb_mgr->ps_dpb_short_term_head != NULL)
{
ih264_dpb_mgr_delete_ref_frame(ps_dpb_mgr,SHORT_TERM_REF);
}
/* Loop over long-term linked list. */
while(ps_dpb_mgr->ps_dpb_long_term_head != NULL)
{
ih264_dpb_mgr_delete_ref_frame(ps_dpb_mgr,LONG_TERM_REF);
}
return 0;
}
void ih264_dpb_mgr_reset(dpb_mgr_t *ps_dpb_mgr, buf_mgr_t *ps_buf_mgr)
{
WORD32 i;
dpb_info_t *ps_dpb_info;
ASSERT(0);
ps_dpb_info = ps_dpb_mgr->as_dpb_info;
for(i = 0; i < MAX_DPB_BUFS; i++)
{
if(ps_dpb_info[i].ps_pic_buf->i4_used_as_ref)
{
ps_dpb_info[i].ps_pic_buf->i4_used_as_ref = UNUSED_FOR_REF;
ps_dpb_info[i].ps_prev_dpb = NULL;
//Release physical buffer
ih264_buf_mgr_release(ps_buf_mgr, ps_dpb_info[i].ps_pic_buf->i4_buf_id,
BUF_MGR_REF);
ps_dpb_info[i].ps_pic_buf = NULL;
}
}
ps_dpb_mgr->u1_num_short_term_ref_bufs = 0;
ps_dpb_mgr->u1_num_long_term_ref_bufs = 0;
ps_dpb_mgr->ps_dpb_short_term_head = NULL;
ps_dpb_mgr->ps_dpb_long_term_head = NULL;
}
/**
*******************************************************************************
*
* @brief
* deletes all pictures from DPB
*
* @par Description:
* Deletes all pictures present in the DPB manager
*
* @param[in] ps_buf_mgr
* Pointer to buffer manager structure
*
* @param[in] u1_disp_bufs
* Number of buffers to be deleted
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/
void ih264_dpb_mgr_release_pics(buf_mgr_t *ps_buf_mgr, UWORD8 u1_disp_bufs)
{
WORD8 i;
UWORD32 buf_status;
ASSERT(0);
for(i = 0; i < u1_disp_bufs; i++)
{
buf_status = ih264_buf_mgr_get_status(ps_buf_mgr, i);
if(0 != buf_status)
{
ih264_buf_mgr_release((buf_mgr_t *)ps_buf_mgr, i, BUF_MGR_REF);
}
}
}
Reference in New Issue View Git Blame Copy Permalink