---

QryBasedSampler.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 _QRYBASEDSAMPLER_HPP
#define _QRYBASEDSAMPLER_HPP



#include "FreqCounter.hpp"
#include "DBManager.hpp"

//typedef stringset docidset;
typedef set<docid_t, less<string> > docidset;

#define T_NDOCS 1

#define T_NWORDS 2

#define T_NQRYS 4

class QryBasedSampler {
public:
  QryBasedSampler();
  ~QryBasedSampler();

  bool probe(const char * initQuery);
  
  void setDBManager(const DBManager * database);

  const DBManager * getDBManager() const;


  void setFreqCounter(FreqCounter * counter);

  const FreqCounter * getFreqCounter() const;


  void setOutputPrefix(const string &prefix);
  
  const string &getOutputPrefix() const;

  void setNumDocs(int n);

  int getNumDocs() const;


  void setNumWords(int n);

  int getNumWords() const;


  void setNumQueries(int n);

  int getNumQueries() const;


  void setTermMode(int m);

  int getTermMode() const;
  

  void setDocsPerQuery(int n);
  
  int getDocsPerQuery() const;


private:

  /* for querying a db */
  const DBManager * db;


  /* for building a description of a db */
  FreqCounter * freqCounter;


  /* output prefix for filenames */
  string outputPrefix;


  /* termination mode of the probe -
   * either T_NDOCS or T_NWORDS */
  int termMode;
  
  /* number unique docs to retrieve - only used if
   * termMode == T_NDOCS */
  int numDocs;

  /* number unique words to retrieve - only used if
   * termMode == T_NWORDS */
  int numWords;

  /* number of queries to run - only used if
   * termMode == T_NQRYS */
  int numQueries;

  /* documents per query to use */
  int docsPerQuery;

  /* stores the ids of the document already retrieved
   * from the system.  used to prevent parsing
   * a document multiple times */
  docidset seenDocs;
};

#endif

---