ClusterRep.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
 *
 *==========================================================================
 */
/*
  Author: dmf 11/2002
*/

#ifndef _CLUSTERREP_HPP
#define _CLUSTERREP_HPP
#include "common_headers.hpp"
#include "FloatFreqVector.hpp"
#include "Index.hpp"
namespace lemur 
{
  namespace cluster
  {
    
    class ClusterRep {
    public:
      ClusterRep(const lemur::api::Index &ind) : index(ind) {
        rep = new lemur::utility::FloatFreqVector();
      }
      ClusterRep(lemur::api::DOCID_T did, const lemur::api::Index &ind);
      ClusterRep(lemur::api::TermInfoList *tList, 
                 const lemur::api::Index &ind);
      ClusterRep(vector<lemur::api::DOCID_T> &dids, 
                 const lemur::api::Index &ind);
      ClusterRep(lemur::utility::FloatFreqVector *v, 
                 const lemur::api::Index &ind);
      ClusterRep(const ClusterRep *old);
      virtual ~ClusterRep();  
      virtual void normalize();
      double sum2() const {return rep->sum2();}
      // This should not be exposed...
      lemur::utility::FloatFreqVector *getRep() const {return rep;}
      void weigh(const double *vals){rep->weigh(vals);} 
      void weigh(double val){rep->weigh(val);} 
      void addVal(int id, double val) {rep->addVal(id, val);}  
      void addTerms(const vector<lemur::api::TERMID_T> &tids) {
        for (int i = 0; i < tids.size(); i++) {
          addVal(tids[i], 1);
        }
      }
      void addVec(const ClusterRep *v2){rep->addVec(v2->rep);}
      void subtract(const ClusterRep *v2){rep->subtract(v2->rep);}
      double dotProd(const ClusterRep *v2) const {
        return rep->dotProd(v2->rep);
      }
    private:
      lemur::utility::FloatFreqVector *rep;
      const lemur::api::Index &index;
    };
  }
}

#endif /* _CLUSTERREP_HPP */

---