예제 #1
0
  // inherit javadoc
  public TopDocs search(Query query, Filter filter, final int nDocs) throws IOException {
    Scorer scorer = query.weight(this).scorer(reader);
    if (scorer == null) return new TopDocs(0, new ScoreDoc[0]);

    final BitSet bits = filter != null ? filter.bits(reader) : null;
    final HitQueue hq = new HitQueue(nDocs);
    final int[] totalHits = new int[1];
    scorer.score(
        new HitCollector() {
          private float minScore = 0.0f;

          public final void collect(int doc, float score) {
            if (score > 0.0f
                && // ignore zeroed buckets
                (bits == null || bits.get(doc))) { // skip docs not in bits
              totalHits[0]++;
              if (hq.size() < nDocs || score >= minScore) {
                hq.insert(new ScoreDoc(doc, score));
                minScore = ((ScoreDoc) hq.top()).score; // maintain minScore
              }
            }
          }
        });

    ScoreDoc[] scoreDocs = new ScoreDoc[hq.size()];
    for (int i = hq.size() - 1; i >= 0; i--) // put docs in array
    scoreDocs[i] = (ScoreDoc) hq.pop();

    return new TopDocs(totalHits[0], scoreDocs);
  }
예제 #2
0
  // inherit javadoc
  public TopFieldDocs search(Query query, Filter filter, final int nDocs, Sort sort)
      throws IOException {
    Scorer scorer = query.weight(this).scorer(reader);
    if (scorer == null) return new TopFieldDocs(0, new ScoreDoc[0], sort.fields);

    final BitSet bits = filter != null ? filter.bits(reader) : null;
    final FieldSortedHitQueue hq = new FieldSortedHitQueue(reader, sort.fields, nDocs);
    final int[] totalHits = new int[1];
    scorer.score(
        new HitCollector() {
          public final void collect(int doc, float score) {
            if (score > 0.0f
                && // ignore zeroed buckets
                (bits == null || bits.get(doc))) { // skip docs not in bits
              totalHits[0]++;
              hq.insert(new FieldDoc(doc, score));
            }
          }
        });

    ScoreDoc[] scoreDocs = new ScoreDoc[hq.size()];
    for (int i = hq.size() - 1; i >= 0; i--) // put docs in array
    scoreDocs[i] = hq.fillFields((FieldDoc) hq.pop());

    return new TopFieldDocs(totalHits[0], scoreDocs, hq.getFields());
  }
예제 #3
0
 Hits(Searcher s, Query q, Filter f) throws IOException {
   weight = q.weight(s);
   searcher = s;
   filter = f;
   nDeletions = countDeletions(s);
   getMoreDocs(50); // retrieve 100 initially
   lengthAtStart = length;
 }
예제 #4
0
  // inherit javadoc
  public void search(Query query, Filter filter, final HitCollector results) throws IOException {
    HitCollector collector = results;
    if (filter != null) {
      final BitSet bits = filter.bits(reader);
      collector =
          new HitCollector() {
            public final void collect(int doc, float score) {
              if (bits.get(doc)) { // skip docs not in bits
                results.collect(doc, score);
              }
            }
          };
    }

    Scorer scorer = query.weight(this).scorer(reader);
    if (scorer == null) return;
    scorer.score(collector);
  }
예제 #5
0
 public Explanation explain(Query query, int doc) throws IOException {
   return query.weight(this).explain(reader, doc);
 }