forked from apache/lucenenet
-
Notifications
You must be signed in to change notification settings - Fork 3
/
ValueSourceScorer.cs
113 lines (100 loc) · 3.19 KB
/
ValueSourceScorer.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
using Lucene.Net.Index;
using Lucene.Net.Search;
using Lucene.Net.Util;
namespace Lucene.Net.Queries.Function
{
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.
*/
/// <summary>
/// <see cref="Scorer"/> which returns the result of <see cref="FunctionValues.SingleVal(int)"/> as
/// the score for a document.
/// </summary>
public class ValueSourceScorer : Scorer
{
protected readonly IndexReader m_reader;
private int doc = -1;
protected readonly int m_maxDoc;
protected readonly FunctionValues m_values;
protected bool m_checkDeletes;
private readonly IBits liveDocs;
protected internal ValueSourceScorer(IndexReader reader, FunctionValues values)
: base(null)
{
this.m_reader = reader;
this.m_maxDoc = reader.MaxDoc;
this.m_values = values;
SetCheckDeletes(true);
this.liveDocs = MultiFields.GetLiveDocs(reader);
}
public virtual IndexReader Reader
{
get
{
return m_reader;
}
}
public virtual void SetCheckDeletes(bool checkDeletes)
{
this.m_checkDeletes = checkDeletes && m_reader.HasDeletions;
}
public virtual bool Matches(int doc)
{
return (!m_checkDeletes || liveDocs.Get(doc)) && MatchesValue(doc);
}
public virtual bool MatchesValue(int doc)
{
return true;
}
public override int DocID
{
get { return doc; }
}
public override int NextDoc()
{
for (; ; )
{
doc++;
if (doc >= m_maxDoc)
{
return doc = NO_MORE_DOCS;
}
if (Matches(doc))
{
return doc;
}
}
}
public override int Advance(int target)
{
// also works fine when target==NO_MORE_DOCS
doc = target - 1;
return NextDoc();
}
public override float GetScore()
{
return m_values.SingleVal(doc);
}
public override int Freq
{
get { return 1; }
}
public override long GetCost()
{
return m_maxDoc;
}
}
}