---

DocUnigramCounter.hpp

Go to the documentation of this file.

/*==========================================================================
 * Copyright (c) 2001 Carnegie Mellon University.  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
 *
 *==========================================================================
*/


#ifndef _DOCUNIGRAMCOUNTER_HPP
#define _DOCUNIGRAMCOUNTER_HPP

#include "common_headers.hpp"
#include "WeightedIDSet.hpp"
#include "Counter.hpp"
#include "Index.hpp"


class DocUnigramCounter : public ArrayCounter <int> {
public:
  DocUnigramCounter(int docID, Index &homeIndex);

  DocUnigramCounter(vector<int> &docSet, Index &homeIndex);

  DocUnigramCounter(WeightedIDSet &docSet, Index &homeIndex);

  DocUnigramCounter(Index &collectionIndex);

  virtual ~DocUnigramCounter() {}
  const char *lexiconID() { return (ind.termLexiconID());}

protected:

  void countDocUnigram(int docID, double weight=1);

  Index &ind;
};


#endif /* _DOCUNIGRAMCOUNTER_HPP */

---