package org.apache.lucene.search;
/**
* Copyright 2005 Apache Software Foundation.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import java.io.IOException;
/** A Scorer for queries with a required part and an optional part.
* Delays skipTo() on the optional part until a score() is needed.
* <br>
* This <code>Scorer</code> implements {@link Scorer#skipTo(int)}.
*/
public class ReqOptSumScorer extends Scorer {
/** The scorers passed from the constructor.
* These are set to null as soon as their next() or skipTo() returns false.
*/
private Scorer reqScorer;
private Scorer optScorer;
/** Construct a <code>ReqOptScorer</code>.
* @param reqScorer The required scorer. This must match.
* @param optScorer The optional scorer. This is used for scoring only.
*/
public ReqOptSumScorer(
Scorer reqScorer,
Scorer optScorer)
{
super(null); // No similarity used.
this.reqScorer = reqScorer;
this.optScorer = optScorer;
}
private boolean firstTimeOptScorer = true;
public boolean next() throws IOException {
return reqScorer.next();
}
public boolean skipTo(int target) throws IOException {
return reqScorer.skipTo(target);
}
public int doc() {
return reqScorer.doc();
}
/** Returns the score of the current document matching the query.
* Initially invalid, until {@link #next()} is called the first time.
* @return The score of the required scorer, eventually increased by the score
* of the optional scorer when it also matches the current document.
*/
public float score() throws IOException {
int curDoc = reqScorer.doc();
float reqScore = reqScorer.score();
if (firstTimeOptScorer) {
firstTimeOptScorer = false;
if (! optScorer.skipTo(curDoc)) {
optScorer = null;
return reqScore;
}
} else if (optScorer == null) {
return reqScore;
} else if ((optScorer.doc() < curDoc) && (! optScorer.skipTo(curDoc))) {
optScorer = null;
return reqScore;
}
// assert (optScorer != null) && (optScorer.doc() >= curDoc);
return (optScorer.doc() == curDoc)
? reqScore + optScorer.score()
: reqScore;
}
/** Explain the score of a document.
* @todo Also show the total score.
* See BooleanScorer.explain() on how to do this.
*/
public Explanation explain(int doc) throws IOException {
Explanation res = new Explanation();
res.setDescription("required, optional");
res.addDetail(reqScorer.explain(doc));
res.addDetail(optScorer.explain(doc));
return res;
}
}
|