Temperature Accretech prober: Updated accretech_uf library from wafer…

… prober driver

src/Temperature-Accretech_UFseries/libs/accretech_uf.py

@@ -34,10 +34,12 @@
 
 import inspect
 from datetime import datetime
+import time
 
 import pysweepme.Ports
-from pysweepme.ErrorMessage import debug
+from pysweepme.ErrorMessage import debug, error
 from pyvisa import constants
+import pyvisa
 
 try:
     from pysweepme.UserInterface import get_input
@@ -255,7 +257,8 @@ def wait_until_status_byte(self, stb_success: int | tuple, timeout: float = 5.0)
         stb = None
         while stb not in stb_success:
             stb = self.acquire_status_byte(timeout)
-
+            if stb == 99:  # Abnormal end always indicate that we can break here
+                break
         return stb
 
     def acquire_status_byte(self, timeout: float = 10.0) -> int:
@@ -276,7 +279,19 @@ def acquire_status_byte(self, timeout: float = 10.0) -> int:
                 print("-->", now.strftime("%H:%M:%S"), "Function:", function_calling_name)
 
         # Wait for the SRQ event to occur
-        response = self.port.port.wait_on_event(self.event_type, int(timeout * 1000))  # conversion from s to ms
+        starttime = time.time()
+        while True: 
+            if time.time()-starttime < timeout:
+                try:
+                    # response = self.port.port.wait_on_event(self.event_type, int(timeout * 1000))  # conversion from s to ms
+                    response = self.port.port.wait_on_event(self.event_type, int(1000))  # waiting 1000 ms
+                    break
+                except pyvisa.errors.VisaIOError:
+                    # Timeout error
+                    pass
+            else:
+                msg = "Timeout reached during waiting for status byte"
+                raise Exception(msg)
 
         # The event mechanism was changed after pyvisa 1.9.0 and the WaitResponse structure is different
         if hasattr(response, "event"):
@@ -634,15 +649,15 @@ def unload(self) -> int:
             # 71
         """
         self.port.write("U")
-        return self.wait_until_status_byte(71, timeout=180.0)
+        return self.wait_until_status_byte(71, timeout=300.0)
 
     def unload_all_wafers(self) -> int:
         """Returns:
         int: status byte
             # 94
         """
         self.port.write("U0")
-        return self.wait_until_status_byte(94, timeout=30.0)
+        return self.wait_until_status_byte((71, 94), timeout=300.0)
 
     def unload_to_inspection_tray(self) -> int:
         """Returns:
@@ -653,7 +668,9 @@ def unload_to_inspection_tray(self) -> int:
         return self.wait_until_status_byte(71, timeout=120.0)
 
     def load_specified_wafer(self, cassette, slot) -> int:
-        """This command can be used to load a wafer from a cassette. It indirectly starts a lot process and is not used with 'start'.
+        """This command can be used to load a wafer from a cassette.
+
+        It indirectly starts a lot process and is not used with 'start'.
 
         You can terminate the lot process by using cassette = 9 and slot = 99
 
@@ -683,7 +700,7 @@ def load_specified_wafer(self, cassette, slot) -> int:
             raise Exception(msg)
 
         self.port.write("j2%i%02d" % (int(cassette), int(slot)))
-        return self.wait_until_status_byte((70, 94), timeout=600.0)
+        return self.wait_until_status_byte((70, 94), timeout=300.0)
 
     def preload_specified_wafer(self, cassette, slot) -> int:
         """This command ("j3") can be used to preload a wafer from a cassette to the subchuck.
@@ -1020,10 +1037,34 @@ def set_chuck_temperature(self, temperature: float) -> None:
             msg = f"Accretech UF series: Temperature must be between {min_temperature} and {max_temperature} °C."
             raise Exception(msg)
 
-        print("h%04d" % (temperature * 10))
         self.port.write("h%04d" % (temperature * 10))
         answer = self.wait_until_status_byte((93, 99))
 
         correct_status = 93
         if answer != correct_status:  # Abnormal end of command  # noqa: PLR2004
-            self.raise_error(answer)
+            self.raise_error(answer)
+
+    def request_device_name_list(self, storage="c"):
+        """Requests the list of names of saved device parameter sets
+
+        Args:
+            storage: str
+
+        Returns:
+            str: list of all available device parameter set names
+        """
+
+        answer = self.query("d" + storage)
+        return answer
+
+    def request_cassette_lock_status(self):
+        """Requests cassette lock status
+
+        Returns:
+            int:
+                0: cassette is in the unlock status
+                1: cassette is in the lock status
+        """
+
+        answer = self.query("cls")
+        return int(answer)