phylodiversitycommand.h

   1 #ifndef PHYLODIVERSITYCOMMAND_H
   2 #define PHYLODIVERSITYCOMMAND_H
   3
   4 /*
   5  *  phylodiversitycommand.h
   6  *  Mothur
   7  *
   8  *  Created by westcott on 4/30/10.
   9  *  Copyright 2010 Schloss Lab. All rights reserved.
  10  *
  11  */
  12
  13 #include "command.hpp"
  14 #include "treemap.h"
  15 #include "readtree.h"
  16 #include "sharedutilities.h"
  17
  18
  19 class PhyloDiversityCommand : public Command {
  20
  21         public:
  22                 PhyloDiversityCommand(string);
  23                 PhyloDiversityCommand();
  24                 ~PhyloDiversityCommand(){}
  25
  26                 vector<string> setParameters();
  27                 string getCommandName()                 { return "phylo.diversity";                     }
  28                 string getCommandCategory()             { return "Hypothesis Testing";          }
  29                 string getHelpString();
  30
  31                 int execute();
  32                 void help() { m->mothurOut(getHelpString()); }
  33 private:
  34                 ReadTree* read;
  35                 TreeMap* tmap;
  36                 float freq;
  37                 int iters, processors, numUniquesInName;
  38                 bool abort, rarefy, summary, collect, scale;
  39                 string groups, outputDir, treefile, groupfile, namefile;
  40                 vector<string> Groups, outputNames; //holds groups to be used, and outputFile names
  41                 map<string, string> nameMap;
  42
  43                 int readNamesFile();
  44                 void printData(set<int>&, map< string, vector<float> >&, ofstream&, int);
  45                 void printSumData(map< string, vector<float> >&, ofstream&, int);
  46                 vector<float> calcBranchLength(Tree*, int, map< string, set<int> >&);
  47                 int driver(Tree*, map< string, vector<float> >&, map<string, vector<float> >&, int, int, vector<int>&, set<int>&, ofstream&, ofstream&, bool);
  48                 int createProcesses(vector<int>&, Tree*, map< string, vector<float> >&, map<string, vector<float> >&, int, int, vector<int>&, set<int>&, ofstream&, ofstream&);
  49
  50 };
  51
  52 #endif
  53