X-Git-Url: https://git.donarmstrong.com/?p=mothur.git;a=blobdiff_plain;f=cluster.hpp;h=dff55e61fcf7b571da5c0b895a5f46910f58f8c6;hp=d7c2737a05ed1a0198be9713aa4345f288ec3976;hb=a8e2df1b96a57f5f29576b08361b86a96a8eff4f;hpb=0caf3fbabaa3ece404f8ce77f4c883dc5b1bf1dc diff --git a/cluster.hpp b/cluster.hpp index d7c2737..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 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 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 seq2Bin; string method; - vector 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); + SingleLinkage(RAbundVector*, ListVector*, SparseDistanceMatrix*, float, string); void update(double&); - bool updateDistance(MatData& colCell, MatData& rowCell); + 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,15 +83,15 @@ private: int rowBin; int colBin; int totalBin; - + }; /***********************************************************************/ class WeightedLinkage : public Cluster { public: - WeightedLinkage(RAbundVector*, ListVector*, SparseMatrix*, float, string); - bool updateDistance(MatData& colCell, MatData& rowCell); + WeightedLinkage(RAbundVector*, ListVector*, SparseDistanceMatrix*, float, string); + bool updateDistance(PDistCell& colCell, PDistCell& rowCell); string getTag(); private: @@ -107,4 +101,6 @@ private: /***********************************************************************/ + + #endif