ENH: group decay node in decay LaTeX rendering

ComPWA · Apr 27, 2024 · 1636635 · 1636635
1 parent 379c56c
commit 1636635
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/ampform_dpd/io.py b/src/ampform_dpd/io.py
@@ -102,7 +102,10 @@ def render_arrow(node: IsobarNode) -> str:
     to = render_arrow(obj)
     child1 = aslatex(obj.child1, **kwargs)
     child2 = aslatex(obj.child2, **kwargs)
-    return Rf"{parent} {to} {child1} {child2}"
+    latex = Rf"{parent} {to} {child1} {child2}"
+    if isinstance(obj.parent, State):
+        return latex
+    return Rf"\left({latex}\right)"
 
 
 @aslatex.register(ThreeBodyDecay)