From 2a11441a9baca65bef23a7a6222d302ae91b3437 Mon Sep 17 00:00:00 2001
From: SGR <13600406+sgr_VMT@user.noreply.gitee.com>
Date: Sat, 15 Jun 2024 02:39:36 +0000
Subject: [PATCH] fix probe

---
 idm.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/idm.py b/idm.py
index 9061296..ae84d91 100644
--- a/idm.py
+++ b/idm.py
@@ -110,6 +110,8 @@ class IDMProbe:
                                             self._handle_mcu_identify)
         self._mcu.register_config_callback(self._build_config)
         self._mcu.register_response(self._handle_idm_data, "idm_data")
+        # Probe results
+        self.results = []
         # Register webhooks
         webhooks = self.printer.lookup_object("webhooks")
         self._api_dump_helper = APIDumpHelper(self)
@@ -271,7 +273,9 @@ class IDMProbe:
 
         self._start_streaming()
         try:
-            return self._probe(speed, allow_faulty=allow_faulty)
+            epos = self._probe(speed, allow_faulty=allow_faulty)
+            self.results.append(epos)
+            return epos
         finally:
             self._stop_streaming()
 
@@ -1289,9 +1293,15 @@ class IDMProbeWrapper:
             'lift_speed': self.idm.lift_speed}
     def start_probe_session(self, gcmd):
         self.multi_probe_begin()
+        self.idm.results=[]
         return self
     def end_probe_session(self):
+        self.idm.results=[]
         self.multi_probe_end()
+    def pull_probed_results(self):
+        res = self.idm.results
+        self.idm.results = []
+        return res
 class IDMTempWrapper:
     def __init__(self, idm):
         self.idm = idm