@iyulab/u-numflow 0.4.0 → 0.5.0

package/README.md

@@ -19,6 +19,7 @@ u-numflow provides foundational mathematical, statistical, and probabilistic bui
 | `distributions` | Probability distributions: Uniform, Triangular, PERT, Normal, LogNormal |
 | `special` | Special functions: normal/t/F/chi² CDF, inverse normal CDF, regularized incomplete beta/gamma, erf |
 | `transforms` | Data transformations: Box-Cox (λ via MLE golden-section search), inverse Box-Cox |
+| `fourier` | Discrete Fourier transform of any length (radix-2 for powers of two, Bluestein otherwise): `fft`, `ifft`, `rfft`, `Complex` |
 | `matrix` | Dense matrix operations: determinant, inverse, Cholesky decomposition, Jacobi eigenvalue decomposition |
 | `random` | Seeded RNG, Fisher-Yates shuffle, weighted sampling, random subset selection |
 | `collections` | Specialized data structures: Union-Find with path compression and union-by-rank |
@@ -62,6 +63,12 @@ use u_numflow::transforms::{estimate_lambda, box_cox};
 let data = [1.0, 2.0, 4.0, 8.0, 16.0];
 let lambda = estimate_lambda(&data, -2.0, 2.0).unwrap(); // MLE via golden-section
 let transformed = box_cox(&data, lambda).unwrap();
+
+// Discrete Fourier transform of any length
+use u_numflow::fourier::rfft;
+let signal: Vec<f64> = (0..30).map(|j| (2.0 * std::f64::consts::PI * 3.0 * j as f64 / 30.0).sin()).collect();
+let spectrum = rfft(&signal);           // 30 complex bins; bin 3 carries the energy
+assert!(spectrum[3].norm() > 14.0);
 ```
 
 ## Build & Test
@@ -93,6 +100,10 @@ The package resolves per environment via a conditional `exports` map:
 | Bundlers (webpack, Vite, …) | ESM + WebAssembly ESM-integration (`default` condition) |
 | Node.js — `require()`, ESM `import`, CJS TS runners (`tsx`, `ts-node`) | CJS glue loading the wasm from the filesystem (`node` condition) — no loader hooks or flags |
 
+Exported functions: `mean`, `std_dev`, `variance`, `normal_cdf`, `box_cox`, `estimate_lambda`, and
+`rfft(data) -> Float64Array` — the DFT of a real sequence of any length, interleaved as
+`[re0, im0, re1, im1, …]` (bins `k` and `n − k` are conjugates, so `0..=n/2` describes the spectrum).
+
 ## Related
 
 - [u-metaheur](https://github.com/iyulab/u-metaheur) — Metaheuristic optimization (GA, SA, ALNS, CP)

package/node/u_numflow.cjs

@@ -79,6 +79,25 @@ function normal_cdf(x) {
 }
 exports.normal_cdf = normal_cdf;
 
+/**
+ * Forward DFT of a real sequence of any length.
+ *
+ * Returns the `n` complex bins interleaved as `[re0, im0, re1, im1, …]`
+ * (length `2n`). Bins `k` and `n − k` are conjugates, so the spectrum of a
+ * real signal is fully described by bins `0..=n/2`.
+ * @param {Float64Array} data
+ * @returns {Float64Array}
+ */
+function rfft(data) {
+    const ptr0 = passArrayF64ToWasm0(data, wasm.__wbindgen_malloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ret = wasm.rfft(ptr0, len0);
+    var v2 = getArrayF64FromWasm0(ret[0], ret[1]).slice();
+    wasm.__wbindgen_free(ret[0], ret[1] * 8, 8);
+    return v2;
+}
+exports.rfft = rfft;
+
 /**
  * Sample standard deviation of `data` (Bessel-corrected, denominator n−1).
  *

package/node/u_numflow.d.cts

@@ -39,6 +39,15 @@ export function mean(data: Float64Array): number;
  */
 export function normal_cdf(x: number): number;
 
+/**
+ * Forward DFT of a real sequence of any length.
+ *
+ * Returns the `n` complex bins interleaved as `[re0, im0, re1, im1, …]`
+ * (length `2n`). Bins `k` and `n − k` are conjugates, so the spectrum of a
+ * real signal is fully described by bins `0..=n/2`.
+ */
+export function rfft(data: Float64Array): Float64Array;
+
 /**
  * Sample standard deviation of `data` (Bessel-corrected, denominator n−1).
  *

package/package.json

@@ -4,8 +4,8 @@
   "collaborators": [
     "iyulab"
   ],
-  "description": "Mathematical primitives for the U-Engine ecosystem: statistics, probability, random sampling, and collections.",
-  "version": "0.4.0",
+  "description": "Foundational mathematical primitives: statistics, probability distributions, random sampling, and collections.",
+  "version": "0.5.0",
   "license": "MIT",
   "repository": {
     "type": "git",

package/u_numflow.d.ts

@@ -39,6 +39,15 @@ export function mean(data: Float64Array): number;
  */
 export function normal_cdf(x: number): number;
 
+/**
+ * Forward DFT of a real sequence of any length.
+ *
+ * Returns the `n` complex bins interleaved as `[re0, im0, re1, im1, …]`
+ * (length `2n`). Bins `k` and `n − k` are conjugates, so the spectrum of a
+ * real signal is fully described by bins `0..=n/2`.
+ */
+export function rfft(data: Float64Array): Float64Array;
+
 /**
  * Sample standard deviation of `data` (Bessel-corrected, denominator n−1).
  *

package/u_numflow.js

@@ -5,5 +5,5 @@ import { __wbg_set_wasm } from "./u_numflow_bg.js";
 __wbg_set_wasm(wasm);
 wasm.__wbindgen_start();
 export {
-    box_cox, estimate_lambda, mean, normal_cdf, std_dev, variance
+    box_cox, estimate_lambda, mean, normal_cdf, rfft, std_dev, variance
 } from "./u_numflow_bg.js";

package/u_numflow_bg.js

@@ -73,6 +73,24 @@ export function normal_cdf(x) {
     return ret;
 }
 
+/**
+ * Forward DFT of a real sequence of any length.
+ *
+ * Returns the `n` complex bins interleaved as `[re0, im0, re1, im1, …]`
+ * (length `2n`). Bins `k` and `n − k` are conjugates, so the spectrum of a
+ * real signal is fully described by bins `0..=n/2`.
+ * @param {Float64Array} data
+ * @returns {Float64Array}
+ */
+export function rfft(data) {
+    const ptr0 = passArrayF64ToWasm0(data, wasm.__wbindgen_malloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ret = wasm.rfft(ptr0, len0);
+    var v2 = getArrayF64FromWasm0(ret[0], ret[1]).slice();
+    wasm.__wbindgen_free(ret[0], ret[1] * 8, 8);
+    return v2;
+}
+
 /**
  * Sample standard deviation of `data` (Bessel-corrected, denominator n−1).
  *