Fixed nan gap in line segment

Author: jommme

mpldxf/backend_dxf.py

@@ -131,9 +131,6 @@ def _clip_mpl(self, gc, vertices, obj):
                 [bbox.x0, bbox.y1],
             ]
 
-            # Remove any NaN values from the vertices
-            vertices = vertices[~np.isnan(vertices).any(axis=1)]
-
             if obj == "patch":
                 vertices = ClippingRect2d(cliprect[0], cliprect[2]).clip_polyline(
                     vertices
@@ -168,6 +165,25 @@ def _draw_mpl_lwpoly(self, gc, path, transform, obj):
         dxfattribs = self._get_polyline_attribs(gc)
         vertices = path.transformed(transform).vertices
 
+        # Check if vertices hold NaN values
+        if np.isnan(vertices).any():
+            nan_rows = np.isnan(vertices).all(axis=1)
+            split_indices = np.where(nan_rows)[0]
+
+            # Split the array at NaN indices
+            list_of_split_vertices = np.split(vertices, split_indices)
+
+            # Remove NaN values from sub-arrays
+            list_of_split_vertices = [
+                arr[~np.isnan(arr).any(axis=1)] for arr in list_of_split_vertices
+            ]
+
+            for split_vertices in list_of_split_vertices:
+                self._clip_and_add_mpl_lwpoly(gc, dxfattribs, split_vertices, obj)
+        else:
+            self._clip_and_add_mpl_lwpoly(gc, dxfattribs, vertices, obj)
+
+    def _clip_and_add_mpl_lwpoly(self, gc, dxfattribs, vertices, obj):
         # clip the polygon if clip rectangle present
         if len(vertices) > 0:
             if isinstance(vertices[0][0], float or np.float64):

tests/test_backend_ezdxf.py

@@ -56,7 +56,7 @@ def test_plot_line_with_no_axis(self):
         modelspace = doc.modelspace()
         entities = list(modelspace)
         assert len(entities) == 1  # 1 line and the bounding box of the plot
-        
+
     def test_plot_line(self):
         """Test a simple line-plot command."""
         plt.gca().patch.set_visible(False)
@@ -163,7 +163,7 @@ def test_contourf(self):
         X, Y = np.meshgrid(x, y)
         Z = np.sin(np.sqrt(X**2 + Y**2))
         plt.contourf(X, Y, Z)
-        
+
         plt.savefig("tests/files/test_contourf.png")
 
         try:
@@ -172,7 +172,7 @@ def test_contourf(self):
 
         finally:
             plt.close()
-      
+
     def test_plot_with_nans(self):
         """Test a plot with NaNs."""
         plt.gca().patch.set_visible(False)
@@ -194,5 +194,5 @@ def test_plot_with_nans(self):
         modelspace = doc.modelspace()
         entities = list(modelspace)
         assert (
-            len(entities) == 1
-        )  # ideally we should have two lines (i.e. one broken line), but one interpolated line works as a hotfix
+            len(entities) == 2
+        )  # ideally we should have two lines (i.e. one broken line), but one interpolated line works as a hotfix