updated kruskalwallis command

[mothur.git] / kruskalwalliscommand.h

diff --git a/kruskalwalliscommand.h b/kruskalwalliscommand.h
index 7e526137bc42daa1928c7e0988e1b52d546dbf2a..003ea91c37fa29eb5f46825f12f898a6b842ad55 100644 (file)
--- a/kruskalwalliscommand.h
+++ b/kruskalwalliscommand.h
@@ -10,10 +10,16 @@
 
 #include "command.hpp"
 
-class KruskalWallisCommand : public Command {
+    struct groupRank {
+        sstring group;
+        double value;
+        double rank;        
+    };
 
+class KruskalWallisCommand : public Command {
+   
 public:
-       
+           
        KruskalWallisCommand(string);   
        KruskalWallisCommand();
        ~KruskalWallisCommand(){}
@@ -28,7 +34,7 @@ public:
     
     int execute(); 
        void help() { m->mothurOut(getHelpString()); }
-    multimap<double,double> getRank(vector<KruskalWallisCommand::groupRank>);
+    void assignRank(vector<groupRank>);
     
 private:
     string outputDir;
@@ -36,6 +42,14 @@ private:
     vector<double> rankSums;
     vector<double> rankMeans;
     
+  
+        
+    bool comparevalue(const groupRank &a, const groupRank &b) { return a.value < b.value; }
+    bool equalvalue(const groupRank &a, const groupRank &b) { return a.value == b.value; }
+    bool comparerank(const groupRank &a, const groupRank &b) { return a.rank < b.rank; }
+    bool equalrank(const groupRank &a, const groupRank &b) { return a.rank == b.rank; }
+    bool equalgroup(const groupRank &a, const groupRank &b) { return a.group == b.group; }
+    
 };
 
 #endif /* KRUSKALWALLISCOMMAND_H */