Пример #1
0
  /**
   * Prints the results for the given search request, using the specified destination.
   *
   * @param pw PrintWriter the destination where to save the results.
   * @param q SearchRequest the object encapsulating the query and the results.
   */
  public void printResults(
      final PrintWriter pw,
      final SearchRequest q,
      String method,
      String iteration,
      int _RESULTS_LENGTH)
      throws IOException {
    final ResultSet set = q.getResultSet();
    final String metaIndexDocumentKey =
        ApplicationSetup.getProperty("trec.querying.outputformat.docno.meta.key", "docno");
    final double[] scores = set.getScores();
    if (set.getResultSize() == 0) {
      logger.warn("No results retrieved for query " + q.getQueryID());
      return;
    }
    String[] docnos = obtainDocnos(metaIndexDocumentKey, q, set);

    final int maximum =
        _RESULTS_LENGTH > set.getResultSize() || _RESULTS_LENGTH == 0
            ? set.getResultSize()
            : _RESULTS_LENGTH;
    logger.debug("Writing " + maximum + " results for query " + q.getQueryID());

    // if the minimum number of documents is more than the
    // number of documents in the results, aw.length, then
    // set minimum = aw.length

    // if (minimum > set.getResultSize())
    // minimum = set.getResultSize();
    // final String iteration = ITERATION + "0";
    final String queryIdExpanded = q.getQueryID() + " " + iteration + " ";
    final String methodExpanded = " " + method + ApplicationSetup.EOL;
    StringBuilder sbuffer = new StringBuilder();
    // the results are ordered in desceding order
    // with respect to the score.
    int limit = 10000;
    int counter = 0;
    for (int i = 0; i < maximum; i++) {
      if (scores[i] == Double.NEGATIVE_INFINITY) continue;
      sbuffer.append(queryIdExpanded);
      sbuffer.append(docnos[i]);
      sbuffer.append(" ");
      sbuffer.append(i);
      sbuffer.append(" ");
      sbuffer.append(scores[i]);
      sbuffer.append(methodExpanded);
      counter++;
      if (counter % limit == 0) {
        pw.write(sbuffer.toString());
        sbuffer = null;
        sbuffer = new StringBuilder();
        pw.flush();
      }
    }
    pw.write(sbuffer.toString());
    pw.flush();
  }
Пример #2
0
 /**
  * According to the given parameters, it sets up the correct matching class.
  *
  * @param queryId String the query identifier to use.
  * @param query String the query to process.
  * @param cParameter double the value of the parameter to use.
  */
 public void processQuery(String queryId, String query, double cParameter) {
   SearchRequest srq = queryingManager.newSearchRequest(queryId, query);
   srq.setControl("c", Double.toString(cParameter));
   srq.addMatchingModel(mModel, wModel);
   matchingCount++;
   queryingManager.runPreProcessing(srq);
   queryingManager.runMatching(srq);
   queryingManager.runPostProcessing(srq);
   queryingManager.runPostFilters(srq);
   try {
     printResults(resultFile, srq);
   } catch (IOException ioe) {
     logger.error("Problem displaying results", ioe);
   }
 }
Пример #3
0
  /**
   * Prints the results
   *
   * @param pw PrintWriter the file to write the results to.
   * @param q SearchRequest the search request to get results from.
   */
  public void printResults(PrintWriter pw, SearchRequest q) throws IOException {
    ResultSet set = q.getResultSet();
    int[] docids = set.getDocids();
    double[] scores = set.getScores();
    int minimum = RESULTS_LENGTH;
    // if the minimum number of documents is more than the
    // number of documents in the results, aw.length, then
    // set minimum = aw.length
    if (minimum > set.getResultSize()) minimum = set.getResultSize();
    if (verbose)
      if (set.getResultSize() > 0)
        pw.write("\n\tDisplaying 1-" + set.getResultSize() + " results\n");
      else pw.write("\n\tNo results\n");
    if (set.getResultSize() == 0) return;

    int metaKeyId = 0;
    final int metaKeyCount = metaKeys.length;
    String[][] docNames = new String[metaKeyCount][];
    for (String metaIndexDocumentKey : metaKeys) {
      if (set.hasMetaItems(metaIndexDocumentKey)) {
        docNames[metaKeyId] = set.getMetaItems(metaIndexDocumentKey);
      } else {
        final MetaIndex metaIndex = index.getMetaIndex();
        docNames[metaKeyId] = metaIndex.getItems(metaIndexDocumentKey, docids);
      }
      metaKeyId++;
    }

    StringBuilder sbuffer = new StringBuilder();
    // the results are ordered in asceding order
    // with respect to the score. For example, the
    // document with the highest score has score
    // score[scores.length-1] and its docid is
    // docid[docids.length-1].
    int start = 0;
    int end = minimum;
    for (int i = start; i < end; i++) {
      if (scores[i] <= 0d) continue;
      sbuffer.append(i);
      sbuffer.append(" ");
      for (metaKeyId = 0; metaKeyId < metaKeyCount; metaKeyId++) {
        sbuffer.append(docNames[metaKeyId][i]);
        sbuffer.append(" ");
      }
      sbuffer.append(docids[i]);
      sbuffer.append(" ");
      sbuffer.append(scores[i]);
      sbuffer.append('\n');
    }
    // System.out.println(sbuffer.toString());
    pw.write(sbuffer.toString());
    pw.flush();
    // pw.write("finished outputting\n");
  }