WordSet.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.lemurproject.org/license.html
 *
 *==========================================================================
 */

#ifndef _WORDSET_HPP
#define _WORDSET_HPP
#include "Exception.hpp"
#include <set>
#include<cstring>
#include "common_headers.hpp"
namespace lemur 
{
  namespace utility 
  {

#ifndef _LT_STR
#define _LT_STR

    /* for the set */
    struct lt_str
    {
      bool operator()(const char* s1, const char* s2) const
      {
        return strcmp(s1, s2) < 0;
      }
    };
#endif



    class WordSet {

    public:
      WordSet();
      WordSet(const string &filename);
      ~WordSet();

      void load(const string &filename);
      void add(const char * word);
      bool contains(const char * word) const;

      void clear();


    protected:
  
      set<char *, lt_str> words;
      string lastfile;

    };
  }
}

#endif

---