figures for doc:build structure

Author: zhangxwww

dhg/visualization/structure/draw.py

@@ -105,7 +105,7 @@ def draw_digraph(
     pull_e_strength: float = 1.0,
     pull_center_strength: float = 1.0,
 ):
-    r"""Draw the directed graph structure. 
+    r"""Draw the directed graph structure.
 
     Args:
         ``g`` (``dhg.DiGraph``): The DHG's directed graph object.
@@ -178,7 +178,7 @@ def draw_bigraph(
     pull_v_center_strength: float = 1.0,
 ):
     r"""Draw the bipartite graph structure.
-    
+
     Args:
         ``g`` (``dhg.BiGraph``): The DHG's bipartite graph object.
         ``e_style`` (``str``): The edge style. The supported edge styles are only ``'line'``. Defaults to ``'line'``.
@@ -255,7 +255,7 @@ def draw_bigraph(
     draw_vertex(
         ax,
         np.vstack([u_coor, v_coor]),
-        u_label + v_label if u_label is not None and v_label is not None else None,
+        list(u_label) + list(v_label) if u_label is not None and v_label is not None else None,
         u_font_size + v_font_size,
         font_family,
         u_size + v_size,
@@ -287,7 +287,7 @@ def draw_hypergraph(
     pull_center_strength: float = 1.0,
 ):
     r"""Draw the hypergraph structure.
-    
+
     Args:
         ``hg`` (``dhg.Hypergraph``): The DHG's hypergraph object.
         ``e_style`` (``str``): The style of hyperedges. The available styles are only ``'circle'``. Defaults to ``'circle'``.

dhg/visualization/structure/utils.py

@@ -76,10 +76,14 @@ def draw_circle_edge(
     for eidx, lines in enumerate(line_paths):
         pathdata = []
         for line in lines:
+            if len(line) == 0:
+                continue
             start_pos, end_pos = line
             pathdata.append((Path.MOVETO, start_pos.tolist()))
             pathdata.append((Path.LINETO, end_pos.tolist()))
 
+        if len(list(zip(*pathdata))) == 0:
+            continue
         codes, verts = zip(*pathdata)
         path = Path(verts, codes)
         ax.add_patch(
@@ -165,6 +169,15 @@ def hull_layout(n_v, e_list, pos, v_size, radius_increment=0.3):
         line_path_for_e = []
         arc_path_for_e = []
 
+        if len(edge) == 1:
+            arc_path_for_e.append([pos[edge[0]], 0, 360, vertices_radius[edge[0]]])
+
+            vertices_radius[edge] += vertices_increased_radius[edge]
+
+            line_paths[e_idx] = line_path_for_e
+            arc_paths[e_idx] = arc_path_for_e
+            continue
+
         pos_in_edge = pos[edge]
         if len(edge) == 2:
             vertices_index = np.array((0, 1), dtype=np.int64)