image_diff_metric.cpp File Reference

#include "include/codec/SkCodec.h"
#include "include/core/SkBitmap.h"
#include "include/core/SkData.h"
#include "include/core/SkPixmap.h"
#include "include/core/SkSize.h"
#include <cmath>
#include <cstdio>

Go to the source code of this file.

Functions
int	main (int argc, char **argv)

Function Documentation

main()

int main	(	int	argc,
		char **	argv
	)

Definition at line 18 of file image_diff_metric.cpp.

                                {
    if (argc != 3) {
        const char usage[] = "\nUsage:\n  %s {FILE1}.png {FILE2}.png\n\n";
        fprintf(stderr, usage, argv[0]);
        return 1;
    }
    SkBitmap bm[2];
    for (int i = 0; i < 2; ++i) {
        const char* path = argv[i + 1];
        if (std::unique_ptr<SkCodec> codec =
                SkCodec::MakeFromData(SkData::MakeFromFileName(path))) {
            bm[i].allocN32Pixels(codec->dimensions().fWidth, codec->dimensions().fHeight);
            if (SkCodec::kSuccess == codec->getPixels(bm[i].pixmap())) {
                continue;
            }
        }
        fprintf(stderr, "\nBad file: '%s'.\n\n", path);
        return 2;
    }
    SkISize dim = bm[0].dimensions();
    if (dim != bm[1].dimensions()) {
        fprintf(stderr, "\nImages must be same size: (%d,%d) != (%d,%d)\n\n",
                dim.fWidth, dim.fHeight, bm[1].dimensions().fWidth, bm[1].dimensions().fHeight);
        return 3;
    }
    int64_t totalDiffs = 0; // Manhattan distance in ARGB color-space.
    for (int y = 0; y < dim.fHeight; ++y) {
        const uint8_t* row1 = reinterpret_cast<const uint8_t*>(bm[0].pixmap().addr32(0, y));
        const uint8_t* row2 = reinterpret_cast<const uint8_t*>(bm[1].pixmap().addr32(0, y));
        for (size_t i = 0; i < (size_t)dim.fWidth * (size_t)4; ++i) {
            totalDiffs += std::abs((int)row1[i] - (int)row2[i]);
        }
    }
    printf("%g\n", (double)totalDiffs /
                   ((uint64_t)255 * 4 * (uint64_t)dim.fWidth * (uint64_t)dim.fHeight));
    return 0;
}