oakscriptjs 0.9.5 → 0.10.0

package/dist/cjs/matrix/index.js

@@ -82,12 +82,14 @@ function new_matrix(rows2, columns2, initial_value) {
   return { rows: rows2, columns: columns2, data };
 }
 function get(id, row2, column) {
+  if (Number.isNaN(row2) || Number.isNaN(column)) return NaN;
   if (row2 < 0 || row2 >= id.rows || column < 0 || column >= id.columns) {
     throw new Error(`Matrix index out of bounds: [${row2}, ${column}] for matrix of size [${id.rows}, ${id.columns}]`);
   }
   return id.data[row2][column];
 }
 function set(id, row2, column, value) {
+  if (Number.isNaN(row2) || Number.isNaN(column)) return;
   if (row2 < 0 || row2 >= id.rows || column < 0 || column >= id.columns) {
     throw new Error(`Matrix index out of bounds: [${row2}, ${column}] for matrix of size [${id.rows}, ${id.columns}]`);
   }
@@ -749,12 +751,6 @@ function det(id) {
   if (n === 0) {
     return 1;
   }
-  if (n === 1) {
-    return id.data[0][0];
-  }
-  if (n === 2) {
-    return id.data[0][0] * id.data[1][1] - id.data[0][1] * id.data[1][0];
-  }
   const a = id.data.map((row2) => [...row2]);
   let determinant = 1;
   let swapCount = 0;
@@ -775,7 +771,7 @@ function det(id) {
       swapCount++;
     }
     const pivot = a[col2][col2];
-    if (Math.abs(pivot) < EPSILON) {
+    if (Math.abs(pivot) < 1e-11) {
       return 0;
     }
     determinant *= pivot;
@@ -895,38 +891,48 @@ function rank(id) {
   if (id.rows === 0 || id.columns === 0) {
     return 0;
   }
-  const a = id.data.map((row2) => [...row2]);
-  const m = id.rows;
-  const n = id.columns;
-  let r = 0;
-  for (let col2 = 0; col2 < n && r < m; col2++) {
-    let maxRow = r;
-    let maxVal = Math.abs(a[r][col2]);
-    for (let row2 = r + 1; row2 < m; row2++) {
-      const absVal = Math.abs(a[row2][col2]);
-      if (absVal > maxVal) {
-        maxVal = absVal;
-        maxRow = row2;
-      }
-    }
-    if (maxVal < EPSILON) {
-      continue;
-    }
-    if (maxRow !== r) {
-      const temp = a[r];
-      a[r] = a[maxRow];
-      a[maxRow] = temp;
-    }
-    const pivot = a[r][col2];
-    for (let row2 = r + 1; row2 < m; row2++) {
-      const factor = a[row2][col2] / pivot;
-      for (let j = col2; j < n; j++) {
-        a[row2][j] = a[row2][j] - factor * a[r][j];
+  const s = singularValues(id.data);
+  const largest = Math.max(...s);
+  if (!(largest > 0)) return 0;
+  const tol = Math.max(id.rows, id.columns) * largest * 2 ** -52;
+  return s.filter((v) => v > tol).length;
+}
+function singularValues(data) {
+  const tall = data.length >= (data[0]?.length ?? 0);
+  const a = tall ? data.map((row2) => [...row2]) : data[0].map((_, j) => data.map((row2) => row2[j]));
+  const m = a.length;
+  const n = a[0].length;
+  for (let sweep = 0; sweep < 60; sweep++) {
+    let rotated = false;
+    for (let p = 0; p < n - 1; p++) {
+      for (let q = p + 1; q < n; q++) {
+        let alpha = 0;
+        let beta = 0;
+        let gamma = 0;
+        for (let i = 0; i < m; i++) {
+          const x = a[i][p];
+          const y = a[i][q];
+          alpha += x * x;
+          beta += y * y;
+          gamma += x * y;
+        }
+        if (gamma === 0 || Math.abs(gamma) <= 1e-15 * Math.sqrt(alpha * beta)) continue;
+        rotated = true;
+        const zeta = (beta - alpha) / (2 * gamma);
+        const tan = Math.sign(zeta || 1) / (Math.abs(zeta) + Math.sqrt(1 + zeta * zeta));
+        const cos = 1 / Math.sqrt(1 + tan * tan);
+        const sin = cos * tan;
+        for (let i = 0; i < m; i++) {
+          const x = a[i][p];
+          const y = a[i][q];
+          a[i][p] = cos * x - sin * y;
+          a[i][q] = sin * x + cos * y;
+        }
       }
     }
-    r++;
+    if (!rotated) break;
   }
-  return r;
+  return Array.from({ length: n }, (_, j) => Math.sqrt(a.reduce((sum2, row2) => sum2 + row2[j] * row2[j], 0)));
 }
 function eigenvalues(id) {
   if (!is_square(id)) {