diff --git a/src-python/models/overlay/overlay.py b/src-python/models/overlay/overlay.py index e83a341a..0888a298 100644 --- a/src-python/models/overlay/overlay.py +++ b/src-python/models/overlay/overlay.py @@ -1,3 +1,4 @@ +import os import ctypes import time from psutil import process_iter @@ -91,7 +92,7 @@ class Overlay: self.system = openvr.init(openvr.VRApplication_Background) self.overlay = openvr.IVROverlay() self.overlay_system = openvr.IVRSystem() - self.handle = self.overlay.createOverlay("Overlay_Speaker2log", "SOverlay_Speaker2log_UI") + self.handle = self.overlay.createOverlay("Overlay_Speaker2log", "Overlay_Speaker2log_UI") self.overlay.showOverlay(self.handle) self.initialized = True @@ -116,7 +117,16 @@ class Overlay: width, height = img.size img = img.tobytes() img = (ctypes.c_char * len(img)).from_buffer_copy(img) - self.overlay.setOverlayRaw(self.handle, img, width, height, 4) + + try: + self.overlay.setOverlayRaw(self.handle, img, width, height, 4) + except Exception as e: + printLog("error:Could not update image", e) + self.initialized = False + self.reStartOverlay() + while self.initialized is False: + time.sleep(0.1) + self.overlay.setOverlayRaw(self.handle, img, width, height, 4) self.updateOpacity(self.settings["opacity"]) self.lastUpdate = time.monotonic() @@ -253,6 +263,10 @@ class Overlay: self.system = None self.initialized = False + def reStartOverlay(self): + self.shutdownOverlay() + self.startOverlay() + @staticmethod def checkSteamvrRunning() -> bool: _proc_name = "vrmonitor.exe" if os.name == "nt" else "vrmonitor"