summaryrefslogtreecommitdiff
path: root/toolkit/modules/NLP.jsm
blob: f596ad9da9f734d988b5211880d30e95c28f459b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

"use strict";

this.EXPORTED_SYMBOLS = ["NLP"];

/**
 * NLP, which stands for Natural Language Processing, is a module that provides
 * an entry point to various methods to interface with human language.
 *
 * At least, that's the goal. Eventually. Right now, the find toolbar only really
 * needs the Levenshtein distance algorithm.
 */
this.NLP = {
  /**
   * Calculate the Levenshtein distance between two words.
   * The implementation of this method was heavily inspired by
   * http://locutus.io/php/strings/levenshtein/index.html
   * License: MIT.
   *
   * @param  {String} word1   Word to compare against
   * @param  {String} word2   Word that may be different
   * @param  {Number} costIns The cost to insert a character
   * @param  {Number} costRep The cost to replace a character
   * @param  {Number} costDel The cost to delete a character
   * @return {Number}
   */
  levenshtein(word1 = "", word2 = "", costIns = 1, costRep = 1, costDel = 1) {
    if (word1 === word2)
      return 0;

    let l1 = word1.length;
    let l2 = word2.length;
    if (!l1)
      return l2 * costIns;
    if (!l2)
      return l1 * costDel;

    let p1 = new Array(l2 + 1)
    let p2 = new Array(l2 + 1)

    let i1, i2, c0, c1, c2, tmp;

    for (i2 = 0; i2 <= l2; i2++)
      p1[i2] = i2 * costIns;

    for (i1 = 0; i1 < l1; i1++) {
      p2[0] = p1[0] + costDel;

      for (i2 = 0; i2 < l2; i2++) {
        c0 = p1[i2] + ((word1[i1] === word2[i2]) ? 0 : costRep);
        c1 = p1[i2 + 1] + costDel;

        if (c1 < c0)
          c0 = c1;

        c2 = p2[i2] + costIns;

        if (c2 < c0)
          c0 = c2;

        p2[i2 + 1] = c0;
      }

      tmp = p1;
      p1 = p2;
      p2 = tmp;
    }

    c0 = p1[l2];

    return c0;
  }
};