1 | var TreeModel = require('tree-model');
|
2 | var _ = require('lodash');
|
3 | var extensions = require('./treemodelExtensions');
|
4 |
|
5 | module.exports = {
|
6 | tree: function(genetree) {
|
7 | function leftIndexComparator(a, b) {
|
8 | return a.left_index > b.left_index ? 1 : -1;
|
9 | }
|
10 | var tree = new TreeModel({modelComparatorFn: leftIndexComparator}).parse(genetree[0]);
|
11 | extensions.indexTree(tree, ['protein_stable_id', 'gene_stable_id']);
|
12 | extensions.decorateTree(tree);
|
13 | extensions.addPrototypeDecorations(tree);
|
14 | tree.geneCount = tree.leafNodes().length;
|
15 | tree._id = tree.model._id;
|
16 | return tree;
|
17 | }
|
18 | }
|