From 08a06d61107b980f11e75e6be216b89a2fb8573d Mon Sep 17 00:00:00 2001
From: Daniel Boros <dancixx@gmail.com>
Date: Thu, 29 Aug 2024 14:38:37 +0200
Subject: [PATCH 1/6] feat: add perf improvements

---
 Cargo.toml        |   3 +
 flamegraph.svg    | 491 ++++++++++++++++++++++++++++++++++++++++++++++
 src/main.rs       |  66 +------
 src/noises/fgn.rs |  21 +-
 4 files changed, 509 insertions(+), 72 deletions(-)
 create mode 100644 flamegraph.svg

diff --git a/Cargo.toml b/Cargo.toml
index e296c2d..a6e94c9 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -33,3 +33,6 @@ derive_builder = "0.20.0"
 name = "stochastic_rs"
 crate-type = ["cdylib", "lib"]
 path = "src/lib.rs"
+
+[profile.release]
+debug = true
\ No newline at end of file
diff --git a/flamegraph.svg b/flamegraph.svg
new file mode 100644
index 0000000..4169efa
--- /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="278" onload="init(evt)" viewBox="0 0 1200 278" 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="278" 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="261.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="261.00"> </text><svg id="frames" x="10" width="1180" total_samples="605"><g><title>libsystem_platform.dylib`_platform_memmove (29 samples, 4.79%)</title><rect x="0.0000%" y="101" width="4.7934%" height="15" fill="rgb(227,0,7)" fg:x="0" fg:w="29"/><text x="0.2500%" y="111.50">libsys..</text></g><g><title>stochastic-rs`&lt;alloc::vec::Vec&lt;T&gt; as alloc::vec::spec_from_iter::SpecFromIter&lt;T,I&gt;&gt;::from_iter (10 samples, 1.65%)</title><rect x="4.7934%" y="101" width="1.6529%" height="15" fill="rgb(217,0,24)" fg:x="29" fg:w="10"/><text x="5.0434%" y="111.50"></text></g><g><title>stochastic-rs`__rdl_dealloc (1 samples, 0.17%)</title><rect x="6.2810%" y="85" width="0.1653%" height="15" fill="rgb(221,193,54)" fg:x="38" fg:w="1"/><text x="6.5310%" y="95.50"></text></g><g><title>libsystem_malloc.dylib`free_medium (5 samples, 0.83%)</title><rect x="6.4463%" y="85" width="0.8264%" height="15" fill="rgb(248,212,6)" fg:x="39" fg:w="5"/><text x="6.6963%" y="95.50"></text></g><g><title>libsystem_kernel.dylib`madvise (5 samples, 0.83%)</title><rect x="6.4463%" y="69" width="0.8264%" height="15" fill="rgb(208,68,35)" fg:x="39" fg:w="5"/><text x="6.6963%" y="79.50"></text></g><g><title>libsystem_platform.dylib`__bzero (21 samples, 3.47%)</title><rect x="7.2727%" y="85" width="3.4711%" height="15" fill="rgb(232,128,0)" fg:x="44" fg:w="21"/><text x="7.5227%" y="95.50">lib..</text></g><g><title>libsystem_platform.dylib`_platform_memmove (18 samples, 2.98%)</title><rect x="10.7438%" y="85" width="2.9752%" height="15" fill="rgb(207,160,47)" fg:x="65" fg:w="18"/><text x="10.9938%" y="95.50">lib..</text></g><g><title>stochastic-rs`&lt;rand_distr::normal::StandardNormal as rand::distributions::distribution::Distribution&lt;f64&gt;&gt;::sample (92 samples, 15.21%)</title><rect x="13.7190%" y="85" width="15.2066%" height="15" fill="rgb(228,23,34)" fg:x="83" fg:w="92"/><text x="13.9690%" y="95.50">stochastic-rs`&lt;rand_dis..</text></g><g><title>stochastic-rs`ndarray::impl_methods::_&lt;impl ndarray::ArrayBase&lt;S,D&gt;&gt;::map (1 samples, 0.17%)</title><rect x="28.9256%" y="85" width="0.1653%" height="15" fill="rgb(218,30,26)" fg:x="175" fg:w="1"/><text x="29.1756%" y="95.50"></text></g><g><title>stochastic-rs`ndarray::impl_ops::arithmetic_ops::_&lt;impl core::ops::arith::Mul&lt;&amp;ndarray::ArrayBase&lt;S2,E&gt;&gt; for &amp;ndarray::ArrayBase&lt;S,D&gt;&gt;::mul (38 samples, 6.28%)</title><rect x="29.0909%" y="85" width="6.2810%" height="15" fill="rgb(220,122,19)" fg:x="176" fg:w="38"/><text x="29.3409%" y="95.50">stochast..</text></g><g><title>stochastic-rs`ndarray::zip::Zip&lt;(P1,P2,PLast),D&gt;::collect_with_partial (38 samples, 6.28%)</title><rect x="29.0909%" y="69" width="6.2810%" height="15" fill="rgb(250,228,42)" fg:x="176" fg:w="38"/><text x="29.3409%" y="79.50">stochast..</text></g><g><title>stochastic-rs`ndarray::iterators::to_vec_mapped (167 samples, 27.60%)</title><rect x="35.3719%" y="85" width="27.6033%" height="15" fill="rgb(240,193,28)" fg:x="214" fg:w="167"/><text x="35.6219%" y="95.50">stochastic-rs`ndarray::iterators::to_vec_map..</text></g><g><title>stochastic-rs`&lt;rand_distr::normal::StandardNormal as rand::distributions::distribution::Distribution&lt;f64&gt;&gt;::sample (156 samples, 25.79%)</title><rect x="37.1901%" y="69" width="25.7851%" height="15" fill="rgb(216,20,37)" fg:x="225" fg:w="156"/><text x="37.4401%" y="79.50">stochastic-rs`&lt;rand_distr::normal::Standa..</text></g><g><title>libsystem_m.dylib`exp (2 samples, 0.33%)</title><rect x="62.6446%" y="53" width="0.3306%" height="15" fill="rgb(206,188,39)" fg:x="379" fg:w="2"/><text x="62.8946%" y="63.50"></text></g><g><title>stochastic-rs`rustfft::array_utils::iter_chunks (13 samples, 2.15%)</title><rect x="62.9752%" y="69" width="2.1488%" height="15" fill="rgb(217,207,13)" fg:x="381" fg:w="13"/><text x="63.2252%" y="79.50">s..</text></g><g><title>stochastic-rs`rustfft::algorithm::radix4::bitreversed_transpose (19 samples, 3.14%)</title><rect x="87.4380%" y="53" width="3.1405%" height="15" fill="rgb(231,73,38)" fg:x="529" fg:w="19"/><text x="87.6880%" y="63.50">sto..</text></g><g><title>stochastic-rs`&lt;stochastic_rs::noises::fgn::FgnFft as stochastic_rs::utils::Generator&gt;::sample (519 samples, 85.79%)</title><rect x="6.4463%" y="101" width="85.7851%" height="15" fill="rgb(225,20,46)" fg:x="39" fg:w="519"/><text x="6.6963%" y="111.50">stochastic-rs`&lt;stochastic_rs::noises::fgn::FgnFft as stochastic_rs::utils::Generator&gt;::sample</text></g><g><title>stochastic-rs`rustfft::Fft::process (177 samples, 29.26%)</title><rect x="62.9752%" y="85" width="29.2562%" height="15" fill="rgb(210,31,41)" fg:x="381" fg:w="177"/><text x="63.2252%" y="95.50">stochastic-rs`rustfft::Fft::process</text></g><g><title>stochastic-rs`rustfft::neon::neon_radix4::Neon64Radix4&lt;T&gt;::perform_fft_out_of_place (164 samples, 27.11%)</title><rect x="65.1240%" y="69" width="27.1074%" height="15" fill="rgb(221,200,47)" fg:x="394" fg:w="164"/><text x="65.3740%" y="79.50">stochastic-rs`rustfft::neon::neon_radix4::N..</text></g><g><title>stochastic-rs`rustfft::algorithm::radix4::reverse_bits (10 samples, 1.65%)</title><rect x="90.5785%" y="53" width="1.6529%" height="15" fill="rgb(226,26,5)" fg:x="548" fg:w="10"/><text x="90.8285%" y="63.50"></text></g><g><title>stochastic-rs`&lt;stochastic_rs::processes::fbm::Fbm as stochastic_rs::utils::Generator&gt;::sample (598 samples, 98.84%)</title><rect x="0.0000%" y="117" width="98.8430%" height="15" fill="rgb(249,33,26)" fg:x="0" fg:w="598"/><text x="0.2500%" y="127.50">stochastic-rs`&lt;stochastic_rs::processes::fbm::Fbm as stochastic_rs::utils::Generator&gt;::sample</text></g><g><title>stochastic-rs`ndarray::impl_methods::_&lt;impl ndarray::ArrayBase&lt;S,D&gt;&gt;::accumulate_axis_inplace (40 samples, 6.61%)</title><rect x="92.2314%" y="101" width="6.6116%" height="15" fill="rgb(235,183,28)" fg:x="558" fg:w="40"/><text x="92.4814%" y="111.50">stochasti..</text></g><g><title>dyld`start (600 samples, 99.17%)</title><rect x="0.0000%" y="213" width="99.1736%" height="15" fill="rgb(221,5,38)" fg:x="0" fg:w="600"/><text x="0.2500%" y="223.50">dyld`start</text></g><g><title>stochastic-rs`main (600 samples, 99.17%)</title><rect x="0.0000%" y="197" width="99.1736%" height="15" fill="rgb(247,18,42)" fg:x="0" fg:w="600"/><text x="0.2500%" y="207.50">stochastic-rs`main</text></g><g><title>stochastic-rs`std::rt::lang_start_internal (600 samples, 99.17%)</title><rect x="0.0000%" y="181" width="99.1736%" height="15" fill="rgb(241,131,45)" fg:x="0" fg:w="600"/><text x="0.2500%" y="191.50">stochastic-rs`std::rt::lang_start_internal</text></g><g><title>stochastic-rs`std::rt::lang_start::_{{closure}} (600 samples, 99.17%)</title><rect x="0.0000%" y="165" width="99.1736%" height="15" fill="rgb(249,31,29)" fg:x="0" fg:w="600"/><text x="0.2500%" y="175.50">stochastic-rs`std::rt::lang_start::_{{closure}}</text></g><g><title>stochastic-rs`std::sys::backtrace::__rust_begin_short_backtrace (600 samples, 99.17%)</title><rect x="0.0000%" y="149" width="99.1736%" height="15" fill="rgb(225,111,53)" fg:x="0" fg:w="600"/><text x="0.2500%" y="159.50">stochastic-rs`std::sys::backtrace::__rust_begin_short_backtrace</text></g><g><title>stochastic-rs`stochastic_rs::main (600 samples, 99.17%)</title><rect x="0.0000%" y="133" width="99.1736%" height="15" fill="rgb(238,160,17)" fg:x="0" fg:w="600"/><text x="0.2500%" y="143.50">stochastic-rs`stochastic_rs::main</text></g><g><title>stochastic-rs`stochastic_rs::processes::fbm::Fbm::new (2 samples, 0.33%)</title><rect x="98.8430%" y="117" width="0.3306%" height="15" fill="rgb(214,148,48)" fg:x="598" fg:w="2"/><text x="99.0930%" y="127.50"></text></g><g><title>stochastic-rs`stochastic_rs::noises::fgn::FgnFft::new (2 samples, 0.33%)</title><rect x="98.8430%" y="101" width="0.3306%" height="15" fill="rgb(232,36,49)" fg:x="598" fg:w="2"/><text x="99.0930%" y="111.50"></text></g><g><title>stochastic-rs`ndrustfft::FftHandler&lt;T&gt;::new (2 samples, 0.33%)</title><rect x="98.8430%" y="85" width="0.3306%" height="15" fill="rgb(209,103,24)" fg:x="598" fg:w="2"/><text x="99.0930%" y="95.50"></text></g><g><title>stochastic-rs`rustfft::neon::neon_planner::FftPlannerNeon&lt;T&gt;::plan_fft (2 samples, 0.33%)</title><rect x="98.8430%" y="69" width="0.3306%" height="15" fill="rgb(229,88,8)" fg:x="598" fg:w="2"/><text x="99.0930%" y="79.50"></text></g><g><title>stochastic-rs`rustfft::neon::neon_planner::FftPlannerNeon&lt;T&gt;::build_fft (2 samples, 0.33%)</title><rect x="98.8430%" y="53" width="0.3306%" height="15" fill="rgb(213,181,19)" fg:x="598" fg:w="2"/><text x="99.0930%" y="63.50"></text></g><g><title>stochastic-rs`rustfft::neon::neon_radix4::Neon64Radix4&lt;T&gt;::new (2 samples, 0.33%)</title><rect x="98.8430%" y="37" width="0.3306%" height="15" fill="rgb(254,191,54)" fg:x="598" fg:w="2"/><text x="99.0930%" y="47.50"></text></g><g><title>libsystem_kernel.dylib`__exit (1 samples, 0.17%)</title><rect x="99.1736%" y="213" width="0.1653%" height="15" fill="rgb(241,83,37)" fg:x="600" fg:w="1"/><text x="99.4236%" y="223.50"></text></g><g><title>stochastic-rs`&lt;rayon_core::registry::WorkerThread as core::convert::From&lt;rayon_core::registry::ThreadBuilder&gt;&gt;::from (1 samples, 0.17%)</title><rect x="99.3388%" y="117" width="0.1653%" height="15" fill="rgb(233,36,39)" fg:x="601" fg:w="1"/><text x="99.5888%" y="127.50"></text></g><g><title>libsystem_malloc.dylib`szone_malloc_should_clear (1 samples, 0.17%)</title><rect x="99.3388%" y="101" width="0.1653%" height="15" fill="rgb(226,3,54)" fg:x="601" fg:w="1"/><text x="99.5888%" y="111.50"></text></g><g><title>libsystem_malloc.dylib`small_malloc_should_clear (1 samples, 0.17%)</title><rect x="99.3388%" y="85" width="0.1653%" height="15" fill="rgb(245,192,40)" fg:x="601" fg:w="1"/><text x="99.5888%" y="95.50"></text></g><g><title>libsystem_malloc.dylib`mvm_allocate_pages (1 samples, 0.17%)</title><rect x="99.3388%" y="69" width="0.1653%" height="15" fill="rgb(238,167,29)" fg:x="601" fg:w="1"/><text x="99.5888%" y="79.50"></text></g><g><title>libsystem_kernel.dylib`_kernelrpc_mach_vm_map_trap (1 samples, 0.17%)</title><rect x="99.3388%" y="53" width="0.1653%" height="15" fill="rgb(232,182,51)" fg:x="601" fg:w="1"/><text x="99.5888%" y="63.50"></text></g><g><title>libsystem_kernel.dylib`swtch_pri (2 samples, 0.33%)</title><rect x="99.5041%" y="101" width="0.3306%" height="15" fill="rgb(231,60,39)" fg:x="602" fg:w="2"/><text x="99.7541%" y="111.50"></text></g><g><title>all (605 samples, 100%)</title><rect x="0.0000%" y="229" width="100.0000%" height="15" fill="rgb(208,69,12)" fg:x="0" fg:w="605"/><text x="0.2500%" y="239.50"></text></g><g><title>libsystem_pthread.dylib`thread_start (4 samples, 0.66%)</title><rect x="99.3388%" y="213" width="0.6612%" height="15" fill="rgb(235,93,37)" fg:x="601" fg:w="4"/><text x="99.5888%" y="223.50"></text></g><g><title>libsystem_pthread.dylib`_pthread_start (4 samples, 0.66%)</title><rect x="99.3388%" y="197" width="0.6612%" height="15" fill="rgb(213,116,39)" fg:x="601" fg:w="4"/><text x="99.5888%" y="207.50"></text></g><g><title>stochastic-rs`std::sys::pal::unix::thread::Thread::new::thread_start (4 samples, 0.66%)</title><rect x="99.3388%" y="181" width="0.6612%" height="15" fill="rgb(222,207,29)" fg:x="601" fg:w="4"/><text x="99.5888%" y="191.50"></text></g><g><title>stochastic-rs`core::ops::function::FnOnce::call_once{{vtable.shim}} (4 samples, 0.66%)</title><rect x="99.3388%" y="165" width="0.6612%" height="15" fill="rgb(206,96,30)" fg:x="601" fg:w="4"/><text x="99.5888%" y="175.50"></text></g><g><title>stochastic-rs`std::sys::backtrace::__rust_begin_short_backtrace (4 samples, 0.66%)</title><rect x="99.3388%" y="149" width="0.6612%" height="15" fill="rgb(218,138,4)" fg:x="601" fg:w="4"/><text x="99.5888%" y="159.50"></text></g><g><title>stochastic-rs`rayon_core::registry::ThreadBuilder::run (4 samples, 0.66%)</title><rect x="99.3388%" y="133" width="0.6612%" height="15" fill="rgb(250,191,14)" fg:x="601" fg:w="4"/><text x="99.5888%" y="143.50"></text></g><g><title>stochastic-rs`rayon_core::registry::WorkerThread::wait_until_cold (3 samples, 0.50%)</title><rect x="99.5041%" y="117" width="0.4959%" height="15" fill="rgb(239,60,40)" fg:x="602" fg:w="3"/><text x="99.7541%" y="127.50"></text></g><g><title>stochastic-rs`rayon_core::join::join_context::_{{closure}} (1 samples, 0.17%)</title><rect x="99.8347%" y="101" width="0.1653%" height="15" fill="rgb(206,27,48)" fg:x="604" fg:w="1"/><text x="100.0847%" y="111.50"></text></g></svg></svg>
\ No newline at end of file
diff --git a/src/main.rs b/src/main.rs
index 7d35425..7e3fa08 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,70 +1,12 @@
 use std::time::Instant;
 
-use plotly::{common::Line, Plot, Scatter};
-use rand_distr::{Gamma, Normal, StandardNormal};
-use stochastic_rs::{
-  diffusions::fou::{fou, Fou},
-  jumps::{
-    bates::{bates_1996, Bates1996},
-    jump_fou::{jump_fou, JumpFou},
-  },
-  processes::fbm::Fbm,
-  utils::Generator,
-};
+use stochastic_rs::{processes::fbm::Fbm, utils::Generator};
 
 fn main() {
   let start = Instant::now();
-
-  let mut plot = Plot::new();
-
-  let jdistr = Gamma::new(1.45, 2.43).unwrap();
-  let jtdistr = Gamma::new(1.45, 2.43).unwrap();
-  let params = JumpFou {
-    hurst: 0.7,
-    mu: 4.0,
-    sigma: 1.0,
-    theta: 2.0,
-    n: 1000,
-    lambda: Some(0.5),
-    x0: Some(0.0),
-    t: Some(1.0),
-  };
-
-  // let bates_params = Bates1996 {
-  //   mu: Some(0.0),
-  //   b: Some(0.0),
-  //   r: Some(0.0),
-  //   r_f: Some(0.0),
-  //   lambda: 0.5,
-  //   k: 0.5,
-  //   alpha: 0.5,
-  //   beta: 0.5,
-  //   sigma: 0.5,
-  //   rho: 0.5,
-  //   n: 1000,
-  //   s0: Some(0.0),
-  //   v0: Some(0.0),
-  //   t: Some(1.0),
-  //   use_sym: Some(false),
-  // };
-  for _ in 0..1 {
-    let s = jump_fou(&params, StandardNormal);
-    let fbm = Fbm::new(0.95, 200, Some(200.0), None);
-    //let bates = bates_1996(&bates_params, StandardNormal);
-    //let f = fbm.sample();
-
-    let trace = Scatter::new((0..s.len()).collect::<Vec<_>>(), s.to_vec())
-      .mode(plotly::common::Mode::Lines)
-      .line(
-        Line::new()
-          .color("orange")
-          .shape(plotly::common::LineShape::Linear),
-      )
-      .name("Bates");
-    plot.add_trace(trace);
+  let fbm = Fbm::new(0.95, 10000, Some(1.0), None);
+  for _ in 0..1000 {
+    fbm.sample();
   }
-
-  plot.show();
-
   println!("{}", start.elapsed().as_secs_f64());
 }
diff --git a/src/noises/fgn.rs b/src/noises/fgn.rs
index 2eaf2c8..2e48344 100644
--- a/src/noises/fgn.rs
+++ b/src/noises/fgn.rs
@@ -3,6 +3,8 @@
 //! The `FgnFft` struct provides methods to generate fractional Gaussian noise (FGN)
 //! using the Fast Fourier Transform (FFT) approach.
 
+use std::sync::{Arc, Mutex};
+
 use crate::utils::Generator;
 use ndarray::parallel::prelude::*;
 use ndarray::{concatenate, prelude::*};
@@ -17,10 +19,10 @@ pub struct FgnFft {
   n: usize,
   offset: usize,
   t: f64,
-  sqrt_eigenvalues: Array1<Complex<f64>>,
+  sqrt_eigenvalues: Arc<Array1<Complex<f64>>>, // Cache-be tárolva
   m: Option<usize>,
-  fft_handler: FftHandler<f64>,
-  fft_fgn: Array1<Complex<f64>>,
+  fft_handler: Arc<FftHandler<f64>>,         // Cache-be tárolva
+  fft_fgn: Arc<Mutex<Array1<Complex<f64>>>>, // Mutex védelem
 }
 
 impl FgnFft {
@@ -79,10 +81,10 @@ impl FgnFft {
       n,
       offset,
       t: t.unwrap_or(1.0),
-      sqrt_eigenvalues,
+      sqrt_eigenvalues: Arc::new(sqrt_eigenvalues), // Cache-be tárolva
       m,
-      fft_handler: FftHandler::new(2 * n),
-      fft_fgn: Array1::<Complex<f64>>::zeros(2 * n),
+      fft_handler: Arc::new(FftHandler::new(2 * n)), // Cache-be tárolva
+      fft_fgn: Arc::new(Mutex::new(Array1::<Complex<f64>>::zeros(2 * n))), // Mutex védelem
     }
   }
 }
@@ -105,10 +107,9 @@ impl Generator for FgnFft {
       2 * self.n,
       ComplexDistribution::new(StandardNormal, StandardNormal),
     );
-    let fgn = &self.sqrt_eigenvalues * &rnd;
-    let fft_handler = self.fft_handler.clone();
-    let mut fgn_fft = self.fft_fgn.clone();
-    ndfft_par(&fgn, &mut fgn_fft, &fft_handler, 0);
+    let fgn = &*self.sqrt_eigenvalues * &rnd;
+    let mut fgn_fft = self.fft_fgn.lock().unwrap().clone();
+    ndfft_par(&fgn, &mut fgn_fft, &*self.fft_handler, 0);
     let fgn = fgn_fft
       .slice(s![1..self.n - self.offset + 1])
       .mapv(|x: Complex<f64>| (x.re * (self.n as f64).powf(-self.hurst)) * self.t.powf(self.hurst));

From 8209c83e27a5e39ff3d7023468fe53a9ddda974a Mon Sep 17 00:00:00 2001
From: Daniel Boros <dancixx@gmail.com>
Date: Sun, 1 Sep 2024 01:11:06 +0200
Subject: [PATCH 2/6] feat: speed up fbm

---
 Cargo.toml           | 10 +++++++++-
 src/lib.rs           | 10 ++++++++++
 src/main.rs          | 28 ++++++++++++++++++++++++++--
 src/noises/fgn.rs    | 29 +++++++++++++++--------------
 src/processes/fbm.rs | 14 ++++++++++----
 5 files changed, 70 insertions(+), 21 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index a6e94c9..78ac604 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -25,9 +25,16 @@ plotly = "0.9.0"
 ndarray-rand = "0.15.0"
 ndrustfft = "0.5.0"
 derive_builder = "0.20.0"
+tikv-jemallocator = {version = "0.6.0", optional = true}
+mimalloc = {version = "0.1.43", optional = true}
 
 [dev-dependencies]
 
+[features]
+mimalloc = ["dep:mimalloc"]
+jemalloc = ["dep:tikv-jemallocator"]
+default = ["jemalloc"]
+
 
 [lib]
 name = "stochastic_rs"
@@ -35,4 +42,5 @@ crate-type = ["cdylib", "lib"]
 path = "src/lib.rs"
 
 [profile.release]
-debug = true
\ No newline at end of file
+debug = false
+codegen-units = 1
\ No newline at end of file
diff --git a/src/lib.rs b/src/lib.rs
index 3bdfd1c..441e27f 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -53,6 +53,16 @@
 //! - Developed by [dancixx](https://github.com/dancixx).
 //! - Contributions and feedback are welcome!
 
+#![feature(portable_simd)]
+
+#[cfg(feature = "mimalloc")]
+#[global_allocator]
+static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
+
+#[cfg(feature = "jemalloc")]
+#[global_allocator]
+static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
+
 pub mod diffusions;
 pub mod interest;
 pub mod jumps;
diff --git a/src/main.rs b/src/main.rs
index 7e3fa08..27e90c6 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -5,8 +5,32 @@ use stochastic_rs::{processes::fbm::Fbm, utils::Generator};
 fn main() {
   let start = Instant::now();
   let fbm = Fbm::new(0.95, 10000, Some(1.0), None);
-  for _ in 0..1000 {
+  (0..1000).for_each(|_| {
     fbm.sample();
-  }
+  });
+  println!("{}", start.elapsed().as_secs_f64());
+
+  let start = Instant::now();
+  (0..1000).for_each(|_| {
+    fbm.sample();
+  });
+  println!("{}", start.elapsed().as_secs_f64());
+
+  let start = Instant::now();
+  (0..1000).for_each(|_| {
+    fbm.sample();
+  });
+  println!("{}", start.elapsed().as_secs_f64());
+
+  let start = Instant::now();
+  (0..1000).for_each(|_| {
+    fbm.sample();
+  });
+  println!("{}", start.elapsed().as_secs_f64());
+
+  let start = Instant::now();
+  (0..1000).for_each(|_| {
+    fbm.sample();
+  });
   println!("{}", start.elapsed().as_secs_f64());
 }
diff --git a/src/noises/fgn.rs b/src/noises/fgn.rs
index 2e48344..1c7ecc2 100644
--- a/src/noises/fgn.rs
+++ b/src/noises/fgn.rs
@@ -3,14 +3,14 @@
 //! The `FgnFft` struct provides methods to generate fractional Gaussian noise (FGN)
 //! using the Fast Fourier Transform (FFT) approach.
 
-use std::sync::{Arc, Mutex};
+use std::sync::{Arc, RwLock};
 
 use crate::utils::Generator;
 use ndarray::parallel::prelude::*;
 use ndarray::{concatenate, prelude::*};
 use ndarray_rand::rand_distr::StandardNormal;
 use ndarray_rand::RandomExt;
-use ndrustfft::{ndfft_par, FftHandler};
+use ndrustfft::{ndfft, FftHandler};
 use num_complex::{Complex, ComplexDistribution};
 
 /// Struct for generating Fractional Gaussian Noise (FGN) using FFT.
@@ -19,10 +19,10 @@ pub struct FgnFft {
   n: usize,
   offset: usize,
   t: f64,
-  sqrt_eigenvalues: Arc<Array1<Complex<f64>>>, // Cache-be tárolva
+  sqrt_eigenvalues: Arc<Array1<Complex<f64>>>,
   m: Option<usize>,
-  fft_handler: Arc<FftHandler<f64>>,         // Cache-be tárolva
-  fft_fgn: Arc<Mutex<Array1<Complex<f64>>>>, // Mutex védelem
+  fft_handler: Arc<FftHandler<f64>>,
+  fft_fgn: Arc<RwLock<Array1<Complex<f64>>>>,
 }
 
 impl FgnFft {
@@ -56,7 +56,7 @@ impl FgnFft {
     let offset = n_ - n;
     let n = n_;
     let mut r = Array1::linspace(0.0, n as f64, n + 1);
-    r.par_mapv_inplace(|x| {
+    r.mapv_inplace(|x| {
       if x == 0.0 {
         1.0
       } else {
@@ -73,18 +73,18 @@ impl FgnFft {
     let data = r.mapv(|v| Complex::new(v, 0.0));
     let r_fft = FftHandler::new(r.len());
     let mut sqrt_eigenvalues = Array1::<Complex<f64>>::zeros(r.len());
-    ndfft_par(&data, &mut sqrt_eigenvalues, &r_fft, 0);
-    sqrt_eigenvalues.par_mapv_inplace(|x| Complex::new((x.re / (2.0 * n as f64)).sqrt(), x.im));
+    ndfft(&data, &mut sqrt_eigenvalues, &r_fft, 0);
+    sqrt_eigenvalues.mapv_inplace(|x| Complex::new((x.re / (2.0 * n as f64)).sqrt(), x.im));
 
     Self {
       hurst,
       n,
       offset,
       t: t.unwrap_or(1.0),
-      sqrt_eigenvalues: Arc::new(sqrt_eigenvalues), // Cache-be tárolva
+      sqrt_eigenvalues: Arc::new(sqrt_eigenvalues),
       m,
-      fft_handler: Arc::new(FftHandler::new(2 * n)), // Cache-be tárolva
-      fft_fgn: Arc::new(Mutex::new(Array1::<Complex<f64>>::zeros(2 * n))), // Mutex védelem
+      fft_handler: Arc::new(FftHandler::new(2 * n)),
+      fft_fgn: Arc::new(RwLock::new(Array1::<Complex<f64>>::zeros(2 * n))),
     }
   }
 }
@@ -108,11 +108,12 @@ impl Generator for FgnFft {
       ComplexDistribution::new(StandardNormal, StandardNormal),
     );
     let fgn = &*self.sqrt_eigenvalues * &rnd;
-    let mut fgn_fft = self.fft_fgn.lock().unwrap().clone();
-    ndfft_par(&fgn, &mut fgn_fft, &*self.fft_handler, 0);
+    let mut fgn_fft = self.fft_fgn.read().unwrap().clone();
+    ndfft(&fgn, &mut fgn_fft, &*self.fft_handler, 0);
+    let scale = (self.n as f64).powf(-self.hurst) * self.t.powf(self.hurst);
     let fgn = fgn_fft
       .slice(s![1..self.n - self.offset + 1])
-      .mapv(|x: Complex<f64>| (x.re * (self.n as f64).powf(-self.hurst)) * self.t.powf(self.hurst));
+      .mapv(|x: Complex<f64>| x.re * scale);
     fgn
   }
 
diff --git a/src/processes/fbm.rs b/src/processes/fbm.rs
index b470d24..d9f97a8 100644
--- a/src/processes/fbm.rs
+++ b/src/processes/fbm.rs
@@ -1,7 +1,7 @@
 //! Fractional Brownian Motion (fBM) generator.
 
 use crate::{noises::fgn::FgnFft, utils::Generator};
-use ndarray::{Array1, Array2, Axis};
+use ndarray::{s, Array1, Array2, Axis};
 use rayon::prelude::*;
 
 /// Struct for generating Fractional Brownian Motion (fBM).
@@ -67,9 +67,15 @@ impl Generator for Fbm {
   /// ```
   fn sample(&self) -> Array1<f64> {
     let fgn = self.fgn.as_ref().unwrap().sample();
-    let mut fbm = Array1::<f64>::from(fgn);
-    fbm.accumulate_axis_inplace(Axis(0), |&x, y| *y += x);
-    vec![0.0].into_iter().chain(fbm).collect()
+
+    let mut fbm = Array1::<f64>::zeros(self.n);
+    fbm.slice_mut(s![1..]).assign(&fgn);
+
+    for i in 1..self.n {
+      fbm[i] += fbm[i - 1];
+    }
+
+    fbm
   }
 
   /// Generates parallel samples of fractional Brownian motion (fBM).

From cf0fb3196b4ed684f69fa0b3f09cbbd2f6683156 Mon Sep 17 00:00:00 2001
From: Daniel Boros <dancixx@gmail.com>
Date: Sun, 1 Sep 2024 01:14:39 +0200
Subject: [PATCH 3/6] feat: add fbm plot test

---
 src/main.rs          | 37 +------------------------------------
 src/processes/fbm.rs | 26 ++++++++++++++++++++++++++
 2 files changed, 27 insertions(+), 36 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 27e90c6..f328e4d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,36 +1 @@
-use std::time::Instant;
-
-use stochastic_rs::{processes::fbm::Fbm, utils::Generator};
-
-fn main() {
-  let start = Instant::now();
-  let fbm = Fbm::new(0.95, 10000, Some(1.0), None);
-  (0..1000).for_each(|_| {
-    fbm.sample();
-  });
-  println!("{}", start.elapsed().as_secs_f64());
-
-  let start = Instant::now();
-  (0..1000).for_each(|_| {
-    fbm.sample();
-  });
-  println!("{}", start.elapsed().as_secs_f64());
-
-  let start = Instant::now();
-  (0..1000).for_each(|_| {
-    fbm.sample();
-  });
-  println!("{}", start.elapsed().as_secs_f64());
-
-  let start = Instant::now();
-  (0..1000).for_each(|_| {
-    fbm.sample();
-  });
-  println!("{}", start.elapsed().as_secs_f64());
-
-  let start = Instant::now();
-  (0..1000).for_each(|_| {
-    fbm.sample();
-  });
-  println!("{}", start.elapsed().as_secs_f64());
-}
+fn main() {}
diff --git a/src/processes/fbm.rs b/src/processes/fbm.rs
index d9f97a8..690a23c 100644
--- a/src/processes/fbm.rs
+++ b/src/processes/fbm.rs
@@ -108,3 +108,29 @@ impl Generator for Fbm {
     xs
   }
 }
+
+#[cfg(test)]
+mod tests {
+  use plotly::{common::Line, Plot, Scatter};
+
+  use super::*;
+
+  #[test]
+  fn plot() {
+    let fbm = Fbm::new(0.45, 1000, Some(1.0), Some(10));
+    let mut plot = Plot::new();
+    let d = fbm.sample_par();
+    for data in d.axis_iter(Axis(0)) {
+      let trace = Scatter::new((0..data.len()).collect::<Vec<_>>(), data.to_vec())
+        .mode(plotly::common::Mode::Lines)
+        .line(
+          Line::new()
+            .color("orange")
+            .shape(plotly::common::LineShape::Linear),
+        )
+        .name("Fbm");
+      plot.add_trace(trace);
+    }
+    plot.show();
+  }
+}

From 578ca84e484fe73e2aa937c01e117ce8f8c516ad Mon Sep 17 00:00:00 2001
From: Daniel Boros <dancixx@gmail.com>
Date: Sun, 1 Sep 2024 01:22:31 +0200
Subject: [PATCH 4/6] feat: add fgn plot tes

---
 src/noises/fgn.rs | 30 ++++++++++++++++++++++++++++--
 1 file changed, 28 insertions(+), 2 deletions(-)

diff --git a/src/noises/fgn.rs b/src/noises/fgn.rs
index 1c7ecc2..3efa876 100644
--- a/src/noises/fgn.rs
+++ b/src/noises/fgn.rs
@@ -138,12 +138,38 @@ impl Generator for FgnFft {
       panic!("m must be specified for parallel sampling");
     }
 
-    let mut xs = Array2::zeros((self.m.unwrap(), self.n));
+    let mut xs = Array2::zeros((self.m.unwrap(), self.n - self.offset));
 
     xs.axis_iter_mut(Axis(0)).into_par_iter().for_each(|mut x| {
-      x.assign(&self.sample());
+      x.assign(&self.sample().slice(s![..self.n - self.offset]));
     });
 
     xs
   }
 }
+
+#[cfg(test)]
+mod tests {
+  use plotly::{common::Line, Plot, Scatter};
+
+  use super::*;
+
+  #[test]
+  fn plot() {
+    let fgn = FgnFft::new(0.9, 1000, Some(1.0), Some(1));
+    let mut plot = Plot::new();
+    let d = fgn.sample_par();
+    for data in d.axis_iter(Axis(0)) {
+      let trace = Scatter::new((0..data.len()).collect::<Vec<_>>(), data.to_vec())
+        .mode(plotly::common::Mode::Lines)
+        .line(
+          Line::new()
+            .color("orange")
+            .shape(plotly::common::LineShape::Linear),
+        )
+        .name("Fgn");
+      plot.add_trace(trace);
+    }
+    plot.show();
+  }
+}

From 2733992a87a39104dba555ef7a5c98edf2743ba7 Mon Sep 17 00:00:00 2001
From: Daniel Boros <dancixx@gmail.com>
Date: Sun, 1 Sep 2024 13:10:43 +0200
Subject: [PATCH 5/6] feat: update deps

---
 .gitignore     |   1 +
 Cargo.toml     |  13 +-
 flamegraph.svg | 491 -------------------------------------------------
 3 files changed, 7 insertions(+), 498 deletions(-)
 delete mode 100644 flamegraph.svg

diff --git a/.gitignore b/.gitignore
index 4fffb2f..fd3233f 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,3 @@
 /target
 /Cargo.lock
+flamegraph.svg
\ No newline at end of file
diff --git a/Cargo.toml b/Cargo.toml
index 78ac604..065090a 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "stochastic-rs"
-version = "0.6.5"
+version = "0.7.0"
 edition = "2021"
 license = "MIT"
 description = "A Rust library for stochastic processes"
@@ -20,20 +20,19 @@ num-complex = { version = "0.4.6", features = ["rand"] }
 rand = "0.8.5"
 rand_distr = "0.4.3"
 rayon = "1.10.0"
-indicatif = "0.17.7"
 plotly = "0.9.0"
 ndarray-rand = "0.15.0"
 ndrustfft = "0.5.0"
-derive_builder = "0.20.0"
-tikv-jemallocator = {version = "0.6.0", optional = true}
-mimalloc = {version = "0.1.43", optional = true}
+derive_builder = "0.20.1"
+tikv-jemallocator = { version = "0.6.0", optional = true }
+mimalloc = { version = "0.1.43", optional = true }
 
 [dev-dependencies]
 
 [features]
 mimalloc = ["dep:mimalloc"]
 jemalloc = ["dep:tikv-jemallocator"]
-default = ["jemalloc"]
+default = []
 
 
 [lib]
@@ -43,4 +42,4 @@ path = "src/lib.rs"
 
 [profile.release]
 debug = false
-codegen-units = 1
\ No newline at end of file
+codegen-units = 1
diff --git a/flamegraph.svg b/flamegraph.svg
deleted file mode 100644
index 4169efa..0000000
--- a/flamegraph.svg
+++ /dev/null
@@ -1,491 +0,0 @@
-<?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="278" onload="init(evt)" viewBox="0 0 1200 278" 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="278" 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="261.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="261.00"> </text><svg id="frames" x="10" width="1180" total_samples="605"><g><title>libsystem_platform.dylib`_platform_memmove (29 samples, 4.79%)</title><rect x="0.0000%" y="101" width="4.7934%" height="15" fill="rgb(227,0,7)" fg:x="0" fg:w="29"/><text x="0.2500%" y="111.50">libsys..</text></g><g><title>stochastic-rs`&lt;alloc::vec::Vec&lt;T&gt; as alloc::vec::spec_from_iter::SpecFromIter&lt;T,I&gt;&gt;::from_iter (10 samples, 1.65%)</title><rect x="4.7934%" y="101" width="1.6529%" height="15" fill="rgb(217,0,24)" fg:x="29" fg:w="10"/><text x="5.0434%" y="111.50"></text></g><g><title>stochastic-rs`__rdl_dealloc (1 samples, 0.17%)</title><rect x="6.2810%" y="85" width="0.1653%" height="15" fill="rgb(221,193,54)" fg:x="38" fg:w="1"/><text x="6.5310%" y="95.50"></text></g><g><title>libsystem_malloc.dylib`free_medium (5 samples, 0.83%)</title><rect x="6.4463%" y="85" width="0.8264%" height="15" fill="rgb(248,212,6)" fg:x="39" fg:w="5"/><text x="6.6963%" y="95.50"></text></g><g><title>libsystem_kernel.dylib`madvise (5 samples, 0.83%)</title><rect x="6.4463%" y="69" width="0.8264%" height="15" fill="rgb(208,68,35)" fg:x="39" fg:w="5"/><text x="6.6963%" y="79.50"></text></g><g><title>libsystem_platform.dylib`__bzero (21 samples, 3.47%)</title><rect x="7.2727%" y="85" width="3.4711%" height="15" fill="rgb(232,128,0)" fg:x="44" fg:w="21"/><text x="7.5227%" y="95.50">lib..</text></g><g><title>libsystem_platform.dylib`_platform_memmove (18 samples, 2.98%)</title><rect x="10.7438%" y="85" width="2.9752%" height="15" fill="rgb(207,160,47)" fg:x="65" fg:w="18"/><text x="10.9938%" y="95.50">lib..</text></g><g><title>stochastic-rs`&lt;rand_distr::normal::StandardNormal as rand::distributions::distribution::Distribution&lt;f64&gt;&gt;::sample (92 samples, 15.21%)</title><rect x="13.7190%" y="85" width="15.2066%" height="15" fill="rgb(228,23,34)" fg:x="83" fg:w="92"/><text x="13.9690%" y="95.50">stochastic-rs`&lt;rand_dis..</text></g><g><title>stochastic-rs`ndarray::impl_methods::_&lt;impl ndarray::ArrayBase&lt;S,D&gt;&gt;::map (1 samples, 0.17%)</title><rect x="28.9256%" y="85" width="0.1653%" height="15" fill="rgb(218,30,26)" fg:x="175" fg:w="1"/><text x="29.1756%" y="95.50"></text></g><g><title>stochastic-rs`ndarray::impl_ops::arithmetic_ops::_&lt;impl core::ops::arith::Mul&lt;&amp;ndarray::ArrayBase&lt;S2,E&gt;&gt; for &amp;ndarray::ArrayBase&lt;S,D&gt;&gt;::mul (38 samples, 6.28%)</title><rect x="29.0909%" y="85" width="6.2810%" height="15" fill="rgb(220,122,19)" fg:x="176" fg:w="38"/><text x="29.3409%" y="95.50">stochast..</text></g><g><title>stochastic-rs`ndarray::zip::Zip&lt;(P1,P2,PLast),D&gt;::collect_with_partial (38 samples, 6.28%)</title><rect x="29.0909%" y="69" width="6.2810%" height="15" fill="rgb(250,228,42)" fg:x="176" fg:w="38"/><text x="29.3409%" y="79.50">stochast..</text></g><g><title>stochastic-rs`ndarray::iterators::to_vec_mapped (167 samples, 27.60%)</title><rect x="35.3719%" y="85" width="27.6033%" height="15" fill="rgb(240,193,28)" fg:x="214" fg:w="167"/><text x="35.6219%" y="95.50">stochastic-rs`ndarray::iterators::to_vec_map..</text></g><g><title>stochastic-rs`&lt;rand_distr::normal::StandardNormal as rand::distributions::distribution::Distribution&lt;f64&gt;&gt;::sample (156 samples, 25.79%)</title><rect x="37.1901%" y="69" width="25.7851%" height="15" fill="rgb(216,20,37)" fg:x="225" fg:w="156"/><text x="37.4401%" y="79.50">stochastic-rs`&lt;rand_distr::normal::Standa..</text></g><g><title>libsystem_m.dylib`exp (2 samples, 0.33%)</title><rect x="62.6446%" y="53" width="0.3306%" height="15" fill="rgb(206,188,39)" fg:x="379" fg:w="2"/><text x="62.8946%" y="63.50"></text></g><g><title>stochastic-rs`rustfft::array_utils::iter_chunks (13 samples, 2.15%)</title><rect x="62.9752%" y="69" width="2.1488%" height="15" fill="rgb(217,207,13)" fg:x="381" fg:w="13"/><text x="63.2252%" y="79.50">s..</text></g><g><title>stochastic-rs`rustfft::algorithm::radix4::bitreversed_transpose (19 samples, 3.14%)</title><rect x="87.4380%" y="53" width="3.1405%" height="15" fill="rgb(231,73,38)" fg:x="529" fg:w="19"/><text x="87.6880%" y="63.50">sto..</text></g><g><title>stochastic-rs`&lt;stochastic_rs::noises::fgn::FgnFft as stochastic_rs::utils::Generator&gt;::sample (519 samples, 85.79%)</title><rect x="6.4463%" y="101" width="85.7851%" height="15" fill="rgb(225,20,46)" fg:x="39" fg:w="519"/><text x="6.6963%" y="111.50">stochastic-rs`&lt;stochastic_rs::noises::fgn::FgnFft as stochastic_rs::utils::Generator&gt;::sample</text></g><g><title>stochastic-rs`rustfft::Fft::process (177 samples, 29.26%)</title><rect x="62.9752%" y="85" width="29.2562%" height="15" fill="rgb(210,31,41)" fg:x="381" fg:w="177"/><text x="63.2252%" y="95.50">stochastic-rs`rustfft::Fft::process</text></g><g><title>stochastic-rs`rustfft::neon::neon_radix4::Neon64Radix4&lt;T&gt;::perform_fft_out_of_place (164 samples, 27.11%)</title><rect x="65.1240%" y="69" width="27.1074%" height="15" fill="rgb(221,200,47)" fg:x="394" fg:w="164"/><text x="65.3740%" y="79.50">stochastic-rs`rustfft::neon::neon_radix4::N..</text></g><g><title>stochastic-rs`rustfft::algorithm::radix4::reverse_bits (10 samples, 1.65%)</title><rect x="90.5785%" y="53" width="1.6529%" height="15" fill="rgb(226,26,5)" fg:x="548" fg:w="10"/><text x="90.8285%" y="63.50"></text></g><g><title>stochastic-rs`&lt;stochastic_rs::processes::fbm::Fbm as stochastic_rs::utils::Generator&gt;::sample (598 samples, 98.84%)</title><rect x="0.0000%" y="117" width="98.8430%" height="15" fill="rgb(249,33,26)" fg:x="0" fg:w="598"/><text x="0.2500%" y="127.50">stochastic-rs`&lt;stochastic_rs::processes::fbm::Fbm as stochastic_rs::utils::Generator&gt;::sample</text></g><g><title>stochastic-rs`ndarray::impl_methods::_&lt;impl ndarray::ArrayBase&lt;S,D&gt;&gt;::accumulate_axis_inplace (40 samples, 6.61%)</title><rect x="92.2314%" y="101" width="6.6116%" height="15" fill="rgb(235,183,28)" fg:x="558" fg:w="40"/><text x="92.4814%" y="111.50">stochasti..</text></g><g><title>dyld`start (600 samples, 99.17%)</title><rect x="0.0000%" y="213" width="99.1736%" height="15" fill="rgb(221,5,38)" fg:x="0" fg:w="600"/><text x="0.2500%" y="223.50">dyld`start</text></g><g><title>stochastic-rs`main (600 samples, 99.17%)</title><rect x="0.0000%" y="197" width="99.1736%" height="15" fill="rgb(247,18,42)" fg:x="0" fg:w="600"/><text x="0.2500%" y="207.50">stochastic-rs`main</text></g><g><title>stochastic-rs`std::rt::lang_start_internal (600 samples, 99.17%)</title><rect x="0.0000%" y="181" width="99.1736%" height="15" fill="rgb(241,131,45)" fg:x="0" fg:w="600"/><text x="0.2500%" y="191.50">stochastic-rs`std::rt::lang_start_internal</text></g><g><title>stochastic-rs`std::rt::lang_start::_{{closure}} (600 samples, 99.17%)</title><rect x="0.0000%" y="165" width="99.1736%" height="15" fill="rgb(249,31,29)" fg:x="0" fg:w="600"/><text x="0.2500%" y="175.50">stochastic-rs`std::rt::lang_start::_{{closure}}</text></g><g><title>stochastic-rs`std::sys::backtrace::__rust_begin_short_backtrace (600 samples, 99.17%)</title><rect x="0.0000%" y="149" width="99.1736%" height="15" fill="rgb(225,111,53)" fg:x="0" fg:w="600"/><text x="0.2500%" y="159.50">stochastic-rs`std::sys::backtrace::__rust_begin_short_backtrace</text></g><g><title>stochastic-rs`stochastic_rs::main (600 samples, 99.17%)</title><rect x="0.0000%" y="133" width="99.1736%" height="15" fill="rgb(238,160,17)" fg:x="0" fg:w="600"/><text x="0.2500%" y="143.50">stochastic-rs`stochastic_rs::main</text></g><g><title>stochastic-rs`stochastic_rs::processes::fbm::Fbm::new (2 samples, 0.33%)</title><rect x="98.8430%" y="117" width="0.3306%" height="15" fill="rgb(214,148,48)" fg:x="598" fg:w="2"/><text x="99.0930%" y="127.50"></text></g><g><title>stochastic-rs`stochastic_rs::noises::fgn::FgnFft::new (2 samples, 0.33%)</title><rect x="98.8430%" y="101" width="0.3306%" height="15" fill="rgb(232,36,49)" fg:x="598" fg:w="2"/><text x="99.0930%" y="111.50"></text></g><g><title>stochastic-rs`ndrustfft::FftHandler&lt;T&gt;::new (2 samples, 0.33%)</title><rect x="98.8430%" y="85" width="0.3306%" height="15" fill="rgb(209,103,24)" fg:x="598" fg:w="2"/><text x="99.0930%" y="95.50"></text></g><g><title>stochastic-rs`rustfft::neon::neon_planner::FftPlannerNeon&lt;T&gt;::plan_fft (2 samples, 0.33%)</title><rect x="98.8430%" y="69" width="0.3306%" height="15" fill="rgb(229,88,8)" fg:x="598" fg:w="2"/><text x="99.0930%" y="79.50"></text></g><g><title>stochastic-rs`rustfft::neon::neon_planner::FftPlannerNeon&lt;T&gt;::build_fft (2 samples, 0.33%)</title><rect x="98.8430%" y="53" width="0.3306%" height="15" fill="rgb(213,181,19)" fg:x="598" fg:w="2"/><text x="99.0930%" y="63.50"></text></g><g><title>stochastic-rs`rustfft::neon::neon_radix4::Neon64Radix4&lt;T&gt;::new (2 samples, 0.33%)</title><rect x="98.8430%" y="37" width="0.3306%" height="15" fill="rgb(254,191,54)" fg:x="598" fg:w="2"/><text x="99.0930%" y="47.50"></text></g><g><title>libsystem_kernel.dylib`__exit (1 samples, 0.17%)</title><rect x="99.1736%" y="213" width="0.1653%" height="15" fill="rgb(241,83,37)" fg:x="600" fg:w="1"/><text x="99.4236%" y="223.50"></text></g><g><title>stochastic-rs`&lt;rayon_core::registry::WorkerThread as core::convert::From&lt;rayon_core::registry::ThreadBuilder&gt;&gt;::from (1 samples, 0.17%)</title><rect x="99.3388%" y="117" width="0.1653%" height="15" fill="rgb(233,36,39)" fg:x="601" fg:w="1"/><text x="99.5888%" y="127.50"></text></g><g><title>libsystem_malloc.dylib`szone_malloc_should_clear (1 samples, 0.17%)</title><rect x="99.3388%" y="101" width="0.1653%" height="15" fill="rgb(226,3,54)" fg:x="601" fg:w="1"/><text x="99.5888%" y="111.50"></text></g><g><title>libsystem_malloc.dylib`small_malloc_should_clear (1 samples, 0.17%)</title><rect x="99.3388%" y="85" width="0.1653%" height="15" fill="rgb(245,192,40)" fg:x="601" fg:w="1"/><text x="99.5888%" y="95.50"></text></g><g><title>libsystem_malloc.dylib`mvm_allocate_pages (1 samples, 0.17%)</title><rect x="99.3388%" y="69" width="0.1653%" height="15" fill="rgb(238,167,29)" fg:x="601" fg:w="1"/><text x="99.5888%" y="79.50"></text></g><g><title>libsystem_kernel.dylib`_kernelrpc_mach_vm_map_trap (1 samples, 0.17%)</title><rect x="99.3388%" y="53" width="0.1653%" height="15" fill="rgb(232,182,51)" fg:x="601" fg:w="1"/><text x="99.5888%" y="63.50"></text></g><g><title>libsystem_kernel.dylib`swtch_pri (2 samples, 0.33%)</title><rect x="99.5041%" y="101" width="0.3306%" height="15" fill="rgb(231,60,39)" fg:x="602" fg:w="2"/><text x="99.7541%" y="111.50"></text></g><g><title>all (605 samples, 100%)</title><rect x="0.0000%" y="229" width="100.0000%" height="15" fill="rgb(208,69,12)" fg:x="0" fg:w="605"/><text x="0.2500%" y="239.50"></text></g><g><title>libsystem_pthread.dylib`thread_start (4 samples, 0.66%)</title><rect x="99.3388%" y="213" width="0.6612%" height="15" fill="rgb(235,93,37)" fg:x="601" fg:w="4"/><text x="99.5888%" y="223.50"></text></g><g><title>libsystem_pthread.dylib`_pthread_start (4 samples, 0.66%)</title><rect x="99.3388%" y="197" width="0.6612%" height="15" fill="rgb(213,116,39)" fg:x="601" fg:w="4"/><text x="99.5888%" y="207.50"></text></g><g><title>stochastic-rs`std::sys::pal::unix::thread::Thread::new::thread_start (4 samples, 0.66%)</title><rect x="99.3388%" y="181" width="0.6612%" height="15" fill="rgb(222,207,29)" fg:x="601" fg:w="4"/><text x="99.5888%" y="191.50"></text></g><g><title>stochastic-rs`core::ops::function::FnOnce::call_once{{vtable.shim}} (4 samples, 0.66%)</title><rect x="99.3388%" y="165" width="0.6612%" height="15" fill="rgb(206,96,30)" fg:x="601" fg:w="4"/><text x="99.5888%" y="175.50"></text></g><g><title>stochastic-rs`std::sys::backtrace::__rust_begin_short_backtrace (4 samples, 0.66%)</title><rect x="99.3388%" y="149" width="0.6612%" height="15" fill="rgb(218,138,4)" fg:x="601" fg:w="4"/><text x="99.5888%" y="159.50"></text></g><g><title>stochastic-rs`rayon_core::registry::ThreadBuilder::run (4 samples, 0.66%)</title><rect x="99.3388%" y="133" width="0.6612%" height="15" fill="rgb(250,191,14)" fg:x="601" fg:w="4"/><text x="99.5888%" y="143.50"></text></g><g><title>stochastic-rs`rayon_core::registry::WorkerThread::wait_until_cold (3 samples, 0.50%)</title><rect x="99.5041%" y="117" width="0.4959%" height="15" fill="rgb(239,60,40)" fg:x="602" fg:w="3"/><text x="99.7541%" y="127.50"></text></g><g><title>stochastic-rs`rayon_core::join::join_context::_{{closure}} (1 samples, 0.17%)</title><rect x="99.8347%" y="101" width="0.1653%" height="15" fill="rgb(206,27,48)" fg:x="604" fg:w="1"/><text x="100.0847%" y="111.50"></text></g></svg></svg>
\ No newline at end of file

From ac41b0366008451c01c0b2991b5038c80e228643 Mon Sep 17 00:00:00 2001
From: Daniel Boros <dancixx@gmail.com>
Date: Sun, 1 Sep 2024 13:37:38 +0200
Subject: [PATCH 6/6] chore: stop doctest

---
 Cargo.toml        |  3 ++-
 src/main.rs       | 25 ++++++++++++++++++++++++-
 src/noises/fgn.rs |  6 ++----
 3 files changed, 28 insertions(+), 6 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index 065090a..9d02490 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -32,13 +32,14 @@ mimalloc = { version = "0.1.43", optional = true }
 [features]
 mimalloc = ["dep:mimalloc"]
 jemalloc = ["dep:tikv-jemallocator"]
-default = []
+default = ["jemalloc"]
 
 
 [lib]
 name = "stochastic_rs"
 crate-type = ["cdylib", "lib"]
 path = "src/lib.rs"
+doctest = false
 
 [profile.release]
 debug = false
diff --git a/src/main.rs b/src/main.rs
index f328e4d..16dbb49 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1 +1,24 @@
-fn main() {}
+use stochastic_rs::{processes::fbm::Fbm, utils::Generator};
+
+fn main() {
+  let fbm = Fbm::new(0.75, 10000, Some(1.0), None);
+  let mut runs = Vec::new();
+
+  for _ in 0..20 {
+    let start = std::time::Instant::now();
+    for _ in 0..1000 {
+      let _ = fbm.sample();
+    }
+
+    let duration = start.elapsed();
+    println!(
+      "Time elapsed in expensive_function() is: {:?}",
+      duration.as_secs_f32()
+    );
+    runs.push(duration.as_secs_f32());
+  }
+
+  let sum: f32 = runs.iter().sum();
+  let average = sum / runs.len() as f32;
+  println!("Average time: {}", average);
+}
diff --git a/src/noises/fgn.rs b/src/noises/fgn.rs
index 3efa876..306cbd3 100644
--- a/src/noises/fgn.rs
+++ b/src/noises/fgn.rs
@@ -3,7 +3,7 @@
 //! The `FgnFft` struct provides methods to generate fractional Gaussian noise (FGN)
 //! using the Fast Fourier Transform (FFT) approach.
 
-use std::sync::{Arc, RwLock};
+use std::sync::Arc;
 
 use crate::utils::Generator;
 use ndarray::parallel::prelude::*;
@@ -22,7 +22,6 @@ pub struct FgnFft {
   sqrt_eigenvalues: Arc<Array1<Complex<f64>>>,
   m: Option<usize>,
   fft_handler: Arc<FftHandler<f64>>,
-  fft_fgn: Arc<RwLock<Array1<Complex<f64>>>>,
 }
 
 impl FgnFft {
@@ -84,7 +83,6 @@ impl FgnFft {
       sqrt_eigenvalues: Arc::new(sqrt_eigenvalues),
       m,
       fft_handler: Arc::new(FftHandler::new(2 * n)),
-      fft_fgn: Arc::new(RwLock::new(Array1::<Complex<f64>>::zeros(2 * n))),
     }
   }
 }
@@ -108,7 +106,7 @@ impl Generator for FgnFft {
       ComplexDistribution::new(StandardNormal, StandardNormal),
     );
     let fgn = &*self.sqrt_eigenvalues * &rnd;
-    let mut fgn_fft = self.fft_fgn.read().unwrap().clone();
+    let mut fgn_fft = Array1::<Complex<f64>>::zeros(2 * self.n);
     ndfft(&fgn, &mut fgn_fft, &*self.fft_handler, 0);
     let scale = (self.n as f64).powf(-self.hurst) * self.t.powf(self.hurst);
     let fgn = fgn_fft