---

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.cs.cmu.edu/~lemur/license.html
 *
 *==========================================================================
*/


#ifndef _RESULTFILE_HPP
#define _RESULTFILE_HPP

#include "common_headers.hpp"
#include "Index.hpp"
#include "IndexedReal.hpp"
#include "ISet.hpp"
#include "String.hpp"

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);}  IndexedRealVector *res;
};


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

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

  void getResult(char *expectedQID, IndexedRealVector &res);

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


  void writeResults(char *queryID, IndexedRealVector *results, int maxCountOfResult);
  
private:

  bool readLine();

  bool trecFmt;

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

  ISet<ResultEntry> * resTable;

};


#endif /* _RESULTFILE_HPP */

---