Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit 193e5a9

Browse files
ianthomas23ianthomas23
committed
Use CLOSEPOLY kind code to close tricontourf polygons
1 parent 3b3fc57 commit 193e5a9

File tree

2 files changed

+76
-12
lines changed

2 files changed

+76
-12
lines changed

‎lib/matplotlib/tests/test_triangulation.py

Lines changed: 66 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1340,3 +1340,69 @@ def test_triplot_label():
13401340
assert labels == ['label']
13411341
assert len(handles) == 1
13421342
assert handles[0] is lines
1343+
1344+
1345+
def test_tricontour_path():
1346+
x = [0, 4, 4, 0, 2]
1347+
y = [0, 0, 4, 4, 2]
1348+
triang = mtri.Triangulation(x, y)
1349+
_, ax = plt.subplots()
1350+
1351+
# Line strip from boundary to boundary
1352+
cs = ax.tricontour(triang, [1, 0, 0, 0, 0], levels=[0.5])
1353+
assert len(cs.collections) == 1
1354+
paths = cs.collections[0].get_paths()
1355+
assert len(paths) == 1
1356+
expected_vertices = [[2, 0], [1, 1], [0, 2]]
1357+
assert_array_almost_equal(paths[0].vertices, expected_vertices)
1358+
assert_array_equal(paths[0].codes, [1, 2, 2])
1359+
assert_array_almost_equal(
1360+
paths[0].to_polygons(closed_only=False), [expected_vertices])
1361+
1362+
# Closed line loop inside domain
1363+
cs = ax.tricontour(triang, [0, 0, 0, 0, 1], levels=[0.5])
1364+
assert len(cs.collections) == 1
1365+
paths = cs.collections[0].get_paths()
1366+
assert len(paths) == 1
1367+
expected_vertices = [[3, 1], [3, 3], [1, 3], [1, 1], [3, 1]]
1368+
assert_array_almost_equal(paths[0].vertices, expected_vertices)
1369+
assert_array_equal(paths[0].codes, [1, 2, 2, 2, 79])
1370+
assert_array_almost_equal(paths[0].to_polygons(), [expected_vertices])
1371+
1372+
1373+
def test_tricontourf_path():
1374+
x = [0, 4, 4, 0, 2]
1375+
y = [0, 0, 4, 4, 2]
1376+
triang = mtri.Triangulation(x, y)
1377+
_, ax = plt.subplots()
1378+
1379+
# Polygon inside domain
1380+
cs = ax.tricontourf(triang, [0, 0, 0, 0, 1], levels=[0.5, 1.5])
1381+
assert len(cs.collections) == 1
1382+
paths = cs.collections[0].get_paths()
1383+
assert len(paths) == 1
1384+
expected_vertices = [[3, 1], [3, 3], [1, 3], [1, 1], [3, 1]]
1385+
assert_array_almost_equal(paths[0].vertices, expected_vertices)
1386+
assert_array_equal(paths[0].codes, [1, 2, 2, 2, 79])
1387+
assert_array_almost_equal(paths[0].to_polygons(), [expected_vertices])
1388+
1389+
# Polygon following boundary and inside domain
1390+
cs = ax.tricontourf(triang, [1, 0, 0, 0, 0], levels=[0.5, 1.5])
1391+
assert len(cs.collections) == 1
1392+
paths = cs.collections[0].get_paths()
1393+
assert len(paths) == 1
1394+
expected_vertices = [[2, 0], [1, 1], [0, 2], [0, 0], [2, 0]]
1395+
assert_array_almost_equal(paths[0].vertices, expected_vertices)
1396+
assert_array_equal(paths[0].codes, [1, 2, 2, 2, 79])
1397+
assert_array_almost_equal(paths[0].to_polygons(), [expected_vertices])
1398+
1399+
# Polygon is outer boundary with hole
1400+
cs = ax.tricontourf(triang, [0, 0, 0, 0, 1], levels=[-0.5, 0.5])
1401+
assert len(cs.collections) == 1
1402+
paths = cs.collections[0].get_paths()
1403+
assert len(paths) == 1
1404+
expected_vertices = [[0, 0], [4, 0], [4, 4], [0, 4], [0, 0],
1405+
[1, 1], [1, 3], [3, 3], [3, 1], [1, 1]]
1406+
assert_array_almost_equal(paths[0].vertices, expected_vertices)
1407+
assert_array_equal(paths[0].codes, [1, 2, 2, 2, 79, 1, 2, 2, 2, 79])
1408+
assert_array_almost_equal(paths[0].to_polygons(), np.split(expected_vertices, [5]))

‎src/tri/_tri.cpp

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -733,6 +733,9 @@ py::tuple TriContourGenerator::contour_to_segs_and_kinds(const Contour& contour)
733733
*segs_ptr++ = point->y;
734734
*codes_ptr++ = (point == line->begin() ? MOVETO : LINETO);
735735
}
736+
737+
if (line->size() > 1)
738+
*(codes_ptr-1) = CLOSEPOLY;
736739
}
737740

738741
py::list vertices_list(1);
@@ -849,11 +852,8 @@ void TriContourGenerator::find_boundary_lines_filled(Contour& contour,
849852
lower_level, upper_level, on_upper);
850853
} while (tri_edge != start_tri_edge);
851854

852-
// Filled contour lines must not have same first and last
853-
// points.
854-
if (contour_line.size() > 1 &&
855-
contour_line.front() == contour_line.back())
856-
contour_line.pop_back();
855+
// Close polygon.
856+
contour_line.push_back(contour_line.front());
857857
}
858858
}
859859
}
@@ -872,6 +872,9 @@ void TriContourGenerator::find_boundary_lines_filled(Contour& contour,
872872
for (Boundary::size_type j = 0; j < boundary.size(); ++j)
873873
contour_line.push_back(triang.get_point_coords(
874874
triang.get_triangle_point(boundary[j])));
875+
876+
// Close polygon.
877+
contour_line.push_back(contour_line.front());
875878
}
876879
}
877880
}
@@ -904,13 +907,8 @@ void TriContourGenerator::find_interior_lines(Contour& contour,
904907
TriEdge tri_edge = triang.get_neighbor_edge(tri, edge);
905908
follow_interior(contour_line, tri_edge, false, level, on_upper);
906909

907-
if (!filled)
908-
// Non-filled contour lines must be closed.
909-
contour_line.push_back(contour_line.front());
910-
else if (contour_line.size() > 1 &&
911-
contour_line.front() == contour_line.back())
912-
// Filled contour lines must not have same first and last points.
913-
contour_line.pop_back();
910+
// Close line loop
911+
contour_line.push_back(contour_line.front());
914912
}
915913
}
916914

0 commit comments

Comments
 (0)