weighted.h

   1 #ifndef WEIGHTED_H
   2 #define WEIGHTED_H
   3
   4
   5 /*
   6  *  weighted.h
   7  *  Mothur
   8  *
   9  *  Created by Sarah Westcott on 2/9/09.
  10  *  Copyright 2009 Schloss Lab UMASS Amherst. All rights reserved.
  11  *
  12  */
  13
  14 #include "treecalculator.h"
  15 #include "counttable.h"
  16
  17 /***********************************************************************/
  18
  19 class Weighted : public TreeCalculator  {
  20
  21         public:
  22         Weighted( bool r) : includeRoot(r) {};
  23                 ~Weighted() {};
  24
  25                 EstOutput getValues(Tree*, string, string);
  26                 EstOutput getValues(Tree*, int, string);
  27
  28         private:
  29                 struct linePair {
  30                         int start;
  31                         int num;
  32                         linePair(int i, int j) : start(i), num(j) {}
  33                 };
  34                 vector<linePair> lines;
  35
  36                 EstOutput data;
  37                 map<string, int>::iterator it;
  38                 map<string, double> WScore; //a score for each group combination i.e. AB, AC, BC.
  39                 int processors;
  40                 string outputDir;
  41                 map< vector<string>, set<int> > rootForGrouping;  //maps a grouping combo to the root for that combo
  42                 bool includeRoot;
  43
  44                 EstOutput driver(Tree*, vector< vector<string> >, int, int, CountTable*);
  45                 EstOutput createProcesses(Tree*, vector< vector<string> >, CountTable*);
  46                 double getLengthToRoot(Tree*, int, string, string);
  47 };
  48
  49 /***********************************************************************/
  50
  51
  52 #endif