Skip to main content

autocore_std/
lib.rs

1//! # AutoCore Standard Library
2//!
3//! The standard library for writing AutoCore control programs. This crate provides
4//! everything you need to build real-time control applications that integrate with
5//! the AutoCore server ecosystem.
6//!
7//! ## Overview
8//!
9//! AutoCore control programs run as separate processes that communicate with the
10//! autocore-server via shared memory and IPC. This library handles all the low-level
11//! details, allowing you to focus on your control logic.
12//!
13//! ```text
14//! ┌─────────────────────────┐     ┌─────────────────────────┐
15//! │   autocore-server       │     │   Your Control Program  │
16//! │                         │     │                         │
17//! │  ┌─────────────────┐    │     │  ┌─────────────────┐    │
18//! │  │ Shared Memory   │◄───┼─────┼──│ ControlRunner   │    │
19//! │  │ (GlobalMemory)  │    │     │  │                 │    │
20//! │  └─────────────────┘    │     │  │ ┌─────────────┐ │    │
21//! │                         │     │  │ │ Your Logic  │ │    │
22//! │  ┌─────────────────┐    │     │  │ └─────────────┘ │    │
23//! │  │ Tick Signal     │────┼─────┼──│                 │    │
24//! │  └─────────────────┘    │     │  └─────────────────┘    │
25//! └─────────────────────────┘     └─────────────────────────┘
26//! ```
27//!
28//! ## Quick Start
29//!
30//! 1. Create a new control project using `acctl`:
31//!    ```bash
32//!    acctl clone <server-ip> <project-name>
33//!    ```
34//!
35//! 2. Implement the [`ControlProgram`] trait:
36//!    ```ignore
37//!    use autocore_std::ControlProgram;
38//!    use autocore_std::fb::RTrig;
39//!
40//!    // GlobalMemory is generated from your project.json
41//!    mod gm;
42//!    use gm::GlobalMemory;
43//!
44//!    pub struct MyProgram {
45//!        start_button: RTrig,
46//!    }
47//!
48//!    impl MyProgram {
49//!        pub fn new() -> Self {
50//!            Self {
51//!                start_button: RTrig::new(),
52//!            }
53//!        }
54//!    }
55//!
56//!    impl ControlProgram for MyProgram {
57//!        type Memory = GlobalMemory;
58//!
59//!        fn process_tick(&mut self, mem: &mut GlobalMemory, _cycle: u64) {
60//!            // Detect rising edge on start button
61//!            if self.start_button.call(mem.inputs.start_button) {
62//!                mem.outputs.motor_running = true;
63//!                autocore_std::log::info!("Motor started!");
64//!            }
65//!        }
66//!    }
67//!    ```
68//!
69//! 3. Use the [`autocore_main!`] macro for the entry point:
70//!    ```ignore
71//!    autocore_std::autocore_main!(MyProgram, "my_project_shm", "tick");
72//!    ```
73//!
74//! ## Function Blocks (IEC 61131-3 Inspired)
75//!
76//! This library includes standard function blocks commonly used in PLC programming:
77//!
78//! - [`fb::RTrig`] - Rising edge detector (false→true transition)
79//! - [`fb::FTrig`] - Falling edge detector (true→false transition)
80//! - [`fb::Ton`] - Timer On Delay (output after delay)
81//! - [`fb::BitResetOnDelay`] - Resets a boolean after it has been true for a duration
82//! - [`fb::SimpleTimer`] - Simple one-shot timer (NOT IEC 61131-3, for imperative use)
83//! - [`fb::StateMachine`] - State machine helper with automatic timer management
84//! - [`fb::RunningAverage`] - Accumulates values and computes their arithmetic mean
85//! - [`fb::Beeper`] - Audible beeper controller with configurable beep sequences
86//! - [`fb::Heartbeat`] - Monitors a remote heartbeat counter for connection loss
87//!
88//! ### Example: Edge Detection
89//!
90//! ```
91//! use autocore_std::fb::RTrig;
92//!
93//! let mut trigger = RTrig::new();
94//!
95//! // First call with false - no edge
96//! assert_eq!(trigger.call(false), false);
97//!
98//! // Rising edge detected!
99//! assert_eq!(trigger.call(true), true);
100//!
101//! // Still true, but no edge (already high)
102//! assert_eq!(trigger.call(true), false);
103//!
104//! // Back to false
105//! assert_eq!(trigger.call(false), false);
106//!
107//! // Another rising edge
108//! assert_eq!(trigger.call(true), true);
109//! ```
110//!
111//! ### Example: Timer
112//!
113//! ```
114//! use autocore_std::fb::Ton;
115//! use std::time::Duration;
116//!
117//! let mut timer = Ton::new();
118//! let delay = Duration::from_millis(100);
119//!
120//! // Timer not enabled - output is false
121//! assert_eq!(timer.call(false, delay), false);
122//!
123//! // Enable timer - starts counting
124//! assert_eq!(timer.call(true, delay), false);
125//!
126//! // Still counting...
127//! std::thread::sleep(Duration::from_millis(50));
128//! assert_eq!(timer.call(true, delay), false);
129//! assert!(timer.et < delay); // Elapsed time < preset
130//!
131//! // After delay elapsed
132//! std::thread::sleep(Duration::from_millis(60));
133//! assert_eq!(timer.call(true, delay), true); // Output is now true!
134//! ```
135//!
136//! ## Logging
137//!
138//! Control programs can send log messages to the autocore-server for display in the
139//! web console. Logging is handled automatically when using [`ControlRunner`].
140//!
141//! ```ignore
142//! use autocore_std::log;
143//!
144//! log::trace!("Detailed trace message");
145//! log::debug!("Debug information");
146//! log::info!("Normal operation message");
147//! log::warn!("Warning condition detected");
148//! log::error!("Error occurred!");
149//! ```
150//!
151//! See the [`logger`] module for advanced configuration.
152//!
153//! ## Memory Synchronization
154//!
155//! The [`ControlRunner`] handles all shared memory synchronization automatically:
156//!
157//! 1. **Wait for tick** - Blocks until the server signals a new cycle
158//! 2. **Read inputs** - Copies shared memory to local buffer (atomic snapshot)
159//! 3. **Execute logic** - Your `process_tick` runs on the local buffer
160//! 4. **Write outputs** - Copies local buffer back to shared memory
161//!
162//! This ensures your control logic always sees a consistent view of the data,
163//! even when other processes are modifying shared memory.
164
165#![warn(missing_docs)]
166#![warn(rustdoc::missing_crate_level_docs)]
167#![doc(html_root_url = "https://docs.rs/autocore-std/3.3.0")]
168
169use anyhow::{anyhow, Result};
170use futures_util::{SinkExt, StreamExt};
171use log::LevelFilter;
172use mechutil::ipc::{CommandMessage, MessageType};
173use raw_sync::events::{Event, EventInit, EventState};
174use raw_sync::Timeout;
175use shared_memory::ShmemConf;
176use std::collections::HashMap;
177use std::sync::atomic::{fence, Ordering, AtomicBool};
178use std::sync::Arc;
179use std::time::Duration;
180use tokio_tungstenite::{connect_async, tungstenite::Message};
181
182/// UDP logger for sending log messages to autocore-server.
183///
184/// This module provides a non-blocking logger implementation that sends log messages
185/// via UDP to the autocore-server. Messages are batched and sent asynchronously to
186/// avoid impacting the control loop timing.
187///
188/// # Example
189///
190/// ```ignore
191/// use autocore_std::logger;
192/// use log::LevelFilter;
193///
194/// // Initialize the logger (done automatically by ControlRunner)
195/// logger::init_udp_logger("127.0.0.1", 39101, LevelFilter::Info, "control")?;
196///
197/// // Now you can use the log macros
198/// log::info!("System initialized");
199/// ```
200pub mod logger;
201
202// Re-export log crate for convenience - control programs can use autocore_std::log::info!() etc.
203pub use log;
204
205/// Function blocks for control programs (IEC 61131-3 inspired).
206pub mod fb;
207
208/// Interface protocols for communication between control programs and external sources.
209pub mod iface;
210
211/// Shared memory utilities for external modules.
212pub mod shm;
213
214// ============================================================================
215// Core Framework
216// ============================================================================
217
218/// Marker trait for generated GlobalMemory structs.
219///
220/// This trait is implemented by the auto-generated `GlobalMemory` struct
221/// that represents the shared memory layout. It serves as a marker for
222/// type safety in the control framework.
223///
224/// You don't need to implement this trait yourself - it's automatically
225/// implemented by the code generator.
226pub trait AutoCoreMemory {}
227
228/// Trait for detecting changes in memory structures.
229pub trait ChangeTracker {
230    /// Compare self with a previous state and return a list of changed fields.
231    /// Returns a vector of (field_name, new_value).
232    fn get_changes(&self, prev: &Self) -> Vec<(&'static str, serde_json::Value)>;
233}
234
235/// The trait that defines a control program's logic.
236///
237/// Implement this trait to create your control program. The associated `Memory`
238/// type should be the generated `GlobalMemory` struct from your project.
239///
240/// # Memory Type Requirements
241///
242/// The `Memory` type must implement `Copy` to allow efficient synchronization
243/// between shared memory and local buffers. This is automatically satisfied
244/// by the generated `GlobalMemory` struct.
245///
246/// # Lifecycle
247///
248/// 1. `initialize` is called once at startup
249/// 2. `process_tick` is called repeatedly in the control loop
250///
251/// # Example
252///
253/// ```ignore
254/// use autocore_std::ControlProgram;
255///
256/// mod gm;
257/// use gm::GlobalMemory;
258///
259/// pub struct MyController {
260///     cycle_counter: u64,
261/// }
262///
263/// impl MyController {
264///     pub fn new() -> Self {
265///         Self { cycle_counter: 0 }
266///     }
267/// }
268///
269/// impl ControlProgram for MyController {
270///     type Memory = GlobalMemory;
271///
272///     fn initialize(&mut self, mem: &mut GlobalMemory) {
273///         // Set initial output states
274///         mem.outputs.ready = true;
275///         log::info!("Controller initialized");
276///     }
277///
278///     fn process_tick(&mut self, mem: &mut GlobalMemory, cycle: u64) {
279///         self.cycle_counter = cycle;
280///
281///         // Your control logic here
282///         if mem.inputs.start && !mem.inputs.estop {
283///             mem.outputs.running = true;
284///         }
285///     }
286/// }
287/// ```
288pub trait ControlProgram {
289    /// The shared memory structure type (usually the generated `GlobalMemory`).
290    ///
291    /// Must implement `Copy` to allow efficient memory synchronization.
292    type Memory: Copy + ChangeTracker;
293
294    /// Called once when the control program starts.
295    ///
296    /// Use this to initialize output states, reset counters, or perform
297    /// any one-time setup. The default implementation does nothing.
298    ///
299    /// # Arguments
300    ///
301    /// * `mem` - Mutable reference to the shared memory. Changes are written
302    ///           back to shared memory after this method returns.
303    fn initialize(&mut self, _mem: &mut Self::Memory) {}
304
305    /// The main control loop - called once per scan cycle.
306    ///
307    /// This is where your control logic lives. Read inputs from `mem`,
308    /// perform calculations, and write outputs back to `mem`.
309    ///
310    /// # Arguments
311    ///
312    /// * `mem` - Mutable reference to a local copy of the shared memory.
313    ///           Changes made here are written back to shared memory after
314    ///           this method returns.
315    /// * `cycle` - The current cycle number (increments each tick, starting at 1).
316    ///
317    /// # Timing
318    ///
319    /// This method should complete within the scan cycle time. Long-running
320    /// operations will cause cycle overruns.
321    fn process_tick(&mut self, mem: &mut Self::Memory, cycle: u64);
322}
323
324/// Configuration for the [`ControlRunner`].
325///
326/// Specifies connection parameters, shared memory names, and logging settings.
327/// Use [`Default::default()`] for typical configurations.
328///
329/// # Example
330///
331/// ```
332/// use autocore_std::RunnerConfig;
333/// use log::LevelFilter;
334///
335/// let config = RunnerConfig {
336///     server_host: "192.168.1.100".to_string(),
337///     module_name: "my_controller".to_string(),
338///     shm_name: "my_project_shm".to_string(),
339///     tick_signal_name: "tick".to_string(),
340///     busy_signal_name: Some("busy".to_string()),
341///     log_level: LevelFilter::Debug,
342///     ..Default::default()
343/// };
344/// ```
345#[derive(Debug, Clone)]
346pub struct RunnerConfig {
347    /// Server host address (default: "127.0.0.1")
348    pub server_host: String,
349    /// WebSocket port for commands (default: 11969)
350    pub ws_port: u16,
351    /// Module name for identification (default: "control")
352    pub module_name: String,
353    /// Shared memory segment name (must match server configuration)
354    pub shm_name: String,
355    /// Name of the tick signal in shared memory (triggers each scan cycle)
356    pub tick_signal_name: String,
357    /// Optional name of the busy signal (set when cycle completes)
358    pub busy_signal_name: Option<String>,
359    /// Minimum log level to send to the server (default: Info)
360    pub log_level: LevelFilter,
361    /// UDP port for sending logs to the server (default: 39101)
362    pub log_udp_port: u16,
363}
364
365/// Default WebSocket port for autocore-server
366pub const DEFAULT_WS_PORT: u16 = 11969;
367
368impl Default for RunnerConfig {
369    fn default() -> Self {
370        Self {
371            server_host: "127.0.0.1".to_string(),
372            ws_port: DEFAULT_WS_PORT,
373            module_name: "control".to_string(),
374            shm_name: "autocore_cyclic".to_string(),
375            tick_signal_name: "tick".to_string(),
376            busy_signal_name: None,
377            log_level: LevelFilter::Info,
378            log_udp_port: logger::DEFAULT_LOG_UDP_PORT,
379        }
380    }
381}
382
383
384/// The main execution engine for control programs.
385///
386/// `ControlRunner` handles all the infrastructure required to run a control program:
387///
388/// - Reading memory layout from the server's layout file
389/// - Opening and mapping shared memory
390/// - Setting up synchronization signals
391/// - Running the real-time control loop
392/// - Sending log messages to the server
393///
394/// # Usage
395///
396/// ```ignore
397/// use autocore_std::{ControlRunner, RunnerConfig};
398///
399/// let config = RunnerConfig {
400///     shm_name: "my_project_shm".to_string(),
401///     tick_signal_name: "tick".to_string(),
402///     ..Default::default()
403/// };
404///
405/// ControlRunner::new(MyProgram::new())
406///     .config(config)
407///     .run()?;  // Blocks forever
408/// ```
409///
410/// # Control Loop
411///
412/// The runner executes a synchronous control loop:
413///
414/// 1. **Wait** - Blocks until the tick signal is set by the server
415/// 2. **Read** - Copies shared memory to a local buffer (acquire barrier)
416/// 3. **Execute** - Calls your `process_tick` method
417/// 4. **Write** - Copies local buffer back to shared memory (release barrier)
418/// 5. **Signal** - Sets the busy signal (if configured) to indicate completion
419///
420/// This ensures your code always sees a consistent snapshot of the data
421/// and that your writes are atomically visible to other processes.
422pub struct ControlRunner<P: ControlProgram> {
423    config: RunnerConfig,
424    program: P,
425}
426
427impl<P: ControlProgram> ControlRunner<P> {
428    /// Creates a new runner for the given control program.
429    ///
430    /// Uses default configuration. Call [`.config()`](Self::config) to customize.
431    ///
432    /// # Arguments
433    ///
434    /// * `program` - Your control program instance
435    ///
436    /// # Example
437    ///
438    /// ```ignore
439    /// let runner = ControlRunner::new(MyProgram::new());
440    /// ```
441    pub fn new(program: P) -> Self {
442        Self {
443            config: RunnerConfig::default(),
444            program,
445        }
446    }
447
448    /// Sets the configuration for this runner.
449    ///
450    /// # Arguments
451    ///
452    /// * `config` - The configuration to use
453    ///
454    /// # Example
455    ///
456    /// ```ignore
457    /// ControlRunner::new(MyProgram::new())
458    ///     .config(RunnerConfig {
459    ///         shm_name: "custom_shm".to_string(),
460    ///         ..Default::default()
461    ///     })
462    ///     .run()?;
463    /// ```
464    pub fn config(mut self, config: RunnerConfig) -> Self {
465        self.config = config;
466        self
467    }
468
469    /// Starts the control loop.
470    ///
471    /// This method blocks indefinitely, running the control loop until
472    /// an error occurs or the process is terminated.
473    ///
474    /// # Returns
475    ///
476    /// Returns `Ok(())` only if the loop exits cleanly (which typically
477    /// doesn't happen). Returns an error if:
478    ///
479    /// - IPC connection fails
480    /// - Shared memory cannot be opened
481    /// - Signal offsets cannot be found
482    /// - A critical error occurs during execution
483    ///
484    /// # Example
485    ///
486    /// ```ignore
487    /// fn main() -> anyhow::Result<()> {
488    ///     ControlRunner::new(MyProgram::new())
489    ///         .config(config)
490    ///         .run()
491    /// }
492    /// ```
493    pub fn run(mut self) -> Result<()> {
494        // Initialize UDP logger FIRST (before any log statements)
495        if let Err(e) = logger::init_udp_logger(
496            &self.config.server_host,
497            self.config.log_udp_port,
498            self.config.log_level,
499            "control",
500        ) {
501            eprintln!("Warning: Failed to initialize UDP logger: {}", e);
502            // Continue anyway - logging will just go nowhere
503        }
504
505        // We use a dedicated runtime for the setup phase
506        let rt = tokio::runtime::Builder::new_current_thread()
507            .enable_all()
508            .build()?;
509
510        rt.block_on(async {
511            log::info!("AutoCore Control Runner Starting...");
512
513            // 1. Connect to server via WebSocket and get layout
514            let ws_url = format!("ws://{}:{}/ws/", self.config.server_host, self.config.ws_port);
515            log::info!("Connecting to server at {}", ws_url);
516
517            let (ws_stream, _) = connect_async(&ws_url).await
518                .map_err(|e| anyhow!("Failed to connect to server at {}: {}", ws_url, e))?;
519
520            let (mut write, mut read) = ws_stream.split();
521
522            // Send gm.get_layout request
523            let request = CommandMessage::request("gm.get_layout", serde_json::Value::Null);
524            let transaction_id = request.transaction_id;
525            let request_json = serde_json::to_string(&request)?;
526
527            write.send(Message::Text(request_json)).await
528                .map_err(|e| anyhow!("Failed to send layout request: {}", e))?;
529
530            // Wait for response with matching transaction_id
531            let timeout = Duration::from_secs(10);
532            let start = std::time::Instant::now();
533            let mut layout: Option<HashMap<String, serde_json::Value>> = None;
534
535            while start.elapsed() < timeout {
536                match tokio::time::timeout(Duration::from_secs(1), read.next()).await {
537                    Ok(Some(Ok(Message::Text(text)))) => {
538                        if let Ok(response) = serde_json::from_str::<CommandMessage>(&text) {
539                            if response.transaction_id == transaction_id {
540                                if !response.success {
541                                    return Err(anyhow!("Server error: {}", response.error_message));
542                                }
543                                layout = Some(serde_json::from_value(response.data)?);
544                                break;
545                            }
546                            // Skip broadcasts and other messages
547                            if response.message_type == MessageType::Broadcast {
548                                continue;
549                            }
550                        }
551                    }
552                    Ok(Some(Ok(_))) => continue,
553                    Ok(Some(Err(e))) => return Err(anyhow!("WebSocket error: {}", e)),
554                    Ok(None) => return Err(anyhow!("Server closed connection")),
555                    Err(_) => continue, // Timeout on single read, keep trying
556                }
557            }
558
559            let layout = layout.ok_or_else(|| anyhow!("Timeout waiting for layout response"))?;
560            log::info!("Layout received with {} entries.", layout.len());
561
562            // We keep the WebSocket open for sending updates
563            // let _ = write.close().await;
564
565            // 2. Find Signal Offsets
566            let tick_offset = self.find_offset(&layout, &self.config.tick_signal_name)?;
567            let busy_offset = if let Some(name) = &self.config.busy_signal_name {
568                Some(self.find_offset(&layout, name)?)
569            } else {
570                None
571            };
572
573            // 4. Open Shared Memory
574            let shmem = ShmemConf::new().os_id(&self.config.shm_name).open()?;
575            let base_ptr = shmem.as_ptr();
576            log::info!("Shared Memory '{}' mapped.", self.config.shm_name);
577
578            // 5. Setup Pointers
579            // SAFETY: We trust the server's layout matches the generated GlobalMemory struct.
580            let gm = unsafe { &mut *(base_ptr as *mut P::Memory) };
581
582            // Get tick event from shared memory
583            log::info!("Setting up tick event at offset {} (base_ptr: {:p})", tick_offset, base_ptr);
584            let (tick_event, _) = unsafe {
585                Event::from_existing(base_ptr.add(tick_offset))
586            }.map_err(|e| anyhow!("Failed to open tick event: {:?}", e))?;
587            log::info!("Tick event ready");
588
589            // Busy signal event (optional)
590            let busy_event = busy_offset.map(|offset| {
591                unsafe { Event::from_existing(base_ptr.add(offset)) }
592                    .map(|(event, _)| event)
593                    .ok()
594            }).flatten();
595
596            // 6. Initialize local memory buffer and user program
597            // We use a local copy for the control loop to ensure:
598            // - Consistent snapshot of inputs at start of cycle
599            // - Atomic commit of outputs at end of cycle
600            // - Proper memory barriers for cross-process visibility
601            let mut local_mem: P::Memory = unsafe { std::ptr::read_volatile(gm) };
602            let mut prev_mem: P::Memory = local_mem; // Snapshot for change detection
603
604            fence(Ordering::Acquire); // Ensure we see all prior writes from other processes
605
606            self.program.initialize(&mut local_mem);
607
608            // Write back any changes from initialize
609            fence(Ordering::Release);
610            unsafe { std::ptr::write_volatile(gm, local_mem) };
611
612            // Set up signal handler for graceful shutdown
613            let running = Arc::new(AtomicBool::new(true));
614            let r = running.clone();
615            
616            // Only set handler if not already set
617            if let Err(e) = ctrlc::set_handler(move || {
618                r.store(false, Ordering::SeqCst);
619            }) {
620                log::warn!("Failed to set signal handler: {}", e);
621            }
622
623            log::info!("Entering Control Loop - waiting for first tick...");
624            let mut cycle_count: u64 = 0;
625
626            while running.load(Ordering::SeqCst) {
627                // Wait for Tick - Event-based synchronization
628                // Use a timeout (1s) to allow checking the running flag periodically
629                match tick_event.wait(Timeout::Val(Duration::from_secs(1))) {
630                    Ok(_) => {},
631                    Err(e) => {
632                        // Check for timeout
633                        let err_str = format!("{:?}", e);
634                        if err_str.contains("Timeout") {
635                            continue;
636                        }
637                        return Err(anyhow!("Tick wait failed: {:?}", e));
638                    }
639                }
640
641                if !running.load(Ordering::SeqCst) {
642                    log::info!("Shutdown signal received, exiting control loop.");
643                    break;
644                }
645
646                cycle_count += 1;
647                if cycle_count == 1 {
648                    log::info!("First tick received!");
649                }
650
651                // === INPUT PHASE ===
652                // Read all variables from shared memory into local buffer.
653                // This gives us a consistent snapshot of inputs for this cycle.
654                // Acquire fence ensures we see all writes from other processes (server, modules).
655                local_mem = unsafe { std::ptr::read_volatile(gm) };
656                
657                // Update prev_mem before execution to track changes made IN THIS CYCLE
658                // Actually, we want to know what changed in SHM relative to what we last knew,
659                // OR what WE changed relative to what we read?
660                // The user wants "writes on shared variables" to be broadcast.
661                // Typically outputs.
662                // If inputs changed (from other source), broadcasting them again is fine too.
663                // Let's capture state BEFORE execution (which is what we just read from SHM).
664                prev_mem = local_mem;
665
666                fence(Ordering::Acquire);
667
668                // === EXECUTE PHASE ===
669                // Execute user logic on the local copy.
670                // All reads/writes during process_tick operate on local_mem.
671                self.program.process_tick(&mut local_mem, cycle_count);
672
673                // === OUTPUT PHASE ===
674                // Write all variables from local buffer back to shared memory.
675                // Release fence ensures our writes are visible to other processes.
676                fence(Ordering::Release);
677                unsafe { std::ptr::write_volatile(gm, local_mem) };
678
679                // === CHANGE DETECTION & NOTIFICATION ===
680                let changes = local_mem.get_changes(&prev_mem);
681                if !changes.is_empty() {
682                    // Construct bulk write message
683                    let mut data_map = serde_json::Map::new();
684                    for (key, val) in changes {
685                        data_map.insert(key.to_string(), val);
686                    }
687                    
688                    let msg = CommandMessage::request("gm.write", serde_json::Value::Object(data_map));
689                    let msg_json = serde_json::to_string(&msg).unwrap_or_default();
690                    
691                    // Send via WebSocket (fire and forget, don't block)
692                    // Note: WebSocket send is async. We are in block_on.
693                    // We can await it. If it takes too long, it might delay the cycle.
694                    // Ideally we should spawn this? But spawn requires 'static or Arc.
695                    // For now, let's await with a very short timeout or just await.
696                    // write is Sink.
697                    if let Err(e) = write.send(Message::Text(msg_json)).await {
698                        log::error!("Failed to send updates: {}", e);
699                    }
700                }
701
702                // Signal Busy/Done event
703                if let Some(ref busy_ev) = busy_event {
704                    let _ = busy_ev.set(EventState::Signaled);
705                }
706            }
707
708            Ok(())
709        })
710    }
711
712    fn find_offset(&self, layout: &HashMap<String, serde_json::Value>, name: &str) -> Result<usize> {
713        let info = layout.get(name).ok_or_else(|| anyhow!("Signal '{}' not found in layout", name))?;
714        info.get("offset")
715            .and_then(|v| v.as_u64())
716            .map(|v| v as usize)
717            .ok_or_else(|| anyhow!("Invalid offset for '{}'", name))
718    }
719}
720
721/// Generates the standard `main` function for a control program.
722///
723/// This macro reduces boilerplate by creating a properly configured `main`
724/// function that initializes and runs your control program.
725///
726/// # Arguments
727///
728/// * `$prog_type` - The type of your control program (must implement [`ControlProgram`])
729/// * `$shm_name` - The shared memory segment name (string literal)
730/// * `$tick_signal` - The tick signal name in shared memory (string literal)
731///
732/// # Example
733///
734/// ```ignore
735/// mod gm;
736/// use gm::GlobalMemory;
737///
738/// pub struct MyProgram;
739///
740/// impl MyProgram {
741///     pub fn new() -> Self { Self }
742/// }
743///
744/// impl autocore_std::ControlProgram for MyProgram {
745///     type Memory = GlobalMemory;
746///
747///     fn process_tick(&mut self, mem: &mut GlobalMemory, _cycle: u64) {
748///         // Your logic here
749///     }
750/// }
751///
752/// // This generates the main function
753/// autocore_std::autocore_main!(MyProgram, "my_project_shm", "tick");
754/// ```
755///
756/// # Generated Code
757///
758/// The macro expands to:
759///
760/// ```ignore
761/// fn main() -> anyhow::Result<()> {
762///     let config = autocore_std::RunnerConfig {
763///         server_host: "127.0.0.1".to_string(),
764///         ws_port: autocore_std::DEFAULT_WS_PORT,
765///         module_name: "control".to_string(),
766///         shm_name: "my_project_shm".to_string(),
767///         tick_signal_name: "tick".to_string(),
768///         busy_signal_name: None,
769///         log_level: log::LevelFilter::Info,
770///         log_udp_port: autocore_std::logger::DEFAULT_LOG_UDP_PORT,
771///     };
772///
773///     autocore_std::ControlRunner::new(MyProgram::new())
774///         .config(config)
775///         .run()
776/// }
777/// ```
778#[macro_export]
779macro_rules! autocore_main {
780    ($prog_type:ty, $shm_name:expr, $tick_signal:expr) => {
781        fn main() -> anyhow::Result<()> {
782            let config = autocore_std::RunnerConfig {
783                server_host: "127.0.0.1".to_string(),
784                ws_port: autocore_std::DEFAULT_WS_PORT,
785                module_name: "control".to_string(),
786                shm_name: $shm_name.to_string(),
787                tick_signal_name: $tick_signal.to_string(),
788                busy_signal_name: None,
789                log_level: log::LevelFilter::Info,
790                log_udp_port: autocore_std::logger::DEFAULT_LOG_UDP_PORT,
791            };
792
793            autocore_std::ControlRunner::new(<$prog_type>::new())
794                .config(config)
795                .run()
796        }
797    };
798}
799