[TASK] added name to different async tasks, added pv's for periodic tune

Author: hzb-ps

src/dt4acc/core/bl/delay_execution.py

@@ -39,7 +39,7 @@ async def request_execution(self):
             self.pending_task.cancel()
 
         # Schedule a new execution task after the delay
-        self.pending_task = asyncio.create_task(self._delayed_execution())
+        self.pending_task = asyncio.create_task(self._delayed_execution(), name="delayed-execution")
 
     async def _delayed_execution(self):
         """

src/dt4acc/custom_epics/ioc/pv_setup.py

@@ -3,7 +3,6 @@
     LatticeElementPropertyID,
     DevicePropertyID,
 )
-from p4p.asLib.yacc import start
 
 from .handlers import handle_device_update, update_manager
 from ..data.constants import config, special_pvs, cavity_names
@@ -116,7 +115,7 @@ def add_pc_pvs(builder, pc_name, prefix):
     rdbk = builder.aOut(f"{pc_name}:rdbk", initial_value=start_val, PREC=2)
 
     async def handle_pc_update(device_id: str, property_id: str, value: float):
-        logger.debug("%s:%s updating setpoint val=%s", device_id, property_id, value)
+        logger.warning("%s:%s updating setpoint val=%s", device_id, property_id, value)
         r = await handle_device_update(
             device_id=device_id, property_id=property_id, value=value
         )
@@ -148,6 +147,12 @@ def initialize_orbit_pvs(builder):
     builder.aOut(f"beam:orbit:found", initial_value=0)
 
 
+def initialize_tune_pvs(builder):
+    for axis in ["x", "y"]:
+        builder.aOut(f"TUNECC:{axis}", initial_value=0.0, PREC=9)
+    builder.longOut(f"TUNECC:count", initial_value=0)
+
+
 def initialize_twiss_pvs(builder):
     """
     Initializes PVs for Twiss parameters, which describe beam optics.

src/dt4acc/custom_epics/ioc/server.py

@@ -4,6 +4,7 @@
 
 from softioc import softioc, builder, asyncio_dispatcher
 
+from ...core.utils.logger import get_logger
 from .tasks import monitor_heartbeat
 from .pv_setup import (
     initialize_power_converter_pvs,
@@ -13,9 +14,12 @@
     initialize_bpm_pvs,
     initialize_orbit_object_pvs,
     initialize_twiss_pvs,
+    initialize_tune_pvs,
     initialize_other_pvs,
 )
 
+logger = get_logger()
+
 # Create an asyncio dispatcher to handle asynchronous PV updates
 dispatcher = asyncio_dispatcher.AsyncioDispatcher()
 
@@ -39,14 +43,16 @@ def startup():
     initialize_bpm_pvs(builder)  # Initialize Beam Position Monitor PVs
     initialize_orbit_object_pvs(builder)  # Initialize PV's of the new orbit object ... collection of bpms
     initialize_twiss_pvs(builder)  # Initialize Twiss parameter PVs
+    initialize_tune_pvs(builder)
+    logger.warning("All pvs set up")
 
     # Load the database of PVs defined above into the SoftIOC server
     builder.LoadDatabase()
     # Start the SoftIOC server to handle PV interactions
     softioc.iocInit(dispatcher)
 
     # Start monitoring the heartbeat to ensure the server is running correctly
-    asyncio.create_task(monitor_heartbeat())
+    asyncio.create_task(monitor_heartbeat(), name="server-heartbeat-loop")
 
 
 def main():

src/dt4acc/custom_epics/ioc/tasks.py

@@ -1,5 +1,6 @@
 import asyncio
 import getpass
+from datetime import datetime
 
 from p4p.client.asyncio import Context
 
@@ -25,7 +26,7 @@ async def monitor_heartbeat():
             await ctx.put(f'{getpass.getuser()}:VS3P2T5R:set', 1e-3)
             await ctx.put(f'{getpass.getuser()}:VS3P2T5R:set', 0e-3)
             logger.warning("Heartbeat loop initial start or it was terminated unexpectedly. Restarting...")
-            heartbeat_task = asyncio.create_task(heartbeat_loop())
+            heartbeat_task = asyncio.create_task(heartbeat_loop(), name="heart-beat-loop")
         await asyncio.sleep(1)
 
 
@@ -36,12 +37,14 @@ async def heartbeat_loop():
     Runs indefinitely and logs any errors encountered.
     """
     while True:
+        logger.info(f"Heart beat loop executing at  {datetime.now()}")
         try:
             await asyncio.sleep(1)  # Run every second
-            # logger.warning("Heartbeat loop was running ...")
+            logger.debug(f"Heart beat loop executed at {datetime.now()}")
             await view.heart_beat()
         except asyncio.CancelledError:
             logger.warning("Heartbeat loop was cancelled.")
             break  # Exit cleanly on cancellation
         except Exception as exc:
             logger.error(f"Heartbeat encountered an error: {exc}")
+            raise exc