1use anyhow::Context as _;
 2use blade_graphics as gpu;
 3use std::sync::Arc;
 4use util::ResultExt;
 5
 6#[cfg_attr(target_os = "macos", derive(Clone))]
 7pub struct BladeContext {
 8    pub(super) gpu: Arc<gpu::Context>,
 9}
10
11impl BladeContext {
12    pub fn new() -> anyhow::Result<Self> {
13        let device_id_forced = match std::env::var("ZED_DEVICE_ID") {
14            Ok(val) => parse_pci_id(&val)
15                .context("Failed to parse device ID from `ZED_DEVICE_ID` environment variable")
16                .log_err(),
17            Err(std::env::VarError::NotPresent) => None,
18            err => {
19                err.context("Failed to read value of `ZED_DEVICE_ID` environment variable")
20                    .log_err();
21                None
22            }
23        };
24        let gpu = Arc::new(
25            unsafe {
26                gpu::Context::init(gpu::ContextDesc {
27                    presentation: true,
28                    validation: false,
29                    device_id: device_id_forced.unwrap_or(0),
30                    ..Default::default()
31                })
32            }
33            .map_err(|e| anyhow::anyhow!("{e:?}"))?,
34        );
35        Ok(Self { gpu })
36    }
37}
38
39fn parse_pci_id(id: &str) -> anyhow::Result<u32> {
40    let mut id = id.trim();
41
42    if id.starts_with("0x") || id.starts_with("0X") {
43        id = &id[2..];
44    }
45    let is_hex_string = id.chars().all(|c| c.is_ascii_hexdigit());
46    let is_4_chars = id.len() == 4;
47    anyhow::ensure!(
48        is_4_chars && is_hex_string,
49        "Expected a 4 digit PCI ID in hexadecimal format"
50    );
51
52    u32::from_str_radix(id, 16).context("parsing PCI ID as hex")
53}
54
55#[cfg(test)]
56mod tests {
57    use super::parse_pci_id;
58
59    #[test]
60    fn test_parse_device_id() {
61        assert!(parse_pci_id("0xABCD").is_ok());
62        assert!(parse_pci_id("ABCD").is_ok());
63        assert!(parse_pci_id("abcd").is_ok());
64        assert!(parse_pci_id("1234").is_ok());
65        assert!(parse_pci_id("123").is_err());
66        assert_eq!(
67            parse_pci_id(&format!("{:x}", 0x1234)).unwrap(),
68            parse_pci_id(&format!("{:X}", 0x1234)).unwrap(),
69        );
70
71        assert_eq!(
72            parse_pci_id(&format!("{:#x}", 0x1234)).unwrap(),
73            parse_pci_id(&format!("{:#X}", 0x1234)).unwrap(),
74        );
75        assert_eq!(
76            parse_pci_id(&format!("{:#x}", 0x1234)).unwrap(),
77            parse_pci_id(&format!("{:#X}", 0x1234)).unwrap(),
78        );
79    }
80}