update plot script for debugging

tests/plot/plot.py

@@ -46,13 +46,28 @@ def main(polygons_file, figure):
 
     num_edges_children = 4
     num_nodes_children = 4
-    tree = polygons.build_tree(ps, num_edges_children, num_nodes_children)
+    tree = polygons.build_search_tree(ps, num_edges_children, num_nodes_children)
+
+    points = []
+
+    # num_points = 50000
+    # points = [
+    #     (random.uniform(xmin, xmax), random.uniform(ymin, ymax))
+    #     for _ in range(num_points)
+    # ]
+
+    # add random points which have the same y-coordinate as the vertices of the polygons
+    for polygon in ps:
+        for _, y in polygon:
+            for _x in range(500):
+                points.append((random.uniform(xmin, xmax), y))
+
+    # add more random points
+    for _y in range(20):
+        y = random.uniform(ymin, ymax)
+        for _x in range(500):
+            points.append((random.uniform(xmin, xmax), y))
 
-    num_points = 50000
-    points = [
-        (random.uniform(xmin, xmax), random.uniform(ymin, ymax))
-        for _ in range(num_points)
-    ]
     inside = polygons.points_are_inside(tree, points)
 
     xs_inside = []

tests/plot/requirements.txt

@@ -1,3 +1,2 @@
 matplotlib
-black
 click

tests/plot/run.sh

@@ -1,3 +1,3 @@
 #!/usr/bin/env bash
 
-python plot.py --polygons-file ../case-1/islands.txt --figure example.png
+python plot.py --polygons-file ../case-2/boundary.txt --figure example.png