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);


  const PointSet& centers() const;


  int centers_size() const;


  int membership(int index) const;

  int membership_size() 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

---