testsuite/ssim.c

   1
   2 #ifdef HAVE_CONFIG_H
   3 #include "config.h"
   4 #endif
   5
   6 #include <stdio.h>
   7 #include <string.h>
   8 #include <stdlib.h>
   9 #include <schroedinger/schro.h>
  10 #include <schroedinger/schrodebug.h>
  11 #include <schroedinger/schroutils.h>
  12
  13 void
  14 create_pattern (SchroFrame *frame)
  15 {
  16   int i,j;
  17   uint8_t *data;
  18
  19   for(j=0;j<frame->height;j++){
  20     data = OFFSET(frame->components[0].data, j*frame->components[0].stride);
  21     for(i=0;i<frame->width;i++){
  22       data[i] = (((i>>4)&1) ^ ((j>>4)&1)) * 255;
  23     }
  24   }
  25   memset (frame->components[1].data, 0, frame->components[1].length);
  26   memset (frame->components[2].data, 0, frame->components[2].length);
  27   schro_frame_filter_lowpass2 (frame, 5.0);
  28 }
  29
  30 void
  31 distort (SchroFrame *frame)
  32 {
  33   int i,j;
  34   uint8_t *data;
  35
  36   for(j=0;j<frame->height;j++){
  37     data = OFFSET(frame->components[0].data, j*frame->components[0].stride);
  38     for(i=0;i<frame->width;i++){
  39       data[i] &= 0xfc;
  40     }
  41   }
  42 }
  43
  44
  45 void
  46 test (void)
  47 {
  48   SchroFrame *frame1;
  49   SchroFrame *frame2;
  50
  51   frame1 = schro_frame_new_and_alloc (NULL, SCHRO_FRAME_FORMAT_U8_420, 256, 256);
  52   frame2 = schro_frame_new_and_alloc (NULL, SCHRO_FRAME_FORMAT_U8_420, 256, 256);
  53
  54   create_pattern (frame1);
  55   create_pattern (frame2);
  56
  57   distort (frame2);
  58
  59   schro_frame_ssim (frame1, frame2);
  60
  61   schro_frame_unref (frame1);
  62   schro_frame_unref (frame2);
  63 }
  64
  65 int
  66 main (int argc, char *argv[])
  67 {
  68   schro_init();
  69
  70   test();
  71
  72   return 0;
  73 }
  74
  75