X-Git-Url: https://git.donarmstrong.com/?p=mothur.git;a=blobdiff_plain;f=cluster.hpp;h=dff55e61fcf7b571da5c0b895a5f46910f58f8c6;hp=72d3611a093296b3dc18d9ddb7bf12d0a0a9a0e5;hb=a8e2df1b96a57f5f29576b08361b86a96a8eff4f;hpb=74844a60d80c6dd06e3fb02ee9b928424f9019b0

diff --git a/cluster.hpp b/cluster.hpp
index 72d3611..dff55e6 100644
--- a/cluster.hpp
+++ b/cluster.hpp
@@ -2,49 +2,43 @@
 #define CLUSTER_H
 
 
+
 #include "mothur.h"
-#include "sparsematrix.hpp"
+#include "sparsedistancematrix.h"
 #include "mothurout.h"
 
 class RAbundVector;
 class ListVector;
 
-typedef vector<MatData> MatVec;
-
 class Cluster {
 	
 public:
-	Cluster(RAbundVector*, ListVector*, SparseMatrix*, float, string);
+	Cluster(RAbundVector*, ListVector*, SparseDistanceMatrix*, float, string);
+    virtual ~Cluster() {}
     virtual void update(double&);				
 	virtual string getTag() = 0;
 	virtual void setMapWanted(bool m);  
 	virtual map<string, int> getSeqtoBin()  {  return seq2Bin;	}
-
-protected:	
-	void getRowColCells();
-    void removeCell(const MatData& cell, int vrow, int vcol, bool rmMatrix=true);
-
-	virtual bool updateDistance(MatData& colCell, MatData& rowCell) = 0;
-
+    
+protected:	    
+	virtual bool updateDistance(PDistCell& colCell, PDistCell& rowCell) = 0;
+    
 	virtual void clusterBins();
 	virtual void clusterNames();
 	virtual void updateMap();
 	
 	RAbundVector* rabund;
 	ListVector* list;
-	SparseMatrix* dMatrix;	
+	SparseDistanceMatrix* dMatrix;	
 	
-	int smallRow;
-	int smallCol;
+	ull smallRow;
+	ull smallCol;
 	float smallDist;
 	bool mapWanted;
 	float cutoff;
 	map<string, int> seq2Bin;
 	string method;
 	
-	vector<MatVec> seqVec;		// contains vectors of cells related to a certain sequence
-	MatVec rowCells;
-	MatVec colCells;
 	ull nRowCells;
 	ull nColCells;
 	MothurOut* m;
@@ -54,33 +48,33 @@ protected:
 
 class CompleteLinkage : public Cluster {
 public:
-	CompleteLinkage(RAbundVector*, ListVector*, SparseMatrix*, float, string);
-	bool updateDistance(MatData& colCell, MatData& rowCell);
+	CompleteLinkage(RAbundVector*, ListVector*, SparseDistanceMatrix*, float, string);
+	bool updateDistance(PDistCell& colCell, PDistCell& rowCell);
 	string getTag();
 	
 private:
-		
+    
 };
 
 /***********************************************************************/
 
 class SingleLinkage : public Cluster {
 public:
-	SingleLinkage(RAbundVector*, ListVector*, SparseMatrix*, float, string);
-    void update();
-	bool updateDistance(MatData& colCell, MatData& rowCell);
+	SingleLinkage(RAbundVector*, ListVector*, SparseDistanceMatrix*, float, string);
+    void update(double&);
+	bool updateDistance(PDistCell& colCell, PDistCell& rowCell);
 	string getTag();
 	
 private:
-		
+    
 };
 
 /***********************************************************************/
 
 class AverageLinkage : public Cluster {
 public:
-	AverageLinkage(RAbundVector*, ListVector*, SparseMatrix*, float, string);
-	bool updateDistance(MatData& colCell, MatData& rowCell);
+	AverageLinkage(RAbundVector*, ListVector*, SparseDistanceMatrix*, float, string);
+	bool updateDistance(PDistCell& colCell, PDistCell& rowCell);
 	string getTag();
 	
 private:
@@ -89,9 +83,24 @@ private:
 	int rowBin;
 	int colBin;
 	int totalBin;
+    
+};
+
+/***********************************************************************/
 
+class WeightedLinkage : public Cluster {
+public:
+	WeightedLinkage(RAbundVector*, ListVector*, SparseDistanceMatrix*, float, string);
+	bool updateDistance(PDistCell& colCell, PDistCell& rowCell);
+	string getTag();
+	
+private:
+	int saveRow;
+	int saveCol;	
 };
 
 /***********************************************************************/
 
+
+
 #endif