---

KeyfileClusterDB.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.lemurproject.org/license.html
 *
 *==========================================================================
*/

// David Fisher
// init: 05/11/2004
#ifndef _KEYFILECLUSTERDB_HPP
#define _KEYFILECLUSTERDB_HPP

#include "ClusterDB.hpp"
#include "IndexedReal.hpp"
#include "Keyfile.hpp"
#include <string>
class KeyfileClusterDB : public ClusterDB {

public:
  KeyfileClusterDB(const Index *ind, 
                    const string &name,                    
                    double threshold = 0.25,
                    enum ClusterParam::simTypes simType = ClusterParam::COS,
                    enum ClusterParam::clusterTypes 
                      clusterType = ClusterParam::CENTROID,
                    enum ClusterParam::docModes docMode = ClusterParam::DMAX);

  virtual ~KeyfileClusterDB();

  virtual vector<Cluster*> getDocCluster(DOCID_T docId) const ;

  virtual int countClusters() const {
  //clustersKey.get("NUMCLUSTERS");
    int clusterCount = 0, actual;
    clustersKey.get("NUMCLUSTERS", &clusterCount, actual, sizeof(int));
    return  clusterCount;
  }

  virtual int maxID() const {// get
  //clustersKey.get("MAXID");
    int clusterIDCounter = 0, actual;
    clustersKey.get("MAXID", &clusterIDCounter, actual, sizeof(int));
    return  clusterIDCounter;
  }
  

  virtual Cluster *getCluster(int clusterId) const ;

  virtual vector<int> getDocClusterId(DOCID_T docId) const ;

  virtual int addToCluster(DOCID_T docId, int clusterId, double score);

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

  virtual int removeFromCluster(DOCID_T docId, int clusterID);

  virtual int deleteCluster(int clusterID);

  virtual int deleteCluster(Cluster * cluster);

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

  virtual int mergeClusters(int cid1, int cid2);

protected:
  virtual Cluster * newCluster();

private:
  mutable Keyfile doc2cluster;
  mutable Keyfile clustersKey;
  void init(const string &name);
  void addC2D(int cid, const vector<DOCID_T> &dids);
};
#endif

---