root/tools/yuv-distortion.cc

DEFINITIONS

1. ssim
2. main

/*
 * H.265 video codec.
 * Copyright (c) 2013-2014 struktur AG, Dirk Farin <farin@struktur.de>
 *
 * This file is part of libde265.
 *
 * libde265 is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * libde265 is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with libde265.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <stdio.h>
#include <stdlib.h>

#include <libde265/quality.h>

#if HAVE_VIDEOGFX
#include <libvideogfx.hh>
using namespace videogfx;
#endif


float ssim(const uint8_t* img1,
           const uint8_t* img2,
           int width, int height)
{
#if HAVE_VIDEOGFX
  Bitmap<Pixel> ref, coded;
  ref  .Create(width, height); // reference image
  coded.Create(width, height); // coded image

  for (int y=0;y<height;y++) {
    memcpy(coded[y], img1 + y*width, width);
    memcpy(ref[y],   img2 + y*width, width);
  }

  SSIM ssimAlgo;
  return ssimAlgo.calcMSSIM(ref,coded);
#else
  return 0;
#endif
}


int main(int argc, char** argv)
{
  if (argc != 5) {
    fprintf(stderr,"need two YUV files and image size as input: FILE1 FILE2 WIDTH HEIGHT\n");
    exit(5);
  }


  FILE* fh_ref = fopen(argv[1],"rb");
  FILE* fh_cmp = fopen(argv[2],"rb");

  int width  = atoi(argv[3]);
  int height = atoi(argv[4]);

  uint8_t* yp_ref = (uint8_t*)malloc(width*height);
  uint8_t* yp_cmp = (uint8_t*)malloc(width*height);

  double mse_y=0.0, ssim_y=0.0;
  int nFrames=0;

  for (;;)
    {
      if (fread(yp_ref,1,width*height,fh_ref) != width*height) {
        break;
      }
      if (fread(yp_cmp,1,width*height,fh_cmp) != width*height) {
        break;
      }

      if (feof(fh_ref)) break;
      if (feof(fh_cmp)) break;

      fprintf(stderr,"yuv-distortion processing frame %d\r",nFrames+1);

      fseek(fh_ref,width*height/2,SEEK_CUR);
      fseek(fh_cmp,width*height/2,SEEK_CUR);

      double curr_mse_y = MSE(yp_ref, width,  yp_cmp, width,  width, height);
      mse_y += curr_mse_y;

      double curr_ssim_y = ssim(yp_ref, yp_cmp, width, height);
      ssim_y += curr_ssim_y;

      printf("%4d %f %f\n",nFrames,PSNR(curr_mse_y),curr_ssim_y);

      nFrames++;
    }

  printf("total: %f %f\n",PSNR(mse_y/nFrames),ssim_y/nFrames);
  fprintf(stderr,"\n");

  fclose(fh_ref);
  fclose(fh_cmp);

  return 0;
}