From 53c24b1a142e4c27cb3f6c523fa1ba73d2d3cfb4 Mon Sep 17 00:00:00 2001
From: Mikolaj Wielgus <wielgusmikolaj@gmail.com>
Date: Mon, 4 Sep 2023 04:51:27 +0200
Subject: [PATCH] Remove created dot if drawing new seg or bend fails

---
 src/draw.rs | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/src/draw.rs b/src/draw.rs
index 54abd1d..7d18551 100644
--- a/src/draw.rs
+++ b/src/draw.rs
@@ -154,9 +154,16 @@ impl<'a> Draw<'a> {
             .add_dot(self.layout.primitive(head.dot).weight())?;
         let net = self.layout.primitive(head.dot).weight().net;
 
-        let bend = self
+        let bend = match self
             .layout
-            .add_bend(head.dot, bend_to, around, BendWeight { net, cw })?;
+            .add_bend(head.dot, bend_to, around, BendWeight { net, cw })
+        {
+            Err(err) => {
+                self.layout.remove(bend_to);
+                return Err(err);
+            }
+            Ok(bend) => bend,
+        };
         Ok(Head {
             dot: bend_to,
             segbend: Some(Segbend { bend, dot, seg }),
@@ -192,9 +199,16 @@ impl<'a> Draw<'a> {
                 r: width / 2.0,
             },
         })?;
-        let seg = self
+        let seg = match self
             .layout
-            .add_seg(head.dot, to_index, SegWeight { net, width })?;
+            .add_seg(head.dot, to_index, SegWeight { net, width })
+        {
+            Err(err) => {
+                self.layout.remove(to_index);
+                return Err(err);
+            }
+            Ok(bend) => bend,
+        };
         Ok((
             Head {
                 dot: to_index,