blob: 0f05ba94775e376434ae72e3e12f73deb9016025 (
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
|
/*
* Contains classes used for representing tree-of-life data.
*/
// Represents a tree-of-life node/tree
export class TolNode {
name: string;
children: TolNode[];
parent: TolNode | null;
constructor(name: string, children: TolNode[] = [], parent = null){
this.name = name;
this.children = children;
this.parent = parent;
}
}
// Represents a tree-of-life node obtained from tol.json
export class TolNodeRaw {
name: string;
children?: TolNodeRaw[];
constructor(name: string, children: TolNodeRaw[] = []){
this.name = name;
this.children = children;
}
}
// Converts a TolNodeRaw tree to a TolNode tree
export function tolFromRaw(node: TolNodeRaw): TolNode {
function helper(node: TolNodeRaw, parent: TolNode | null){
let tolNode = new TolNode(node.name);
if (node.children == null){
tolNode.children = [];
} else {
tolNode.children = Array(node.children.length);
node.children.forEach((child, idx) => {tolNode.children[idx] = helper(child, tolNode)});
}
tolNode.parent = parent;
return tolNode;
}
return helper(node, null);
}
// Returns a mapping from TolNode names to TolNodes in a given tree
export function getTolMap(tolTree: TolNode): Map<string, TolNode> {
function helper(node: TolNode, map: Map<string, TolNode>){
map.set(node.name, node);
node.children.forEach(child => helper(child, map));
}
let map = new Map();
helper(tolTree, map);
return map;
}
|
