You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
599 B
21 lines
599 B
import numpy as np
|
|
import os
|
|
import csv
|
|
path = os.path.dirname(os.path.abspath(__file__))
|
|
csv_file_name = path + '/rg_cities1000.csv'
|
|
|
|
# right hand drive is the same as left hand traffic
|
|
LHT_COUNTRIES = ['AU', 'IN', 'IE', 'JP', 'MU', 'MY', 'NZ', 'UK', 'ZA']
|
|
|
|
|
|
def get_city(lat, lon):
|
|
cities = np.array(list(csv.reader(open(csv_file_name))))[1:]
|
|
positions = cities[:,:2].astype(np.float32)
|
|
idx = np.argmin(np.linalg.norm((positions - np.array([lat, lon])), axis=1))
|
|
return cities[idx]
|
|
|
|
|
|
def is_lht(lat, lon):
|
|
city = get_city(lat, lon)
|
|
country = city[-1]
|
|
return country in LHT_COUNTRIES
|
|
|