---

ClusterDB.hpp

Go to the documentation of this file.

/*==========================================================================
 * Copyright (c) 2002 University of Massachusetts.  All Rights Reserved.
 *
 * Use of the Lemur Toolkit for Language Modeling and Information Retrieval
 * is subject to the terms of the software license set forth in the LICENSE
 * file included with this software, and also available at
 * http://www.cs.cmu.edu/~lemur/license.html
 *
 *==========================================================================
*/

// David Fisher, Peter Amstutz
// init: 11/18/2002
#ifndef _CLUSTERDB_HPP
#define _CLUSTERDB_HPP
#include <stdexcept>
#include "common_headers.hpp"
#include "Index.hpp"
#include "IndexedReal.hpp"
#include "ClusterParam.hpp"
#include "ClusterFactory.hpp"
#include "SimFactory.hpp"
#include "ThresholdFcn.hpp"

class ClusterDBError : public runtime_error
{
public:
    ClusterDBError(const string& s) : runtime_error(s) { };
};


class ClusterDB
{
public:
  ClusterDB(const Index *ind, 
            double threshold = 0.25,
            enum ClusterParam::simTypes simType = ClusterParam::COS,
            enum ClusterParam::clusterTypes clusterType=ClusterParam::CENTROID,
            enum ClusterParam::docModes docMode = ClusterParam::DMAX);

  virtual ~ClusterDB();

  virtual int countClusters() const = 0;

  virtual int maxID() const = 0;

  virtual vector<Cluster*> getDocCluster(int docId) const = 0;

  virtual Cluster *getCluster(int clusterId) const = 0;

  virtual vector<int> getDocClusterId(int docId) const = 0;

  virtual int addToCluster(int docId, int clusterId, double score) = 0;

  virtual int addToCluster(int docId, Cluster *cluster, double score) = 0;

  virtual int removeFromCluster(int docId, int clusterID) = 0;
  virtual int deleteCluster(int clusterID) = 0;

  virtual int deleteCluster (Cluster *target) = 0;

  virtual vector<int> splitCluster(int cid, int num = 2) = 0;

  virtual int mergeClusters(int cid1, int cid2) = 0;

  virtual void printClusters() const;

  virtual int cluster(int docId);

  virtual int cluster(int docId, double &finalScore);

  virtual string getKeyWords(int cid, int numTerms = 10) const;

protected:
  const Index *index;
  int numDocs;
  int numTerms;
  double threshold; // fix this.
  mutable vector<Cluster *> clusters;
  const SimilarityMethod *sim;
  ClusterFactory *factory;
  virtual Cluster * newCluster() = 0;
  Cluster* allocateCluster(int clusterID) const;
  ThresholdFcn *thresh;
};
#endif

---