From 316751d78044a5571cd2260ba41271a41cded990 Mon Sep 17 00:00:00 2001
From: shilapi <shilapi@outlook.com>
Date: Wed, 28 Aug 2024 21:51:50 +0800
Subject: [PATCH] :bug: fix a bug causing lags

---
 pydglab/service.py | 35 ++++++++++++++++++++++++++++++-----
 1 file changed, 30 insertions(+), 5 deletions(-)

diff --git a/pydglab/service.py b/pydglab/service.py
index f0fdd0f..221312c 100644
--- a/pydglab/service.py
+++ b/pydglab/service.py
@@ -4,6 +4,7 @@
 from pydglab.model import *
 from pydglab.uuid import *
 from pydglab.bthandler import *
+import time
 
 logger = logging.getLogger(__name__)
 
@@ -313,20 +314,44 @@ async def set_wave_sync(
         """
         self.channelA_wave_set = [(waveX_A, waveY_A, waveZ_A)]
         self.channelB_wave_set = [(waveX_B, waveY_B, waveZ_B)]
+        r = await set_wave_(
+            self.client, self.coyote.ChannelA, self.characteristics
+        ), await set_wave_(self.client, self.coyote.ChannelB, self.characteristics)
         return (waveX_A, waveY_A, waveZ_A), (waveX_B, waveY_B, waveZ_B)
 
     async def _keep_wave(self) -> None:
         """
         Don't use this function directly.
         """
+        last_time = time.time()
+        last_time_local = time.time()
         while True:
-            r = await set_wave_(
-                self.client, self.coyote.ChannelA, self.characteristics
-            ), await set_wave_(self.client, self.coyote.ChannelB, self.characteristics)
-            logger.debug(f"Set wave response: {r}")
             try:
-                await asyncio.sleep(0.1)
+                # logger.debug(f"Time elapsed: {time.time() - last_time}")
+                if time.time() - last_time >= 0.1:
+                    
+                    # Record time for loop
+                    last_time = time.time()
+                    
+                    # Refresh A and B channel by turn
+                    last_refreshing = ChannelA
+                    
+                    if self.coyote.ChannelA.strength == 0:
+                        r = (0,0,0), await set_wave_(self.client, self.coyote.ChannelB, self.characteristics)
+                    elif self.coyote.ChannelB.strength == 0:
+                        r = await set_wave_(self.client, self.coyote.ChannelA, self.characteristics), (0,0,0)
+                    else:
+                        if last_refreshing == ChannelA:
+                            r = (0,0,0), await set_wave_(self.client, self.coyote.ChannelB, self.characteristics)
+                            last_refreshing = ChannelB
+                        elif last_refreshing == ChannelB:
+                            r = await set_wave_(self.client, self.coyote.ChannelA, self.characteristics), (0,0,0)
+                            last_refreshing = ChannelA
+                    logger.debug(f"Set wave response: {r}")
+                    logger.debug(f"Time elapsed: {time.time() - last_time_local}")
+                    last_time_local = time.time()
             except asyncio.exceptions.CancelledError:
+                logger.error("Cancelled error")
                 break
         return None