public override bool Score(ICollector collector, int max)
{
RandomOrderCollector randomCollector = new RandomOrderCollector(Random, collector);
bool remaining = @in.Score(randomCollector, max);
randomCollector.Flush();
return(remaining);
}
/// <summary>
/// Lower-level search API.
///
/// <p>
/// <seealso cref="Collector#collect(int)"/> is called for every document. <br>
///
/// <p>
/// NOTE: this method executes the searches on all given leaves exclusively.
/// To search across all the searchers leaves use <seealso cref="#leafContexts"/>.
/// </summary>
/// <param name="leaves">
/// the searchers leaves to execute the searches on </param>
/// <param name="weight">
/// to match documents </param>
/// <param name="collector">
/// to receive hits </param>
/// <exception cref="BooleanQuery.TooManyClauses"> If a query would exceed
/// <seealso cref="BooleanQuery#getMaxClauseCount()"/> clauses. </exception>
protected internal virtual void Search(IList <AtomicReaderContext> leaves, Weight weight, Collector collector)
{
// TODO: should we make this
// threaded...? the Collector could be sync'd?
// always use single thread:
foreach (AtomicReaderContext ctx in leaves) // search each subreader
{
try
{
collector.NextReader = ctx;
}
catch (CollectionTerminatedException)
{
// there is no doc of interest in this reader context
// continue with the following leaf
continue;
}
BulkScorer scorer = weight.BulkScorer(ctx, !collector.AcceptsDocsOutOfOrder(), ctx.AtomicReader.LiveDocs);
if (scorer != null)
{
try
{
scorer.Score(collector);
}
catch (CollectionTerminatedException)
{
// collection was terminated prematurely
// continue with the following leaf
}
}
}
}
public virtual void Test()
{
Term allTerm = new Term(FIELD, "all");
TermQuery termQuery = new TermQuery(allTerm);
Weight weight = IndexSearcher.CreateNormalizedWeight(termQuery);
Assert.IsTrue(IndexSearcher.TopReaderContext is AtomicReaderContext);
AtomicReaderContext context = (AtomicReaderContext)IndexSearcher.TopReaderContext;
BulkScorer ts = weight.GetBulkScorer(context, true, (context.AtomicReader).LiveDocs);
// we have 2 documents with the term all in them, one document for all the
// other values
IList <TestHit> docs = new List <TestHit>();
// must call next first
ts.Score(new CollectorAnonymousInnerClassHelper(this, context, docs));
Assert.IsTrue(docs.Count == 2, "docs Size: " + docs.Count + " is not: " + 2);
TestHit doc0 = docs[0];
TestHit doc5 = docs[1];
// The scores should be the same
Assert.IsTrue(doc0.Score == doc5.Score, doc0.Score + " does not equal: " + doc5.Score);
/*
* Score should be (based on Default Sim.: All floats are approximate tf = 1
* numDocs = 6 docFreq(all) = 2 idf = ln(6/3) + 1 = 1.693147 idf ^ 2 =
* 2.8667 boost = 1 lengthNorm = 1 //there is 1 term in every document coord
* = 1 sumOfSquaredWeights = (idf * boost) ^ 2 = 1.693147 ^ 2 = 2.8667
* queryNorm = 1 / (sumOfSquaredWeights)^0.5 = 1 /(1.693147) = 0.590
*
* score = 1 * 2.8667 * 1 * 1 * 0.590 = 1.69
*/
Assert.IsTrue(doc0.Score == 1.6931472f, doc0.Score + " does not equal: " + 1.6931472f);
}
public override void Score(Collector collector)
{
if (Random.NextBoolean())
{
try
{
bool remaining = @in.Score(collector, DocsEnum.NO_MORE_DOCS);
Debug.Assert(!remaining);
}
catch (System.NotSupportedException e)
{
@in.Score(collector);
}
}
else
{
@in.Score(collector);
}
}
public override void Score(ICollector collector)
{
if (random.NextBoolean())
{
try
{
bool remaining = @in.Score(collector, DocsEnum.NO_MORE_DOCS);
Debug.Assert(!remaining);
}
#pragma warning disable 168
catch (NotSupportedException e)
#pragma warning restore 168
{
@in.Score(collector);
}
}
else
{
@in.Score(collector);
}
}
public override void Score(ICollector collector)
{
if (random.NextBoolean())
{
try
{
bool remaining = @in.Score(collector, DocsEnum.NO_MORE_DOCS);
if (Debugging.AssertsEnabled)
{
Debugging.Assert(!remaining);
}
}
catch (Exception e) when(e.IsUnsupportedOperationException())
{
@in.Score(collector);
}
}
else
{
@in.Score(collector);
}
}
public virtual void TestBooleanScorerMax()
{
Directory dir = NewDirectory();
RandomIndexWriter riw = new RandomIndexWriter(Random(), dir, NewIndexWriterConfig(TEST_VERSION_CURRENT, new MockAnalyzer(Random())));
int docCount = AtLeast(10000);
for (int i = 0; i < docCount; i++)
{
Document doc = new Document();
doc.Add(NewField("field", "a", TextField.TYPE_NOT_STORED));
riw.AddDocument(doc);
}
riw.ForceMerge(1);
IndexReader r = riw.Reader;
riw.Dispose();
IndexSearcher s = NewSearcher(r);
BooleanQuery bq = new BooleanQuery();
bq.Add(new TermQuery(new Term("field", "a")), BooleanClause.Occur.SHOULD);
bq.Add(new TermQuery(new Term("field", "a")), BooleanClause.Occur.SHOULD);
Weight w = s.CreateNormalizedWeight(bq);
Assert.AreEqual(1, s.IndexReader.Leaves.Count);
BulkScorer scorer = w.BulkScorer(s.IndexReader.Leaves[0], false, null);
FixedBitSet hits = new FixedBitSet(docCount);
AtomicInteger end = new AtomicInteger();
Collector c = new CollectorAnonymousInnerClassHelper(this, scorer, hits, end);
while (end.Get() < docCount)
{
int inc = TestUtil.NextInt(Random(), 1, 1000);
end.AddAndGet(inc);
scorer.Score(c, end.Get());
}
Assert.AreEqual(docCount, hits.Cardinality());
r.Dispose();
dir.Dispose();
}
