Revert to previous commit

[mothur.git] / heatmap.h

diff --git a/heatmap.h b/heatmap.h
new file mode 100644 (file)
index 0000000..2b49176
--- /dev/null
+++ b/heatmap.h
@@ -0,0 +1,80 @@
+#ifndef HEATMAP_H
+#define HEATMAP_H
+/*
+ *  heatmap.h
+ *  Mothur
+ *
+ *  Created by Sarah Westcott on 3/25/09.
+ *  Copyright 2009 Schloss Lab UMASS Amherst. All rights reserved.
+ *
+ */
+
+#include "rabundvector.hpp"
+#include "sharedrabundvector.h"
+#include "sharedrabundfloatvector.h"
+#include "datavector.hpp"
+
+
+/***********************************************************************/
+struct binCount {
+               int bin;
+               int abund;
+               binCount(int i, int j) : bin(i), abund(j) {}
+};
+/***********************************************************************/
+struct binCountFloat {
+               int bin;
+               float abund;
+               binCountFloat(int i, float j) : bin(i), abund(j) {}
+};
+
+/***********************************************************************/
+//sorts highest abund to lowest
+inline bool comparebinCounts(binCount left, binCount right){
+       return (left.abund > right.abund);      
+}
+/***********************************************************************/
+//sorts highest abund to lowest
+inline bool comparebinFloatCounts(binCountFloat left, binCountFloat right){
+       return (left.abund > right.abund);      
+}
+/***********************************************************************/
+
+class HeatMap {
+       
+       public:
+               HeatMap(string, string, int, int, string, string);
+               ~HeatMap(){};
+       
+               string getPic(RAbundVector*);
+               string getPic(vector<SharedRAbundVector*>);
+               string getPic(vector<SharedRAbundFloatVector*>);
+
+       private:
+               int sortSharedVectors(vector<SharedRAbundVector*>& );
+               int sortSharedVectors(vector<SharedRAbundFloatVector*>& );
+               int sortRabund(RAbundVector*&);
+               void printLegend(int, float);
+
+               string format, sorted, groupComb, scaler, outputDir, inputfile;
+               ofstream outsvg;
+               MothurOut* m;
+               int numOTU, fontSize;
+               
+               map<int, int> orderTopGroup(vector<SharedRAbundVector*>&);
+               map<int, int> orderTopOtu(vector<SharedRAbundVector*>&);
+               map<int, int> orderShared(vector<SharedRAbundVector*>&);
+               map<int, int> orderTopGroup(vector<SharedRAbundFloatVector*>&);
+               map<int, int> orderTopOtu(vector<SharedRAbundFloatVector*>&);
+               map<int, int> orderShared(vector<SharedRAbundFloatVector*>&);
+
+                       
+};
+
+/***********************************************************************/
+
+#endif
+
+
+
+