Go to the documentation of this file.
36 #define FFSWAP(type,a,b) do{type SWAP_tmp= b; b= a; a= SWAP_tmp;}while(0)
37 #define FFMIN(a,b) ((a) > (b) ? (b) : (a))
40 #define PIXEL_MAX ((1 << BIT_DEPTH)-1)
47 const pixel *pix2, intptr_t stride2,
52 for( z = 0; z < 2; z++ )
54 uint32_t
s1 = 0,
s2 = 0,
ss = 0, s12 = 0;
55 for( y = 0; y < 4; y++ )
56 for( x = 0; x < 4; x++ )
58 int a = pix1[x+y*stride1];
59 int b = pix2[x+y*stride2];
93 type vars = fss*64 - fs1*fs1 - fs2*fs2;
94 type covar = fs12*64 - fs1*fs2;
95 return (
float)(2*fs1*fs2 + ssim_c1) * (
float)(2*covar + ssim_c2)
96 / ((
float)(fs1*fs1 + fs2*fs2 + ssim_c1) * (
float)(
vars + ssim_c2));
105 ssim +=
ssim_end1( sum0[
i][0] + sum0[
i+1][0] + sum1[
i][0] + sum1[
i+1][0],
106 sum0[
i][1] + sum0[
i+1][1] + sum1[
i][1] + sum1[
i+1][1],
107 sum0[
i][2] + sum0[
i+1][2] + sum1[
i][2] + sum1[
i+1][2],
108 sum0[
i][3] + sum0[
i+1][3] + sum1[
i][3] + sum1[
i+1][3] );
113 pixel *pix1, intptr_t stride1,
114 pixel *pix2, intptr_t stride2,
120 int (*sum0)[4] = buf;
121 int (*sum1)[4] = sum0 + (
width >> 2) + 3;
124 for( y = 1; y <
height; y++ )
128 FFSWAP(
void*, sum0, sum1 );
129 for( x = 0; x <
width; x+=2 )
130 ssim_4x4x2_core( &pix1[4*(x+z*stride1)], stride1, &pix2[4*(x+z*stride2)], stride2, &sum0[x] );
132 for( x = 0; x <
width-1; x += 4 )
146 int d = pix1[
i] - pix2[
i];
154 return -10*
log((
double)ssd/(denom*255*255))/
log(10);
164 printf(
"PSNR Y:%.3f U:%.3f V:%.3f All:%.3f | ",
169 printf(
"SSIM Y:%.5f U:%.5f V:%.5f All:%.5f (%.5f)",
173 (ssim[0]*4 + ssim[1] + ssim[2]) / (
frames*6),
177 int main(
int argc,
char* argv[])
180 uint8_t *buf[2], *plane[2][3];
182 uint64_t ssd[3] = {0,0,0};
183 double ssim[3] = {0,0,0};
188 if( argc<4 || 2 != sscanf(argv[3],
"%dx%d", &
w, &
h) )
190 printf(
"tiny_ssim <file1.yuv> <file2.yuv> <width>x<height> [<seek>]\n");
194 f[0] = fopen(argv[1],
"rb");
195 f[1] = fopen(argv[2],
"rb");
196 sscanf(argv[3],
"%dx%d", &
w, &
h);
198 if (
w<=0 ||
h<=0 ||
w*(int64_t)
h >= INT_MAX/3 || 2LL*
w+12 >= INT_MAX /
sizeof(*
temp)) {
199 fprintf(stderr,
"Dimensions are too large, or invalid\n");
207 plane[
i][0] = buf[
i];
208 plane[
i][1] = plane[
i][0] +
w*
h;
209 plane[
i][2] = plane[
i][1] +
w*
h/4;
212 seek = argc<5 ? 0 : atoi(argv[4]);
219 if( fread(buf[0],
frame_size, 1,
f[0]) != 1)
break;
220 if( fread(buf[1],
frame_size, 1,
f[1]) != 1)
break;
227 ssd[
i] += ssd_one[
i];
228 ssim[
i] += ssim_one[
i];
static void print_results(uint64_t ssd[3], double ssim[3], int frames, int w, int h)
float ssim_plane(pixel *pix1, intptr_t stride1, pixel *pix2, intptr_t stride2, int width, int height, void *buf, int *cnt)
#define FFSWAP(type, a, b)
if it could not because there are no more frames
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf type
#define ss(width, name, subs,...)
int main(int argc, char *argv[])
static const uint8_t vars[2][12]
static void BS_FUNC() seek(BSCTX *bc, unsigned pos)
Seek to the given bit position.
static double ssim_db(double ssim, double weight)
static int weight(int i, int blen, int offset)
printf("static const uint8_t my_array[100] = {\n")
The reader does not expect b to be semantically here and if the code is changed by maybe adding a a division or other the signedness will almost certainly be mistaken To avoid this confusion a new type was SUINT is the C unsigned type but it holds a signed int to use the same example SUINT a
uint64_t ssd_plane(const uint8_t *pix1, const uint8_t *pix2, int size)
static double ssd_to_psnr(uint64_t ssd, uint64_t denom)
#define i(width, name, range_min, range_max)
static float ssim_end4(int sum0[5][4], int sum1[5][4], int width)
static void ssim_4x4x2_core(const pixel *pix1, intptr_t stride1, const pixel *pix2, intptr_t stride2, int sums[2][4])
static float ssim_end1(int s1, int s2, int ss, int s12)