Skip to content

Compatible with Py3k, fixed cpp code warning and styles #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Compatible with Py3k, fixed cpp code warning and styles #1

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a671df2
ported to python 3.4.0
zonyitoo May 17, 2014
5fbc4e2
fixed warnings
zonyitoo May 17, 2014
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
342 changes: 177 additions & 165 deletions geohash.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,17 +2,21 @@
"""
Copyright (C) 2009 Hiroaki Kawai <[email protected]>
"""

from __future__ import absolute_import, print_function, \
division, unicode_literals

try:
import _geohash
except ImportError,e:
_geohash = None
import _geohash
except ImportError as e:
_geohash = None

__all__ = ['encode','decode','decode_exactly','bbox', 'neighbors', 'expand']
__all__ = ['encode', 'decode', 'decode_exactly', 'bbox', 'neighbors', 'expand']

_base32 = '0123456789bcdefghjkmnpqrstuvwxyz'
_base32_map = {}
for i in range(len(_base32)):
_base32_map[_base32[i]] = i
_base32_map[_base32[i]] = i
del i

# def _binstr(b, length=32):
Expand All @@ -24,181 +28,189 @@
# t.append('0')
# b=b>>1
# length-=1
#
#
# t.reverse()
# return ''.join(t)

def _encode_i2c(lat,lon,lat_length,lon_length):
precision=(lat_length+lon_length)/5
if lat_length < lon_length:
a = lon
b = lat
else:
a = lat
b = lon

boost = (0,1,4,5,16,17,20,21)
ret = ''
for i in range(precision):
ret+=_base32[(boost[a&7]+(boost[b&3]<<1))&0x1F]
t = a>>3
a = b>>2
b = t

return ret[::-1]

def _encode_i2c(lat, lon, lat_length, lon_length):
precision = (lat_length + lon_length) // 5
if lat_length < lon_length:
a = lon
b = lat
else:
a = lat
b = lon

boost = (0, 1, 4, 5, 16, 17, 20, 21)
ret = ''
for i in range(precision):
ret += _base32[(boost[a & 7] + (boost[b & 3] << 1)) & 0x1F]
t = a >> 3
a = b >> 2
b = t

return ret[::-1]


def encode(latitude, longitude, precision=12):
if latitude >= 90.0 or latitude < -90.0:
raise Exception("invalid latitude.")
while longitude < -180.0:
longitude += 360.0
while longitude >= 180.0:
longitude -= 360.0

if _geohash:
basecode=_geohash.encode(latitude,longitude)
if len(basecode)>precision:
return basecode[0:precision]
return basecode+'0'*(precision-len(basecode))

lat = latitude/180.0
lon = longitude/360.0

xprecision=precision+1
lat_length=lon_length=xprecision*5/2
if xprecision%2==1:
lon_length+=1

if lat>0:
lat = int((1<<lat_length)*lat)+(1<<(lat_length-1))
else:
lat = (1<<lat_length-1)-int((1<<lat_length)*(-lat))

if lon>0:
lon = int((1<<lon_length)*lon)+(1<<(lon_length-1))
else:
lon = (1<<lon_length-1)-int((1<<lon_length)*(-lon))

return _encode_i2c(lat,lon,lat_length,lon_length)[:precision]
if latitude >= 90.0 or latitude < -90.0:
raise Exception("invalid latitude.")
while longitude < -180.0:
longitude += 360.0
while longitude >= 180.0:
longitude -= 360.0

if _geohash:
basecode = _geohash.encode(latitude, longitude)
if len(basecode) > precision:
return basecode[0:precision]
return basecode + '0' * (precision - len(basecode))

lat = latitude / 180.0
lon = longitude / 360.0

xprecision = precision + 1
lat_length = lon_length = xprecision * 5 // 2
if xprecision % 2 == 1:
lon_length += 1

if lat > 0:
lat = int((1 << lat_length) * lat) + (1 << (lat_length - 1))
else:
lat = (1 << lat_length - 1) - int((1 << lat_length) * (-lat))

if lon > 0:
lon = int((1 << lon_length) * lon) + (1 << (lon_length - 1))
else:
lon = (1 << lon_length - 1) - int((1 << lon_length) * (-lon))

return _encode_i2c(lat, lon, lat_length, lon_length)[:precision]


def _decode_c2i(hashcode):
lon = 0
lat = 0
bit_length = 0
lat_length = 0
lon_length = 0
for i in hashcode:
t = _base32_map[i]
if bit_length%2==0:
lon = lon<<3
lat = lat<<2
lon += (t>>2)&4
lat += (t>>2)&2
lon += (t>>1)&2
lat += (t>>1)&1
lon += t&1
lon_length+=3
lat_length+=2
else:
lon = lon<<2
lat = lat<<3
lat += (t>>2)&4
lon += (t>>2)&2
lat += (t>>1)&2
lon += (t>>1)&1
lat += t&1
lon_length+=2
lat_length+=3

bit_length+=5

return (lat,lon,lat_length,lon_length)
lon = 0
lat = 0
bit_length = 0
lat_length = 0
lon_length = 0
for i in hashcode:
t = _base32_map[i]
if bit_length % 2 == 0:
lon = lon << 3
lat = lat << 2
lon += (t >> 2) & 4
lat += (t >> 2) & 2
lon += (t >> 1) & 2
lat += (t >> 1) & 1
lon += t & 1
lon_length += 3
lat_length += 2
else:
lon = lon << 2
lat = lat << 3
lat += (t >> 2) & 4
lon += (t >> 2) & 2
lat += (t >> 1) & 2
lon += (t >> 1) & 1
lat += t & 1
lon_length += 2
lat_length += 3

bit_length += 5

return (lat, lon, lat_length, lon_length)


def decode(hashcode, delta=False):
'''
decode a hashcode and get center coordinate, and distance between center and outer border
'''
if _geohash:
(lat,lon,lat_bits,lon_bits) = _geohash.decode(hashcode)
latitude_delta = 180.0/(2<<lat_bits)
longitude_delta = 360.0/(2<<lon_bits)
latitude = lat + latitude_delta
longitude = lon + longitude_delta
if delta:
return latitude,longitude,latitude_delta,longitude_delta
return latitude,longitude

(lat,lon,lat_length,lon_length) = _decode_c2i(hashcode)

lat = (lat<<1) + 1
lon = (lon<<1) + 1
lat_length += 1
lon_length += 1

latitude = 180.0*(lat-(1<<(lat_length-1)))/(1<<lat_length)
longitude = 360.0*(lon-(1<<(lon_length-1)))/(1<<lon_length)
if delta:
latitude_delta = 180.0/(1<<lat_length)
longitude_delta = 360.0/(1<<lon_length)
return latitude,longitude,latitude_delta,longitude_delta

return latitude,longitude
'''
decode a hashcode and get center coordinate, and distance between center and outer border
'''
if _geohash:
(lat, lon, lat_bits, lon_bits) = _geohash.decode(hashcode)
latitude_delta = 180.0 / (2 << lat_bits)
longitude_delta = 360.0 / (2 << lon_bits)
latitude = lat + latitude_delta
longitude = lon + longitude_delta
if delta:
return latitude, longitude, latitude_delta, longitude_delta
return latitude, longitude

(lat, lon, lat_length, lon_length) = _decode_c2i(hashcode)

lat = (lat << 1) + 1
lon = (lon << 1) + 1
lat_length += 1
lon_length += 1

latitude = 180.0 * (lat - (1 << (lat_length - 1))) / (1 << lat_length)
longitude = 360.0 * (lon - (1 << (lon_length - 1))) / (1 << lon_length)
if delta:
latitude_delta = 180.0 / (1 << lat_length)
longitude_delta = 360.0 / (1 << lon_length)
return latitude, longitude, latitude_delta, longitude_delta

return latitude, longitude


def decode_exactly(hashcode):
return decode(hashcode, True)
return decode(hashcode, True)

# hashcode operations below

## hashcode operations below

def bbox(hashcode):
'''
decode a hashcode and get north, south, east and west border.
'''
if _geohash:
(lat,lon,lat_bits,lon_bits) = _geohash.decode(hashcode)
latitude_delta = 180.0/(1<<lat_bits)
longitude_delta = 360.0/(1<<lon_bits)
return {'s':lat,'w':lon,'n':lat+latitude_delta,'e':lon+longitude_delta}

(lat,lon,lat_length,lon_length) = _decode_c2i(hashcode)
ret={}
if lat_length:
ret['n'] = 180.0*(lat+1-(1<<(lat_length-1)))/(1<<lat_length)
ret['s'] = 180.0*(lat-(1<<(lat_length-1)))/(1<<lat_length)
else: # can't calculate the half with bit shifts (negative shift)
ret['n'] = 90.0
ret['s'] = -90.0

if lon_length:
ret['e'] = 360.0*(lon+1-(1<<(lon_length-1)))/(1<<lon_length)
ret['w'] = 360.0*(lon-(1<<(lon_length-1)))/(1<<lon_length)
else: # can't calculate the half with bit shifts (negative shift)
ret['e'] = 180.0
ret['w'] = -180.0

return ret
'''
decode a hashcode and get north, south, east and west border.
'''
if _geohash:
(lat, lon, lat_bits, lon_bits) = _geohash.decode(hashcode)
latitude_delta = 180.0 / (1 << lat_bits)
longitude_delta = 360.0 / (1 << lon_bits)
return {'s': lat, 'w': lon, 'n': lat + latitude_delta, 'e': lon + longitude_delta}

(lat, lon, lat_length, lon_length) = _decode_c2i(hashcode)
ret = {}
if lat_length:
ret['n'] = 180.0 * (lat + 1 - (1 << (lat_length - 1))) / (1 << lat_length)
ret['s'] = 180.0 * (lat - (1 << (lat_length - 1))) / (1 << lat_length)
else: # can't calculate the half with bit shifts (negative shift)
ret['n'] = 90.0
ret['s'] = -90.0

if lon_length:
ret['e'] = 360.0 * (lon + 1 - (1 << (lon_length - 1))) / (1 << lon_length)
ret['w'] = 360.0 * (lon - (1 << (lon_length - 1))) / (1 << lon_length)
else: # can't calculate the half with bit shifts (negative shift)
ret['e'] = 180.0
ret['w'] = -180.0

return ret


def neighbors(hashcode):
if _geohash and len(hashcode)<25:
return _geohash.neighbors(hashcode)
(lat,lon,lat_length,lon_length) = _decode_c2i(hashcode)
ret = []
tlat = lat
for tlon in (lon-1, lon+1):
ret.append(_encode_i2c(tlat,tlon,lat_length,lon_length))

tlat = lat+1
if not tlat >> lat_length:
for tlon in (lon-1, lon, lon+1):
ret.append(_encode_i2c(tlat,tlon,lat_length,lon_length))

tlat = lat-1
if tlat >= 0:
for tlon in (lon-1, lon, lon+1):
ret.append(_encode_i2c(tlat,tlon,lat_length,lon_length))

return ret
if _geohash and len(hashcode) < 25:
return _geohash.neighbors(hashcode)
(lat, lon, lat_length, lon_length) = _decode_c2i(hashcode)
ret = []
tlat = lat
for tlon in (lon - 1, lon + 1):
ret.append(_encode_i2c(tlat, tlon, lat_length, lon_length))

tlat = lat + 1
if not tlat >> lat_length:
for tlon in (lon - 1, lon, lon + 1):
ret.append(_encode_i2c(tlat, tlon, lat_length, lon_length))

tlat = lat - 1
if tlat >= 0:
for tlon in (lon - 1, lon, lon + 1):
ret.append(_encode_i2c(tlat, tlon, lat_length, lon_length))

return ret


def expand(hashcode):
ret = neighbors(hashcode)
ret.append(hashcode)
return ret
ret = neighbors(hashcode)
ret.append(hashcode)
return ret
Loading