Geohash

geo_coding.py

# -*- coding: utf-8 -*-

import string
import geohash


base32_string = (string.digits + string.lowercase) \
    .replace('a', '').replace('i', '').replace('o', '').replace('l', '')


def convert_to_binary(latitude, longitude, length=25):
    bin_longitude = ''
    bin_latitude = ''
    static_longitude = [-180.0, 180.0]
    static_latitude = [-90.0, 90.0]
    for i in range(length):
        middle_longitude = sum(static_longitude) / 2
        if longitude < middle_longitude:
            bin_longitude += '0'
            static_longitude = [static_longitude[0], middle_longitude]
        else:
            bin_longitude += '1'
            static_longitude = [middle_longitude, static_longitude[1]]

        middle_latitude = sum(static_latitude) / 2
        if latitude < middle_latitude:
            bin_latitude += '0'
            static_latitude = [static_latitude[0], middle_latitude]
        else:
            bin_latitude += '1'
            static_latitude = [middle_latitude, static_latitude[1]]

    return bin_latitude, bin_longitude


def base32_encode(bin_latitude, bin_longitude):

    bin_geo = ''

    for i in range(len(bin_longitude)):
        bin_geo += bin_longitude[i]
        bin_geo += bin_latitude[i]

    geo_hash = ''
    for i in range(0, len(bin_geo), 5):
        geo_hash += base32_string[int(bin_geo[i:i+5], 2)]

    return geo_hash


def base32_decode(geo_hash):
    bin_string = "".join(["{:05b}".format(base32_string.index(i))
                          for i in geo_hash])

    bin_longitude = "".join([bin_string[i]
                             for i in range(0, len(bin_string), 2)])
    bin_latitude = "".join([bin_string[i]
                            for i in range(1, len(bin_string), 2)])

    return bin_latitude, bin_longitude


def convert_from_binary(bin_latitude, bin_longitude):
    static_longitude = [-180.0, 180.0]
    static_latitude = [-90.0, 90.0]
    for i in bin_longitude:
        middle_longitude = sum(static_longitude) / 2
        if i == '0':
            static_longitude = [static_longitude[0], middle_longitude]
        else:
            static_longitude = [middle_longitude, static_longitude[1]]

    middle_longitude = sum(static_longitude) / 2

    for i in bin_latitude:
        middle_latitude = sum(static_latitude) / 2
        if i == '0':
            static_latitude = [static_latitude[0], middle_latitude]
        else:
            static_latitude = [middle_latitude, static_latitude[1]]

    middle_latitude = sum(static_latitude) / 2
    return middle_latitude, middle_longitude


def geo_encode(latitude, longitude, length=25):
    return base32_encode(*convert_to_binary(latitude, longitude, length))


def geo_decode(geo_hash):
    return convert_from_binary(*base32_decode(geo_hash))


if __name__ == '__main__':
    longitude = 110.082600
    latitude = 34.477861

    length = 30
    print latitude, longitude
    bin_lat, bin_lon = convert_to_binary(latitude, longitude, length)
    print bin_lat, bin_lon
    print base32_encode(bin_lat, bin_lon)
    print base32_decode(base32_encode(bin_lat, bin_lon))
    print convert_from_binary(*base32_decode(base32_encode(bin_lat, bin_lon)))

    print geohash.encode(36.255833, 117.103367)
    print geo_encode(36.255833, 117.103367)
    print geohash.encode(36.255833, 117.103367)
    print geo_encode(36.255833, 117.103367)
    print geohash.encode(36.255833, 117.103367)
    print geo_encode(36.255833, 117.103367)

    print geohash.decode('wx0csn0ng8')
    print geo_decode('wx0csn0ng8')
    print geohash.decode('ww0k7k0et7')
    print geo_decode('ww0k7k0et7')

默认的 python bin 函数不支持负数，负数的正确写法应该是首位表示负数位，但是这样的负数表示依赖于整形的位数，下面是一个增强版的 bin 函数。

tobin = lambda x, count=8: "".join(map(lambda y:str((x>>y)&1), range(count-1, -1, -1)))