---

InQueryRetMethod.hpp

Go to the documentation of this file.

/*==========================================================================
 * Copyright (c) 2002 University of Massachusetts.  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
 *
 *==========================================================================
*/
/*
  Author: fff
 */

#ifndef _INQUERYRETMETHOD_HPP
#define _INQUERYRETMETHOD_HPP

#include "StructQueryRetMethod.hpp"
#include "StructQryDocRep.hpp"

namespace InQueryParameter {
  struct docParam {
    double defaultBelief;
  };
  struct qryParam {
    double fbCoeff;
    int fbTermCount;
  };

  static double defaultFBCoeff = 0.5;
  static int defaultFBTermCount = 50;
  static double defaultBelief = 0.4;
};

class InQueryRep : public StructQueryRep {
public:
  InQueryRep(StructQuery &qry, InvFPIndex &dbIndex) :  StructQueryRep(qry, dbIndex, InQueryParameter::defaultBelief), ind(dbIndex) {}
  virtual ~InQueryRep() {}

protected:
  double *idf;
  InvFPIndex &ind;
};



class InQueryRetMethod : public StructQueryRetMethod {
public:

  InQueryRetMethod(InvFPIndex &dbIndex, ScoreAccumulator &accumulator);
  virtual ~InQueryRetMethod() {delete [] idfV; delete scFunc;}

  virtual StructQueryRep *computeStructQueryRep(StructQuery &qry) {
    return (new InQueryRep(qry, index));
  }

  virtual DocumentRep *computeDocRep(int docID) { 
    return (new StructQryDocRep(docID, index, idfV));
  }
  virtual ScoreFunction *scoreFunc() {
    return (scFunc);
  }
  virtual void updateStructQuery(StructQueryRep &qryRep, DocIDSet &relDocs);

  void setDocINQParam(InQueryParameter::docParam &dParam);
  void setQueryINQParam(InQueryParameter::qryParam &qParam);

protected:
  double *idfV;
  ScoreFunction *scFunc; 
  InQueryParameter::qryParam qryPrm;
  InQueryParameter::docParam docPrm;

};

inline void InQueryRetMethod::setDocINQParam(InQueryParameter::docParam &dParam)
{
  docPrm = dParam;
}



inline void InQueryRetMethod::setQueryINQParam(InQueryParameter::qryParam &qParam)
{
  qryPrm = qParam;
}

#endif /* _INQUERYRETMETHOD_HPP */

---