'extras/get_disto' example: compute PSNR between two files

Change-Id: I40e693e52d8eb4e32b86273952dac2e5eefe6e7c

extras/Makefile.am

@@ -9,3 +9,11 @@ libwebpextras_la_SOURCES += extras.c extras.h
 
 libwebpextras_la_CPPFLAGS = $(AM_CPPFLAGS)
 libwebpextras_la_LIBADD = ../src/libwebp.la
+
+noinst_PROGRAMS = get_disto
+
+get_disto_SOURCES = get_disto.c
+get_disto_CPPFLAGS  = $(AM_CPPFLAGS)
+get_disto_LDADD = ../imageio/libexample_util.la ../imageio/libimagedec.la
+get_disto_LDADD += ../src/libwebp.la
+get_disto_LDADD += $(PNG_LIBS) $(JPEG_LIBS) $(TIFF_LIBS)

extras/get_disto.c

@@ -0,0 +1,124 @@
+// Copyright 2016 Google Inc. All Rights Reserved.
+//
+// Use of this source code is governed by a BSD-style license
+// that can be found in the COPYING file in the root of the source
+// tree. An additional intellectual property rights grant can be found
+// in the file PATENTS. All contributing project authors may
+// be found in the AUTHORS file in the root of the source tree.
+// -----------------------------------------------------------------------------
+//
+// Simple tool to load two webp/png/jpg/tiff files and compute PSNR/SSIM.
+// This is mostly a wrapper around WebPPictureDistortion().
+//
+/*
+ gcc -o get_disto get_disto.c -O3 -I../ -L../examples -L../imageio \
+    -lexample_util -limagedec -lwebp -L/opt/local/lib \
+    -lpng -lz -ljpeg -ltiff -lm -lpthread
+*/
+//
+// Author: Skal (pascal.massimino@gmail.com)
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "webp/encode.h"
+#include "../imageio/example_util.h"
+#include "../imageio/image_dec.h"
+
+static size_t ReadPicture(const char* const filename, WebPPicture* const pic,
+                          int keep_alpha) {
+  const uint8_t* data = NULL;
+  size_t data_size = 0;
+  WebPImageReader reader = NULL;
+  int ok = ExUtilReadFile(filename, &data, &data_size);
+  if (!ok) goto Error;
+
+  pic->use_argb = 1;  // force ARGB
+
+  reader = WebPGuessImageReader(data, data_size);
+  ok = (reader != NULL) && reader(data, data_size, pic, keep_alpha, NULL);
+
+ Error:
+  if (!ok) {
+    fprintf(stderr, "Error! Could not process file %s\n", filename);
+  }
+  free((void*)data);
+  return ok ? data_size : 0;
+}
+
+static void Help(void) {
+  fprintf(stderr,
+          "Usage: get_disto [-ssim][-psnr][-alpha] compressed.webp orig.webp\n"
+          "  -ssim ..... print SSIM distortion\n"
+          "  -psnr ..... print PSNR distortion (default)\n"
+          "  -alpha .... preserve alpha plane\n"
+          "  -h ........ this message\n"
+          " Also handles PNG, JPG and TIFF files, in addition to WebP.\n");
+}
+
+int main(int argc, const char *argv[]) {
+  WebPPicture pic1, pic2;
+  int ret = 1;
+  float disto[5];
+  size_t size1 = 0, size2 = 0;
+  int type = 0;
+  int c;
+  int help = 0;
+  int keep_alpha = 0;
+  const char* name1 = NULL;
+  const char* name2 = NULL;
+
+  if (!WebPPictureInit(&pic1) || !WebPPictureInit(&pic2)) {
+    fprintf(stderr, "Can't init pictures\n");
+    return 1;
+  }
+
+  for (c = 1; c < argc; ++c) {
+    if (!strcmp(argv[c], "-ssim")) {
+      type = 1;
+    } else if (!strcmp(argv[c], "-psnr")) {
+      type = 0;
+    } else if (!strcmp(argv[c], "-alpha")) {
+      keep_alpha = 1;
+    } else if (!strcmp(argv[c], "-h")) {
+      help = 1;
+      ret = 0;
+    } else if (name1 == NULL) {
+      name1 = argv[c];
+    } else {
+      name2 = argv[c];
+    }
+  }
+  if (help || name1 == NULL || name2 == NULL) {
+    if (!help) {
+      fprintf(stderr, "Error: missing arguments.\n");
+    }
+    Help();
+    goto End;
+  }
+  if ((size1 = ReadPicture(name1, &pic1, 1)) == 0) {
+    goto End;
+  }
+  if ((size2 = ReadPicture(name2, &pic2, 1)) == 0) {
+    goto End;
+  }
+  if (!keep_alpha) {
+    WebPBlendAlpha(&pic1, 0x00000000);
+    WebPBlendAlpha(&pic2, 0x00000000);
+  }
+
+  if (!WebPPictureDistortion(&pic1, &pic2, type, disto)) {
+    fprintf(stderr, "Error while computing the distortion.\n");
+    goto End;
+  }
+  printf("%u %.2f    %.2f %.2f %.2f %.2f\n",
+         (unsigned int)size1, disto[4],
+         disto[0], disto[1], disto[2], disto[3]);
+  ret = 0;
+
+ End:
+  WebPPictureFree(&pic1);
+  WebPPictureFree(&pic2);
+  return ret;
+}

makefile.unix

@@ -306,11 +306,12 @@ HDRS = \
     src/webp/format_constants.h \
     $(HDRS_INSTALLED) \
 
-OUT_LIBS = imageio/libexample_util.a src/libwebpdecoder.a src/libwebp.a
+OUT_LIBS = imageio/libexample_util.a imageio/libimagedec.a src/libwebpdecoder.a
+OUT_LIBS += src/libwebp.a
 EXTRA_LIB = extras/libwebpextras.a
 OUT_EXAMPLES = examples/cwebp examples/dwebp
 EXTRA_EXAMPLES = examples/gif2webp examples/vwebp examples/webpmux \
-                 examples/anim_diff
+                 examples/anim_diff extras/get_disto
 
 OUTPUT = $(OUT_LIBS) $(OUT_EXAMPLES)
 ifeq ($(MAKECMDGOALS),clean)
@@ -337,10 +338,10 @@ src/utils/bit_writer.o: src/utils/endian_inl.h
 	$(CC) $(CFLAGS) $(CPPFLAGS) -c $< -o $@
 
 examples/libanim_util.a: $(ANIM_UTIL_OBJS)
-examples/libexample_dec.a: $(EX_FORMAT_DEC_OBJS)
 examples/libgifdec.a: $(GIFDEC_OBJS)
 extras/libwebpextras.a: $(LIBWEBPEXTRA_OBJS)
 imageio/libexample_util.a: $(EX_UTIL_OBJS)
+imageio/libimagedec.a: $(EX_FORMAT_DEC_OBJS)
 src/libwebpdecoder.a: $(LIBWEBPDECODER_OBJS)
 src/libwebp.a: $(LIBWEBP_OBJS)
 src/mux/libwebpmux.a: $(LIBWEBPMUX_OBJS)
@@ -361,7 +362,7 @@ examples/anim_diff: src/demux/libwebpdemux.a imageio/libexample_util.a
 examples/anim_diff: src/libwebp.a
 examples/anim_diff: EXTRA_LIBS += $(GIF_LIBS)
 examples/anim_diff: EXTRA_FLAGS += -DWEBP_HAVE_GIF
-examples/cwebp: imageio/libexample_util.a examples/libexample_dec.a
+examples/cwebp: imageio/libexample_util.a imageio/libimagedec.a
 examples/cwebp: src/libwebp.a
 examples/cwebp: EXTRA_LIBS += $(CWEBP_LIBS)
 examples/dwebp: imageio/libexample_util.a src/libwebpdecoder.a
@@ -377,6 +378,11 @@ examples/vwebp: EXTRA_FLAGS += -DWEBP_HAVE_GL
 examples/webpmux: imageio/libexample_util.a src/mux/libwebpmux.a
 examples/webpmux: src/libwebpdecoder.a
 
+extras/get_disto: extras/get_disto.o
+extras/get_disto: imageio/libimagedec.a imageio/libexampledec.a
+extras/get_disto: imageio/libexample_util.a
+extras/get_disto: src/libwebp.a $(CWEBP_LIBS)
+
 $(OUT_EXAMPLES) $(EXTRA_EXAMPLES):
 	$(CC) -o $@ $^ $(LDFLAGS)