Skip to content

Commit 1f42140

Browse files
authored
Performance improvement (#1075)
1 parent 4bc7048 commit 1f42140

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

PathPlanning/WavefrontCPP/wavefront_coverage_path_planner.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def transform(
6464
is_visited = np.zeros_like(transform_matrix, dtype=bool)
6565
is_visited[src[0], src[1]] = True
6666
traversal_queue = [src]
67-
calculated = [(src[0] - 1) * n_cols + src[1]]
67+
calculated = set([(src[0] - 1) * n_cols + src[1]])
6868

6969
def is_valid_neighbor(g_i, g_j):
7070
return 0 <= g_i < n_rows and 0 <= g_j < n_cols \
@@ -86,7 +86,7 @@ def is_valid_neighbor(g_i, g_j):
8686
if not is_visited[ni][nj] \
8787
and ((ni - 1) * n_cols + nj) not in calculated:
8888
traversal_queue.append((ni, nj))
89-
calculated.append((ni - 1) * n_cols + nj)
89+
calculated.add((ni - 1) * n_cols + nj)
9090

9191
return transform_matrix
9292

0 commit comments

Comments
 (0)