tommy/libpostal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[rtree] Language polygon index returns polygons from most specific admin level to least specific

Browse Source
This commit is contained in:
Al
2015-07-13 00:58:47 -04:00
parent 86fe289320
commit fcff210d77
1 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
scripts/geodata/polygons/language_polys.py
View File

@@ -56,15 +56,22 @@ class LanguagePolygonIndex(RTreePolygonIndex):
admin_level = properties['qs_level'] admin_level = properties['qs_level']
level_num = None
if admin_level == 'adm1': if admin_level == 'adm1':
name_key = 'qs_a1' name_key = 'qs_a1'
code_key = 'qs_a1_lc' code_key = 'qs_a1_lc'
level_num = 1
elif admin_level == 'adm1_region': elif admin_level == 'adm1_region':
name_key = 'qs_a1r' name_key = 'qs_a1r'
code_key = 'qs_a1r_lc' code_key = 'qs_a1r_lc'
elif admin_level != 'adm0': level_num = 1
elif admin_level == 'adm0':
level_num = 0
else:
continue continue
assert level_num is not None
if admin_level != 'adm0': if admin_level != 'adm0':
admin1 = properties.get(name_key) admin1 = properties.get(name_key)
admin1_code = properties.get(code_key) admin1_code = properties.get(code_key)
@@ -91,6 +98,7 @@ class LanguagePolygonIndex(RTreePolygonIndex):
properties['languages'] = [{'lang': lang, 'default': default} properties['languages'] = [{'lang': lang, 'default': default}
for lang, default in languages] for lang, default in languages]
properties['admin_level'] = level_num
poly_type = rec['geometry']['type'] poly_type = rec['geometry']['type']
if poly_type == 'Polygon': if poly_type == 'Polygon':
@@ -110,3 +118,11 @@ class LanguagePolygonIndex(RTreePolygonIndex):
i += 1 i += 1
return index return index
def admin_level(self, i):
props, p = self.polygons[i]
return props['admin_level']
def get_candidate_polygons(self, lat, lon):
candidates = OrderedDict.fromkeys(self.index.intersection((lon, lat, lon, lat))).keys()
return sorted(candidates, key=self.admin_level, reverse=True)