forked from TrueCloudLab/frostfs-node
[#316] locode: Define UN/LOCODE table data types
Define the data types needed to work with LOCODE's in NeoFS (country code, location code, coordinates). Implement string parsers for new types. Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
parent
137ef25d3e
commit
0be35859ed
4 changed files with 268 additions and 0 deletions
193
pkg/util/locode/column/coordinates.go
Normal file
193
pkg/util/locode/column/coordinates.go
Normal file
|
@ -0,0 +1,193 @@
|
||||||
|
package locodecolumn
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/util/locode"
|
||||||
|
"github.com/pkg/errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
minutesDigits = 2
|
||||||
|
hemisphereSymbols = 1
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
latDegDigits = 2
|
||||||
|
lngDegDigits = 3
|
||||||
|
)
|
||||||
|
|
||||||
|
type coordinateCode struct {
|
||||||
|
degDigits int
|
||||||
|
value []uint8
|
||||||
|
}
|
||||||
|
|
||||||
|
// LongitudeCode represents the value of the longitude
|
||||||
|
// of the location conforming to UN/LOCODE specification.
|
||||||
|
type LongitudeCode coordinateCode
|
||||||
|
|
||||||
|
// LongitudeHemisphere represents hemisphere of the earth
|
||||||
|
// // along the Greenwich meridian.
|
||||||
|
type LongitudeHemisphere [hemisphereSymbols]uint8
|
||||||
|
|
||||||
|
// LatitudeCode represents the value of the latitude
|
||||||
|
// of the location conforming to UN/LOCODE specification.
|
||||||
|
type LatitudeCode coordinateCode
|
||||||
|
|
||||||
|
// LatitudeHemisphere represents hemisphere of the earth
|
||||||
|
// along the equator.
|
||||||
|
type LatitudeHemisphere [hemisphereSymbols]uint8
|
||||||
|
|
||||||
|
func coordinateFromString(s string, degDigits int, hemisphereAlphabet []uint8) (*coordinateCode, error) {
|
||||||
|
if len(s) != degDigits+minutesDigits+hemisphereSymbols {
|
||||||
|
return nil, locode.ErrInvalidString
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := range s[:degDigits+minutesDigits] {
|
||||||
|
if !isDigit(s[i]) {
|
||||||
|
return nil, locode.ErrInvalidString
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
loop:
|
||||||
|
for _, sym := range s[degDigits+minutesDigits:] {
|
||||||
|
for j := range hemisphereAlphabet {
|
||||||
|
if hemisphereAlphabet[j] == uint8(sym) {
|
||||||
|
continue loop
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, locode.ErrInvalidString
|
||||||
|
}
|
||||||
|
|
||||||
|
return &coordinateCode{
|
||||||
|
degDigits: degDigits,
|
||||||
|
value: []uint8(s),
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// LongitudeFromString parses string and returns location's longitude.
|
||||||
|
func LongitudeFromString(s string) (*LongitudeCode, error) {
|
||||||
|
cc, err := coordinateFromString(s, lngDegDigits, []uint8{'W', 'E'})
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return (*LongitudeCode)(cc), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// LatitudeFromString parses string and returns location's latitude.
|
||||||
|
func LatitudeFromString(s string) (*LatitudeCode, error) {
|
||||||
|
cc, err := coordinateFromString(s, latDegDigits, []uint8{'N', 'S'})
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return (*LatitudeCode)(cc), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cc *coordinateCode) degrees() []uint8 {
|
||||||
|
return cc.value[:cc.degDigits]
|
||||||
|
}
|
||||||
|
|
||||||
|
// Degrees returns longitude's degrees.
|
||||||
|
func (lc *LongitudeCode) Degrees() (l [lngDegDigits]uint8) {
|
||||||
|
copy(l[:], (*coordinateCode)(lc).degrees())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Degrees returns latitude's degrees.
|
||||||
|
func (lc *LatitudeCode) Degrees() (l [latDegDigits]uint8) {
|
||||||
|
copy(l[:], (*coordinateCode)(lc).degrees())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cc *coordinateCode) minutes() (mnt [minutesDigits]uint8) {
|
||||||
|
for i := 0; i < minutesDigits; i++ {
|
||||||
|
mnt[i] = cc.value[cc.degDigits+i]
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Minutes returns longitude's minutes.
|
||||||
|
func (lc *LongitudeCode) Minutes() [minutesDigits]uint8 {
|
||||||
|
return (*coordinateCode)(lc).minutes()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Minutes returns latitude's minutes.
|
||||||
|
func (lc *LatitudeCode) Minutes() [minutesDigits]uint8 {
|
||||||
|
return (*coordinateCode)(lc).minutes()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Hemisphere returns longitude's hemisphere code.
|
||||||
|
func (lc *LongitudeCode) Hemisphere() LongitudeHemisphere {
|
||||||
|
return (*coordinateCode)(lc).hemisphere()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Hemisphere returns latitude's hemisphere code.
|
||||||
|
func (lc *LatitudeCode) Hemisphere() LatitudeHemisphere {
|
||||||
|
return (*coordinateCode)(lc).hemisphere()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cc *coordinateCode) hemisphere() (h [hemisphereSymbols]uint8) {
|
||||||
|
for i := 0; i < hemisphereSymbols; i++ {
|
||||||
|
h[i] = cc.value[cc.degDigits+minutesDigits+i]
|
||||||
|
}
|
||||||
|
|
||||||
|
return h
|
||||||
|
}
|
||||||
|
|
||||||
|
// North returns true for the northern hemisphere.
|
||||||
|
func (h LatitudeHemisphere) North() bool {
|
||||||
|
return h[0] == 'W'
|
||||||
|
}
|
||||||
|
|
||||||
|
// East returns true for the eastern hemisphere.
|
||||||
|
func (h LongitudeHemisphere) East() bool {
|
||||||
|
return h[0] == 'E'
|
||||||
|
}
|
||||||
|
|
||||||
|
// Coordinates represents coordinates of the location from UN/LOCODE table.
|
||||||
|
type Coordinates struct {
|
||||||
|
lat *LatitudeCode
|
||||||
|
|
||||||
|
lng *LongitudeCode
|
||||||
|
}
|
||||||
|
|
||||||
|
// Latitude returns location's latitude.
|
||||||
|
func (c *Coordinates) Latitude() *LatitudeCode {
|
||||||
|
return c.lat
|
||||||
|
}
|
||||||
|
|
||||||
|
// Longitude returns location's longitude.
|
||||||
|
func (c *Coordinates) Longitude() *LongitudeCode {
|
||||||
|
return c.lng
|
||||||
|
}
|
||||||
|
|
||||||
|
// CoordinatesFromString parses string and returns location's coordinates.
|
||||||
|
func CoordinatesFromString(s string) (*Coordinates, error) {
|
||||||
|
if len(s) == 0 {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
strs := strings.Split(s, " ")
|
||||||
|
if len(strs) != 2 {
|
||||||
|
return nil, locode.ErrInvalidString
|
||||||
|
}
|
||||||
|
|
||||||
|
lat, err := LatitudeFromString(strs[0])
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "could not parse latitude")
|
||||||
|
}
|
||||||
|
|
||||||
|
lng, err := LongitudeFromString(strs[1])
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "could not parse longitude")
|
||||||
|
}
|
||||||
|
|
||||||
|
return &Coordinates{
|
||||||
|
lat: lat,
|
||||||
|
lng: lng,
|
||||||
|
}, nil
|
||||||
|
}
|
33
pkg/util/locode/column/country.go
Normal file
33
pkg/util/locode/column/country.go
Normal file
|
@ -0,0 +1,33 @@
|
||||||
|
package locodecolumn
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/util/locode"
|
||||||
|
)
|
||||||
|
|
||||||
|
const countryCodeLen = 2
|
||||||
|
|
||||||
|
// CountryCode represents ISO 3166 alpha-2 Country Code.
|
||||||
|
type CountryCode [countryCodeLen]uint8
|
||||||
|
|
||||||
|
// Symbols returns digits of the country code.
|
||||||
|
func (cc *CountryCode) Symbols() [countryCodeLen]uint8 {
|
||||||
|
return *cc
|
||||||
|
}
|
||||||
|
|
||||||
|
// CountryCodeFromString parses string and returns country code.
|
||||||
|
func CountryCodeFromString(s string) (*CountryCode, error) {
|
||||||
|
if len(s) != countryCodeLen {
|
||||||
|
return nil, locode.ErrInvalidString
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := range s {
|
||||||
|
if !isUpperAlpha(s[i]) {
|
||||||
|
return nil, locode.ErrInvalidString
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
cc := CountryCode{}
|
||||||
|
copy(cc[:], s)
|
||||||
|
|
||||||
|
return &cc, nil
|
||||||
|
}
|
33
pkg/util/locode/column/location.go
Normal file
33
pkg/util/locode/column/location.go
Normal file
|
@ -0,0 +1,33 @@
|
||||||
|
package locodecolumn
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/util/locode"
|
||||||
|
)
|
||||||
|
|
||||||
|
const locationCodeLen = 3
|
||||||
|
|
||||||
|
// LocationCode represents 3-character code for the location.
|
||||||
|
type LocationCode [locationCodeLen]uint8
|
||||||
|
|
||||||
|
// Symbols returns characters of the location code.
|
||||||
|
func (lc *LocationCode) Symbols() [locationCodeLen]uint8 {
|
||||||
|
return *lc
|
||||||
|
}
|
||||||
|
|
||||||
|
// LocationCodeFromString parses string and returns location code.
|
||||||
|
func LocationCodeFromString(s string) (*LocationCode, error) {
|
||||||
|
if len(s) != locationCodeLen {
|
||||||
|
return nil, locode.ErrInvalidString
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := range s {
|
||||||
|
if !isUpperAlpha(s[i]) && !isDigit(s[i]) {
|
||||||
|
return nil, locode.ErrInvalidString
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
lc := LocationCode{}
|
||||||
|
copy(lc[:], s)
|
||||||
|
|
||||||
|
return &lc, nil
|
||||||
|
}
|
9
pkg/util/locode/column/util.go
Normal file
9
pkg/util/locode/column/util.go
Normal file
|
@ -0,0 +1,9 @@
|
||||||
|
package locodecolumn
|
||||||
|
|
||||||
|
func isDigit(sym uint8) bool {
|
||||||
|
return sym >= '0' && sym <= '9'
|
||||||
|
}
|
||||||
|
|
||||||
|
func isUpperAlpha(sym uint8) bool {
|
||||||
|
return sym >= 'A' && sym <= 'Z'
|
||||||
|
}
|
Loading…
Reference in a new issue