BD02 Tests

bd_generate.py

import json
import random
import requests

AK = 'USE_YOUR_AK'

def bdconv(lat, lng):
    print('converting %f, %f...' % (lat, lng), end='')
    resp = requests.get('http://api.map.baidu.com/geoconv/v1/?coords=%f,%f&from=3&to=5&ak=%s' % (lng, lat, AK))
    r = json.loads(resp.text)
    print('status', r['status'])
    return (r['result'][0]['y'], r['result'][0]['x'])

with open('bd_test.csv', 'w') as f:
    f.write('GCJ_LAT,GCJ_LNG,BD_LAT,BD_LNG\n')
    for i in range(500):
        lat, lng = random.uniform(25.3, 36.4), random.uniform(108.6, 118.5)
        blat, blng = bdconv(lat, lng)
        f.write('%f,%f,%f,%f\n' % (lat, lng, blat, blng))

bd_test.py

from eviltransform import bd2gcj, gcj2bd
from math import sqrt

def calc_encryption_error(glat, glng, blat, blng):
    rblat, rblng = gcj2bd(glat, glng)
    return ((rblat - blat) ** 2, (rblng - blng) ** 2)

def calc_decryption_error(glat, glng, blat, blng):
    rglat, rglng = bd2gcj(blat, blng)
    return ((rglat - glat) ** 2, (rglng - glng) ** 2)

with open('bd_testdata.csv') as f:
    data = list(map(lambda row: list(map(float, row.split(','))),
        f.read().splitlines()[1:]))

latEE, lngEE, latDE, lngDE, N = 0, 0, 0, 0, len(data)
for entry in data:
    lat_eerr, lng_eerr = calc_encryption_error(*entry)
    lat_derr, lng_derr = calc_decryption_error(*entry)
    latEE += lat_eerr
    lngEE += lng_eerr
    latDE += lat_derr
    lngDE += lng_derr

print('Encryption: Lat RMSE %f, Lng RMSE %f; Decryption: Lat RMSE %f, Lng RMSE %f' % (sqrt(latEE / N), sqrt(lngEE / N), sqrt(latDE / N), sqrt(lngDE / N)))

bd_testdata.csv