---

LemurDBManager.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 _LEMURDBMANAGER_HPP
#define _LEMURDBMANAGER_HPP

#include "DBManager.hpp"

#include "InvFPIndex.hpp"

#include "RetrievalMethod.hpp"
#include "LemurMemParser.hpp"
#include "RetMethodManager.hpp"

namespace LemurParameter {
  static RetMethodManager::RetModel mod;
  static String dbname;

  static void get() {
    // default is KL divergence model
    mod = (RetMethodManager::RetModel) ParamGetInt("retModel", RetMethodManager::KL); 
    dbname = ParamGetString("index");
    RetrievalParameter::get();  
  }
};

class LemurDBManager : public DBManager {

public:
  void open(char * dbname);

  results_t * query (char * query, int numdocs);

  MemParser * getParser();

  doc_t * getDoc(docid_t docid);

  void output(docid_t docid);
  void setOutputFile(char * filename);

  void close();

private:
  
  Index * index;
  RetrievalMethod * model;
  ScoreAccumulator * accumulator;
  IndexedRealVector * results;
  LemurMemParser * parser;
  ofstream * outfile;

};

#endif

---