---

Counter.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 _COUNTER_HPP
#define _COUNTER_HPP



class Counter {
public:

  virtual ~Counter() {}
  // access functions
  
  virtual double count(int eventIndex) = 0;
  virtual double sum() = 0;
  
  virtual void startIteration() = 0;
  virtual bool hasMore() = 0;
  virtual void nextCount(int &eventIndex, double &count) = 0;
};

class ModifiableCounter : public Counter {
public: 
  // manipulation functions
  virtual ~ModifiableCounter() {}
  virtual void incCount(int eventIndex, double count) = 0;
  virtual void setCount(int eventIndex, double count)  = 0;
};


template<class T>
class ArrayCounter : public ModifiableCounter {
public:
  
  ArrayCounter(int size) : sz(size), ct(new T[size]), total(0) {
    for (int i=0; i<size; i++) ct[i]=0;
  }
  virtual ~ArrayCounter() { delete [] ct;}

  virtual double count(int eventIndex) {
    return ct[eventIndex];
  }

  virtual double sum() {
    return total;
  }

  virtual void incCount(int eventIndex, double count) {
    ct[eventIndex] += (T)count;
    total += (T)count;
  }

  virtual void setCount(int eventIndex, double count) {
    total = total - ct[eventIndex]+ (T)count;
    ct[eventIndex] = (T)count;
  }


  virtual void startIteration() {
    pos=0;
  }

  virtual bool hasMore() {
    while ((pos < sz) && (ct[pos] == 0))
      pos++;
    return (pos<sz); 
  }

  virtual void nextCount(int &eventIndex, double &count) {
    eventIndex = pos;
    count = ct[pos];
    pos++;
  }

protected:
  T *ct;
  int sz;
  T total;
  int pos;
};

#endif /* _COUNTER_HPP */

---