waterkit-screen 0.1.1

Screen capture with wgpu texture output
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

import Foundation
import UIKit

// MARK: - Screenshot Capture

/// Capture the main screen and return encoded image data.
/// - Parameter format: 0 = PNG, 1 = AVIF, 2 = HEIF
public func capture_screenshot(format: UInt8) -> RustVec<UInt8> {
    var result: RustVec<UInt8>? = nil

    DispatchQueue.main.sync {
        guard let window = UIApplication.shared.windows.first(where: { $0.isKeyWindow }) else {
            return
        }

        let renderer = UIGraphicsImageRenderer(bounds: window.bounds)
        let image = renderer.image { ctx in
            window.drawHierarchy(in: window.bounds, afterScreenUpdates: true)
        }

        var data: Data?
        switch format {
        case 1: // AVIF
            data = encodeAsAVIF(image)
        case 2: // HEIF
            data = encodeAsHEIF(image)
        default: // PNG
            data = image.pngData()
        }

        if let data = data {
            let vec = RustVec<UInt8>()
            for byte in data {
                vec.push(value: byte)
            }
            result = vec
        }
    }

    return result ?? RustVec()
}

private func encodeAsHEIF(_ image: UIImage) -> Data? {
    guard let cgImage = image.cgImage else { return nil }

    let data = NSMutableData()
    let uti = "public.heic" as CFString
    guard let dest = CGImageDestinationCreateWithData(data, uti, 1, nil) else {
        return nil
    }

    CGImageDestinationAddImage(dest, cgImage, nil)
    if CGImageDestinationFinalize(dest) {
        return data as Data
    }
    return nil
}

private func encodeAsAVIF(_ image: UIImage) -> Data? {
    guard let cgImage = image.cgImage else { return nil }

    let data = NSMutableData()
    let uti = "public.avif" as CFString
    guard let dest = CGImageDestinationCreateWithData(data, uti, 1, nil) else {
        return nil
    }

    CGImageDestinationAddImage(dest, cgImage, nil)
    if CGImageDestinationFinalize(dest) {
        return data as Data
    }
    return nil
}

// MARK: - Brightness Control

public func get_screen_brightness() -> Float {
    return Float(UIScreen.main.brightness)
}

public func set_screen_brightness(value: Float) -> Bool {
    UIScreen.main.brightness = CGFloat(value)
    return true
}

// MARK: - Screen Stream (Not supported on iOS)

public func init_screen_stream(display_id: UInt32, target_fps: UInt32, show_cursor: Bool) -> Bool {
    return false
}

public func stop_screen_stream() {
    // No-op
}

public func get_iosurface_ptr() -> UInt64 {
    return 0
}

public func get_iosurface_sequence() -> UInt32 {
    return 0
}

public func get_frame_width() -> UInt32 {
    return 0
}

public func get_frame_height() -> UInt32 {
    return 0
}

public func get_frame_timestamp_ns() -> UInt64 {
    return 0
}