Add combo ui

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "arc_util"
-version = "0.11.2"
+version = "0.11.3"
 edition = "2021"
 authors = ["Zerthox"]
 repository = "https://github.com/zerthox/arcdps-utils"
@@ -13,6 +13,7 @@ open = { version = "5.0.0", optional = true }
 semver = { version = "1.0.17", optional = true }
 serde = { version = "1.0.152", features = ["derive"], optional = true }
 serde_json = { version = "1.0.93", optional = true }
+strum = "0.26.1"
 
 [dependencies.windows]
 version = "0.52.0"

src/ui/render/combo.rs

@@ -0,0 +1,58 @@
+use std::borrow::Cow;
+
+use arcdps::imgui::{Selectable, StyleColor, Ui};
+use strum::IntoEnumIterator;
+
+/// Renders a combo box for items from an iterator.
+// TODO: make more generic?
+pub fn combo<T>(
+    ui: &Ui,
+    label: impl AsRef<str>,
+    all: impl IntoIterator<Item = T>,
+    current: &mut T,
+    item_label: impl Fn(&T) -> Cow<str>,
+    item_color: impl Fn(&T) -> Option<[f32; 4]>,
+) -> bool
+where
+    T: PartialEq,
+{
+    let mut changed = false;
+    if let Some(_token) = ui.begin_combo(label, item_label(current).as_ref()) {
+        for entry in all {
+            let selected = entry == *current;
+
+            // apply color to selectable
+            let style =
+                item_color(&entry).map(|color| ui.push_style_color(StyleColor::Text, color));
+            if Selectable::new(item_label(&entry).as_ref())
+                .selected(selected)
+                .build(ui)
+            {
+                changed = true;
+                *current = entry;
+            }
+            drop(style);
+
+            // handle focus
+            if selected {
+                ui.set_item_default_focus();
+            }
+        }
+    }
+    changed
+}
+
+/// Renders a combo box for an enum implementing [`IntoEnumIterator`].
+pub fn enum_combo<T>(ui: &Ui, label: impl AsRef<str>, current: &mut T) -> bool
+where
+    T: PartialEq + AsRef<str> + IntoEnumIterator,
+{
+    combo(
+        ui,
+        label,
+        T::iter(),
+        current,
+        |item| item.as_ref().into(),
+        |_| None,
+    )
+}

src/ui/render/mod.rs

@@ -1,11 +1,13 @@
 //! Predefined primitive UI elements & render helpers.
 
 mod button;
+mod combo;
 mod context_menu;
 mod input;
 mod table;
 
 pub use self::button::*;
+pub use self::combo::*;
 pub use self::context_menu::*;
 pub use self::input::*;
 pub use self::table::*;