]> git.donarmstrong.com Git - mothur.git/blob - unifracweightedcommand.h
unifrac.weighted
[mothur.git] / unifracweightedcommand.h
1 #ifndef UNIFRACWEIGHTEDCOMMAND_H
2 #define UNIFRACWEIGHTEDCOMMAND_H
3
4 /*
5  *  unifracweightedcommand.h
6  *  Mothur
7  *
8  *  Created by Sarah Westcott on 2/9/09.
9  *  Copyright 2009 Schloss Lab UMASS Amherst. All rights reserved.
10  *
11  */
12
13 #include "command.hpp"
14 #include "weighted.h"
15 #include "treemap.h"
16
17 using namespace std;
18
19 class GlobalData;
20
21 class UnifracWeightedCommand : public Command {
22         
23         public:
24                 UnifracWeightedCommand();       
25                 ~UnifracWeightedCommand() { delete weighted; }
26                 int execute();  
27         
28         private:
29                 GlobalData* globaldata;
30                 vector<Tree*> T;           //user trees
31                 vector<double> utreeScores;  //user tree unweighted scores
32                 vector<double> WScoreSig;  //tree weighted score signifigance when compared to random trees - percentage of random trees with that score or lower.
33                 vector<string> groupComb; // AB. AC, BC...
34                 Tree* randT;  //random tree
35                 TreeMap* tmap;
36                 Weighted* weighted;
37                 string weightedFile, weightedFileout, sumFile;
38                 int iters, numGroups, numComp, counter;
39                 EstOutput userData;                     //weighted score info for user tree
40                 EstOutput randomData;           //weighted score info for random trees
41                 vector< vector<double> > rScores;  //vector<weighted scores for random trees.> each group comb has an entry
42                 vector< vector<double> > uScores;  //vector<weighted scores for user trees.> each group comb has an entry
43                 vector< map<float, float> > rScoreFreq;  //map <weighted score, number of random trees with that score.> -vector entry for each combination.
44                 vector< map<float, float> > rCumul;  //map <weighted score, cumulative percentage of number of random trees with that score or higher.> -vector entry for each c                                                                
45                 map<float, float>  validScores;  //map contains scores from random
46                 map<float, float>::iterator it2;
47                 map<float, float>::iterator it;
48                 
49                 ofstream outSum, out;
50                 ifstream inFile;
51                 
52                 void printWSummaryFile();
53                 void printWeightedFile();  
54                 //void removeValidScoresDuplicates();
55                 int findIndex(float, int);
56                 void setGroups(); 
57                 void calculateFreqsCumuls();
58                 void initFile(string);
59                 void output(vector<double>);
60                 void resetFile();
61
62 };
63
64
65
66 #endif