| /* |
| * Copyright (c) 2021, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 3-Clause Clear License |
| * and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear |
| * License was not distributed with this source code in the LICENSE file, you |
| * can obtain it at aomedia.org/license/software-license/bsd-3-c-c/. If the |
| * Alliance for Open Media Patent License 1.0 was not distributed with this |
| * source code in the PATENTS file, you can obtain it at |
| * aomedia.org/license/patent-license/. |
| */ |
| |
| #include <stdio.h> |
| |
| #include "av1/common/av1_common_int.h" |
| #include "av1/common/blockd.h" |
| #include "av1/common/enums.h" |
| |
| static void log_frame_info(AV1_COMMON *cm, const char *str, FILE *f) { |
| fprintf(f, "%s", str); |
| fprintf(f, "(Frame %u, Show:%d, Q:%d): \n", cm->current_frame.frame_number, |
| cm->show_frame, cm->quant_params.base_qindex); |
| } |
| /* This function dereferences a pointer to the mbmi structure |
| * and uses the passed in member offset to print out the value of an integer |
| * for each mbmi member value in the mi structure. |
| */ |
| static void print_mi_data(AV1_COMMON *cm, FILE *file, const char *descriptor, |
| size_t member_offset) { |
| const CommonModeInfoParams *const mi_params = &cm->mi_params; |
| MB_MODE_INFO **mi = mi_params->mi_grid_base; |
| int rows = mi_params->mi_rows; |
| int cols = mi_params->mi_cols; |
| char prefix = descriptor[0]; |
| |
| log_frame_info(cm, descriptor, file); |
| for (int mi_row = 0; mi_row < rows; mi_row++) { |
| fprintf(file, "%c ", prefix); |
| for (int mi_col = 0; mi_col < cols; mi_col++) { |
| fprintf(file, "%2d ", *((char *)((char *)(mi[0]) + member_offset))); |
| mi++; |
| } |
| fprintf(file, "\n"); |
| mi += mi_params->mi_stride - cols; |
| } |
| fprintf(file, "\n"); |
| } |
| |
| void av1_print_modes_and_motion_vectors(AV1_COMMON *cm, const char *file) { |
| CommonModeInfoParams *mi_params = &cm->mi_params; |
| FILE *mvs = fopen(file, "a"); |
| MB_MODE_INFO **mi = mi_params->mi_grid_base; |
| const int rows = mi_params->mi_rows; |
| const int cols = mi_params->mi_cols; |
| #if CONFIG_SDP |
| print_mi_data(cm, mvs, "Partitions:", offsetof(MB_MODE_INFO, sb_type[0])); |
| #else |
| print_mi_data(cm, mvs, "Partitions:", offsetof(MB_MODE_INFO, sb_type)); |
| #endif |
| print_mi_data(cm, mvs, "Modes:", offsetof(MB_MODE_INFO, mode)); |
| print_mi_data(cm, mvs, "Ref frame:", offsetof(MB_MODE_INFO, ref_frame[0])); |
| print_mi_data(cm, mvs, "Transform:", offsetof(MB_MODE_INFO, tx_size)); |
| print_mi_data(cm, mvs, "UV Modes:", offsetof(MB_MODE_INFO, uv_mode)); |
| |
| // output skip infomation. |
| log_frame_info(cm, "Skips:", mvs); |
| for (int mi_row = 0; mi_row < rows; mi_row++) { |
| fprintf(mvs, "S "); |
| for (int mi_col = 0; mi_col < cols; mi_col++) { |
| #if CONFIG_SDP |
| fprintf(mvs, "%2d ", mi[0]->skip_txfm[0]); |
| #else |
| fprintf(mvs, "%2d ", mi[0]->skip_txfm); |
| #endif |
| mi++; |
| } |
| fprintf(mvs, "\n"); |
| mi += mi_params->mi_stride - cols; |
| } |
| fprintf(mvs, "\n"); |
| |
| // output motion vectors. |
| log_frame_info(cm, "Vectors ", mvs); |
| mi = mi_params->mi_grid_base; |
| for (int mi_row = 0; mi_row < rows; mi_row++) { |
| fprintf(mvs, "V "); |
| for (int mi_col = 0; mi_col < cols; mi_col++) { |
| fprintf(mvs, "%4d:%4d ", mi[0]->mv[0].as_mv.row, mi[0]->mv[0].as_mv.col); |
| mi++; |
| } |
| fprintf(mvs, "\n"); |
| mi += mi_params->mi_stride - cols; |
| } |
| fprintf(mvs, "\n"); |
| |
| fclose(mvs); |
| } |
| |
| void av1_print_uncompressed_frame_header(const uint8_t *data, int size, |
| const char *filename) { |
| FILE *hdrFile = fopen(filename, "w"); |
| fwrite(data, size, sizeof(uint8_t), hdrFile); |
| |
| // Reset order hints(7bit + a previous bit) to 0, so that all camera frame |
| // headers are identical in large scale coding. |
| uint8_t zero = 0; |
| fseek(hdrFile, 1, SEEK_SET); |
| // Reset second byte. |
| fwrite(&zero, 1, sizeof(uint8_t), hdrFile); |
| fclose(hdrFile); |
| } |
| |
| void av1_print_frame_contexts(const FRAME_CONTEXT *fc, const char *filename) { |
| FILE *fcFile = fopen(filename, "w"); |
| const uint16_t *fcp = (uint16_t *)fc; |
| const unsigned int n_contexts = sizeof(FRAME_CONTEXT) / sizeof(uint16_t); |
| unsigned int i; |
| |
| for (i = 0; i < n_contexts; ++i) fprintf(fcFile, "%d ", *fcp++); |
| fclose(fcFile); |
| } |