Coverage for pygeodesy/gars.py: 98%
142 statements
« prev ^ index » next coverage.py v7.2.2, created at 2024-06-27 20:21 -0400
« prev ^ index » next coverage.py v7.2.2, created at 2024-06-27 20:21 -0400
2# -*- coding: utf-8 -*-
4u'''I{Global Area Reference System} (GARS) en-/decoding.
6Classes L{Garef} and L{GARSError} and several functions to encode,
7decode and inspect I{Global Area Reference System} (GARS) references.
9Transcoded from C++ class U{GARS
10<https://GeographicLib.SourceForge.io/C++/doc/classGeographicLib_1_1GARS.html>}
11by I{Charles Karney}. See also U{Global Area Reference System
12<https://WikiPedia.org/wiki/Global_Area_Reference_System>} and U{NGA (GARS)
13<https://Earth-Info.NGA.mil/GandG/coordsys/grids/gars.html>}.
14'''
16# from pygeodesy.basics import isstr # from .named
17from pygeodesy.constants import _off90, _1_over, _0_5, \
18 _1_0 # PYCHOK used!
19from pygeodesy.dms import parse3llh, Fmt # parseDMS2
20from pygeodesy.errors import _ValueError, _xkwds, _xStrError
21from pygeodesy.interns import NN, _0to9_, _AtoZnoIO_, _COMMA_
22from pygeodesy.lazily import _ALL_LAZY, _ALL_OTHER
23from pygeodesy.named import _name__, isstr, Property_RO
24from pygeodesy.namedTuples import LatLon2Tuple, LatLonPrec3Tuple
25# from pygeodesy.props import Property_RO # from .named
26# from pygeodesy.streprs import Fmt # from .dms
27from pygeodesy.units import Int_, Lat, Lon, Precision_, Scalar_, Str
29from math import floor
31__all__ = _ALL_LAZY.gars
32__version__ = '24.06.15'
34_Digits = _0to9_
35_LatLen = 2
36_LatOrig = -90
37_Letters = _AtoZnoIO_
38_LonLen = 3
39_LonOrig = -180
40_MaxPrec = 2
42_MinLen = _LonLen + _LatLen
43_MaxLen = _MinLen + _MaxPrec
45_M1 = _M2 = 2
46_M3 = 3
47_M4 = _M1 * _M2 * _M3
49_LatOrig_M4 = _LatOrig * _M4
50_LatOrig_M1 = _LatOrig * _M1
51_LonOrig_M4 = _LonOrig * _M4
52_LonOrig_M1_1 = _LonOrig * _M1 - 1
54_Resolutions = _1_over(_M1), _1_over(_M1 * _M2), _1_over(_M4)
57def _2divmod2(ll, _Orig_M4):
58 x = int(floor(ll * _M4)) - _Orig_M4
59 i = (x * _M1) // _M4
60 x -= i * _M4 // _M1
61 return i, x
64def _2fll(lat, lon, *unused):
65 '''(INTERNAL) Convert lat, lon.
66 '''
67 # lat, lon = parseDMS2(lat, lon)
68 return (Lat(lat, Error=GARSError),
69 Lon(lon, Error=GARSError))
72# def _2Garef(garef):
73# '''(INTERNAL) Check or create a L{Garef} instance.
74# '''
75# if not isinstance(garef, Garef):
76# try:
77# garef = Garef(garef)
78# except (TypeError, ValueError):
79# raise _xStrError(Garef, Str, garef=garef)
80# return garef
83def _2garstr2(garef):
84 '''(INTERNAL) Check a garef string.
85 '''
86 try:
87 n, garstr = len(garef), garef.upper()
88 if n < _MinLen or n > _MaxLen \
89 or garstr[:3] == 'INV' \
90 or not garstr.isalnum():
91 raise ValueError
92 return garstr, _2Precision(n - _MinLen)
94 except (AttributeError, TypeError, ValueError) as x:
95 raise GARSError(Garef.__name__, garef, cause=x)
98def _2Precision(precision):
99 '''(INTERNAL) Return a L{Precision_} instance.
100 '''
101 return Precision_(precision, Error=GARSError, low=0, high=_MaxPrec)
104class GARSError(_ValueError):
105 '''Global Area Reference System (GARS) encode, decode or other L{Garef} issue.
106 '''
107 pass
110class Garef(Str):
111 '''Garef class, a named C{str}.
112 '''
113 # no str.__init__ in Python 3
114 def __new__(cls, cll, precision=1, **name):
115 '''New L{Garef} from an other L{Garef} instance or garef
116 C{str} or from a C{LatLon} instance or lat-/longitude C{str}.
118 @arg cll: Cell or location (L{Garef} or C{str}, C{LatLon}
119 or C{str}).
120 @kwarg precision: Optional, the desired garef resolution
121 and length (C{int} 0..2), see function
122 L{gars.encode} for more details.
123 @kwarg name: Optional C{B{name}=NN} (C{str}).
125 @return: New L{Garef}.
127 @raise RangeError: Invalid B{C{cll}} lat- or longitude.
129 @raise TypeError: Invalid B{C{cll}}.
131 @raise GARSError: INValid or non-alphanumeric B{C{cll}}.
132 '''
133 ll = p = None
135 if isinstance(cll, Garef):
136 g, p = _2garstr2(str(cll))
138 elif isstr(cll):
139 if _COMMA_ in cll:
140 ll = _2fll(*parse3llh(cll))
141 g = encode(*ll, precision=precision) # PYCHOK false
142 else:
143 g = cll.upper()
145 else: # assume LatLon
146 try:
147 ll = _2fll(cll.lat, cll.lon)
148 g = encode(*ll, precision=precision) # PYCHOK false
149 except AttributeError:
150 raise _xStrError(Garef, cll=cll) # Error=GARSError
152 self = Str.__new__(cls, g, name=_name__(name, _or_nameof=cll))
153 self._latlon = ll
154 self._precision = p
155 return self
157 @Property_RO
158 def decoded3(self):
159 '''Get this garef's attributes (L{LatLonPrec3Tuple}).
160 '''
161 lat, lon = self.latlon
162 return LatLonPrec3Tuple(lat, lon, self.precision, name=self.name)
164 @Property_RO
165 def _decoded3(self):
166 '''(INTERNAL) Initial L{LatLonPrec3Tuple}.
167 '''
168 return decode3(self)
170 @Property_RO
171 def latlon(self):
172 '''Get this garef's (center) lat- and longitude (L{LatLon2Tuple}).
173 '''
174 lat, lon = self._latlon or self._decoded3[:2]
175 return LatLon2Tuple(lat, lon, name=self.name)
177 @Property_RO
178 def precision(self):
179 '''Get this garef's precision (C{int}).
180 '''
181 p = self._precision
182 return self._decoded3.precision if p is None else p
184 def toLatLon(self, LatLon=None, **LatLon_kwds):
185 '''Return (the center of) this garef cell as an instance
186 of the supplied C{LatLon} class.
188 @kwarg LatLon: Class to use (C{LatLon} or C{None}).
189 @kwarg LatLon_kwds: Optional, additional B{C{LatLon}}
190 keyword arguments.
192 @return: This garef location as B{C{LatLon}} or if
193 C{B{LatLon} is None} as L{LatLonPrec3Tuple}.
194 '''
195 return self.decoded3 if LatLon is None else LatLon(
196 *self.latlon, **_xkwds(LatLon_kwds, name=self.name))
199def decode3(garef, center=True, **name):
200 '''Decode a C{garef} to lat-, longitude and precision.
202 @arg garef: To be decoded (L{Garef} or C{str}).
203 @kwarg center: If C{True} the center, otherwise the south-west,
204 lower-left corner (C{bool}).
206 @return: A L{LatLonPrec3Tuple}C{(lat, lon, precision)}.
208 @raise GARSError: Invalid B{C{garef}}, INValid, non-alphanumeric
209 or bad length B{C{garef}}.
210 '''
211 def _Error(i):
212 return GARSError(garef=Fmt.SQUARE(repr(garef), i))
214 def _ll(chars, g, i, j, lo, hi):
215 ll, b = 0, len(chars)
216 for i in range(i, j):
217 d = chars.find(g[i])
218 if d < 0:
219 raise _Error(i)
220 ll = ll * b + d
221 if ll < lo or ll > hi:
222 raise _Error(j)
223 return ll
225 def _ll2(lon, lat, g, i, m):
226 d = _Digits.find(g[i])
227 if d < 1 or d > m * m:
228 raise _Error(i)
229 d, r = divmod(d - 1, m)
230 lon = lon * m + r
231 lat = lat * m + (m - 1 - d)
232 return lon, lat
234 g, precision = _2garstr2(garef)
236 lon = _ll(_Digits, g, 0, _LonLen, 1, 720) + _LonOrig_M1_1
237 lat = _ll(_Letters, g, _LonLen, _MinLen, 0, 359) + _LatOrig_M1
238 if precision > 0:
239 lon, lat = _ll2(lon, lat, g, _MinLen, _M2)
240 if precision > 1:
241 lon, lat = _ll2(lon, lat, g, _MinLen + 1, _M3)
243 if center: # ll = (ll * 2 + 1) / 2
244 lon += _0_5
245 lat += _0_5
247 n = _name__(name, _or_nameof=garef)
248 r = _Resolutions[precision] # == 1.0 / unit
249 return LatLonPrec3Tuple(Lat(lat * r, Error=GARSError),
250 Lon(lon * r, Error=GARSError),
251 precision, name=n)
254def encode(lat, lon, precision=1): # MCCABE 14
255 '''Encode a lat-/longitude as a C{garef} of the given precision.
257 @arg lat: Latitude (C{degrees}).
258 @arg lon: Longitude (C{degrees}).
259 @kwarg precision: Optional, the desired C{garef} resolution
260 and length (C{int} 0..2).
262 @return: The C{garef} (C{str}).
264 @raise RangeError: Invalid B{C{lat}} or B{C{lon}}.
266 @raise GARSError: Invalid B{C{precision}}.
268 @note: The C{garef} length is M{precision + 5} and the C{garef}
269 resolution is B{30′} for B{C{precision}} 0, B{15′} for 1
270 and B{5′} for 2, respectively.
271 '''
272 def _digit(x, y, m):
273 return _Digits[m * (m - y - 1) + x + 1],
275 def _str(chars, x, n):
276 s, b = [], len(chars)
277 for i in range(n):
278 x, i = divmod(x, b)
279 s.append(chars[i])
280 return tuple(reversed(s))
282 p = _2Precision(precision)
284 lat, lon = _2fll(lat, lon)
285 lat = _off90(lat)
287 ix, x = _2divmod2(lon, _LonOrig_M4)
288 iy, y = _2divmod2(lat, _LatOrig_M4)
290 g = _str(_Digits, ix + 1, _LonLen) + _str(_Letters, iy, _LatLen)
291 if p > 0:
292 ix, x = divmod(x, _M3)
293 iy, y = divmod(y, _M3)
294 g += _digit(ix, iy, _M2)
295 if p > 1:
296 g += _digit(x, y, _M3)
298 return NN.join(g)
301def precision(res):
302 '''Determine the L{Garef} precision to meet a required (geographic)
303 resolution.
305 @arg res: The required resolution (C{degrees}).
307 @return: The L{Garef} precision (C{int} 0..2).
309 @raise ValueError: Invalid B{C{res}}.
311 @see: Function L{gars.encode} for more C{precision} details.
312 '''
313 r = Scalar_(res=res)
314 for p in range(_MaxPrec):
315 if resolution(p) <= r:
316 return p
317 return _MaxPrec
320def resolution(prec):
321 '''Determine the (geographic) resolution of a given L{Garef} precision.
323 @arg prec: The given precision (C{int}).
325 @return: The (geographic) resolution (C{degrees}).
327 @raise GARSError: Invalid B{C{prec}}.
329 @see: Function L{gars.encode} for more C{precision} details.
330 '''
331 p = Int_(prec=prec, Error=GARSError, low=-1, high=_MaxPrec + 1)
332 return _Resolutions[max(0, min(p, _MaxPrec))]
335__all__ += _ALL_OTHER(decode3, # functions
336 encode, precision, resolution)
338# **) MIT License
339#
340# Copyright (C) 2016-2024 -- mrJean1 at Gmail -- All Rights Reserved.
341#
342# Permission is hereby granted, free of charge, to any person obtaining a
343# copy of this software and associated documentation files (the "Software"),
344# to deal in the Software without restriction, including without limitation
345# the rights to use, copy, modify, merge, publish, distribute, sublicense,
346# and/or sell copies of the Software, and to permit persons to whom the
347# Software is furnished to do so, subject to the following conditions:
348#
349# The above copyright notice and this permission notice shall be included
350# in all copies or substantial portions of the Software.
351#
352# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
353# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
354# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
355# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
356# OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
357# ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
358# OTHER DEALINGS IN THE SOFTWARE.