]> git.donarmstrong.com Git - mothur.git/blob - parsimony.cpp
fixed parsimony for groups
[mothur.git] / parsimony.cpp
1 /*
2  *  parsimony.cpp
3  *  Mothur
4  *
5  *  Created by Sarah Westcott on 1/26/09.
6  *  Copyright 2009 Schloss Lab UMASS Amherst. All rights reserved.
7  *
8  */
9
10 #include "parsimony.h"
11
12 /**************************************************************************************************/
13 EstOutput Parsimony::getValues(Tree* t) {
14         try {
15                 globaldata = GlobalData::getInstance();
16                 
17                 data.resize(1,0);
18                         
19                 int score = 0;
20                                         
21                 for(int i=t->getNumLeaves();i<t->getNumNodes();i++){
22                         int lc = t->tree[i].getLChild();
23                         int rc = t->tree[i].getRChild();
24                         
25                         int iSize = t->tree[i].pGroups.size();
26                         int rcSize = t->tree[rc].pGroups.size();
27                         int lcSize = t->tree[lc].pGroups.size();
28                         
29                         //add in all the groups the users wanted
30                         for (it = t->tree[i].pGroups.begin(); it != t->tree[i].pGroups.end(); it++) {
31                                 if (inUsersGroups(it->first, globaldata->Groups) != true) {  iSize--;  }
32                         }
33                         //add in all the groups the users wanted
34                         for (it = t->tree[rc].pGroups.begin(); it != t->tree[rc].pGroups.end(); it++) {
35                                 if (inUsersGroups(it->first, globaldata->Groups) != true) {  rcSize--;  }
36                         }
37                         
38                         //add in all the groups the users wanted
39                         for (it = t->tree[lc].pGroups.begin(); it != t->tree[lc].pGroups.end(); it++) {
40                                 if (inUsersGroups(it->first, globaldata->Groups) != true) {  lcSize--;  }
41                         }
42                         
43                         //if isize are 0 then that branch is to be ignored
44                         if (iSize == 0) { }
45                         else if ((rcSize == 0) || (lcSize == 0)) { }
46                         //if you have more groups than either of your kids then theres been a change.
47                         else if(iSize > rcSize || iSize > lcSize){
48                                 score++;
49                         }
50                 } 
51                 
52                 data[0] = score;
53                 
54                 return data;
55         }
56         catch(exception& e) {
57                 cout << "Standard Error: " << e.what() << " has occurred in the Parsimony class Function getValues. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
58                 exit(1);
59         }
60         catch(...) {
61                 cout << "An unknown error has occurred in the Parsimony class function getValues. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
62                 exit(1);
63         }
64
65 }
66