| |
| /*! |
| ************************************************************************************* |
| * \file explicit_gop.c |
| * |
| * \brief |
| * Code for explicit gop support and hierarchical coding. |
| * |
| * \author |
| * Main contributors (see contributors.h for copyright, address and affiliation details) |
| * - Alexis Michael Tourapis <alexismt@ieee.org> |
| ************************************************************************************* |
| */ |
| |
| #include <stdlib.h> |
| #include <ctype.h> |
| #include <limits.h> |
| #include "global.h" |
| #include "contributors.h" |
| #include "explicit_gop.h" |
| #include "image.h" |
| #include "nalucommon.h" |
| #include "string.h" |
| |
| |
| /*! |
| ************************************************************************ |
| * \brief |
| * Generation of hierarchical GOP |
| ************************************************************************ |
| */ |
| void create_hierarchy() |
| { |
| int i, j; |
| int centerB=input->successive_Bframe/2; |
| GOP_DATA tmp; |
| |
| if (input->HierarchicalCoding == 1) |
| { |
| for (i=0;i<input->successive_Bframe;i++) |
| { |
| if (i < centerB) |
| { |
| gop_structure[i].slice_type = B_SLICE; |
| gop_structure[i].display_no = i * 2 + 1; |
| gop_structure[i].hierarchy_layer = 1; |
| gop_structure[i].reference_idc = NALU_PRIORITY_HIGH; |
| gop_structure[i].slice_qp = imax(0, (input->qpB + (input->HierarchyLevelQPEnable ? -1: input->qpBRSOffset))); |
| |
| } |
| else |
| { |
| gop_structure[i].slice_type = B_SLICE; |
| gop_structure[i].display_no = (i - centerB) * 2; |
| gop_structure[i].hierarchy_layer = 0; |
| gop_structure[i].reference_idc = NALU_PRIORITY_DISPOSABLE; |
| gop_structure[i].slice_qp = input->qpB; |
| } |
| } |
| img->GopLevels = 2; |
| } |
| else |
| { |
| int GOPlevels = 1; |
| int Bframes = input->successive_Bframe; |
| int *curGOPLevelfrm,*curGOPLeveldist ; |
| int curlevel = GOPlevels ; |
| int i; |
| |
| while (((Bframes + 1 ) >> GOPlevels) > 1) |
| { |
| GOPlevels ++; |
| } |
| |
| curlevel = GOPlevels; |
| img->GopLevels = GOPlevels; |
| if (NULL == (curGOPLevelfrm = (int*)malloc(GOPlevels * sizeof(int)))) no_mem_exit("create_hierarchy:curGOPLevelfrm"); |
| if (NULL == (curGOPLeveldist= (int*)malloc(GOPlevels * sizeof(int)))) no_mem_exit("create_hierarchy:curGOPLeveldist"); |
| |
| for (i=0; i <input->successive_Bframe; i++) |
| { |
| gop_structure[i].display_no = i; |
| gop_structure[i].slice_type = B_SLICE; |
| gop_structure[i].hierarchy_layer = 0; |
| gop_structure[i].reference_idc = NALU_PRIORITY_DISPOSABLE; |
| gop_structure[i].slice_qp = input->qpB; |
| } |
| |
| for (j = 1; j < GOPlevels; j++) |
| { |
| for (i = (1 << j) - 1; i < Bframes + 1 - (1 << j); i += (1 << j)) { |
| gop_structure[i].hierarchy_layer = j; |
| gop_structure[i].reference_idc = NALU_PRIORITY_HIGH; |
| gop_structure[i].slice_qp = imax(0, input->qpB + (input->HierarchyLevelQPEnable ? -j: input->qpBRSOffset)); |
| } |
| } |
| |
| for (i = 1; i < Bframes; i++) |
| { |
| j = i; |
| |
| while (j > 0 && gop_structure[j].hierarchy_layer > gop_structure[j-1].hierarchy_layer) |
| { |
| tmp = gop_structure[j-1]; |
| gop_structure[j-1] = gop_structure[j]; |
| gop_structure[j] = tmp; |
| j--; |
| } |
| } |
| } |
| } |
| |
| |
| /*! |
| ************************************************************************ |
| * \brief |
| * Initialization of GOP structure. |
| * |
| ************************************************************************ |
| */ |
| void init_gop_structure() |
| { |
| int max_gopsize = input->HierarchicalCoding != 3 ? input->successive_Bframe : input->jumpd; |
| |
| gop_structure = calloc(imax(10,max_gopsize), sizeof (GOP_DATA)); // +1 for reordering |
| if (NULL==gop_structure) |
| no_mem_exit("init_gop_structure: gop_structure"); |
| } |
| |
| |
| /*! |
| ************************************************************************ |
| * \brief |
| * Clear GOP structure |
| ************************************************************************ |
| */ |
| void clear_gop_structure() |
| { |
| if (gop_structure) |
| free(gop_structure); |
| } |
| |
| |
| /*! |
| ************************************************************************ |
| * \brief |
| * Interpret GOP struct from input parameters |
| ************************************************************************ |
| */ |
| void interpret_gop_structure() |
| { |
| |
| int nLength = strlen(input->ExplicitHierarchyFormat); |
| int i =0, k, dqp, display_no; |
| int slice_read =0, order_read = 0, stored_read = 0, qp_read =0; |
| int coded_frame = 0; |
| |
| if (nLength > 0) |
| { |
| |
| for (i = 0; i < nLength ; i++) |
| { |
| //! First lets read slice type |
| if (slice_read == 0) |
| { |
| switch (input->ExplicitHierarchyFormat[i]) |
| { |
| case 'P': |
| case 'p': |
| gop_structure[coded_frame].slice_type=P_SLICE; |
| break; |
| case 'B': |
| case 'b': |
| gop_structure[coded_frame].slice_type=B_SLICE; |
| break; |
| case 'I': |
| case 'i': |
| gop_structure[coded_frame].slice_type=I_SLICE; |
| break; |
| default: |
| snprintf(errortext, ET_SIZE, "Slice Type invalid in ExplicitHierarchyFormat param. Please check configuration file."); |
| error (errortext, 400); |
| break; |
| } |
| slice_read = 1; |
| } |
| else |
| { |
| //! Next is Display Order |
| if (order_read == 0) |
| { |
| if (isdigit((int)(*(input->ExplicitHierarchyFormat+i)))) |
| { |
| sscanf(input->ExplicitHierarchyFormat+i,"%d",&display_no); |
| gop_structure[coded_frame].display_no = display_no; |
| order_read = 1; |
| if (display_no<0 || display_no>=input->jumpd) |
| { |
| snprintf(errortext, ET_SIZE, "Invalid Frame Order value. Frame position needs to be in [0,%d] range.",input->jumpd-1); |
| error (errortext, 400); |
| } |
| for (k=0;k<coded_frame;k++) |
| { |
| if (gop_structure[k].display_no == display_no) |
| { |
| snprintf(errortext, ET_SIZE, "Frame Order value %d in frame %d already used for enhancement frame %d.",display_no,coded_frame,k); |
| error (errortext, 400); |
| } |
| } |
| } |
| else |
| { |
| snprintf(errortext, ET_SIZE, "Slice Type needs to be followed by Display Order. Please check configuration file."); |
| error (errortext, 400); |
| } |
| } |
| else if (order_read == 1) |
| { |
| if (stored_read == 0 && !(isdigit((int)(*(input->ExplicitHierarchyFormat+i))))) |
| { |
| switch (input->ExplicitHierarchyFormat[i]) |
| { |
| case 'E': |
| case 'e': |
| gop_structure[coded_frame].reference_idc = NALU_PRIORITY_DISPOSABLE; |
| gop_structure[coded_frame].hierarchy_layer = 0; |
| break; |
| case 'R': |
| case 'r': |
| gop_structure[coded_frame].reference_idc= NALU_PRIORITY_HIGH; |
| gop_structure[coded_frame].hierarchy_layer = 1; |
| img->GopLevels = 2; |
| break; |
| default: |
| snprintf(errortext, ET_SIZE, "Reference_IDC invalid in ExplicitHierarchyFormat param. Please check configuration file."); |
| error (errortext, 400); |
| break; |
| } |
| stored_read = 1; |
| } |
| else if (stored_read == 1 && qp_read == 0) |
| { |
| if (isdigit((int)(*(input->ExplicitHierarchyFormat+i)))) |
| { |
| sscanf(input->ExplicitHierarchyFormat+i,"%d",&dqp); |
| |
| if (gop_structure[coded_frame].slice_type == I_SLICE) |
| gop_structure[coded_frame].slice_qp = input->qp0; |
| else if (gop_structure[coded_frame].slice_type == P_SLICE) |
| gop_structure[coded_frame].slice_qp = input->qpN; |
| else |
| gop_structure[coded_frame].slice_qp = input->qpB; |
| |
| gop_structure[coded_frame].slice_qp = iClip3(-img->bitdepth_luma_qp_scale, 51,gop_structure[coded_frame].slice_qp + dqp); |
| qp_read = 1; |
| } |
| else |
| { |
| snprintf(errortext, ET_SIZE, "Reference_IDC needs to be followed by QP. Please check configuration file."); |
| error (errortext, 400); |
| } |
| } |
| else if (stored_read == 1 && qp_read == 1 && !(isdigit((int)(*(input->ExplicitHierarchyFormat+i)))) && (i < nLength - 2)) |
| { |
| stored_read =0; |
| qp_read=0; |
| order_read=0; |
| slice_read=0; |
| i--; |
| coded_frame ++; |
| if (coded_frame >= input->jumpd ) |
| { |
| snprintf(errortext, ET_SIZE, "Total number of frames in Enhancement GOP need to be fewer or equal to FrameSkip parameter."); |
| error (errortext, 400); |
| } |
| } |
| } |
| } |
| } |
| } |
| else |
| { |
| snprintf(errortext, ET_SIZE, "ExplicitHierarchyFormat is empty. Please check configuration file."); |
| error (errortext, 400); |
| } |
| |
| input->successive_Bframe = coded_frame + 1; |
| } |
| |
| |
| /*! |
| ************************************************************************ |
| * \brief |
| * Encode Enhancement Layer. |
| ************************************************************************ |
| */ |
| void encode_enhancement_layer() |
| { |
| int previous_ref_idc = 1; |
| |
| if ((input->successive_Bframe != 0) && (IMG_NUMBER > 0)) // B-frame(s) to encode |
| { |
| if (input->PReplaceBSlice) |
| img->type = P_SLICE; // set image type to P-frame |
| else |
| img->type = B_SLICE; // set image type to B-frame |
| |
| if (input->NumFramesInELSubSeq == 0) |
| img->layer = 0; |
| else |
| img->layer = 1; |
| |
| if (input->BRefPictures != 1 && input->HierarchicalCoding==0) |
| { |
| img->frame_num++; //increment frame_num once for B-frames |
| img->frame_num %= (1 << (log2_max_frame_num_minus4 + 4)); |
| } |
| img->nal_reference_idc = 0; |
| |
| //if (input->HierarchicalCoding == 3 || input->HierarchicalCoding == 1) |
| if (input->HierarchicalCoding) |
| { |
| for(img->b_frame_to_code=1; img->b_frame_to_code<=input->successive_Bframe; img->b_frame_to_code++) |
| { |
| |
| img->nal_reference_idc = 0; |
| |
| img->type = gop_structure[img->b_frame_to_code - 1].slice_type; |
| |
| if (previous_ref_idc == 1) |
| { |
| img->frame_num++; //increment frame_num for each stored B slice |
| img->frame_num %= (1 << (log2_max_frame_num_minus4 + 4)); |
| } |
| |
| if (gop_structure[img->b_frame_to_code - 1].reference_idc== NALU_PRIORITY_HIGH ) |
| { |
| img->nal_reference_idc = 1; |
| previous_ref_idc = 1; |
| } |
| else |
| previous_ref_idc = 0; |
| |
| img->b_interval = |
| ((double) (input->jumpd + 1) / (input->successive_Bframe + 1.0) ); |
| |
| if (input->HierarchicalCoding == 3) |
| img->b_interval = 1.0; |
| |
| if(input->intra_period && input->idr_enable) |
| img->toppoc = 2*(((IMG_NUMBER%input->intra_period)-1)*(input->jumpd+1) + (int)(img->b_interval * (double)(1 + gop_structure[img->b_frame_to_code - 1].display_no))); |
| else |
| img->toppoc = 2*((IMG_NUMBER-1)*(input->jumpd + 1) + (int)(img->b_interval * (double)(1 + gop_structure[img->b_frame_to_code -1].display_no))); |
| |
| if (img->b_frame_to_code == 1) |
| img->delta_pic_order_cnt[0] = img->toppoc - 2*(start_tr_in_this_IGOP + (IMG_NUMBER)*((input->jumpd+1))); |
| else |
| img->delta_pic_order_cnt[0] = img->toppoc - 2*(start_tr_in_this_IGOP + (IMG_NUMBER-1)*((input->jumpd+1)) + (int) (2.0 *img->b_interval * (double) (1+ gop_structure[img->b_frame_to_code - 2].display_no))); |
| |
| if ((input->PicInterlace==FRAME_CODING)&&(input->MbInterlace==FRAME_CODING)) |
| img->bottompoc = img->toppoc; //progressive |
| else |
| img->bottompoc = img->toppoc+1; |
| |
| img->framepoc = imin (img->toppoc, img->bottompoc); |
| |
| img->delta_pic_order_cnt[1]= 0; // POC200301 |
| |
| encode_one_frame(); // encode one B-frame |
| if (input->ReportFrameStats) |
| report_frame_statistic(); |
| |
| if (gop_structure[img->b_frame_to_code - 1].reference_idc== NALU_PRIORITY_HIGH && img->b_frame_to_code==input->successive_Bframe) |
| { |
| img->frame_num++; //increment frame_num for each stored B slice |
| img->frame_num %= (1 << (log2_max_frame_num_minus4 + 4)); |
| } |
| } |
| img->b_frame_to_code = 0; |
| } |
| else |
| { |
| for(img->b_frame_to_code=1; img->b_frame_to_code<=input->successive_Bframe; img->b_frame_to_code++) |
| { |
| |
| img->nal_reference_idc = 0; |
| if (input->BRefPictures == 1 ) |
| { |
| img->nal_reference_idc = 1; |
| img->frame_num++; //increment frame_num once for B-frames |
| img->frame_num %= (1 << (log2_max_frame_num_minus4 + 4)); |
| } |
| |
| img->b_interval = |
| ((double) (input->jumpd + 1) / (input->successive_Bframe + 1.0) ); |
| |
| if (input->HierarchicalCoding == 3) |
| img->b_interval = 1.0; |
| |
| if(input->intra_period && input->idr_enable) |
| img->toppoc = 2*(((IMG_NUMBER% input->intra_period)-1)*(input->jumpd+1) + (int) (img->b_interval * (double)img->b_frame_to_code)); |
| else |
| img->toppoc = 2*((IMG_NUMBER-1)*(input->jumpd+1) + (int) (img->b_interval * (double)img->b_frame_to_code)); |
| |
| if ((input->PicInterlace==FRAME_CODING)&&(input->MbInterlace==FRAME_CODING)) |
| img->bottompoc = img->toppoc; //progressive |
| else |
| img->bottompoc = img->toppoc+1; |
| |
| img->framepoc = imin (img->toppoc, img->bottompoc); |
| |
| //the following is sent in the slice header |
| if (input->BRefPictures != 1) |
| { |
| img->delta_pic_order_cnt[0]= 2*(img->b_frame_to_code-1); |
| } |
| else |
| { |
| img->delta_pic_order_cnt[0]= -2; |
| } |
| |
| img->delta_pic_order_cnt[1]= 0; // POC200301 |
| |
| encode_one_frame(); // encode one B-frame |
| |
| if (input->BRefPictures == 1 && img->b_frame_to_code==input->successive_Bframe) |
| { |
| img->frame_num++; //increment frame_num for each stored B slice |
| img->frame_num %= (1 << (log2_max_frame_num_minus4 + 4)); |
| } |
| |
| if (input->ReportFrameStats) |
| report_frame_statistic(); |
| } |
| } |
| } |
| img->b_frame_to_code = 0; |
| } |
| |
| |
| void poc_based_ref_management(int current_pic_num) |
| { |
| unsigned i, pic_num = 0; |
| |
| int min_poc=INT_MAX; |
| DecRefPicMarking_t *tmp_drpm,*tmp_drpm2; |
| |
| if (img->dec_ref_pic_marking_buffer!=NULL) |
| return; |
| |
| if ((dpb.ref_frames_in_buffer+dpb.ltref_frames_in_buffer)==0) |
| return; |
| |
| for (i=0; i<dpb.used_size;i++) |
| { |
| if (dpb.fs[i]->is_reference && (!(dpb.fs[i]->is_long_term)) && dpb.fs[i]->poc < min_poc) |
| { |
| min_poc = dpb.fs[i]->frame->poc ; |
| pic_num = dpb.fs[i]->frame->pic_num; |
| } |
| } |
| |
| if (NULL==(tmp_drpm=(DecRefPicMarking_t*)calloc (1,sizeof (DecRefPicMarking_t)))) no_mem_exit("poc_based_ref_management: tmp_drpm"); |
| tmp_drpm->Next=NULL; |
| |
| tmp_drpm->memory_management_control_operation = 0; |
| |
| if (NULL==(tmp_drpm2=(DecRefPicMarking_t*)calloc (1,sizeof (DecRefPicMarking_t)))) no_mem_exit("poc_based_ref_management: tmp_drpm2"); |
| tmp_drpm2->Next=tmp_drpm; |
| |
| tmp_drpm2->memory_management_control_operation = 1; |
| tmp_drpm2->difference_of_pic_nums_minus1 = current_pic_num - pic_num - 1; |
| img->dec_ref_pic_marking_buffer = tmp_drpm2; |
| |
| } |
