| 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 | }
|