Adding upstream version 2.5.1.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
c71cb8b61d
commit
982828099e
GPG key ID:
FBB4F0E80A80222F
783 changed files with 150650 additions and 0 deletions
409
search/searcher/search_geopolygon_test.go
Normal file
409
search/searcher/search_geopolygon_test.go
Normal file
|
|
@ -0,0 +1,409 @@
|
|||
// Copyright (c) 2019 Couchbase, Inc.
|
||||
//
|
||||
// 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.
|
||||
|
||||
package searcher
|
||||
|
||||
import (
|
||||
"context"
|
||||
"reflect"
|
||||
"testing"
|
||||
|
||||
"github.com/blevesearch/bleve/v2/document"
|
||||
"github.com/blevesearch/bleve/v2/geo"
|
||||
"github.com/blevesearch/bleve/v2/index/upsidedown"
|
||||
"github.com/blevesearch/bleve/v2/index/upsidedown/store/gtreap"
|
||||
"github.com/blevesearch/bleve/v2/search"
|
||||
index "github.com/blevesearch/bleve_index_api"
|
||||
)
|
||||
|
||||
func TestSimpleGeoPolygons(t *testing.T) {
|
||||
tests := []struct {
|
||||
polygon []geo.Point
|
||||
field string
|
||||
want []string
|
||||
}{
|
||||
// test points inside a triangle & on vertices
|
||||
// r, s - inside and t,u - on vertices.
|
||||
{[]geo.Point{{Lon: 1.0, Lat: 1.0}, {Lon: 2.0, Lat: 1.9}, {Lon: 2.0, Lat: 1.0}}, "loc", []string{"r", "s", "t", "u"}},
|
||||
// non overlapping polygon for the indexed documents
|
||||
{[]geo.Point{{Lon: 3.0, Lat: 1.0}, {Lon: 4.0, Lat: 2.5}, {Lon: 3.0, Lat: 2}}, "loc", nil},
|
||||
}
|
||||
i := setupGeoPolygonPoints(t)
|
||||
indexReader, err := i.Reader()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
defer func() {
|
||||
err = indexReader.Close()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
for _, test := range tests {
|
||||
got, err := testGeoPolygonSearch(indexReader, test.polygon, test.field)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !reflect.DeepEqual(got, test.want) {
|
||||
t.Errorf("expected %v, got %v for polygon: %+v", test.want, got, test.polygon)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestRealGeoPolygons(t *testing.T) {
|
||||
tests := []struct {
|
||||
polygon []geo.Point
|
||||
field string
|
||||
want []string
|
||||
}{
|
||||
{[]geo.Point{
|
||||
{Lon: -80.881, Lat: 35.282},
|
||||
{Lon: -80.858, Lat: 35.281},
|
||||
{Lon: -80.864, Lat: 35.270},
|
||||
}, "loc", []string{"k", "l"}},
|
||||
{[]geo.Point{
|
||||
{Lon: -82.467, Lat: 36.356},
|
||||
{Lon: -78.127, Lat: 36.321},
|
||||
{Lon: -80.555, Lat: 32.932},
|
||||
{Lon: -84.807, Lat: 33.111},
|
||||
}, "loc", []string{"k", "l", "m"}},
|
||||
// same polygon vertices
|
||||
{[]geo.Point{{Lon: -82.467, Lat: 36.356}, {Lon: -82.467, Lat: 36.356}, {Lon: -82.467, Lat: 36.356}, {Lon: -82.467, Lat: 36.356}}, "loc", nil},
|
||||
// non-overlaping polygon
|
||||
{[]geo.Point{{Lon: -89.113, Lat: 36.400}, {Lon: -93.947, Lat: 36.471}, {Lon: -93.947, Lat: 34.031}}, "loc", nil},
|
||||
// concave polygon with a document `n` residing inside the hands, but outside the polygon
|
||||
{[]geo.Point{{Lon: -71.65, Lat: 42.446}, {Lon: -71.649, Lat: 42.428}, {Lon: -71.640, Lat: 42.445}, {Lon: -71.649, Lat: 42.435}}, "loc", nil},
|
||||
// V like concave polygon with a document 'p' residing inside the bottom corner
|
||||
{[]geo.Point{{Lon: -80.304, Lat: 40.740}, {Lon: -80.038, Lat: 40.239}, {Lon: -79.562, Lat: 40.786}, {Lon: -80.018, Lat: 40.328}}, "loc", []string{"p"}},
|
||||
{[]geo.Point{
|
||||
{Lon: -111.918, Lat: 33.515},
|
||||
{Lon: -111.938, Lat: 33.494},
|
||||
{Lon: -111.944, Lat: 33.481},
|
||||
{Lon: -111.886, Lat: 33.517},
|
||||
{Lon: -111.919, Lat: 33.468},
|
||||
{Lon: -111.929, Lat: 33.508},
|
||||
}, "loc", []string{"q"}},
|
||||
// real points near cb bangalore
|
||||
{[]geo.Point{
|
||||
{Lat: 12.974872, Lon: 77.607749},
|
||||
{Lat: 12.971725, Lon: 77.610110},
|
||||
{Lat: 12.972530, Lon: 77.606912},
|
||||
{Lat: 12.975112, Lon: 77.603780},
|
||||
}, "loc", []string{"amoeba", "communiti"}},
|
||||
}
|
||||
|
||||
i := setupGeoPolygonPoints(t)
|
||||
indexReader, err := i.Reader()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
defer func() {
|
||||
err = indexReader.Close()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
for _, test := range tests {
|
||||
got, err := testGeoPolygonSearch(indexReader, test.polygon, test.field)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !reflect.DeepEqual(got, test.want) {
|
||||
t.Errorf("expected %v, got %v for polygon: %+v", test.want, got, test.polygon)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestGeoRectanglePolygon(t *testing.T) {
|
||||
tests := []struct {
|
||||
polygon []geo.Point
|
||||
field string
|
||||
want []string
|
||||
}{
|
||||
{
|
||||
[]geo.Point{{Lon: 0, Lat: 0}, {Lon: 0, Lat: 50}, {Lon: 50, Lat: 50}, {Lon: 50, Lat: 0}, {Lon: 0, Lat: 0}},
|
||||
"loc",
|
||||
[]string{"a", "b", "c", "d", "e", "f", "g", "h", "i", "j"},
|
||||
},
|
||||
}
|
||||
|
||||
i := setupGeo(t)
|
||||
indexReader, err := i.Reader()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
defer func() {
|
||||
err = indexReader.Close()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
for _, test := range tests {
|
||||
got, err := testGeoPolygonSearch(indexReader, test.polygon, test.field)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !reflect.DeepEqual(got, test.want) {
|
||||
t.Errorf("expected %v, got %v for polygon: %+v", test.want, got, test.polygon)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func testGeoPolygonSearch(i index.IndexReader, polygon []geo.Point, field string) ([]string, error) {
|
||||
var rv []string
|
||||
gbs, err := NewGeoBoundedPolygonSearcher(context.TODO(), i, polygon, field, 1.0, search.SearcherOptions{})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
ctx := &search.SearchContext{
|
||||
DocumentMatchPool: search.NewDocumentMatchPool(gbs.DocumentMatchPoolSize(), 0),
|
||||
}
|
||||
docMatch, err := gbs.Next(ctx)
|
||||
for docMatch != nil && err == nil {
|
||||
rv = append(rv, string(docMatch.IndexInternalID))
|
||||
docMatch, err = gbs.Next(ctx)
|
||||
}
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return rv, nil
|
||||
}
|
||||
|
||||
func setupGeoPolygonPoints(t *testing.T) index.Index {
|
||||
analysisQueue := index.NewAnalysisQueue(1)
|
||||
i, err := upsidedown.NewUpsideDownCouch(
|
||||
gtreap.Name,
|
||||
map[string]interface{}{
|
||||
"path": "",
|
||||
},
|
||||
analysisQueue)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
err = i.Open()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc := document.NewDocument("k")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, -80.86469327, 35.2782))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc = document.NewDocument("l")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, -80.8713, 35.28138))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc = document.NewDocument("m")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, -84.25, 33.153))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc = document.NewDocument("n")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, -89.992, 35.063))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc = document.NewDocument("o")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, -71.648, 42.437))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc = document.NewDocument("p")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, -80.016, 40.314))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc = document.NewDocument("q")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, -111.919, 33.494))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc = document.NewDocument("r")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, 1.5, 1.1))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc = document.NewDocument("s")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, 2, 1.5))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc = document.NewDocument("t")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, 2.0, 1.9))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc = document.NewDocument("u")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, 2.0, 1.0))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc = document.NewDocument("amoeba")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, 77.60490, 12.97467))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
doc = document.NewDocument("communiti")
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, 77.608237, 12.97237))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
return i
|
||||
}
|
||||
|
||||
type geoPoint struct {
|
||||
title string
|
||||
lon float64
|
||||
lat float64
|
||||
}
|
||||
|
||||
// Test points inside a complex self intersecting polygon
|
||||
func TestComplexGeoPolygons(t *testing.T) {
|
||||
tests := []struct {
|
||||
polygon []geo.Point
|
||||
points []geoPoint
|
||||
field string
|
||||
want []string
|
||||
}{
|
||||
/*
|
||||
/\ /\
|
||||
/__\____/__\
|
||||
\ /
|
||||
\/
|
||||
*/
|
||||
// a, b, c - inside and d - on vertices.
|
||||
{
|
||||
[]geo.Point{
|
||||
{Lon: 6.0, Lat: 2.0},
|
||||
{Lon: 3.0, Lat: 4.0},
|
||||
{Lon: 9.0, Lat: 6.0},
|
||||
{Lon: 3.0, Lat: 8.0},
|
||||
{Lon: 6.0, Lat: 10.0},
|
||||
{Lon: 6.0, Lat: 2.0},
|
||||
},
|
||||
[]geoPoint{
|
||||
{title: "a", lon: 3, lat: 4},
|
||||
{title: "b", lon: 7, lat: 6},
|
||||
{title: "c", lon: 4, lat: 8.1},
|
||||
{title: "d", lon: 6, lat: 10.0},
|
||||
{title: "e", lon: 5, lat: 6},
|
||||
{title: "f", lon: 7, lat: 5},
|
||||
},
|
||||
"loc",
|
||||
[]string{"a", "b", "c", "d"},
|
||||
},
|
||||
/*
|
||||
____
|
||||
\ /
|
||||
\/
|
||||
/\
|
||||
/__\
|
||||
*/
|
||||
{
|
||||
[]geo.Point{
|
||||
{Lon: 7.0, Lat: 2.0},
|
||||
{Lon: 1.0, Lat: 8.0},
|
||||
{Lon: 1.0, Lat: 2.0},
|
||||
{Lon: 7.0, Lat: 8.0},
|
||||
{Lon: 7.0, Lat: 2.0},
|
||||
},
|
||||
[]geoPoint{
|
||||
{title: "a", lon: 6, lat: 5},
|
||||
{title: "b", lon: 5, lat: 5},
|
||||
{title: "c", lon: 3, lat: 5.0},
|
||||
{title: "d", lon: 2, lat: 4.0},
|
||||
{title: "e", lon: 5, lat: 3},
|
||||
{title: "f", lon: 4, lat: 4},
|
||||
},
|
||||
"loc",
|
||||
[]string{"a", "b", "c", "d"},
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
i := setupComplexGeoPolygonPoints(t, test.points)
|
||||
indexReader, err := i.Reader()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
got, err := testGeoPolygonSearch(indexReader, test.polygon, test.field)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !reflect.DeepEqual(got, test.want) {
|
||||
t.Errorf("expected %v, got %v for polygon: %+v", test.want, got, test.polygon)
|
||||
}
|
||||
err = indexReader.Close()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func setupComplexGeoPolygonPoints(t *testing.T, points []geoPoint) index.Index {
|
||||
analysisQueue := index.NewAnalysisQueue(1)
|
||||
i, err := upsidedown.NewUpsideDownCouch(
|
||||
gtreap.Name,
|
||||
map[string]interface{}{
|
||||
"path": "",
|
||||
},
|
||||
analysisQueue)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
err = i.Open()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
for _, point := range points {
|
||||
doc := document.NewDocument(point.title)
|
||||
doc.AddField(document.NewGeoPointField("loc", []uint64{}, point.lon, point.lat))
|
||||
err = i.Update(doc)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
return i
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue