blob: 179aac54a5b7c8cdbd63bd4ec3dac131af8f6a3b [file] [log] [blame]
using Lucene.Net.Index;
using Lucene.Net.Queries.Function;
using Lucene.Net.Queries.Function.DocValues;
using Lucene.Net.Search;
using Lucene.Net.Spatial.Queries;
using Spatial4n.Core.Shapes;
using System.Collections;
namespace Lucene.Net.Spatial.Util
{
/*
* 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>
/// A boolean <see cref="ValueSource"/> that compares a shape from a provided <see cref="ValueSource"/> with a given <see cref="IShape">Shape</see> and sees
/// if it matches a given <see cref="SpatialOperation"/> (the predicate).
///
/// @lucene.experimental
/// </summary>
public class ShapePredicateValueSource : ValueSource
{
private readonly ValueSource shapeValuesource;//the left hand side
private readonly SpatialOperation op;
private readonly IShape queryShape;//the right hand side (constant)
/// <summary>
///
/// </summary>
/// <param name="shapeValuesource">
/// Must yield <see cref="IShape"/> instances from it's ObjectVal(doc). If null
/// then the result is false. This is the left-hand (indexed) side.
/// </param>
/// <param name="op">the predicate</param>
/// <param name="queryShape">The shape on the right-hand (query) side.</param>
public ShapePredicateValueSource(ValueSource shapeValuesource, SpatialOperation op, IShape queryShape)
{
this.shapeValuesource = shapeValuesource;
this.op = op;
this.queryShape = queryShape;
}
public override string GetDescription()
{
return shapeValuesource + " " + op + " " + queryShape;
}
public override void CreateWeight(IDictionary context, IndexSearcher searcher)
{
shapeValuesource.CreateWeight(context, searcher);
}
public override FunctionValues GetValues(IDictionary context, AtomicReaderContext readerContext)
{
FunctionValues shapeValues = shapeValuesource.GetValues(context, readerContext);
return new BoolDocValuesAnonymousHelper(this, shapeValues);
}
internal class BoolDocValuesAnonymousHelper : BoolDocValues
{
private readonly ShapePredicateValueSource outerInstance;
private readonly FunctionValues shapeValues;
public BoolDocValuesAnonymousHelper(ShapePredicateValueSource outerInstance, FunctionValues shapeValues)
: base(outerInstance)
{
this.outerInstance = outerInstance;
this.shapeValues = shapeValues;
}
public override bool BoolVal(int doc)
{
IShape indexedShape = (IShape)shapeValues.ObjectVal(doc);
if (indexedShape == null)
return false;
return outerInstance.op.Evaluate(indexedShape, outerInstance.queryShape);
}
public override Explanation Explain(int doc)
{
Explanation exp = base.Explain(doc);
exp.AddDetail(shapeValues.Explain(doc));
return exp;
}
}
public override bool Equals(object o)
{
if (this == o) return true;
if (o == null || GetType() != o.GetType()) return false;
ShapePredicateValueSource that = (ShapePredicateValueSource)o;
if (!shapeValuesource.Equals(that.shapeValuesource)) return false;
if (!op.Equals(that.op)) return false;
if (!queryShape.Equals(that.queryShape)) return false;
return true;
}
public override int GetHashCode()
{
int result = shapeValuesource.GetHashCode();
result = 31 * result + op.GetHashCode();
result = 31 * result + queryShape.GetHashCode();
return result;
}
}
}