external-repos/squoosh
1
0
Fork 1
mirror of https://github.com/GoogleChromeLabs/squoosh.git synced 2025-11-14 09:39:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Keep planarized reference image around across runs

Browse Source
This commit is contained in:
Surma
2020-09-09 12:50:02 +01:00
parent 4b6334a212
commit ebdb00f50d
4 changed files with 113 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
codecs/visdif/visdif.cpp
View File

@@ -5,50 +5,57 @@
using namespace emscripten;
using namespace butteraugli;
double visdif(std::string rgba1, std::string rgba2, int width, int height) {
const char* rgba1p = rgba1.c_str();
const char* rgba2p = rgba2.c_str();
// One Image8 for each color channel + alpha
std::vector<ImageF> img1;
img1.push_back(ImageF(width, height));
img1.push_back(ImageF(width, height));
img1.push_back(ImageF(width, height));
img1.push_back(ImageF(width, height));
std::vector<ImageF> img2;
img2.push_back(ImageF(width, height));
img2.push_back(ImageF(width, height));
img2.push_back(ImageF(width, height));
img2.push_back(ImageF(width, height));
// Convert from interleaved RGBA to separate channels
// Turns an interleaved RGBA buffer into 4 planes for each color channel
void planarize(std::vector<ImageF>& img,
const char* rgba,
int width,
int height,
float gamma = 2.2) {
assert(img.size() == 0);
img.push_back(ImageF(width, height));
img.push_back(ImageF(width, height));
img.push_back(ImageF(width, height));
img.push_back(ImageF(width, height));
for (int y = 0; y < height; y++) {
float* const img1_row_r = img1[0].Row(y);
float* const img1_row_g = img1[1].Row(y);
float* const img1_row_b = img1[2].Row(y);
float* const img1_row_a = img1[3].Row(y);
float* const img2_row_r = img2[0].Row(y);
float* const img2_row_g = img2[1].Row(y);
float* const img2_row_b = img2[2].Row(y);
float* const img2_row_a = img2[3].Row(y);
float* const row_r = img[0].Row(y);
float* const row_g = img[1].Row(y);
float* const row_b = img[2].Row(y);
float* const row_a = img[3].Row(y);
for (int x = 0; x < width; x++) {
img1_row_r[x] = 255.0 * pow(rgba1p[y * width * 4 + x * 4 + 0] / 255.0, 2.2);
img1_row_g[x] = 255.0 * pow(rgba1p[y * width * 4 + x * 4 + 1] / 255.0, 2.2);
img1_row_b[x] = 255.0 * pow(rgba1p[y * width * 4 + x * 4 + 2] / 255.0, 2.2);
img1_row_a[x] = 255.0 * pow(rgba1p[y * width * 4 + x * 4 + 3] / 255.0, 2.2);
img2_row_r[x] = 255.0 * pow(rgba2p[y * width * 4 + x * 4 + 0] / 255.0, 2.2);
img2_row_g[x] = 255.0 * pow(rgba2p[y * width * 4 + x * 4 + 1] / 255.0, 2.2);
img2_row_b[x] = 255.0 * pow(rgba2p[y * width * 4 + x * 4 + 2] / 255.0, 2.2);
img2_row_a[x] = 255.0 * pow(rgba2p[y * width * 4 + x * 4 + 3] / 255.0, 2.2);
row_r[x] = 255.0 * pow(rgba[y * width * 4 + x * 4 + 0] / 255.0, gamma);
row_g[x] = 255.0 * pow(rgba[y * width * 4 + x * 4 + 1] / 255.0, gamma);
row_b[x] = 255.0 * pow(rgba[y * width * 4 + x * 4 + 2] / 255.0, gamma);
row_a[x] = 255.0 * pow(rgba[y * width * 4 + x * 4 + 3] / 255.0, gamma);
}
}
ImageF diffmap;
double diffvalue;
if (!ButteraugliInterface(img1, img2, 1.0, diffmap, diffvalue)) {
return -1.0;
}
return diffvalue;
}
class VisDiff {
private:
std::vector<ImageF> ref_img;
int width;
int height;
public:
VisDiff(std::string ref_img, int width, int height) {
planarize(this->ref_img, ref_img.c_str(), width, height);
this->width = width;
this->height = height;
}
double distance(std::string other_img) {
std::vector<ImageF> img;
planarize(img, other_img.c_str(), width, height);
ImageF diffmap;
double diffvalue;
if (!ButteraugliInterface(ref_img, img, 1.0, diffmap, diffvalue)) {
return -1.0;
}
return diffvalue;
}
};
EMSCRIPTEN_BINDINGS(my_module) {
function("visdif", &visdif);
class_<VisDiff>("VisDiff").constructor<std::string, int, int>().function("distance",
&VisDiff::distance);
}