fixed unifrac.weighted

[mothur.git] / unifracweightedcommand.h

diff --git a/unifracweightedcommand.h b/unifracweightedcommand.h
index 77dd10d3490999def3789b91d405ac76e470f360..547e35235a24ac1ab169c918da544301aea7bd13 100644 (file)
--- a/unifracweightedcommand.h
+++ b/unifracweightedcommand.h
@@ -24,11 +24,23 @@ class UnifracWeightedCommand : public Command {
        
        public:
                UnifracWeightedCommand(string);
+               UnifracWeightedCommand();
                ~UnifracWeightedCommand() { if (abort == false) {  delete weighted; delete util; } }
+               vector<string> getRequiredParameters();
+               vector<string> getValidParameters();
+               vector<string> getRequiredFiles();
+               map<string, vector<string> > getOutputFiles() { return outputTypes; }
                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;
@@ -36,7 +48,6 @@ class UnifracWeightedCommand : public Command {
                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 sumFile, outputDir;
@@ -52,6 +63,8 @@ class UnifracWeightedCommand : public Command {
                bool abort, phylip, random;
                string groups, itersString;
                vector<string> Groups, outputNames; //holds groups to be used
+               map<string, vector<string> > outputTypes;
+               int processors;
 
                
                ofstream outSum;
@@ -62,6 +75,8 @@ class UnifracWeightedCommand : public Command {
                //void removeValidScoresDuplicates();
                int findIndex(float, int);
                void calculateFreqsCumuls();
+               int createProcesses(Tree*,  vector< vector<string> >,  vector< vector<double> >&);
+               int driver(Tree*, vector< vector<string> >, int, int,  vector< vector<double> >&);
                
 };