From fc60ef8c898de02d3634d2364b2ff8adea8eaa99 Mon Sep 17 00:00:00 2001
From: Sven <sven.reiche@psi.ch>
Date: Thu, 9 Jul 2026 16:19:19 +0200
Subject: [PATCH] Adding ability of branching

---
 main.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/main.py b/main.py
index e361797..539875b 100644
--- a/main.py
+++ b/main.py
@@ -275,7 +275,11 @@ class BeamlineEditor(QtWidgets.QMainWindow, Ui_BeamlineGUI):
                 self.elementDB[ele].Prefix=name
                 if ele in self.branchingID.keys():
                     if self.branchingID[ele] is True:
+                        self.elementDB[ele].angle=self.elementDB[ele].design_angle
+                        line.LengthRes=0  # needs to shorten the line container
                         return True
+                    else:
+                        self.elementDB[ele].angle=0
         if secondBranch:
             if tree.secondary:
                 self.unwrap(name, tree.secondary.node_id, line, case_num)