Make bubble wires go straight (#278)

discopy · Apr 15, 2024 · 2bac279 · 2bac279
1 parent c6815c6
commit 2bac279
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/discopy/drawing/legacy.py b/discopy/drawing/legacy.py
@@ -112,6 +112,7 @@ def add_box(scan, box, off, depth, x_pos):
         for i, obj in enumerate(box.cod.inside):
             position = (
                 pos[scan[off + i]][0] if len(box.dom) == len(box.cod)
+                else pos[scan[off + i + 1]][0] if bubble_closing
                 else x_pos - len(box.cod[1:]) / 2 + i,
                 len(diagram) - depth - .75)
             wire = Node("cod", obj=obj, i=i, depth=depth)

diff --git a/test/drawing/legacy.py b/test/drawing/legacy.py
@@ -31,6 +31,11 @@ def merge(x):
         >> merge(white) @ merge(yolk)
 
 
+@draw_and_compare("bubble-straight-wire.png", draw_type_labels=False)
+def test_draw_bubble_wires():
+    return (Ty('x') @ Box('s', Ty(), Ty())).bubble()
+
+
 @draw_and_compare(
     'spiral.png', draw_type_labels=False,
     draw_box_labels=False, aspect='equal')

diff --git a/test/src/imgs/bubble-straight-wire.png b/test/src/imgs/bubble-straight-wire.png