@gkucmierz/utils 1.18.0 → 1.19.0

package/main.mjs

@@ -29,6 +29,9 @@ import {
 import {
   lcm, lcmBI
 } from './src/lcm.mjs'
+import {
+  memoize
+} from './src/memoize.mjs'
 import {
   mod, modBI
 } from './src/mod.mjs'
@@ -58,6 +61,7 @@ export * from './src/get-type.mjs';
 export * from './src/gpn.mjs';
 export * from './src/herons-formula.mjs';
 export * from './src/lcm.mjs';
+export * from './src/memoize.mjs';
 export * from './src/mod.mjs';
 export * from './src/phi.mjs';
 export * from './src/pow-mod.mjs';
@@ -66,5 +70,5 @@ export * from './src/square-root.mjs';
 export * from './src/tonelli-shanks.mjs';
 
 export default [
-  SetCnt, fromBase64, fromBase64Url, toBase64, toBase64Url, bijective2num, bijective2numBI, num2bijective, num2bijectiveBI, egcd, factors, factorsBI, gcd, gcdBI, getType, gpn, gpnBI, heronsFormula, heronsFormulaBI, lcm, lcmBI, mod, modBI, phi, phiBI, powMod, powModBI, array2range, range2array, squareRoot, squareRootBI, tonelliShanksBI
+  SetCnt, fromBase64, fromBase64Url, toBase64, toBase64Url, bijective2num, bijective2numBI, num2bijective, num2bijectiveBI, egcd, factors, factorsBI, gcd, gcdBI, getType, gpn, gpnBI, heronsFormula, heronsFormulaBI, lcm, lcmBI, memoize, mod, modBI, phi, phiBI, powMod, powModBI, array2range, range2array, squareRoot, squareRootBI, tonelliShanksBI
 ];

package/package-lock.json

@@ -1,12 +1,12 @@
 {
   "name": "@gkucmierz/utils",
-  "version": "1.18.0",
+  "version": "1.19.0",
   "lockfileVersion": 2,
   "requires": true,
   "packages": {
     "": {
       "name": "@gkucmierz/utils",
-      "version": "1.18.0",
+      "version": "1.19.0",
       "license": "MIT",
       "devDependencies": {
         "husky": "^8.0.1",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gkucmierz/utils",
-  "version": "1.18.0",
+  "version": "1.19.0",
   "description": "Usefull functions for solving programming tasks",
   "main": "main.mjs",
   "scripts": {

package/spec/memoize.spec.mjs

@@ -0,0 +1,32 @@
+
+import {
+  memoize,
+} from '../src/memoize.mjs';
+
+describe('memoize', () => {
+  it('called once', () => {
+    let cnt = 0;
+    const add = memoize((a, b) => {
+      ++cnt;
+      return a + b;
+    });
+
+    expect(add(1, 2)).toBe(3);
+    expect(cnt).toBe(1);
+    expect(add(1, 2)).toBe(3);
+    expect(cnt).toBe(1);
+
+    expect(add(2, 1)).toBe(3);
+    expect(cnt).toBe(2);
+  });
+
+  it('variable args length', () => {
+    const fn = memoize((...args) => args.length);
+
+    expect(fn()).toBe(0);
+    expect(fn(1)).toBe(1);
+    expect(fn(1, 2)).toBe(2);
+    expect(fn(1, 2, 3)).toBe(3);
+  });
+});
+

package/src/memoize.mjs

@@ -0,0 +1,38 @@
+
+export const memoize = fn => {
+  const maps = [];
+
+  const getLeafMap = args => {
+    if (!maps[args.length]) maps[args.length] = new Map();
+    const map = maps[args.length];
+
+    const leaf = args.reduce((map, arg) => {
+      if (map.has(arg)) return map.get(arg);
+      const tmp = new Map();
+      map.set(arg, tmp);
+      return tmp;
+    }, map);
+
+    return leaf;
+  };
+
+  const zero = {
+    called: false,
+    ret: null,
+  };
+  
+  return function() {
+    const args = [...arguments];
+    if (args.length === 0) {
+      if (zero.called) return zero.ret;
+      zero.called = true;
+      return zero.ret = fn();
+    }
+    const [firstArg, ...restArgs] = args;
+    const map = getLeafMap(restArgs);
+    if (map.has(firstArg)) return map.get(firstArg);
+    const ret = fn(...args);
+    map.set(firstArg, ret);
+    return ret;
+  };
+};