Preview gradient in UI

Author: K0bin

engine/src/renderer/passes/dear_imgui_renderer.rs

@@ -10,6 +10,7 @@ use sourcerenderer_core::gpu::{Texture as _, TexturePlane};
 use sourcerenderer_core::{Matrix4, Vec2, Vec2I, Vec2UI, Vec3, Vec3UI};
 use std::collections::HashMap;
 use std::sync::Arc;
+use crate::asset::{AssetHandle, AssetType};
 
 pub struct DearImguiRenderer {
     next_id: u64,
@@ -19,6 +20,8 @@ pub struct DearImguiRenderer {
 }
 
 impl DearImguiRenderer {
+    const IMGUI_IDENTIFY_MASK: u64 = 1 << 63;
+
     pub fn new(
         _device: &Device,
         _resources: &mut RendererResources,
@@ -207,7 +210,7 @@ impl DearImguiRenderer {
                         Some(&format!("DearImgui view {}", id)),
                     );
 
-                    let imgui_id = dear_imgui_rs::TextureId::new(id);
+                    let imgui_id = dear_imgui_rs::TextureId::new(id | Self::IMGUI_IDENTIFY_MASK);
                     self.textures
                         .insert(texture_request.texture(), (imgui_id, view));
 
@@ -423,12 +426,18 @@ impl DearImguiRenderer {
 
                         let view = match texture {
                             dear_imgui_rs::TextureBinding::Legacy(id) => {
-                                log::warn!("Binding with legacy texture id: {:?}", id);
-                                self.textures
-                                    .iter()
-                                    .find(|(_, (stored_id, _))| id == stored_id)
-                                    .map(|(_, (_, texture))| texture)
-                                    .unwrap()
+                                if (id.id() & Self::IMGUI_IDENTIFY_MASK) != 0 {
+                                    log::warn!("Got a managed texture in the legacy path.");
+                                    self.textures
+                                        .iter()
+                                        .find(|(_, (stored_id, _))| id == stored_id)
+                                        .map(|(_, (_, texture))| texture)
+                                        .unwrap()
+                                } else {
+                                    let asset_handle = AssetHandle::new(id.id(), AssetType::Texture);
+                                    let texture = assets_readonly.get_texture_opt(asset_handle.into());
+                                    texture.map(|t| &t.view).unwrap_or(&assets_readonly.get_placeholder_texture_black().view)
+                                }
                             }
                             dear_imgui_rs::TextureBinding::Managed(id) => {
                                 &(self.textures.get(id).unwrap().1)

game/src/uni_project/ui.rs

@@ -306,6 +306,11 @@ fn color_gradient<const STEPS: usize>(ui: &dear_imgui_rs::Ui, greyscale: bool, p
     let space = unsafe { ui.style() }.item_spacing()[0];
     let size = (ui.content_region_avail()[0] - ((STEPS - 1) as f32 * space)) * (1.0f32 / (STEPS as f32));
 
+    let asset_handle: AssetHandle = texture_handle.into();
+    let imgui_texture_id = dear_imgui_rs::TextureId::new(asset_handle.index());
+    let imgui_texture_ref = dear_imgui_rs::TextureRef::from(imgui_texture_id);
+    ui.image(imgui_texture_ref, [ui.content_region_avail()[0], ui.frame_height() * 2.0f32]);
+
     let mut changed = false;
 
     for i in 0..STEPS {