File: greedy.h

package info (click to toggle)
iqtree 1.6.12%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 12,140 kB
  • sloc: cpp: 111,752; ansic: 53,619; python: 242; sh: 195; makefile: 52
file content (112 lines) | stat: -rw-r--r-- 3,358 bytes parent folder | download | duplicates (5)
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/***************************************************************************
 *   Copyright (C) 2006 by BUI Quang Minh, Steffen Klaere, Arndt von Haeseler   *
 *   minh.bui@univie.ac.at   *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
#ifndef GREEDY_H
#define GREEDY_H

#include "pdtree.h"

/**
Implementation of greedy algorithm with complexity O(n*logk)
@author BUI Quang Minh, Steffen Klaere, Arndt von Haeseler
*/
class Greedy : public PDTree
{
public:
	/**
		construct from program parameters
		@param params program parameters
	*/
    Greedy(Params &params) : 
		PDTree(params) {}

	/**
		construct from a tree
		@param tree a tree class
	*/
    Greedy(PDTree &tree) : 
		PDTree(tree) {}

	/**
		constructor
	*/
	Greedy() : PDTree() {};

	/**
		run the algorithm
		@param params program parameters
		@param taxa_set (OUT) vector of PD sets
	*/
	void run(Params &params, vector<PDTaxaSet> &taxa_set);

	/**
		update the ordered list based on the recent longest path
		@param node the starting node
		@param subtree (OUT) resulted subtree
		@param cur_set the current set
	*/
	void updateOnLongestPath(Node *node, NodeVector &subtree, PDTaxaSet &cur_set);

	/**
		build the initial subtree based on the initial set of taxa
		@param node the starting node, NULL to start from the root
		@param dad dad of the node, used to direct the search
		@param subtree (OUT) resulted subtree
		@param nodestack (TEMP) stack of node, used only by function
	*/
	void buildOnInitialSet(NodeVector &subtree, NodeVector &nodestack, Node *node = NULL, Node *dad = NULL);

	/**
		initialize the ordered list based on the initial subtree structure
		@param subtree vector containing nodes in the subtree
		@return the subtree length
	*/
	double updateOnInitialSet(NodeVector &subtree);

	/**
		@return innodes.begin().
	*/
	NeighborSet::iterator highestNeighbor();

	/**
		add an edge into the NeighborSet
	*/
	void addNeighbor(Neighbor* neigh);

	//NodeSet innodes;

	/**
		neighbor set
	*/
	NeighborSet neighset;

	/**
		list of nodes in the subtree
	*/
	NodeVector subtree;

private:

	/**
		size of list of nodes, used internally during greedy search
	*/
	int list_size;
};

#endif