unifrac parallelization done, changed dist.seqs output=square to calc all distance...

[mothur.git] / unifracweightedcommand.h

diff --git a/unifracweightedcommand.h b/unifracweightedcommand.h
index d309566e45b0d58fb2631eeb1a0ad1816907f437..7317c8a43092a5712a06dda00589d1eb4528c649 100644 (file)
--- a/unifracweightedcommand.h
+++ b/unifracweightedcommand.h
@@ -13,42 +13,65 @@
 #include "command.hpp"
 #include "weighted.h"
 #include "treemap.h"
+#include "progress.hpp"
+#include "sharedutilities.h"
+#include "fileoutput.h"
 
-using namespace std;
 
 class GlobalData;
 
 class UnifracWeightedCommand : public Command {
        
        public:
-               UnifracWeightedCommand();       
-               ~UnifracWeightedCommand() { delete weighted; }
+               UnifracWeightedCommand(string);
+               ~UnifracWeightedCommand() { if (abort == false) {  delete weighted; delete util; } }
                int execute();  
+               void help();
        
        private:
+               struct linePair {
+                       int start;
+                       int num;
+                       linePair(int i, int j) : start(i), num(j) {}
+               };
+               vector<linePair> lines;
+               
                GlobalData* globaldata;
+               SharedUtil* util;
+               FileOutput* output;
                vector<Tree*> T;           //user trees
-               vector<float> utreeScores;  //user tree unweighted scores
-               vector<float> WScoreSig;  //tree weighted score signifigance when compared to random trees - percentage of random trees with that score or lower.
+               vector<double> utreeScores;  //user tree unweighted scores
+               vector<double> WScoreSig;  //tree weighted score signifigance when compared to random trees - percentage of random trees with that score or lower.
                vector<string> groupComb; // AB. AC, BC...
-               Tree* randT;  //random tree
                TreeMap* tmap;
                Weighted* weighted;
-               string weightedFile, sumFile;
-               int iters, numGroups, numComp;
+               string sumFile, outputDir;
+               int iters, numGroups, numComp, counter;
                EstOutput userData;                     //weighted score info for user tree
                EstOutput randomData;           //weighted score info for random trees
-               vector< vector<float> > validScores;  //vector<contains scores from both user and random> each group comb has an entry
-               vector< vector<float> > rScores;  //vector<weighted scores for random trees.> each group comb has an entry
-               vector< vector<float> > uScores;  //vector<weighted scores for user trees.> each group comb has an entry
-                                                               
-               ofstream outSum, out;
+               vector< vector<double> > rScores;  //vector<weighted scores for random trees.> each group comb has an entry
+               vector< vector<double> > uScores;  //vector<weighted scores for user trees.> each group comb has an entry
+               vector< map<float, float> > rScoreFreq;  //map <weighted score, number of random trees with that score.> -vector entry for each combination.
+               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                                                                
+               map<float, float>  validScores;  //map contains scores from random
+               
+               bool abort, phylip, random;
+               string groups, itersString;
+               vector<string> Groups, outputNames; //holds groups to be used
+               int processors;
+
+               
+               ofstream outSum;
                
                void printWSummaryFile();
-       //      void printWeightedFile();  
-               void removeValidScoresDuplicates();
-               int findIndex(float);
-               void setGroups(); 
+               void printWeightedFile();  
+               void createPhylipFile();
+               //void removeValidScoresDuplicates();
+               int findIndex(float, int);
+               void calculateFreqsCumuls();
+               int createProcesses(Tree*,  vector< vector<string> >, vector<double>&, vector< vector<double> >&);
+               int driver(Tree*, vector< vector<string> >, int, int, vector<double>&, vector< vector<double> >&);
+               
 };