cmpstr/dist/esm/metric/Levenshtein.mjs

CmpStr is a lightweight, fast and well performing package for calculating string similarity

// CmpStr v3.2.2 build-bb61120-260311 by Paul Köhler @komed3 / MIT License
import { Pool } from '../utils/Pool.mjs';
import { MetricRegistry, Metric } from './Metric.mjs';

class LevenshteinDistance extends Metric {
  constructor(a, b, opt = {}) {
    super('levenshtein', a, b, opt, true);
  }
  compute(a, b, m, n, maxLen) {
    const len = m + 1;
    const [prev, curr] = Pool.acquireMany('int32', [len, len]);
    try {
      for (let i = 0; i <= m; i++) prev[i] = i;
      for (let j = 1; j <= n; j++) {
        curr[0] = j;
        const cb = b.charCodeAt(j - 1);
        for (let i = 1; i <= m; i++) {
          const cost = a.charCodeAt(i - 1) === cb ? 0 : 1;
          curr[i] = Math.min(curr[i - 1] + 1, prev[i] + 1, prev[i - 1] + cost);
        }
        prev.set(curr);
      }
      const dist = prev[m];
      return {
        res: maxLen === 0 ? 1 : Metric.clamp(1 - dist / maxLen),
        raw: { dist, maxLen }
      };
    } finally {
      Pool.release('int32', prev, len);
      Pool.release('int32', curr, len);
    }
  }
}
MetricRegistry.add('levenshtein', LevenshteinDistance);

export { LevenshteinDistance };