From d23a584b9eaeb06b3ff3d14715e95a36b0f887a3 Mon Sep 17 00:00:00 2001 From: Ge0rg3 <16269580+Ge0rg3@users.noreply.github.com> Date: Thu, 29 Aug 2019 16:17:07 +0100 Subject: [PATCH] Randomize Colour Palette Operation --- src/core/config/Categories.json | 1 + .../operations/RandomizeColourPalette.mjs | 84 +++++++++++++++++++ 2 files changed, 85 insertions(+) create mode 100644 src/core/operations/RandomizeColourPalette.mjs diff --git a/src/core/config/Categories.json b/src/core/config/Categories.json index 6f71cf72..695546f0 100755 --- a/src/core/config/Categories.json +++ b/src/core/config/Categories.json @@ -380,6 +380,7 @@ "Split Colour Channels", "Extract RGBA", "View Bit Plane", + "Randomize Colour Palette", "Extract LSB", "Rotate Image", "Resize Image", diff --git a/src/core/operations/RandomizeColourPalette.mjs b/src/core/operations/RandomizeColourPalette.mjs new file mode 100644 index 00000000..ded8fd49 --- /dev/null +++ b/src/core/operations/RandomizeColourPalette.mjs @@ -0,0 +1,84 @@ +/** + * @author Ge0rg3 [georgeomnet+cyberchef@gmail.com] + * @copyright Crown Copyright 2019 + * @license Apache-2.0 + */ + +import Operation from "../Operation.mjs"; +import OperationError from "../errors/OperationError.mjs"; +import Utils from "../Utils"; +import PseudoRandomNumberGenerator from "./PseudoRandomNumberGenerator.mjs"; +import { isImage } from "../lib/FileType"; +import { runHash } from "../lib/Hash.mjs"; +import { toBase64 } from "../lib/Base64"; +import jimp from "jimp"; + +/** + * Randomize Colour Palette operation + */ +class RandomizeColourPalette extends Operation { + + /** + * RandomizeColourPalette constructor + */ + constructor() { + super(); + + this.name = "Randomize Colour Palette"; + this.module = "Image"; + this.description = "Randomize's each colour in an image's colour palette. This can often reveal text or symbols that were previously a very similar colour to their surroundings."; + this.infoURL = "https://en.wikipedia.org/wiki/Indexed_color"; + this.inputType = "byteArray"; + this.outputType = "byteArray"; + this.presentType = "html"; + this.args = [ + { + name: "Seed", + type: "string", + value: "" + } + ]; + } + + /** + * @param {byteArray} input + * @param {Object[]} args + * @returns {byteArray} + */ + async run(input, args) { + if (!isImage(input)) throw new OperationError("Please enter a valid image file."); + + const seed = args[0] || (new PseudoRandomNumberGenerator()).run("", [5, "Hex"]), + parsedImage = await jimp.read(Buffer.from(input)), + width = parsedImage.bitmap.width, + height = parsedImage.bitmap.height; + + let rgbString, rgbHash, rgbHex; + + parsedImage.scan(0, 0, width, height, function(x, y, idx) { + rgbString = this.bitmap.data.slice(idx, idx+3).join("."); + rgbHash = runHash("md5", Utils.strToArrayBuffer(seed + rgbString)); + rgbHex = rgbHash.substr(0, 6) + "ff"; + parsedImage.setPixelColor(parseInt(rgbHex, 16), x, y); + }); + + const imageBuffer = await parsedImage.getBufferAsync(jimp.AUTO); + + return Array.from(imageBuffer); + } + + /** + * Displays the extracted data as an image for web apps. + * @param {byteArray} data + * @returns {html} + */ + present(data) { + if (!data.length) return ""; + const type = isImage(data); + + return ``; + } + +} + +export default RandomizeColourPalette;