Add quantize plugin

Author: hipstersmoothie

packages/jimp/package.json

@@ -60,6 +60,7 @@
     "@jimp/plugin-resize": "workspace:*",
     "@jimp/plugin-rotate": "workspace:*",
     "@jimp/plugin-threshold": "workspace:*",
+    "@jimp/plugin-quantize": "workspace:*",
     "@jimp/types": "workspace:*",
     "@jimp/utils": "workspace:*"
   },

packages/jimp/src/index.ts

@@ -27,6 +27,7 @@ import * as print from "@jimp/plugin-print";
 import * as resize from "@jimp/plugin-resize";
 import * as rotate from "@jimp/plugin-rotate";
 import * as threshold from "@jimp/plugin-threshold";
+import * as quantize from "@jimp/plugin-quantize";
 
 import { createJimp } from "@jimp/core";
 
@@ -119,6 +120,7 @@ export const Jimp = createJimp({
     resize.methods,
     rotate.methods,
     threshold.methods,
+    quantize.methods,
   ],
 });
 

plugins/quantize/package.json

@@ -0,0 +1,61 @@
+{
+  "name": "@jimp/plugin-quantize",
+  "version": "1.0.1",
+  "repository": "jimp-dev/jimp",
+  "scripts": {
+    "lint": "eslint .",
+    "test": "vitest",
+    "build": "tshy",
+    "dev": "tshy --watch",
+    "clean": "rm -rf node_modules .tshy .tshy-build dist .turbo"
+  },
+  "author": "Andrew Lisowski <[email protected]>",
+  "license": "MIT",
+  "devDependencies": {
+    "@jimp/config-eslint": "workspace:*",
+    "@jimp/config-typescript": "workspace:*",
+    "@jimp/core": "workspace:*",
+    "@jimp/js-jpeg": "workspace:*",
+    "@jimp/js-png": "workspace:*",
+    "@jimp/test-utils": "workspace:*",
+    "@jimp/types": "workspace:*",
+    "eslint": "^8.57.0",
+    "tshy": "^3.0.2",
+    "typescript": "^5.5.4",
+    "vitest": "^1.4.0"
+  },
+  "tshy": {
+    "exclude": [
+      "**/*.test.ts"
+    ],
+    "exports": {
+      "./package.json": "./package.json",
+      ".": "./src/index.ts"
+    }
+  },
+  "exports": {
+    "./package.json": "./package.json",
+    ".": {
+      "import": {
+        "types": "./dist/esm/index.d.ts",
+        "default": "./dist/esm/index.js"
+      },
+      "require": {
+        "types": "./dist/commonjs/index.d.ts",
+        "default": "./dist/commonjs/index.js"
+      }
+    }
+  },
+  "main": "./dist/commonjs/index.js",
+  "types": "./dist/commonjs/index.d.ts",
+  "type": "module",
+  "publishConfig": {
+    "access": "public"
+  },
+  "sideEffects": false,
+  "dependencies": {
+    "image-q": "^4.0.0",
+    "zod": "^3.22.4"
+  },
+  "module": "./dist/esm/index.js"
+}

plugins/quantize/src/__image_snapshots__/index-node-test-ts-src-index-node-test-ts-quantize-defines-default-threshold-for-lighter-backgrounds-1-snap.png

@@ -0,0 +1,23 @@
+import { expect, test, describe } from "vitest";
+import "@jimp/test-utils";
+import "@jimp/test-utils/image-snapshot";
+
+import png from "@jimp/js-png";
+import { createJimp } from "@jimp/core";
+
+import { methods as quantize } from "./index.js";
+
+const jimp = createJimp({ formats: [png], plugins: [quantize] });
+
+describe("Quantize", () => {
+  test("defines default threshold for lighter backgrounds", async () => {
+    const testImage = await jimp.read(__dirname + "/images/colorful.png");
+    const output = await testImage
+      .quantize({
+        colors: 8,
+      })
+      .getBuffer("image/png");
+
+    expect(output).toMatchImageSnapshot();
+  });
+});

plugins/quantize/src/images/colorful.png

@@ -0,0 +1,81 @@
+import { JimpClass } from "@jimp/types";
+import { applyPaletteSync, buildPaletteSync, utils } from "image-q";
+import z from "zod";
+
+const QuantizeOptionsSchema = z.object({
+  colors: z.number().optional(),
+  colorDistanceFormula: z
+    .union([
+      z.literal("cie94-textiles"),
+      z.literal("cie94-graphic-arts"),
+      z.literal("ciede2000"),
+      z.literal("color-metric"),
+      z.literal("euclidean"),
+      z.literal("euclidean-bt709-noalpha"),
+      z.literal("euclidean-bt709"),
+      z.literal("manhattan"),
+      z.literal("manhattan-bt709"),
+      z.literal("manhattan-nommyde"),
+      z.literal("pngquant"),
+    ])
+    .optional(),
+  paletteQuantization: z
+    .union([
+      z.literal("neuquant"),
+      z.literal("neuquant-float"),
+      z.literal("rgbquant"),
+      z.literal("wuquant"),
+    ])
+    .optional(),
+  imageQuantization: z
+    .union([
+      z.literal("nearest"),
+      z.literal("riemersma"),
+      z.literal("floyd-steinberg"),
+      z.literal("false-floyd-steinberg"),
+      z.literal("stucki"),
+      z.literal("atkinson"),
+      z.literal("jarvis"),
+      z.literal("burkes"),
+      z.literal("sierra"),
+      z.literal("two-sierra"),
+      z.literal("sierra-lite"),
+    ])
+    .optional(),
+});
+
+export type QuantizeOptions = z.infer<typeof QuantizeOptionsSchema>;
+
+export const methods = {
+  /**
+   */
+  quantize<I extends JimpClass>(
+    image: I,
+    {
+      colors,
+      colorDistanceFormula,
+      paletteQuantization,
+      imageQuantization,
+    }: QuantizeOptions
+  ) {
+    const inPointContainer = utils.PointContainer.fromUint8Array(
+      image.bitmap.data,
+      image.bitmap.width,
+      image.bitmap.height
+    );
+
+    const palette = buildPaletteSync([inPointContainer], {
+      colors,
+      colorDistanceFormula,
+      paletteQuantization,
+    });
+    const outPointContainer = applyPaletteSync(inPointContainer, palette, {
+      colorDistanceFormula,
+      imageQuantization,
+    });
+
+    image.bitmap.data = Buffer.from(outPointContainer.toUint8Array());
+
+    return image;
+  },
+};

plugins/quantize/src/index.node.test.ts

@@ -0,0 +1,6 @@
+{
+  "extends": "@jimp/config-typescript/base.json",
+  "compilerOptions": {
+    "outDir": "dist"
+  }
+}