erabo/spiegel-keyman
1
0
Fork 0
mirror of https://github.com/keymanapp/keyman.git synced 2026-08-12 03:45:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 13

Fix comment, fix function name, and tweak a few things with sumWeights.

Browse source
This commit is contained in:
Eddie Antonio Santos 2019-06-03 08:59:39 -06:00
parent 7244000412
commit 6f028733ac
No known key found for this signature in database
GPG key ID: DD411EA4D9509D87
1 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
common/predictive-text/worker/models/trie-model.ts
View file

@ -237,7 +237,7 @@
constructor(root: Node, wordform2key: Wordform2Key) {
this.root = root;
this.toKey = wordform2key;
this.totalWeight = countNodes(root);
this.totalWeight = sumWeights(root);
}
/**
@ -354,11 +354,13 @@
}
/**
* O(n) lookup to determine the total amount of nodes in the trie, starting
* at the provided node. Don't use this if you want lookups to be fast.
* @param node
* O(n) traversal to sum the total weight of all leaves in the trie, starting
* at the provided node. Don't use this if you want lookups to be fast!
*
* TODO: Move this functionality to the trie compiler!
* @param node The node to start summing weights.
*/
function countNodes(node: Node): number {
function sumWeights(node: Node): number {
if (node.type === 'leaf') {
return node.entries
.map(entry => entry.weight)
@ -366,8 +368,8 @@
} else {
// @ts-ignore
return Object.values(node.children)
.map(child => countNodes(child))
.reduce((acc, count) => acc + count, 0);
.map((child: Node) => sumWeights(child))
.reduce((acc: number, count: number) => acc + count, 0);
}
}