wita 0.9.6

A window library in Rust for Windows
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

use crate::bindings::Windows::Win32::{
    Controls::*, MenusAndResources::*, SystemServices::*, WindowsAndMessaging::*,
};
use std::path::{Path, PathBuf};

#[inline]
fn make_int_resource(id: u16) -> PWSTR {
    PWSTR(id as _)
}

/// Describes a icon.
#[derive(Clone, Debug)]
pub enum Icon {
    /// A icon from a resource id.
    Resource(u16),
    /// A icon from a file.
    File(PathBuf),
}

impl Icon {
    pub fn from_path(path: impl AsRef<Path>) -> Icon {
        Icon::File(path.as_ref().to_path_buf())
    }
}

fn load_icon_impl(hinst: HINSTANCE, icon: &Icon, cx: i32, cy: i32) -> HICON {
    let icon = unsafe {
        match icon {
            Icon::Resource(id) => LoadImageW(
                hinst,
                make_int_resource(*id),
                GDI_IMAGE_TYPE::IMAGE_ICON,
                cx,
                cy,
                IMAGE_FLAGS::LR_SHARED,
            ),
            Icon::File(path) => {
                let wpath = path
                    .to_string_lossy()
                    .encode_utf16()
                    .chain(Some(0))
                    .collect::<Vec<_>>();
                LoadImageW(
                    HINSTANCE(0),
                    PWSTR(wpath.as_ptr() as _),
                    GDI_IMAGE_TYPE::IMAGE_ICON,
                    cx,
                    cy,
                    IMAGE_FLAGS(
                        IMAGE_FLAGS::LR_SHARED.0
                            | IMAGE_FLAGS::LR_LOADFROMFILE.0,
                    ),
                )
            }
        }
    };
    if icon == HANDLE(0) {
        panic!("cannot load the icon");
    }
    HICON(icon.0)
}

pub(crate) fn load_icon(icon: &Icon, hinst: HINSTANCE) -> HICON {
    unsafe {
        load_icon_impl(
            hinst,
            icon,
            GetSystemMetrics(SYSTEM_METRICS_INDEX::SM_CXICON),
            GetSystemMetrics(SYSTEM_METRICS_INDEX::SM_CYICON),
        )
    }
}

pub(crate) fn load_small_icon(icon: &Icon, hinst: HINSTANCE) -> HICON {
    unsafe {
        load_icon_impl(
            hinst,
            icon,
            GetSystemMetrics(SYSTEM_METRICS_INDEX::SM_CXSMICON),
            GetSystemMetrics(SYSTEM_METRICS_INDEX::SM_CYSMICON),
        )
    }
}