@gkucmierz/utils 1.17.0 → 1.19.0

package/main.mjs

@@ -20,12 +20,18 @@ import {
 import {
   getType
 } from './src/get-type.mjs'
+import {
+  gpn, gpnBI
+} from './src/gpn.mjs'
 import {
   heronsFormula, heronsFormulaBI
 } from './src/herons-formula.mjs'
 import {
   lcm, lcmBI
 } from './src/lcm.mjs'
+import {
+  memoize
+} from './src/memoize.mjs'
 import {
   mod, modBI
 } from './src/mod.mjs'
@@ -52,8 +58,10 @@ export * from './src/egcd.mjs';
 export * from './src/factors.mjs';
 export * from './src/gcd.mjs';
 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';
@@ -62,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, 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.17.0",
+  "version": "1.19.0",
   "lockfileVersion": 2,
   "requires": true,
   "packages": {
     "": {
       "name": "@gkucmierz/utils",
-      "version": "1.17.0",
+      "version": "1.19.0",
       "license": "MIT",
       "devDependencies": {
         "husky": "^8.0.1",

package/package.json

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

package/spec/gpn.spec.mjs

@@ -0,0 +1,35 @@
+
+import {
+  gpn,
+  gpnBI,
+} from '../src/gpn.mjs';
+
+describe('gpn', () => {
+  it('first terms', () => {
+    expect(gpn(0)).toBe(0);
+    expect(gpn(1)).toBe(1);
+    expect(gpn(2)).toBe(2);
+    expect(gpn(3)).toBe(5);
+    expect(gpn(4)).toBe(7);
+    expect(gpn(5)).toBe(12);
+    expect(gpn(6)).toBe(15);
+    expect(gpn(7)).toBe(22);
+    expect(gpn(8)).toBe(26);
+    expect(gpn(9)).toBe(35);
+  });
+});
+
+describe('gpn BI', () => {
+  it('first terms', () => {
+    expect(gpnBI(0n)).toBe(0n);
+    expect(gpnBI(1n)).toBe(1n);
+    expect(gpnBI(2n)).toBe(2n);
+    expect(gpnBI(3n)).toBe(5n);
+    expect(gpnBI(4n)).toBe(7n);
+    expect(gpnBI(5n)).toBe(12n);
+    expect(gpnBI(6n)).toBe(15n);
+    expect(gpnBI(7n)).toBe(22n);
+    expect(gpnBI(8n)).toBe(26n);
+    expect(gpnBI(9n)).toBe(35n);
+  });
+});

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/gpn.mjs

@@ -0,0 +1,14 @@
+
+// Generalized pentagonal numbers
+// https://oeis.org/A001318
+
+const getGpn = (zero, one, two, three) => {
+  const gk = k => k * (three * k - one) / two;
+  return n => {
+    const m = (n + one) / two | zero;
+    return n % two === zero ? gk(-m) : gk(m);
+  };
+};
+
+export const gpn = getGpn(0, 1, 2, 3);
+export const gpnBI = getGpn(0n, 1n, 2n, 3n);

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;
+  };
+};