blob: 87aadfdff468e3ae74895e347876489c231d6c80 [file] [log] [blame]
/*
* 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;
print_mi_data(cm, mvs, "Partitions:", offsetof(MB_MODE_INFO, sb_type[0]));
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++) {
fprintf(mvs, "%2d ", mi[0]->skip_txfm[0]);
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);
}