---

ScoreAccumulator.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 _SCOREACCUMULATOR_HPP
#define _SCOREACCUMULATOR_HPP


//------------------------------------------------------------
//      Abstract Interface for A Score Accumulator
//------------------------------------------------------------

class ScoreAccumulator {
public:
  virtual void reset()=0;
  virtual bool findScore(int id, double &score)=0;
  virtual void setScore(int id, double score)=0;
  virtual void incScore(int id, double score)=0;
  
  virtual void startIteration()=0;
  virtual bool hasMore() = 0;
  virtual void nextScore(int &id, double &score) =0;

};


//------------------------------------------------------------
//      An Array Score Accumulator
//------------------------------------------------------------

class ArrayAccumulator : public ScoreAccumulator {
public:
  ArrayAccumulator(int maxID);

  virtual ~ArrayAccumulator() { delete [] acc; delete [] status;}

  virtual void reset();
  
  virtual bool findScore(int id, double &score);

  virtual void setScore(int id, double score) { 
    acc[id-1] = score;
    status[id-1] =1;
  }

  virtual void incScore(int id, double score) {
    acc[id-1] += score;
    status[id-1] =1;
  }
  
  virtual void startIteration() { p = 0; }
  bool hasMore();
  
  void nextScore(int &id, double &score); 

protected:
  int p;
  int sz;
  double *acc;
  short *status;
};


// ============== inlines ==============================

inline ArrayAccumulator::ArrayAccumulator(int maxID) : sz(maxID) 
{ 
  acc = new double[maxID]; // index 0 refers to docid 1.
  status = new short[maxID]; // using integer making the test faster
}


#endif /* _SCOREACCUMULATOR_HPP */

---