fix error flood during locating
This commit is contained in:
parent
90999daa31
commit
d6402839ab
1 changed files with 4 additions and 0 deletions
|
@ -98,6 +98,8 @@ class Locator:
|
||||||
|
|
||||||
for pk, space, station_ids in good_spaces:
|
for pk, space, station_ids in good_spaces:
|
||||||
point, score = space.get_best_point(scan_values, station_ids, needed_station_id=best_station_id)
|
point, score = space.get_best_point(scan_values, station_ids, needed_station_id=best_station_id)
|
||||||
|
if point is None:
|
||||||
|
continue
|
||||||
if score < best_score:
|
if score < best_score:
|
||||||
location = CustomLocation(router.spaces[pk].level, point.x, point.y,
|
location = CustomLocation(router.spaces[pk].level, point.x, point.y,
|
||||||
permissions=permissions, icon='my_location')
|
permissions=permissions, icon='my_location')
|
||||||
|
@ -143,6 +145,8 @@ class LocatorSpace:
|
||||||
acceptable_points = tuple(
|
acceptable_points = tuple(
|
||||||
np.argwhere(self.levels[:, self.stations_lookup[needed_station_id]] > -90).ravel()
|
np.argwhere(self.levels[:, self.stations_lookup[needed_station_id]] > -90).ravel()
|
||||||
)
|
)
|
||||||
|
if not acceptable_points:
|
||||||
|
return None, None
|
||||||
scores = np.sum((self.levels[np.array(acceptable_points).reshape((-1, 1)),
|
scores = np.sum((self.levels[np.array(acceptable_points).reshape((-1, 1)),
|
||||||
stations]-values)**2, axis=1) / len(stations)
|
stations]-values)**2, axis=1) / len(stations)
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue