---

OneStepMarkovChain.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 _ONESTEPMARKOVCHAIN_HPP
#define _ONESTEPMARKOVCHAIN_HPP


#include "Index.hpp"
#include "WeightedIDSet.hpp"

class OneStepMarkovChain {
public:
  OneStepMarkovChain(const WeightedIDSet &docSet, const Index &homeIndex, double *wdNorm, double stopProbAlpha);
  ~OneStepMarkovChain();
  


  void startFromWordIteration(int toWord) const;
  bool hasMoreFromWord() const { 
    while (itPos <= ind.termCountUnique() && fromWordPr[itPos]==0) 
      itPos++;
    return (itPos<= ind.termCountUnique());
  }
  void nextFromWordProb(int &fromWord, double &prob) const ;

private:
  void computeFromWordProb(int toWord) const;
  const Index &ind;
  double alpha;
  int *dSet;
  double *norm;
  mutable double *fromWordPr;
  mutable int itPos;
  mutable int curToWord;
};


#endif /* _ONESTEPMARKOVCHAIN_HPP */

---