gled 2.6.3

gled is an application for creating animations and effects on artnet or wled light installations
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132

use crossbeam_channel::Sender;
use egui::mutex::Mutex;
use log::{debug, trace, warn};
use once_cell::sync::{Lazy, OnceCell};
use serialport::SerialPort;
use std::{
    collections::{HashMap, hash_map::Entry},
    thread,
    time::{Duration, Instant},
};

static SERIAL_NUMBERS_PORT: Lazy<Mutex<HashMap<String, String>>> =
    Lazy::new(|| Mutex::new(HashMap::new()));
static SENDER: OnceCell<Sender<(String, [u8; 512])>> = OnceCell::new();

pub fn start() {
    find_devices();

    let (tx, rx) = crossbeam_channel::unbounded();
    SENDER.set(tx).expect("SENDER already set");

    debug!("Spawning enttec dmx usb pro thread");
    thread::spawn(move || {
        let mut devices = HashMap::<String, Box<dyn SerialPort>>::new();
        loop {
            let now = Instant::now();
            let mut data_per_serial_number = HashMap::new();
            while let Ok((serial_number, data)) = rx.try_recv() {
                data_per_serial_number.insert(serial_number, data);
            }

            for (serial_number, data) in data_per_serial_number {
                let mut enttec_data = vec![
                    0x7E, // Start of message
                    6,    // DMX output message
                    0,    // Data length LSB
                    2,    // Data length MSB
                ];
                enttec_data.extend_from_slice(&data);
                enttec_data.push(0xE7); // End of message

                log::trace!("Sending data to enttec dmx usb pro: {enttec_data:?}");

                match devices.entry(serial_number) {
                    Entry::Occupied(mut occupied_entry) => {
                        if let Err(err) = occupied_entry.get_mut().write_all(&enttec_data) {
                            warn!("Could not write to port: {err}");
                            occupied_entry.remove();
                        } else {
                            trace!("Send data to enttec dmx usb pro");
                        }
                    }
                    Entry::Vacant(vacant_entry) => {
                        let Some(port) =
                            SERIAL_NUMBERS_PORT.lock().get(vacant_entry.key()).cloned()
                        else {
                            warn!(
                                "Could not find port for serial number {}",
                                vacant_entry.key()
                            );
                            continue;
                        };

                        match serialport::new(port, 115_200).open() {
                            Ok(mut port) => {
                                if let Err(err) = port.set_timeout(Duration::from_millis(50)) {
                                    warn!("Could not set timeout on port: {err:?}");
                                    continue;
                                }
                                if let Err(err) = port.write_all(&enttec_data) {
                                    warn!("Could not write to port: {err}");
                                } else {
                                    trace!("Send data to enttec dmx usb pro");
                                    vacant_entry.insert(port);
                                }
                            }
                            Err(err) => {
                                warn!("Could not open port: {err}");
                            }
                        }
                    }
                }
            }

            std::thread::sleep(Duration::from_millis(1000 / 40).saturating_sub(now.elapsed()));
        }
    });
}

fn find_devices() {
    debug!("Spawning enttec dmx usb pro discovery thread");
    thread::spawn(|| {
        loop {
            let mut serial_numbers_port = HashMap::new();
            let ports = match serialport::available_ports() {
                Ok(ports) => ports,
                Err(err) => {
                    warn!("Error listing serial ports: {err}");
                    std::thread::sleep(Duration::from_secs(60));
                    continue;
                }
            };
            for port in ports {
                if let serialport::SerialPortType::UsbPort(usb_port_info) = port.port_type {
                    if usb_port_info.manufacturer == Some("ENTTEC".to_owned())
                        && usb_port_info.product == Some("DMX USB PRO".to_owned())
                    {
                        if let Some(serial_number) = usb_port_info.serial_number {
                            serial_numbers_port.insert(serial_number, port.port_name);
                        }
                    }
                }
            }
            *SERIAL_NUMBERS_PORT.lock() = serial_numbers_port;
            std::thread::sleep(Duration::from_secs(1));
        }
    });
}

pub fn serial_numbers() -> Vec<String> {
    SERIAL_NUMBERS_PORT.lock().keys().cloned().collect()
}

pub fn send(serial_number: String, data: [u8; 512]) {
    if let Some(sender) = SENDER.get() {
        if let Err(err) = sender.send((serial_number, data)) {
            warn!("Could not send data to enttec dmx usb pro thread: {err}");
        }
    } else {
        warn!("enttec dmx usb pro thread not started");
    }
}