Switch to Butteraugli and simpel CLI

cli/index.js

@@ -0,0 +1,100 @@
+const mozjpeg_dec = require("../codecs/mozjpeg/dec/mozjpeg_dec.js");
+const mozjpeg_enc = require("../codecs/mozjpeg/enc/mozjpeg_enc.js");
+const webp_enc = require("../codecs/webp/enc/webp_enc.js");
+const visdifModule = require("../codecs/visdif/visdif.js");
+
+// Our decoders currently rely on this.
+globalThis.ImageData = require("./image_data.js");
+
+const fsp = require("fs").promises;
+
+async function decodeFile(file) {
+  const buffer = await fsp.readFile(file);
+  const rgba = (await mozjpeg_dec()).decode(new Uint8Array(buffer));
+  return rgba;
+}
+
+const butteraugliGoal = 1.4;
+const maxRounds = 8;
+async function optimize(bitmapIn, encode, decode) {
+  let quality = 50;
+  let inc = 25;
+  let butteraugliDistance = 2;
+  let attempts = 0;
+  let bitmapOut;
+  let binaryOut;
+
+  const { visdif } = await visdifModule();
+  do {
+    binaryOut = await encode(bitmapIn, quality);
+    bitmapOut = await decode(binaryOut);
+    console.log({
+      butteraugliDistance,
+      quality,
+      attempts,
+      binaryOut,
+      bitmapOut
+    });
+    butteraugliDistance = visdif(
+      bitmapIn.data,
+      bitmapOut.data,
+      bitmapIn.width,
+      bitmapIn.height
+    );
+    if (butteraugliDistance > butteraugliGoal) {
+      quality += inc;
+    } else {
+      quality -= inc;
+    }
+    inc /= 2;
+    attempts++;
+  } while (
+    Math.abs(butteraugliDistance - butteraugliGoal) > 0.1 &&
+    attempts < maxRounds
+  );
+
+  return {
+    bitmap: bitmapOut,
+    binary: binaryOut,
+    quality,
+    attempts
+  };
+}
+
+async function main() {
+  const [, , inFile, outFile] = process.argv;
+  const bitmapIn = await decodeFile(inFile);
+
+  const jpegEncModule = await mozjpeg_enc();
+  const jpegEnc = async (bitmap, quality) =>
+    jpegEncModule.encode(bitmap.data.buffer, bitmap.width, bitmap.height, {
+      quality,
+      baseline: false,
+      arithmetic: false,
+      progressive: true,
+      optimize_coding: true,
+      smoothing: 0,
+      color_space: 3 /*YCbCr*/,
+      quant_table: 3,
+      trellis_multipass: false,
+      trellis_opt_zero: false,
+      trellis_opt_table: false,
+      trellis_loops: 1,
+      auto_subsample: true,
+      chroma_subsample: 2,
+      separate_chroma_quality: false,
+      chroma_quality: 75
+    });
+  const jpegDecModule = await mozjpeg_dec();
+  const jpegDec = async binary => jpegDecModule.decode(binary);
+  const jpegOut = await optimize(bitmapIn, jpegEnc, jpegDec);
+  await fsp.writeFile("out.jpg", jpegOut.binary);
+  //await webp_enc();
+  //console.log(bitmapIn);
+}
+
+main().catch(err => {
+  console.error("Something went wrong");
+  console.error(err);
+  process.exit(1);
+});