/*
* Hibernate Search, full-text search for your domain model
*
* License: GNU Lesser General Public License (LGPL), version 2.1 or later
* See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
*/
package org.hibernate.search.spatial.impl;
import java.io.IOException;
import org.apache.lucene.search.DocIdSetIterator;
import org.apache.lucene.search.Scorer;
import org.apache.lucene.search.TwoPhaseIterator;
import org.apache.lucene.search.Weight;
/**
* A copy of Lucene's ConstantScoreWeight implementation,
* necessary because the one in Lucene is marked as "internal".
*
* @author Yoann Rodiere
*/
final class ConstantScoreScorer extends Scorer {
private final float score;
private final TwoPhaseIterator twoPhaseIterator;
private final DocIdSetIterator disi;
/**
* Constructor based on a {@link DocIdSetIterator} which will be used to drive iteration. Two phase iteration will
* not be supported.
*
* @param weight the parent weight
* @param score the score to return on each document
* @param disi the iterator that defines matching documents
*/
public ConstantScoreScorer(Weight weight, float score, DocIdSetIterator disi) {
super( weight );
this.score = score;
this.twoPhaseIterator = null;
this.disi = disi;
}
/**
* Constructor based on a {@link TwoPhaseIterator}. In that case the {@link Scorer} will support two-phase
* iteration.
*
* @param weight the parent weight
* @param score the score to return on each document
* @param twoPhaseIterator the iterator that defines matching documents
*/
public ConstantScoreScorer(Weight weight, float score, TwoPhaseIterator twoPhaseIterator) {
super( weight );
this.score = score;
this.twoPhaseIterator = twoPhaseIterator;
this.disi = TwoPhaseIterator.asDocIdSetIterator( twoPhaseIterator );
}
@Override
public DocIdSetIterator iterator() {
return disi;
}
@Override
public TwoPhaseIterator twoPhaseIterator() {
return twoPhaseIterator;
}
@Override
public int docID() {
return disi.docID();
}
@Override
public float score() throws IOException {
return score;
}
@Override
public int freq() throws IOException {
return 1;
}
}