feat: add new function recursiveUntypeArrays

Author: lpatiny

src/__tests__/__snapshots__/index.test.ts.snap

@@ -170,6 +170,7 @@ exports[`test existence of exported functions 1`] = `
   "isPowerOfTwo",
   "nextPowerOfTwo",
   "recursiveResolve",
+  "recursiveUntypeArrays",
   "stringify",
   "calculateAdaptiveWeights",
 ]

src/utils/__tests__/resursiveUntypeArrays.test.ts

@@ -0,0 +1,34 @@
+import { expect, test } from 'vitest';
+
+import { recursiveUntypeArrays } from '../recursiveUntypeArrays';
+
+test('basic cases', () => {
+  expect(recursiveUntypeArrays(1)).toStrictEqual(1);
+  expect(recursiveUntypeArrays('a')).toStrictEqual('a');
+  expect(recursiveUntypeArrays({})).toStrictEqual({});
+  expect(recursiveUntypeArrays([])).toStrictEqual([]);
+  expect(recursiveUntypeArrays([1, 2, 3])).toStrictEqual([1, 2, 3]);
+  expect(recursiveUntypeArrays(Float32Array.from([1, 2, 3]))).toStrictEqual([
+    1, 2, 3,
+  ]);
+});
+
+test('simple object', async () => {
+  const object = {
+    a: {
+      b: {
+        c: Int32Array.from([1, 2, 3]),
+      },
+    },
+    d: [new Int32Array([1, 2, 3]), new Float32Array([3, 4, 5]), 2],
+  };
+
+  expect(recursiveUntypeArrays(object)).toStrictEqual({
+    a: {
+      b: {
+        c: [1, 2, 3],
+      },
+    },
+    d: [[1, 2, 3], [3, 4, 5], 2],
+  });
+});

src/utils/index.ts

@@ -6,5 +6,6 @@ export * from './getRescaler';
 export * from './isPowerOfTwo';
 export * from './nextPowerOfTwo';
 export * from './recursiveResolve';
+export * from './recursiveUntypeArrays';
 export * from './stringify';
 export * from './calculateAdaptiveWeights';

src/utils/recursiveUntypeArrays.ts

@@ -0,0 +1,28 @@
+import type { NumberArray } from 'cheminfo-types';
+
+/**
+ * Recursively change the typed arrays to normal arrays
+ * The changes are done in-place !
+ * @param object
+ * @returns
+ */
+export function recursiveUntypeArrays(object: unknown) {
+  if (typeof object !== 'object') return object;
+  object = modifier(object);
+  return object;
+}
+
+function modifier(object: any) {
+  if (typeof object !== 'object') return object;
+  if (ArrayBuffer.isView(object)) {
+    return Array.from(object as NumberArray);
+  }
+  for (const key in object) {
+    if (ArrayBuffer.isView(object[key])) {
+      object[key] = Array.from(object[key] as NumberArray);
+    } else if (typeof object[key] === 'object') {
+      modifier(object[key]);
+    }
+  }
+  return object;
+}