/***************************************************************************** * This file is part of uvg266 VVC encoder. * * Copyright (c) 2021, Tampere University, ITU/ISO/IEC, project contributors * All rights reserved. * * Redistribution and use in source and binary forms, with or without modification, * are permitted provided that the following conditions are met: * * * Redistributions of source code must retain the above copyright notice, this * list of conditions and the following disclaimer. * * * Redistributions in binary form must reproduce the above copyright notice, this * list of conditions and the following disclaimer in the documentation and/or * other materials provided with the distribution. * * * Neither the name of the Tampere University or ITU/ISO/IEC nor the names of its * contributors may be used to endorse or promote products derived from * this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES * INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION HOWEVER CAUSED AND ON * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. * INCLUDING NEGLIGENCE OR OTHERWISE ARISING IN ANY WAY OUT OF THE USE OF THIS ****************************************************************************/ /** * \file * Functions for debugging purposes. */ #include "global.h" #include "debug.h" #include "encoderstate.h" #include "inter.h" #ifdef KVZ_DEBUG_PRINT_YUVIEW_CSV #define INITIAL_ALLOC 10000 FILE* yuview_output; struct yuview_data { int poc; int bytes_allocated[DBG_YUVIEW_NUM_ITEMS]; int last_pos[DBG_YUVIEW_NUM_ITEMS]; char* items[DBG_YUVIEW_NUM_ITEMS]; }; struct yuview_data** yuview_frame_data; int yuview_frames; static int yuview_find_poc(int poc) { for (int i = 0; i < yuview_frames; i++) { if (yuview_frame_data[i]->poc == poc) return i; } return -1; } static void yuview_alloc_frame(int poc) { assert(yuview_frame_data != NULL); yuview_frame_data = realloc(yuview_frame_data, (yuview_frames + 1) * sizeof(struct yuview_data*)); struct yuview_data* data = malloc(sizeof(struct yuview_data)); memset(data->last_pos, 0, sizeof(data->last_pos[0]) * DBG_YUVIEW_NUM_ITEMS); for (int i = 0; i < DBG_YUVIEW_NUM_ITEMS; i++) { data->bytes_allocated[i] = INITIAL_ALLOC; data->items[i] = malloc(INITIAL_ALLOC); } data->poc = poc; yuview_frame_data[yuview_frames] = data; yuview_frames++; } void kvz_dbg_yuview_init(const encoder_control_t* const encoder, char* filename, char* sequence) { char buf[100]; snprintf(buf, 100, "%s.csv", filename); yuview_output = fopen(buf, "wb"); yuview_frame_data = malloc(1); yuview_frames = 0; fprintf(yuview_output, "%%;syntax-version;v1.22\r\n"); fprintf(yuview_output, "%%;seq-specs;%s;layer2;%d;%d;%f\r\n", sequence, encoder->in.width, encoder->in.height, encoder->cfg.framerate); fprintf(yuview_output, "%%;type;0;CU-type;range\r\n"); fprintf(yuview_output, "%%;defaultRange;0;2;jet\r\n"); fprintf(yuview_output, "%%;type;1;IntraDirLuma;range\r\n"); fprintf(yuview_output, "%%;defaultRange;0;67;autumn\r\n"); fprintf(yuview_output, "%%;type;2;IntraDirChroma;range\r\n"); fprintf(yuview_output, "%%;defaultRange;0;67;autumn\r\n"); fprintf(yuview_output, "%%;type;3;RefIdxSkipL0;range\r\n"); fprintf(yuview_output, "%%;range;0;4;0;65;75;154;160;255;255;255\r\n"); fprintf(yuview_output, "%%;type;4;RefIdxSkipL1;range\r\n"); fprintf(yuview_output, "%%;range;0;4;53;128;0;66;160;255;255;255\r\n"); fprintf(yuview_output, "%%;type;5;RefIdxMergeL0;range\r\n"); fprintf(yuview_output, "%%;range;0;4;75;154;160;255;0;65;255;255\r\n"); fprintf(yuview_output, "%%;type;6;RefIdxMergeL1;range\r\n"); fprintf(yuview_output, "%%;range;0;4;0;65;160;255;56;132;255;255\r\n"); fprintf(yuview_output, "%%;type;7;RefIdxInterL0;range\r\n"); fprintf(yuview_output, "%%;range;0;4;160;255;113;199;0;65;255;255\r\n"); fprintf(yuview_output, "%%;type;8;RefIdxInterL1;range\r\n"); fprintf(yuview_output, "%%;range;0;4;146;239;160;255;0;65;255;255\r\n"); fprintf(yuview_output, "%%;type;9;MVSkipL0;vector\r\n"); fprintf(yuview_output, "%%;vectorColor;0;0;0;255\r\n"); fprintf(yuview_output, "%%;scaleFactor;16\r\n"); fprintf(yuview_output, "%%;type;10;MVSkipL1;vector\r\n"); fprintf(yuview_output, "%%;vectorColor;255;255;255;255\r\n"); fprintf(yuview_output, "%%;scaleFactor;16\r\n"); fprintf(yuview_output, "%%;type;11;MVMergeL0;vector\r\n"); fprintf(yuview_output, "%%;vectorColor;0;0;0;255\r\n"); fprintf(yuview_output, "%%;scaleFactor;16\r\n"); fprintf(yuview_output, "%%;type;12;MVMergeL1;vector\r\n"); fprintf(yuview_output, "%%;vectorColor;255;255;255;255\r\n"); fprintf(yuview_output, "%%;scaleFactor;16\r\n"); fprintf(yuview_output, "%%;type;13;MVInterL0;vector\r\n"); fprintf(yuview_output, "%%;vectorColor;0;0;0;255\r\n"); fprintf(yuview_output, "%%;scaleFactor;16\r\n"); fprintf(yuview_output, "%%;type;14;MVInterL1;vector\r\n"); fprintf(yuview_output, "%%;vectorColor;255;255;255;255\r\n"); fprintf(yuview_output, "%%;scaleFactor;16\r\n"); fprintf(yuview_output, "%%;type;15;NumSigCoeffY;range\r\n"); fprintf(yuview_output, "%%;range;0;1024;0;255;0;255;0;255;255;255\r\n"); fprintf(yuview_output, "%%;scaleToBlockSize;1\r\n"); fprintf(yuview_output, "%%;type;16;NumSigCoeffU;range\r\n"); fprintf(yuview_output, "%%;range;0;1024;0;255;0;255;0;255;255;255\r\n"); fprintf(yuview_output, "%%;scaleToBlockSize;1\r\n"); fprintf(yuview_output, "%%;type;17;NumSigCoeffV;range\r\n"); fprintf(yuview_output, "%%;range;0;1024;0;255;0;255;0;255;255;255\r\n"); fprintf(yuview_output, "%%;scaleToBlockSize;1\r\n"); } static int yuview_check_allocated_memory(int poc, int type) { int idx = yuview_find_poc(poc); if (idx == -1) yuview_alloc_frame(poc); idx = yuview_frames - 1; if (yuview_frame_data[idx]->bytes_allocated[type] - yuview_frame_data[idx]->last_pos[type] < 1000) { yuview_frame_data[idx]->items[type] = realloc(yuview_frame_data[idx]->items[type], yuview_frame_data[idx]->bytes_allocated[type] * 2); yuview_frame_data[idx]->bytes_allocated[type] *= 2; } return idx; } void kvz_dbg_yuview_add_vector(int poc, int x, int y, int width, int height, int type, int x_vec, int y_vec) { int idx = yuview_check_allocated_memory(poc, type); char* buffer = &yuview_frame_data[idx]->items[type][ yuview_frame_data[idx]->last_pos[type] ]; int space_left = yuview_frame_data[idx]->bytes_allocated[type] - yuview_frame_data[idx]->last_pos[type]; yuview_frame_data[idx]->last_pos[type] += snprintf(buffer, space_left, "%d;%d;%d;%d;%d;%d;%d;%d\r\n", poc, x, y, width, height, type, x_vec, y_vec); } void kvz_dbg_yuview_add(int poc, int x, int y, int width, int height, int type, int val) { int idx = yuview_check_allocated_memory(poc, type); char* buffer = &yuview_frame_data[idx]->items[type][ yuview_frame_data[idx]->last_pos[type] ]; int space_left = yuview_frame_data[idx]->bytes_allocated[type] - yuview_frame_data[idx]->last_pos[type]; yuview_frame_data[idx]->last_pos[type] += snprintf(buffer, space_left, "%d;%d;%d;%d;%d;%d;%d\r\n", poc, x, y, width, height, type, val); } void kvz_dbg_yuview_finish_frame(int poc) { int idx = yuview_find_poc(poc); if (idx == -1) return; struct yuview_data* data = yuview_frame_data[idx]; for (int i = 0; i < DBG_YUVIEW_NUM_ITEMS; i++) { if (data->last_pos[i] == 0) continue; data->items[i][data->last_pos[i]] = '\0'; fprintf(yuview_output, "%s", data->items[i]); free(data->items[i]); } free(yuview_frame_data[idx]); for (int i = idx + 1; i < yuview_frames; i++) { yuview_frame_data[i - 1] = yuview_frame_data[i]; } yuview_frames--; } void kvz_dbg_yuview_cleanup() { fclose(yuview_output); yuview_output = NULL; for (int idx = 0; idx < yuview_frames; idx++) { struct yuview_data* data = yuview_frame_data[idx]; for (int i = 0; i < DBG_YUVIEW_NUM_ITEMS; i++) { free(data->items[i]); } free(yuview_frame_data[idx]); } free(yuview_frame_data); } #endif //KVZ_DEBUG_PRINT_YUVIEW_CSV #ifdef KVZ_DEBUG_PRINT_THREADING_INFO void kvz_dbg_encoder_state_dump_graphviz(const encoder_state_t* const state) { int i; if (!state->parent) { const encoder_control_t* const encoder = state->encoder_control; int y, x; //Empty lines (easier to copy-paste) printf("\n\n\n\n\n"); //Some styling... printf("digraph EncoderStates {\n"); printf(" fontname = \"Bitstream Vera Sans\"\n"); printf(" fontsize = 8\n\n"); printf(" node [\n"); printf(" fontname = \"Bitstream Vera Sans\"\n"); printf(" fontsize = 8\n"); printf(" shape = \"record\"\n"); printf(" ]\n\n"); printf(" edge [\n"); printf(" arrowtail = \"empty\"\n"); printf(" ]\n\n"); printf(" \"Map\" [\n"); printf(" shape=plaintext\n"); printf(" label = <"); printf("", encoder->in.width_in_lcu); for (y = 0; y < encoder->in.height_in_lcu; ++y) { printf(""); for (x = 0; x < encoder->in.width_in_lcu; ++x) { const int lcu_id_rs = y * encoder->in.width_in_lcu + x; printf("", lcu_id_rs); } printf(""); } printf("", encoder->in.width_in_lcu); for (y = 0; y < encoder->in.height_in_lcu; ++y) { printf(""); for (x = 0; x < encoder->in.width_in_lcu; ++x) { const int lcu_id_rs = y * encoder->in.width_in_lcu + x; const int lcu_id_ts = encoder->tiles_ctb_addr_rs_to_ts[lcu_id_rs]; printf("", lcu_id_ts); } printf(""); } printf("", encoder->in.width_in_lcu); for (y = 0; y < encoder->in.height_in_lcu; ++y) { printf(""); for (x = 0; x < encoder->in.width_in_lcu; ++x) { const int lcu_id_rs = y * encoder->in.width_in_lcu + x; const int lcu_id_ts = encoder->tiles_ctb_addr_rs_to_ts[lcu_id_rs]; printf("", encoder->tiles_tile_id[lcu_id_ts]); } printf(""); } printf("", encoder->in.width_in_lcu); for (y = 0; y < encoder->in.height_in_lcu; ++y) { printf(""); for (x = 0; x < encoder->in.width_in_lcu; ++x) { const int lcu_id_rs = y * encoder->in.width_in_lcu + x; const int lcu_id_ts = encoder->tiles_ctb_addr_rs_to_ts[lcu_id_rs]; int slice_id = 0; //Not efficient, but who cares for (i = 0; i < encoder->slice_count; ++i) { if (encoder->slice_addresses_in_ts[i] <= lcu_id_ts) { slice_id = i; } } printf("", slice_id); } printf(""); } printf("
RS Map
%d
TS Map
%d
Tile map
%d
Slice map
%d
>\n ]\n"); } printf(" \"%p\" [\n", state); printf(" label = \"{encoder_state|"); printf("+ type=%c\\l", state->type); if (!state->parent || state->frame != state->parent->global) { printf("|+ global\\l"); } if (!state->parent || state->tile != state->parent->tile) { printf("|+ tile\\l"); printf(" - id = %d\\l", state->tile->id); printf(" - lcu_offset_x = %d\\l", state->tile->lcu_offset_x); printf(" - lcu_offset_y = %d\\l", state->tile->lcu_offset_y); printf(" - lcu_offset_in_ts = %d\\l", state->tile->lcu_offset_in_ts); } if (!state->parent || state->slice != state->parent->slice) { printf("|+ slice\\l"); printf(" - id = %d\\l", state->slice->id); printf(" - start_in_ts = %d\\l", state->slice->start_in_ts); printf(" - end_in_ts = %d\\l", state->slice->end_in_ts); printf(" - start_in_rs = %d\\l", state->slice->start_in_rs); printf(" - end_in_rs = %d\\l", state->slice->end_in_rs); } if (!state->parent || state->wfrow != state->parent->wfrow) { printf("|+ wfrow\\l"); printf(" - lcu_offset_y = %d\\l", state->wfrow->lcu_offset_y); } printf("}\"\n"); printf(" ]\n"); if (state->parent) { printf(" \"%p\" -> \"%p\"\n", state->parent, state); } for (i = 0; state->children[i].encoder_control; ++i) { encoder_state_dump_graphviz(&state->children[i]); } if (!state->parent) { printf("}\n"); //Empty lines (easier to copy-paste) printf("\n\n\n\n\n"); } } #endif //KVZ_DEBUG_PRINT_THREADING_INFO #ifdef KVZ_DEBUG_PRINT_MV_INFO static void lcu_from_cu_array(cu_array_t* src, int src_x, int src_y, lcu_t* dst) { // ToDo: Fix invalid memory access outside of src->data const int dst_stride = src->stride >> 2; for (int y = src_y ? -4 : 0; y < LCU_WIDTH; y += SCU_WIDTH) { for (int x = src_x ? -4 : 0; x < LCU_WIDTH; x += SCU_WIDTH) { const cu_info_t* to_cu = LCU_GET_CU_AT_PX(dst, x, y); const int x_scu = (src_x + x) >> 2; const int y_scu = (src_y + y) >> 2; cu_info_t* from_cu = &src->data[x_scu + y_scu * dst_stride]; memcpy(to_cu, from_cu, sizeof(*to_cu)); } } if (src_x) { const cu_info_t* to_cu = LCU_GET_CU_AT_PX(dst, -1, 64); const int x_scu = (src_x + -1) >> 2; const int y_scu = (src_y + 64) >> 2; cu_info_t* from_cu = &src->data[x_scu + y_scu * dst_stride]; memcpy(to_cu, from_cu, sizeof(*to_cu)); } if (src_y) { const cu_info_t* to_cu = LCU_GET_TOP_RIGHT_CU(dst); const int x_scu = (src_x + 64) >> 2; const int y_scu = (src_y + -1) >> 2; cu_info_t* from_cu = &src->data[x_scu + y_scu * dst_stride]; memcpy(to_cu, from_cu, sizeof(*to_cu)); } } void kvz_print_merge_vectors(const encoder_state_t* const state, uint32_t pic_x, uint32_t pic_y, uint32_t block_width, uint32_t block_height, cu_info_t* cu) { lcu_t lcu; lcu_from_cu_array(state->tile->frame->cu_array, pic_x - SUB_SCU(pic_x), pic_y - SUB_SCU(pic_y), &lcu); static int val = 0; inter_merge_cand_t merge_cand[MRG_MAX_NUM_CANDS] = { 0 }; // Search for merge mode candidates uint8_t num_merge_cand = kvz_inter_get_merge_cand(state, pic_x, pic_y, block_width, block_height, true, true, merge_cand, &lcu ); static FILE* lut = NULL; if (lut == NULL) lut = fopen("uvg_merge2.txt", "w"); fprintf(lut, "%d: (%d,%d) Block (%d,%d) -> %d (", val++, pic_x, pic_y, block_width, block_height, cu->inter.mv_dir); if (cu->inter.mv_dir & 1) fprintf(lut, "%d,%d ", cu->inter.mv[0][0], cu->inter.mv[0][1]); if (cu->inter.mv_dir & 2) fprintf(lut, "%d,%d ", cu->inter.mv[1][0], cu->inter.mv[1][1]); fprintf(lut, ") -> ("); for (int i = 0; i < num_merge_cand; i++) { mv_t mv[2] = { merge_cand[i].mv[0][0], merge_cand[i].mv[0][1] }; mv_t mv2[2] = { merge_cand[i].mv[1][0], merge_cand[i].mv[1][1] }; //kvz_change_precision(4, 2, &mv[0], &mv[1]); fprintf(lut, "{ %d: ", merge_cand[i].dir); if (merge_cand[i].dir & 1) fprintf(lut, "%d,%d, ", mv[0], mv[1]); if (merge_cand[i].dir & 2) fprintf(lut, "%d,%d ", mv2[0], mv2[1]); fprintf(lut, "} "); } fprintf(lut, "%d, %d)\n", cu->merged | cu->skipped, (cu->merged | cu->skipped) ? cu->merge_idx : 0); } #endif // KVZ_DEBUG_PRINT_MV_INFO