ResultFile.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 _RESULTFILE_HPP
#define _RESULTFILE_HPP

#include "common_headers.hpp"
#include "Index.hpp"
#include "IndexedReal.hpp"
#include "ISet.hpp"
#include "String.hpp"
namespace lemur 
{
  namespace utility
  {
    struct ResultEntry {
    public:
      String key;
      unsigned int Hash() const { return key.Hash();}
      unsigned int hash() const { return Hash();} 
      bool operator==(const ResultEntry entry) { return (entry.key ==  this->key);}
      lemur::api::IndexedRealVector *res;
    };
  }
}
namespace lemur 
{
  namespace api
  {

    class ResultFile {
    public:
      ResultFile(bool TRECFormat = true) : trecFmt (TRECFormat), resTable(NULL) {
        resTable = new lemur::utility::ISet<lemur::utility::ResultEntry>(50);
      }
  
      ~ResultFile() { delete resTable;}

      void openForRead(istream &is, Index &index);
  
      void load(istream &is, Index &index);

      void getResult(const string& expectedQID, IndexedRealVector &res);

      bool findResult(const string& queryID, IndexedRealVector *&res);
  
      void openForWrite( ostream &os, Index &index) {
    
        outStr = &os;
        ind = &index;
      }


      void writeResults(const string& queryID, IndexedRealVector *results, int maxCountOfResult);
  
    private:

      bool readLine();

      bool trecFmt;

      //  char curQID[300];
      string curQID;
      char curDID[300];
      double curSC;
      Index *ind;
      istream *inStr;
      ostream *outStr;
      bool eof;

      lemur::utility::ISet<lemur::utility::ResultEntry> * resTable;

    };
  }
}
#endif /* _RESULTFILE_HPP */

---