-rw-r--r--.gitignore1
-rw-r--r--Cargo.lock271
-rw-r--r--Cargo.toml13
-rwxr-xr-xdumbbin0 -> 694584 bytes
-rw-r--r--eoquan.pngbin0 -> 5735707 bytes
-rw-r--r--flamegraph.svg491
-rw-r--r--perf.databin0 -> 174156 bytes
-rw-r--r--perf.data.oldbin0 -> 109236 bytes
-rw-r--r--plane_4096_2048.jpgbin0 -> 3198098 bytes
-rw-r--r--plane_4096_2048.pngbin0 -> 7173748 bytes
-rw-r--r--sqrtless.pngbin0 -> 2661850 bytes
-rw-r--r--src/dumb.rs27
-rw-r--r--src/kd.rs148
-rw-r--r--src/lib.rs34
-rw-r--r--src/main.rs78
-rw-r--r--yeee.pngbin0 -> 1364151 bytes
16 files changed, 1063 insertions, 0 deletions
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..ea8c4bf
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+/target
diff --git a/Cargo.lock b/Cargo.lock
new file mode 100644
index 0000000..07b5657
--- /dev/null
+++ b/Cargo.lock
@@ -0,0 +1,271 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+version = 3
+
+[[package]]
+name = "adler2"
+version = "2.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "512761e0bb2578dd7380c6baaa0f4ce03e84f95e960231d1dec8bf4d7d6e2627"
+
+[[package]]
+name = "atools"
+version = "0.1.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a0b8fc827c2d2ad4c1fa0d458efc9716e2f458c135f3f4f2332ed9ddd85c037e"
+
+[[package]]
+name = "bitflags"
+version = "1.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
+
+[[package]]
+name = "byteorder"
+version = "1.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1fd0f2584146f6f2ef48085050886acf353beff7305ebd1ae69500e27c67f64b"
+
+[[package]]
+name = "cfg-if"
+version = "1.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
+
+[[package]]
+name = "clipline"
+version = "0.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "01c033212f55b799c43650c2fb12866ba8fe873e5786e7e649810c4dc9a76561"
+
+[[package]]
+name = "crc32fast"
+version = "1.4.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a97769d94ddab943e4510d138150169a2758b5ef3eb191a9ee688de3e23ef7b3"
+dependencies = [
+ "cfg-if",
+]
+
+[[package]]
+name = "exoquant"
+version = "0.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5048aeecde83b7455a63b44aa54da5321d6171cff2adb8a5959849e142e5e4aa"
+
+[[package]]
+name = "fdeflate"
+version = "0.3.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "07c6f4c64c1d33a3111c4466f7365ebdcc37c5bd1ea0d62aae2e3d722aacbedb"
+dependencies = [
+ "simd-adler32",
+]
+
+[[package]]
+name = "fimg"
+version = "0.4.43"
+dependencies = [
+ "atools",
+ "clipline",
+ "libc",
+ "mattr",
+ "png",
+ "umath",
+ "vecto",
+]
+
+[[package]]
+name = "flate2"
+version = "1.0.35"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c936bfdafb507ebbf50b8074c54fa31c5be9a1e7e5f467dd659697041407d07c"
+dependencies = [
+ "crc32fast",
+ "miniz_oxide",
+]
+
+[[package]]
+name = "fux_kdtree"
+version = "0.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "216bf8256834a87394bf0364f82a92444ee8a00c757630a02a2bf278456f74e7"
+
+[[package]]
+name = "getrandom"
+version = "0.2.15"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c4567c8db10ae91089c99af84c68c38da3ec2f087c3f82960bcdbf3656b6f4d7"
+dependencies = [
+ "cfg-if",
+ "libc",
+ "wasi",
+]
+
+[[package]]
+name = "libc"
+version = "0.2.167"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "09d6582e104315a817dff97f75133544b2e094ee22447d2acf4a74e189ba06fc"
+
+[[package]]
+name = "mattr"
+version = "0.0.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5f63dc7ec862e5d146c89d104d437548fef5216a6a653f4afc4b87c581970677"
+
+[[package]]
+name = "miniz_oxide"
+version = "0.8.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e2d80299ef12ff69b16a84bb182e3b9df68b5a91574d3d4fa6e41b65deec4df1"
+dependencies = [
+ "adler2",
+ "simd-adler32",
+]
+
+[[package]]
+name = "png"
+version = "0.17.14"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "52f9d46a34a05a6a57566bc2bfae066ef07585a6e3fa30fbbdff5936380623f0"
+dependencies = [
+ "bitflags",
+ "crc32fast",
+ "fdeflate",
+ "flate2",
+ "miniz_oxide",
+]
+
+[[package]]
+name = "ppv-lite86"
+version = "0.2.20"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "77957b295656769bb8ad2b6a6b09d897d94f05c41b069aede1fcdaa675eaea04"
+dependencies = [
+ "zerocopy",
+]
+
+[[package]]
+name = "proc-macro2"
+version = "1.0.92"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "37d3544b3f2748c54e147655edb5025752e2303145b5aefb3c3ea2c78b973bb0"
+dependencies = [
+ "unicode-ident",
+]
+
+[[package]]
+name = "quote"
+version = "1.0.37"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b5b9d34b8991d19d98081b46eacdd8eb58c6f2b201139f7c5f643cc155a633af"
+dependencies = [
+ "proc-macro2",
+]
+
+[[package]]
+name = "rand"
+version = "0.8.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "34af8d1a0e25924bc5b7c43c079c942339d8f0a8b57c39049bef581b46327404"
+dependencies = [
+ "libc",
+ "rand_chacha",
+ "rand_core",
+]
+
+[[package]]
+name = "rand_chacha"
+version = "0.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e6c10a63a0fa32252be49d21e7709d4d4baf8d231c2dbce1eaa8141b9b127d88"
+dependencies = [
+ "ppv-lite86",
+ "rand_core",
+]
+
+[[package]]
+name = "rand_core"
+version = "0.6.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ec0be4795e2f6a28069bec0b5ff3e2ac9bafc99e6a9a7dc3547996c5c816922c"
+dependencies = [
+ "getrandom",
+]
+
+[[package]]
+name = "remapper"
+version = "0.1.0"
+dependencies = [
+ "atools",
+ "exoquant",
+ "fimg",
+ "fux_kdtree",
+ "rand",
+]
+
+[[package]]
+name = "simd-adler32"
+version = "0.3.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d66dc143e6b11c1eddc06d5c423cfc97062865baf299914ab64caa38182078fe"
+
+[[package]]
+name = "syn"
+version = "2.0.90"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "919d3b74a5dd0ccd15aeb8f93e7006bd9e14c295087c9896a110f490752bcf31"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "unicode-ident",
+]
+
+[[package]]
+name = "umath"
+version = "0.0.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "42f74eb7957e3a63fa27bfa53c3d361e7ce3871e66f2518292a011eb8e2c00cc"
+
+[[package]]
+name = "unicode-ident"
+version = "1.0.14"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "adb9e6ca4f869e1180728b7950e35922a7fc6397f7b641499e8f3ef06e50dc83"
+
+[[package]]
+name = "vecto"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c9f1339978a8a20d3ca67ea5d6cbb13e5d06bfac55f5a2a05338fb13e36ab75d"
+dependencies = [
+ "umath",
+]
+
+[[package]]
+name = "wasi"
+version = "0.11.0+wasi-snapshot-preview1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9c8d87e72b64a3b4db28d11ce29237c246188f4f51057d65a7eab63b7987e423"
+
+[[package]]
+name = "zerocopy"
+version = "0.7.35"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1b9b4fd18abc82b8136838da5d50bae7bdea537c574d8dc1a34ed098d6c166f0"
+dependencies = [
+ "byteorder",
+ "zerocopy-derive",
+]
+
+[[package]]
+name = "zerocopy-derive"
+version = "0.7.35"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fa4f8080344d4671fb4e831a13ad1e68092748387dfc4f55e356242fae12ce3e"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn",
+]
diff --git a/Cargo.toml b/Cargo.toml
new file mode 100644
index 0000000..144126e
--- /dev/null
+++ b/Cargo.toml
@@ -0,0 +1,13 @@
+[package]
+name = "remapper"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+atools = "0.1.5"
+exoquant = "0.2.0"
+fimg = { version = "0.4.43", path = "../fimg", default-features = false, features = [
+ "save",
+] }
+fux_kdtree = "0.2.0"
+rand = "0.8.5"
diff --git a/dumb b/dumb
new file mode 100755
index 0000000..2d5c49b
--- /dev/null
+++ b/dumb
Binary files differ
diff --git a/eoquan.png b/eoquan.png
new file mode 100644
index 0000000..0ca7bc3
--- /dev/null
+++ b/eoquan.png
Binary files differ
diff --git a/flamegraph.svg b/flamegraph.svg
new file mode 100644
index 0000000..ccb5eab
--- /dev/null
+++ b/flamegraph.svg
@@ -0,0 +1,491 @@
+<?xml version="1.0" standalone="no"?><!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd"><svg version="1.1" width="1200" height="118" onload="init(evt)" viewBox="0 0 1200 118" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" xmlns:fg="http://github.com/jonhoo/inferno"><!--Flame graph stack visualization. See https://github.com/brendangregg/FlameGraph for latest version, and http://www.brendangregg.com/flamegraphs.html for examples.--><!--NOTES: --><defs><linearGradient id="background" y1="0" y2="1" x1="0" x2="0"><stop stop-color="#eeeeee" offset="5%"/><stop stop-color="#eeeeb0" offset="95%"/></linearGradient></defs><style type="text/css">
+text { font-family:monospace; font-size:12px }
+#title { text-anchor:middle; font-size:17px; }
+#matched { text-anchor:end; }
+#search { text-anchor:end; opacity:0.1; cursor:pointer; }
+#search:hover, #search.show { opacity:1; }
+#subtitle { text-anchor:middle; font-color:rgb(160,160,160); }
+#unzoom { cursor:pointer; }
+#frames > *:hover { stroke:black; stroke-width:0.5; cursor:pointer; }
+.hide { display:none; }
+.parent { opacity:0.5; }
+</style><script type="text/ecmascript"><![CDATA[
+ var nametype = 'Function:';
+ var fontsize = 12;
+ var fontwidth = 0.59;
+ var xpad = 10;
+ var inverted = false;
+ var searchcolor = 'rgb(230,0,230)';
+ var fluiddrawing = true;
+ var truncate_text_right = false;
+ ]]><![CDATA["use strict";
+var details, searchbtn, unzoombtn, matchedtxt, svg, searching, frames, known_font_width;
+function init(evt) {
+ details = document.getElementById("details").firstChild;
+ searchbtn = document.getElementById("search");
+ unzoombtn = document.getElementById("unzoom");
+ matchedtxt = document.getElementById("matched");
+ svg = document.getElementsByTagName("svg")[0];
+ frames = document.getElementById("frames");
+ known_font_width = get_monospace_width(frames);
+ total_samples = parseInt(frames.attributes.total_samples.value);
+ searching = 0;
+
+ // Use GET parameters to restore a flamegraph's state.
+ var restore_state = function() {
+ var params = get_params();
+ if (params.x && params.y)
+ zoom(find_group(document.querySelector('[*|x="' + params.x + '"][y="' + params.y + '"]')));
+ if (params.s)
+ search(params.s);
+ };
+
+ if (fluiddrawing) {
+ // Make width dynamic so the SVG fits its parent's width.
+ svg.removeAttribute("width");
+ // Edge requires us to have a viewBox that gets updated with size changes.
+ var isEdge = /Edge\/\d./i.test(navigator.userAgent);
+ if (!isEdge) {
+ svg.removeAttribute("viewBox");
+ }
+ var update_for_width_change = function() {
+ if (isEdge) {
+ svg.attributes.viewBox.value = "0 0 " + svg.width.baseVal.value + " " + svg.height.baseVal.value;
+ }
+
+ // Keep consistent padding on left and right of frames container.
+ frames.attributes.width.value = svg.width.baseVal.value - xpad * 2;
+
+ // Text truncation needs to be adjusted for the current width.
+ update_text_for_elements(frames.children);
+
+ // Keep search elements at a fixed distance from right edge.
+ var svgWidth = svg.width.baseVal.value;
+ searchbtn.attributes.x.value = svgWidth - xpad;
+ matchedtxt.attributes.x.value = svgWidth - xpad;
+ };
+ window.addEventListener('resize', function() {
+ update_for_width_change();
+ });
+ // This needs to be done asynchronously for Safari to work.
+ setTimeout(function() {
+ unzoom();
+ update_for_width_change();
+ restore_state();
+ }, 0);
+ } else {
+ restore_state();
+ }
+}
+// event listeners
+window.addEventListener("click", function(e) {
+ var target = find_group(e.target);
+ if (target) {
+ if (target.nodeName == "a") {
+ if (e.ctrlKey === false) return;
+ e.preventDefault();
+ }
+ if (target.classList.contains("parent")) unzoom();
+ zoom(target);
+
+ // set parameters for zoom state
+ var el = target.querySelector("rect");
+ if (el && el.attributes && el.attributes.y && el.attributes["fg:x"]) {
+ var params = get_params()
+ params.x = el.attributes["fg:x"].value;
+ params.y = el.attributes.y.value;
+ history.replaceState(null, null, parse_params(params));
+ }
+ }
+ else if (e.target.id == "unzoom") {
+ unzoom();
+
+ // remove zoom state
+ var params = get_params();
+ if (params.x) delete params.x;
+ if (params.y) delete params.y;
+ history.replaceState(null, null, parse_params(params));
+ }
+ else if (e.target.id == "search") search_prompt();
+}, false)
+// mouse-over for info
+// show
+window.addEventListener("mouseover", function(e) {
+ var target = find_group(e.target);
+ if (target) details.nodeValue = nametype + " " + g_to_text(target);
+}, false)
+// clear
+window.addEventListener("mouseout", function(e) {
+ var target = find_group(e.target);
+ if (target) details.nodeValue = ' ';
+}, false)
+// ctrl-F for search
+window.addEventListener("keydown",function (e) {
+ if (e.keyCode === 114 || (e.ctrlKey && e.keyCode === 70)) {
+ e.preventDefault();
+ search_prompt();
+ }
+}, false)
+// functions
+function get_params() {
+ var params = {};
+ var paramsarr = window.location.search.substr(1).split('&');
+ for (var i = 0; i < paramsarr.length; ++i) {
+ var tmp = paramsarr[i].split("=");
+ if (!tmp[0] || !tmp[1]) continue;
+ params[tmp[0]] = decodeURIComponent(tmp[1]);
+ }
+ return params;
+}
+function parse_params(params) {
+ var uri = "?";
+ for (var key in params) {
+ uri += key + '=' + encodeURIComponent(params[key]) + '&';
+ }
+ if (uri.slice(-1) == "&")
+ uri = uri.substring(0, uri.length - 1);
+ if (uri == '?')
+ uri = window.location.href.split('?')[0];
+ return uri;
+}
+function find_child(node, selector) {
+ var children = node.querySelectorAll(selector);
+ if (children.length) return children[0];
+ return;
+}
+function find_group(node) {
+ var parent = node.parentElement;
+ if (!parent) return;
+ if (parent.id == "frames") return node;
+ return find_group(parent);
+}
+function orig_save(e, attr, val) {
+ if (e.attributes["fg:orig_" + attr] != undefined) return;
+ if (e.attributes[attr] == undefined) return;
+ if (val == undefined) val = e.attributes[attr].value;
+ e.setAttribute("fg:orig_" + attr, val);
+}
+function orig_load(e, attr) {
+ if (e.attributes["fg:orig_"+attr] == undefined) return;
+ e.attributes[attr].value = e.attributes["fg:orig_" + attr].value;
+ e.removeAttribute("fg:orig_" + attr);
+}
+function g_to_text(e) {
+ var text = find_child(e, "title").firstChild.nodeValue;
+ return (text)
+}
+function g_to_func(e) {
+ var func = g_to_text(e);
+ // if there's any manipulation we want to do to the function
+ // name before it's searched, do it here before returning.
+ return (func);
+}
+function get_monospace_width(frames) {
+ // Given the id="frames" element, return the width of text characters if
+ // this is a monospace font, otherwise return 0.
+ text = find_child(frames.children[0], "text");
+ originalContent = text.textContent;
+ text.textContent = "!";
+ bangWidth = text.getComputedTextLength();
+ text.textContent = "W";
+ wWidth = text.getComputedTextLength();
+ text.textContent = originalContent;
+ if (bangWidth === wWidth) {
+ return bangWidth;
+ } else {
+ return 0;
+ }
+}
+function update_text_for_elements(elements) {
+ // In order to render quickly in the browser, you want to do one pass of
+ // reading attributes, and one pass of mutating attributes. See
+ // https://web.dev/avoid-large-complex-layouts-and-layout-thrashing/ for details.
+
+ // Fall back to inefficient calculation, if we're variable-width font.
+ // TODO This should be optimized somehow too.
+ if (known_font_width === 0) {
+ for (var i = 0; i < elements.length; i++) {
+ update_text(elements[i]);
+ }
+ return;
+ }
+
+ var textElemNewAttributes = [];
+ for (var i = 0; i < elements.length; i++) {
+ var e = elements[i];
+ var r = find_child(e, "rect");
+ var t = find_child(e, "text");
+ var w = parseFloat(r.attributes.width.value) * frames.attributes.width.value / 100 - 3;
+ var txt = find_child(e, "title").textContent.replace(/\([^(]*\)$/,"");
+ var newX = format_percent((parseFloat(r.attributes.x.value) + (100 * 3 / frames.attributes.width.value)));
+
+ // Smaller than this size won't fit anything
+ if (w < 2 * known_font_width) {
+ textElemNewAttributes.push([newX, ""]);
+ continue;
+ }
+
+ // Fit in full text width
+ if (txt.length * known_font_width < w) {
+ textElemNewAttributes.push([newX, txt]);
+ continue;
+ }
+
+ var substringLength = Math.floor(w / known_font_width) - 2;
+ if (truncate_text_right) {
+ // Truncate the right side of the text.
+ textElemNewAttributes.push([newX, txt.substring(0, substringLength) + ".."]);
+ continue;
+ } else {
+ // Truncate the left side of the text.
+ textElemNewAttributes.push([newX, ".." + txt.substring(txt.length - substringLength, txt.length)]);
+ continue;
+ }
+ }
+
+ console.assert(textElemNewAttributes.length === elements.length, "Resize failed, please file a bug at https://github.com/jonhoo/inferno/");
+
+ // Now that we know new textContent, set it all in one go so we don't refresh a bazillion times.
+ for (var i = 0; i < elements.length; i++) {
+ var e = elements[i];
+ var values = textElemNewAttributes[i];
+ var t = find_child(e, "text");
+ t.attributes.x.value = values[0];
+ t.textContent = values[1];
+ }
+}
+
+function update_text(e) {
+ var r = find_child(e, "rect");
+ var t = find_child(e, "text");
+ var w = parseFloat(r.attributes.width.value) * frames.attributes.width.value / 100 - 3;
+ var txt = find_child(e, "title").textContent.replace(/\([^(]*\)$/,"");
+ t.attributes.x.value = format_percent((parseFloat(r.attributes.x.value) + (100 * 3 / frames.attributes.width.value)));
+
+ // Smaller than this size won't fit anything
+ if (w < 2 * fontsize * fontwidth) {
+ t.textContent = "";
+ return;
+ }
+ t.textContent = txt;
+ // Fit in full text width
+ if (t.getComputedTextLength() < w)
+ return;
+ if (truncate_text_right) {
+ // Truncate the right side of the text.
+ for (var x = txt.length - 2; x > 0; x--) {
+ if (t.getSubStringLength(0, x + 2) <= w) {
+ t.textContent = txt.substring(0, x) + "..";
+ return;
+ }
+ }
+ } else {
+ // Truncate the left side of the text.
+ for (var x = 2; x < txt.length; x++) {
+ if (t.getSubStringLength(x - 2, txt.length) <= w) {
+ t.textContent = ".." + txt.substring(x, txt.length);
+ return;
+ }
+ }
+ }
+ t.textContent = "";
+}
+// zoom
+function zoom_reset(e) {
+ if (e.tagName == "rect") {
+ e.attributes.x.value = format_percent(100 * parseInt(e.attributes["fg:x"].value) / total_samples);
+ e.attributes.width.value = format_percent(100 * parseInt(e.attributes["fg:w"].value) / total_samples);
+ }
+ if (e.childNodes == undefined) return;
+ for(var i = 0, c = e.childNodes; i < c.length; i++) {
+ zoom_reset(c[i]);
+ }
+}
+function zoom_child(e, x, zoomed_width_samples) {
+ if (e.tagName == "text") {
+ var parent_x = parseFloat(find_child(e.parentNode, "rect[x]").attributes.x.value);
+ e.attributes.x.value = format_percent(parent_x + (100 * 3 / frames.attributes.width.value));
+ } else if (e.tagName == "rect") {
+ e.attributes.x.value = format_percent(100 * (parseInt(e.attributes["fg:x"].value) - x) / zoomed_width_samples);
+ e.attributes.width.value = format_percent(100 * parseInt(e.attributes["fg:w"].value) / zoomed_width_samples);
+ }
+ if (e.childNodes == undefined) return;
+ for(var i = 0, c = e.childNodes; i < c.length; i++) {
+ zoom_child(c[i], x, zoomed_width_samples);
+ }
+}
+function zoom_parent(e) {
+ if (e.attributes) {
+ if (e.attributes.x != undefined) {
+ e.attributes.x.value = "0.0%";
+ }
+ if (e.attributes.width != undefined) {
+ e.attributes.width.value = "100.0%";
+ }
+ }
+ if (e.childNodes == undefined) return;
+ for(var i = 0, c = e.childNodes; i < c.length; i++) {
+ zoom_parent(c[i]);
+ }
+}
+function zoom(node) {
+ var attr = find_child(node, "rect").attributes;
+ var width = parseInt(attr["fg:w"].value);
+ var xmin = parseInt(attr["fg:x"].value);
+ var xmax = xmin + width;
+ var ymin = parseFloat(attr.y.value);
+ unzoombtn.classList.remove("hide");
+ var el = frames.children;
+ var to_update_text = [];
+ for (var i = 0; i < el.length; i++) {
+ var e = el[i];
+ var a = find_child(e, "rect").attributes;
+ var ex = parseInt(a["fg:x"].value);
+ var ew = parseInt(a["fg:w"].value);
+ // Is it an ancestor
+ if (!inverted) {
+ var upstack = parseFloat(a.y.value) > ymin;
+ } else {
+ var upstack = parseFloat(a.y.value) < ymin;
+ }
+ if (upstack) {
+ // Direct ancestor
+ if (ex <= xmin && (ex+ew) >= xmax) {
+ e.classList.add("parent");
+ zoom_parent(e);
+ to_update_text.push(e);
+ }
+ // not in current path
+ else
+ e.classList.add("hide");
+ }
+ // Children maybe
+ else {
+ // no common path
+ if (ex < xmin || ex >= xmax) {
+ e.classList.add("hide");
+ }
+ else {
+ zoom_child(e, xmin, width);
+ to_update_text.push(e);
+ }
+ }
+ }
+ update_text_for_elements(to_update_text);
+}
+function unzoom() {
+ unzoombtn.classList.add("hide");
+ var el = frames.children;
+ for(var i = 0; i < el.length; i++) {
+ el[i].classList.remove("parent");
+ el[i].classList.remove("hide");
+ zoom_reset(el[i]);
+ }
+ update_text_for_elements(el);
+}
+// search
+function reset_search() {
+ var el = document.querySelectorAll("#frames rect");
+ for (var i = 0; i < el.length; i++) {
+ orig_load(el[i], "fill")
+ }
+ var params = get_params();
+ delete params.s;
+ history.replaceState(null, null, parse_params(params));
+}
+function search_prompt() {
+ if (!searching) {
+ var term = prompt("Enter a search term (regexp " +
+ "allowed, eg: ^ext4_)", "");
+ if (term != null) {
+ search(term)
+ }
+ } else {
+ reset_search();
+ searching = 0;
+ searchbtn.classList.remove("show");
+ searchbtn.firstChild.nodeValue = "Search"
+ matchedtxt.classList.add("hide");
+ matchedtxt.firstChild.nodeValue = ""
+ }
+}
+function search(term) {
+ var re = new RegExp(term);
+ var el = frames.children;
+ var matches = new Object();
+ var maxwidth = 0;
+ for (var i = 0; i < el.length; i++) {
+ var e = el[i];
+ // Skip over frames which are either not visible, or below the zoomed-to frame
+ if (e.classList.contains("hide") || e.classList.contains("parent")) {
+ continue;
+ }
+ var func = g_to_func(e);
+ var rect = find_child(e, "rect");
+ if (func == null || rect == null)
+ continue;
+ // Save max width. Only works as we have a root frame
+ var w = parseInt(rect.attributes["fg:w"].value);
+ if (w > maxwidth)
+ maxwidth = w;
+ if (func.match(re)) {
+ // highlight
+ var x = parseInt(rect.attributes["fg:x"].value);
+ orig_save(rect, "fill");
+ rect.attributes.fill.value = searchcolor;
+ // remember matches
+ if (matches[x] == undefined) {
+ matches[x] = w;
+ } else {
+ if (w > matches[x]) {
+ // overwrite with parent
+ matches[x] = w;
+ }
+ }
+ searching = 1;
+ }
+ }
+ if (!searching)
+ return;
+ var params = get_params();
+ params.s = term;
+ history.replaceState(null, null, parse_params(params));
+
+ searchbtn.classList.add("show");
+ searchbtn.firstChild.nodeValue = "Reset Search";
+ // calculate percent matched, excluding vertical overlap
+ var count = 0;
+ var lastx = -1;
+ var lastw = 0;
+ var keys = Array();
+ for (k in matches) {
+ if (matches.hasOwnProperty(k))
+ keys.push(k);
+ }
+ // sort the matched frames by their x location
+ // ascending, then width descending
+ keys.sort(function(a, b){
+ return a - b;
+ });
+ // Step through frames saving only the biggest bottom-up frames
+ // thanks to the sort order. This relies on the tree property
+ // where children are always smaller than their parents.
+ for (var k in keys) {
+ var x = parseInt(keys[k]);
+ var w = matches[keys[k]];
+ if (x >= lastx + lastw) {
+ count += w;
+ lastx = x;
+ lastw = w;
+ }
+ }
+ // display matched percent
+ matchedtxt.classList.remove("hide");
+ var pct = 100 * count / maxwidth;
+ if (pct != 100) pct = pct.toFixed(1);
+ matchedtxt.firstChild.nodeValue = "Matched: " + pct + "%";
+}
+function format_percent(n) {
+ return n.toFixed(4) + "%";
+}
+]]></script><rect x="0" y="0" width="100%" height="118" fill="url(#background)"/><text id="title" fill="rgb(0,0,0)" x="50.0000%" y="24.00">Flame Graph</text><text id="details" fill="rgb(0,0,0)" x="10" y="101.00"> </text><text id="unzoom" class="hide" fill="rgb(0,0,0)" x="10" y="24.00">Reset Zoom</text><text id="search" fill="rgb(0,0,0)" x="1190" y="24.00">Search</text><text id="matched" fill="rgb(0,0,0)" x="1190" y="101.00"> </text><svg id="frames" x="10" width="1180" total_samples="1163"><g><title>&lt;alloc::vec::Vec&lt;T&gt; as alloc::vec::spec_from_iter::SpecFromIter&lt;T,I&gt;&gt;::from_iter (23 samples, 1.98%)</title><rect x="0.0000%" y="37" width="1.9776%" height="15" fill="rgb(227,0,7)" fg:x="0" fg:w="23"/><text x="0.2500%" y="47.50">&lt;..</text></g><g><title>[ld-linux-x86-64.so.2] (2 samples, 0.17%)</title><rect x="1.9776%" y="37" width="0.1720%" height="15" fill="rgb(217,0,24)" fg:x="23" fg:w="2"/><text x="2.2276%" y="47.50"></text></g><g><title>[libc.so.6] (16 samples, 1.38%)</title><rect x="2.1496%" y="37" width="1.3758%" height="15" fill="rgb(221,193,54)" fg:x="25" fg:w="16"/><text x="2.3996%" y="47.50"></text></g><g><title>[unknown] (8 samples, 0.69%)</title><rect x="3.5254%" y="37" width="0.6879%" height="15" fill="rgb(248,212,6)" fg:x="41" fg:w="8"/><text x="3.7754%" y="47.50"></text></g><g><title>__cpu_indicator_init@GCC_4.8.0 (1 samples, 0.09%)</title><rect x="4.2132%" y="37" width="0.0860%" height="15" fill="rgb(208,68,35)" fg:x="49" fg:w="1"/><text x="4.4632%" y="47.50"></text></g><g><title>crc32fast::Hasher::update (2 samples, 0.17%)</title><rect x="4.2992%" y="37" width="0.1720%" height="15" fill="rgb(232,128,0)" fg:x="50" fg:w="2"/><text x="4.5492%" y="47.50"></text></g><g><title>fdeflate::compress::Compressor&lt;W&gt;::write_data (20 samples, 1.72%)</title><rect x="4.4712%" y="37" width="1.7197%" height="15" fill="rgb(207,160,47)" fg:x="52" fg:w="20"/><text x="4.7212%" y="47.50"></text></g><g><title>fdeflate::compress::Compressor&lt;W&gt;::write_run (6 samples, 0.52%)</title><rect x="6.1909%" y="37" width="0.5159%" height="15" fill="rgb(228,23,34)" fg:x="72" fg:w="6"/><text x="6.4409%" y="47.50"></text></g><g><title>fdeflate::decompress::Decompressor::read (65 samples, 5.59%)</title><rect x="6.7068%" y="37" width="5.5890%" height="15" fill="rgb(218,30,26)" fg:x="78" fg:w="65"/><text x="6.9568%" y="47.50">fdeflat..</text></g><g><title>fdeflate::huffman::build_table (8 samples, 0.69%)</title><rect x="12.2958%" y="37" width="0.6879%" height="15" fill="rgb(220,122,19)" fg:x="143" fg:w="8"/><text x="12.5458%" y="47.50"></text></g><g><title>fimg::convert::f32s_to_u8s (63 samples, 5.42%)</title><rect x="12.9837%" y="37" width="5.4170%" height="15" fill="rgb(250,228,42)" fg:x="151" fg:w="63"/><text x="13.2337%" y="47.50">fimg::c..</text></g><g><title>fimg::convert::mapping (26 samples, 2.24%)</title><rect x="18.4007%" y="37" width="2.2356%" height="15" fill="rgb(240,193,28)" fg:x="214" fg:w="26"/><text x="18.6507%" y="47.50">f..</text></g><g><title>fimg::convert::u8s_to_f32s (5 samples, 0.43%)</title><rect x="20.6363%" y="37" width="0.4299%" height="15" fill="rgb(216,20,37)" fg:x="240" fg:w="5"/><text x="20.8863%" y="47.50"></text></g><g><title>png::decoder::transform::expand_trns_line (15 samples, 1.29%)</title><rect x="21.0662%" y="37" width="1.2898%" height="15" fill="rgb(206,188,39)" fg:x="245" fg:w="15"/><text x="21.3162%" y="47.50"></text></g><g><title>png::filter::filter_internal (2 samples, 0.17%)</title><rect x="22.3560%" y="37" width="0.1720%" height="15" fill="rgb(217,207,13)" fg:x="260" fg:w="2"/><text x="22.6060%" y="47.50"></text></g><g><title>png::filter::unfilter (4 samples, 0.34%)</title><rect x="22.5279%" y="37" width="0.3439%" height="15" fill="rgb(231,73,38)" fg:x="262" fg:w="4"/><text x="22.7779%" y="47.50"></text></g><g><title>remapper::kd::KDNode::find_nearest (895 samples, 76.96%)</title><rect x="22.8719%" y="37" width="76.9561%" height="15" fill="rgb(225,20,46)" fg:x="266" fg:w="895"/><text x="23.1219%" y="47.50">remapper::kd::KDNode::find_nearest</text></g><g><title>simd_adler32::imp::avx2::imp::update (1 samples, 0.09%)</title><rect x="99.8280%" y="37" width="0.0860%" height="15" fill="rgb(210,31,41)" fg:x="1161" fg:w="1"/><text x="100.0780%" y="47.50"></text></g><g><title>all (1,163 samples, 100%)</title><rect x="0.0000%" y="69" width="100.0000%" height="15" fill="rgb(221,200,47)" fg:x="0" fg:w="1163"/><text x="0.2500%" y="79.50"></text></g><g><title>remapper (1,163 samples, 100.00%)</title><rect x="0.0000%" y="53" width="100.0000%" height="15" fill="rgb(226,26,5)" fg:x="0" fg:w="1163"/><text x="0.2500%" y="63.50">remapper</text></g><g><title>std::sys::sync::once::futex::Once::call (1 samples, 0.09%)</title><rect x="99.9140%" y="37" width="0.0860%" height="15" fill="rgb(249,33,26)" fg:x="1162" fg:w="1"/><text x="100.1640%" y="47.50"></text></g></svg></svg> \ No newline at end of file
diff --git a/perf.data b/perf.data
new file mode 100644
index 0000000..796517d
--- /dev/null
+++ b/perf.data
Binary files differ
diff --git a/perf.data.old b/perf.data.old
new file mode 100644
index 0000000..027596c
--- /dev/null
+++ b/perf.data.old
Binary files differ
diff --git a/plane_4096_2048.jpg b/plane_4096_2048.jpg
new file mode 100644
index 0000000..704ce0d
--- /dev/null
+++ b/plane_4096_2048.jpg
Binary files differ
diff --git a/plane_4096_2048.png b/plane_4096_2048.png
new file mode 100644
index 0000000..dc4cf3f
--- /dev/null
+++ b/plane_4096_2048.png
Binary files differ
diff --git a/sqrtless.png b/sqrtless.png
new file mode 100644
index 0000000..21ec8ae
--- /dev/null
+++ b/sqrtless.png
Binary files differ
diff --git a/src/dumb.rs b/src/dumb.rs
new file mode 100644
index 0000000..12608a7
--- /dev/null
+++ b/src/dumb.rs
@@ -0,0 +1,27 @@
+use atools::prelude::*;
+pub trait Closest {
+ fn closest(&self, color: [f32; 4]) -> (f32, [f32; 4], usize);
+}
+
+fn euclidean_distance(f: [f32; 4], with: [f32; 4]) -> f32 {
+ f.asub(with).map(|x| x * x).sum()
+}
+
+impl Closest for &[[f32; 4]] {
+ fn closest(&self, color: [f32; 4]) -> (f32, [f32; 4], usize) {
+ self.iter()
+ .enumerate()
+ .map(|(i, x)| (euclidean_distance(*x, color), x, i))
+ .min_by(|x, y| x.0.total_cmp(&y.0))
+ .map(|(d, x, i)| (d, *x, i))
+ .unwrap()
+ // let mut best = (euclidean_distance(self[0], color), self[0], 0);
+ // for (&c, i) in self[1..].iter().zip(1..) {
+ // let d = euclidean_distance(c, color);
+ // if d < best.0 {
+ // best = (d, c, i);
+ // }
+ // }
+ // best
+ }
+}
diff --git a/src/kd.rs b/src/kd.rs
new file mode 100644
index 0000000..3b17124
--- /dev/null
+++ b/src/kd.rs
@@ -0,0 +1,148 @@
+/// A data structure for fast nearest color lookups in a palette.
+use atools::prelude::*;
+
+pub struct KD {
+ mid_point: [f32; 4],
+ index: u32,
+ normal: [f32; 4],
+ left: Option<Box<KD>>,
+ right: Option<Box<KD>>,
+}
+
+struct KDNearest {
+ index: u32,
+ distance: f32,
+}
+
+trait Dot {
+ fn dot(self, other: Self) -> f32;
+}
+
+impl Dot for [f32; 4] {
+ fn dot(self, other: Self) -> f32 {
+ self.amul(other).sum()
+ }
+}
+
+impl KD {
+ pub fn new(colors: &[[f32; 4]]) -> Self {
+ assert!(colors.len() < u32::MAX as usize);
+ let mut x = (0..colors.len() as u32).collect::<Vec<_>>();
+
+ Self::_new(&mut x, colors)
+ }
+
+ fn _new(mut indices: &mut [u32], colors: &[[f32; 4]]) -> KD {
+ assert!(indices.len() > 0);
+
+ let middle = indices.len() / 2;
+
+ let mut sum = [0.; 4];
+ let mut sum2 = [0.; 4];
+ for i in indices.iter() {
+ let c = colors[*i as usize];
+ sum = sum.aadd(c);
+ sum2 = sum2.aadd(c.amul(c));
+ }
+ let [r, g, b, a] = { sum2.asub(sum.amul(sum).mul(1.0 / indices.len() as f32)) };
+ let normal = if r > g && r > b && r > a {
+ (1.0).join([0.; 3])
+ } else if g > b && g > a {
+ [0.0, 1.0].couple([0.; 2])
+ } else if b > a {
+ [0.; 2].couple([1., 0.])
+ } else {
+ [0.; 3].join(1.)
+ };
+ indices.sort_by(|a, b| {
+ colors[*a as usize]
+ .dot(normal)
+ .partial_cmp(&colors[*b as usize].dot(normal))
+ .unwrap()
+ });
+ let i = indices.len() / 2;
+ let left = if i > 0 {
+ Some(Box::new(KD::_new(&mut indices[0..i], colors)))
+ } else {
+ None
+ };
+ let right = if i + 1 < indices.len() {
+ Some(Box::new(KD::_new(&mut indices[(i + 1)..], colors)))
+ } else {
+ None
+ };
+ KD {
+ mid_point: colors[indices[i as usize] as usize],
+ index: indices[i as usize],
+ normal: normal,
+ left: left,
+ right: right,
+ }
+ }
+
+ fn _find_nearest(
+ &self,
+ needle: [f32; 4],
+ mut limit: f32,
+ ignore_index: u32,
+ ) -> Option<KDNearest> {
+ let mut result = None;
+
+ let diff = needle.asub(self.mid_point);
+ let distance = diff.dot(diff).sqrt();
+
+ if distance < limit && self.index != ignore_index {
+ limit = distance;
+ result = Some(KDNearest {
+ index: self.index,
+ distance: distance,
+ })
+ }
+
+ let dot = diff.dot(self.normal);
+ if dot <= 0.0 {
+ if let Some(ref left) = self.left {
+ if let Some(nearest) = left._find_nearest(needle, limit, ignore_index) {
+ limit = nearest.distance;
+ result = Some(nearest);
+ }
+ }
+
+ if -dot < limit {
+ if let Some(ref right) = self.right {
+ if let Some(nearest) = right._find_nearest(needle, limit, ignore_index) {
+ result = Some(nearest);
+ }
+ }
+ }
+ } else {
+ if let Some(ref right) = self.right {
+ if let Some(nearest) = right._find_nearest(needle, limit, ignore_index) {
+ limit = nearest.distance;
+ result = Some(nearest);
+ }
+ }
+
+ if dot < limit {
+ if let Some(ref left) = self.left {
+ if let Some(nearest) = left._find_nearest(needle, limit, ignore_index) {
+ result = Some(nearest);
+ }
+ }
+ }
+ }
+
+ result
+ }
+
+ pub fn find_nearest(&self, color: [f32; 4]) -> u32 {
+ self._find_nearest(color, f32::MAX, u32::MAX)
+ .map(|x| x.index)
+ .unwrap_or(0)
+ }
+}
+
+fn occludes(origin: [f32; 4], occluder: [f32; 4], target: [f32; 4]) -> bool {
+ let dir = occluder.asub(origin);
+ dir.dot(dir) * 0.5 <= (target.asub(origin)).dot(dir)
+}
diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..8b83855
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,34 @@
+#![feature(
+ adt_const_params,
+ effects,
+ const_refs_to_cell,
+ generic_const_exprs,
+ core_intrinsics,
+ iter_intersperse,
+ const_trait_impl,
+ maybe_uninit_array_assume_init,
+ array_windows,
+ iter_map_windows
+)]
+mod dumb;
+mod kd;
+use dumb::Closest;
+use exoquant::Remapper;
+use fimg::Image;
+use kd::KD;
+// type KD = kiddo::immutable::float::kdtree::ImmutableKdTree<f32, u64, 4, 32>;
+fn map(colors: &[[f32; 4]]) -> KD {
+ KD::new(colors)
+}
+
+pub fn remap(image: Image<&[f32], 4>, palette: &[[f32; 4]]) -> Image<Box<[f32]>, 4> {
+ let kd = map(palette);
+ Image::build(image.width(), image.height()).buf(
+ image
+ .chunked()
+ .map(|x| palette[kd.find_nearest(*x) as usize])
+ // .map(|&x| palette.closest(x).1)
+ .flatten()
+ .collect(),
+ )
+}
diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..51aa694
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,78 @@
+#![feature(slice_as_chunks, generic_const_exprs)]
+use atools::prelude::*;
+use exoquant::SimpleColorSpace;
+use rand::{Rng, RngCore};
+
+fn main() {
+ reemap();
+ // eomap();
+ // let mut rng = rand::thread_rng();
+ // let pal = std::iter::repeat_with(|| {
+ // let a: [f32; 3] = std::array::from_fn(|_| rng.next_u64() as f32 / u64::MAX as f32);
+ // a.join(1.)
+ // })
+ // .take(200)
+ // .collect::<Vec<_>>();
+}
+
+fn reemap() {
+ // let mut rng = rand::thread_rng();
+ // let pal = std::iter::repeat_with(|| {
+ // let a: [f32; 3] = std::array::from_fn(|_| rng.next_u64() as f32 / u64::MAX as f32);
+ // a.join(1.)
+ // })
+ // .take(5124)
+ // .collect::<Vec<_>>();
+ let pal = fimg::Image::<Box<[f32]>, 4>::from(fimg::Image::open("../endesga.png").as_ref());
+ let pal = pal.flatten();
+
+ /*let pal = [
+ ]
+ .chunked::<3>()
+ .map(|x| x.join(255).map(|x| x as f32 * (1.0 / 255.0)));
+ */
+ // println!("{pal:?}");
+
+ fimg::Image::<Box<[u8]>, 4>::from(
+ remapper::remap(
+ fimg::Image::<Box<[f32]>, 4>::from(
+ fimg::Image::<Vec<u8>, 4>::open("plane_4096_2048.png").as_ref(),
+ )
+ .as_ref(),
+ &pal,
+ )
+ .as_ref(),
+ )
+ .save("yeee.png");
+}
+
+fn eomap() {
+ let x = fimg::Image::<Vec<u8>, 4>::open("../fimg/tdata/cat.png");
+ let pal = fimg::Image::open("../endesga.png");
+ let pal = pal.flatten();
+ let res = exoquant::Remapper::new(
+ &pal.iter()
+ .map(|&[r, g, b, a]| exoquant::Color::new(r, g, b, a))
+ .collect::<Vec<_>>(),
+ &SimpleColorSpace::default(),
+ &exoquant::ditherer::Ordered,
+ )
+ .remap(
+ &x.chunked()
+ .map(|&[r, g, b, a]| exoquant::Color::new(r, g, b, a))
+ .collect::<Vec<_>>(),
+ x.width() as usize,
+ );
+ let (width, height) = (x.width() as usize, x.height() as usize);
+ let pixels = (0..width)
+ .flat_map(|x_| (0..height).map(move |y| (x_, y)))
+ .filter_map(move |(x_, y_)| match res[(height - y_ - 1) * width + x_] {
+ 0 => None,
+ x => Some(((x_, y_), x)),
+ });
+ let mut preview = fimg::Image::build(x.width(), x.height()).fill([0; 3].join(255));
+ for ((x, y), i) in pixels {
+ unsafe { preview.set_pixel(x as _, (height - y - 1) as _, pal[i as usize]) };
+ }
+ preview.save("eoquan.png");
+}
diff --git a/yeee.png b/yeee.png
new file mode 100644
index 0000000..4807ad4
--- /dev/null
+++ b/yeee.png
Binary files differ