99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
/*
|
|
Package s2 encapsulates Google's S2 library.
|
|
|
|
See https://s2geometry.io/
|
|
|
|
Copyright (c) 2018 - 2022 PhotoPrism UG. All rights reserved.
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under Version 3 of the GNU Affero General Public License (the "AGPL"):
|
|
<https://docs.photoprism.app/license/agpl>
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
The AGPL is supplemented by our Trademark and Brand Guidelines,
|
|
which describe how our Brand Assets may be used:
|
|
<https://photoprism.app/trademark>
|
|
|
|
Feel free to send an email to hello@photoprism.app if you have questions,
|
|
want to support our work, or just want to say hello.
|
|
|
|
Additional information can be found in our Developer Guide:
|
|
<https://docs.photoprism.app/developer-guide/>
|
|
*/
|
|
package s2
|
|
|
|
import (
|
|
gs2 "github.com/golang/geo/s2"
|
|
)
|
|
|
|
// Default cell level, see https://s2geometry.io/resources/s2cell_statistics.html.
|
|
var DefaultLevel = 21
|
|
|
|
// Token returns the S2 cell token for coordinates using the default level.
|
|
func Token(lat, lng float64) string {
|
|
return TokenLevel(lat, lng, DefaultLevel)
|
|
}
|
|
|
|
// Token returns the S2 cell token for coordinates.
|
|
func TokenLevel(lat, lng float64, level int) string {
|
|
if lat == 0.0 && lng == 0.0 {
|
|
return ""
|
|
}
|
|
|
|
if lat < -90 || lat > 90 {
|
|
return ""
|
|
}
|
|
|
|
if lng < -180 || lng > 180 {
|
|
return ""
|
|
}
|
|
|
|
l := gs2.LatLngFromDegrees(lat, lng)
|
|
return gs2.CellIDFromLatLng(l).Parent(level).ToToken()
|
|
}
|
|
|
|
// LatLng returns the coordinates for a S2 cell token.
|
|
func LatLng(token string) (lat, lng float64) {
|
|
token = NormalizeToken(token)
|
|
|
|
if len(token) < 3 {
|
|
return 0.0, 0.0
|
|
}
|
|
|
|
c := gs2.CellIDFromToken(token)
|
|
|
|
if !c.IsValid() {
|
|
return 0.0, 0.0
|
|
}
|
|
|
|
l := c.LatLng()
|
|
return l.Lat.Degrees(), l.Lng.Degrees()
|
|
}
|
|
|
|
// IsZero returns true if the coordinates are both empty.
|
|
func IsZero(lat, lng float64) bool {
|
|
return lat == 0.0 && lng == 0.0
|
|
}
|
|
|
|
// Range returns a token range for finding nearby locations.
|
|
func Range(token string, levelUp int) (min, max string) {
|
|
token = NormalizeToken(token)
|
|
|
|
c := gs2.CellIDFromToken(token)
|
|
|
|
if !c.IsValid() {
|
|
return min, max
|
|
}
|
|
|
|
// See https://s2geometry.io/resources/s2cell_statistics.html
|
|
lvl := c.Level()
|
|
|
|
parent := c.Parent(lvl - levelUp)
|
|
|
|
return parent.Prev().ChildBeginAtLevel(lvl).ToToken(), parent.Next().ChildBeginAtLevel(lvl).ToToken()
|
|
}
|