clusterer.h

Go to the documentation of this file.

// Copyright 2007, Massachusetts Institute of Technology.
// The use of this code is permitted for research only. There is
// absolutely no warranty for this software.
//
// Author: John Lee (jjl@mit.edu)
//
#ifndef CLUSTERING_CLUSTERER_H
#define CLUSTERING_CLUSTERER_H

#include <vector>
#include <iostream>
#include "point_set/point-set.h"
#include "point_set/point-ref.h"

using namespace std;

namespace libpmk {

class Clusterer {
 public:
   Clusterer();
   virtual ~Clusterer() { }


   void Cluster(const vector<PointRef>& data);


   PointSet GetClusterCenters() const;


   int GetNumCenters() const;


   vector<int> GetMembership() const;


   void WriteToStream(ostream& output_stream) const;

   void WriteToFile(const char* output_filename) const;


   void ReadFromStream(istream& input_stream);

   void ReadFromFile(const char* input_filename);

 protected:
   auto_ptr<PointSet> cluster_centers_;
   vector<int> membership_;
   bool done_;


   virtual void DoClustering(const vector<PointRef>& data) = 0;
};
}  // namespace libpmk

#endif  // CLUSTERING_CLUSTERER_H

---