hashirama/uvg266
1
0
Fork 0
mirror of https://github.com/ultravideo/uvg266.git synced 2024-11-24 02:24:07 +00:00
Code Issues Actions Projects Releases Wiki Activity

Fix PSNR calculation, broken after the introduction of frame padding

Browse source
This commit is contained in:
Marko Viitanen 2021-02-26 19:38:15 +02:00
parent c6baa8ad62
commit 3c75500cd4
1 changed files with 16 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
src/encmain.c
View file

@ -110,13 +110,26 @@ static void compute_psnr(const kvz_picture *const src,
double sse[3] = { 0.0 };
for (int32_t c = 0; c < colors; ++c) {
kvz_pixel* src_ptr = src->data[c];
kvz_pixel* rec_ptr = rec->data[c];
int32_t width = src->width;
int32_t height = src->height;
int32_t stride = src->stride;
int32_t num_pixels = pixels;
if (c != COLOR_Y) {
width >>= 1;
height >>= 1;
stride >>= 1;
num_pixels >>= 2;
}
for (int32_t i = 0; i < num_pixels; ++i) {
const int32_t error = src->data[c][i] - rec->data[c][i];
sse[c] += error * error;
for (int32_t y = 0; y < height; ++y) {
for (int32_t x = 0; x < width; ++x) {
const int32_t error = src_ptr[x] - rec_ptr[x];
sse[c] += error * error;
}
src_ptr += stride;
rec_ptr += stride;
}
// Avoid division by zero