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

[points] Adding __iter__ and __len__ to point index

Browse Source
This commit is contained in:
Al
2016-08-19 01:01:05 -04:00
parent f43abe0846
commit 9a6ec41ce6
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/geodata/points/index.py
View File

@@ -139,6 +139,14 @@ class PointIndex(object):
point_index.load_properties(os.path.join(d, cls.PROPS_FILENAME)) point_index.load_properties(os.path.join(d, cls.PROPS_FILENAME))
return point_index return point_index
def __iter__(self):
for i in xrange(self.i):
lat, lon = self.points[i * 2], self.points[i * 2 + 1]
yield self.get_properties(i), lat, lon
def __len__(self):
return self.i
def get_candidate_points(self, latitude, longitude): def get_candidate_points(self, latitude, longitude):
code = geohash.encode(latitude, longitude)[:self.precision] code = geohash.encode(latitude, longitude)[:self.precision]
candidates = OrderedDict() candidates = OrderedDict()