-
Notifications
You must be signed in to change notification settings - Fork 3.8k
/
validity_check.go
74 lines (67 loc) · 2.08 KB
/
validity_check.go
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
// Copyright 2020 The Cockroach Authors.
//
// Use of this software is governed by the Business Source License
// included in the file licenses/BSL.txt.
//
// As of the Change Date specified in that file, in accordance with
// the Business Source License, use of this software will be governed
// by the Apache License, Version 2.0, included in the file
// licenses/APL.txt.
package geomfn
import (
"github.com/cockroachdb/cockroach/pkg/geo"
"github.com/cockroachdb/cockroach/pkg/geo/geos"
)
// ValidDetail contains information about the validity of a geometry.
type ValidDetail struct {
IsValid bool
// Reason is only populated if IsValid = true.
Reason string
// InvalidLocation is only populated if IsValid = true.
InvalidLocation *geo.Geometry
}
// IsValid returns whether the given Geometry is valid.
func IsValid(g *geo.Geometry) (bool, error) {
isValid, err := geos.IsValid(g.EWKB())
if err != nil {
return false, err
}
return isValid, nil
}
// IsValidReason returns the reasoning for whether the Geometry is valid or invalid.
func IsValidReason(g *geo.Geometry) (string, error) {
reason, err := geos.IsValidReason(g.EWKB())
if err != nil {
return "", err
}
return reason, nil
}
// IsValidDetail returns information about the validity of a Geometry.
// It takes in a flag parameter which behaves the same as the GEOS module, where 1
// means that self-intersecting rings forming holes are considered valid.
func IsValidDetail(g *geo.Geometry, flags int) (ValidDetail, error) {
isValid, reason, locEWKB, err := geos.IsValidDetail(g.EWKB(), flags)
if err != nil {
return ValidDetail{}, err
}
var loc *geo.Geometry
if len(locEWKB) > 0 {
loc, err = geo.ParseGeometryFromEWKB(locEWKB)
if err != nil {
return ValidDetail{}, err
}
}
return ValidDetail{
IsValid: isValid,
Reason: reason,
InvalidLocation: loc,
}, nil
}
// MakeValid returns a valid form of the given Geometry.
func MakeValid(g *geo.Geometry) (*geo.Geometry, error) {
validEWKB, err := geos.MakeValid(g.EWKB())
if err != nil {
return nil, err
}
return geo.ParseGeometryFromEWKB(validEWKB)
}