#![cfg_attr(not(debug_assertions), windows_subsystem = "windows")]
use bevy_ecs::prelude::*;
use std::sync::Mutex;
use std::sync::mpsc::channel;
use std::thread;
use std::time::Duration;
use windows::core::Result;
use wintf::ecs::Window;
use wintf::ecs::widget::brushes::{Brush, Brushes};
use wintf::ecs::widget::shapes::Rectangle;
use wintf::ecs::{
GraphicsCore, Point, SizeI, SurfaceGraphics, VisualGraphics, WindowGraphics, WindowHandle,
WindowPos,
};
use wintf::*;
type WorldCommand = Box<dyn FnOnce(&mut World) + Send>;
fn main() -> Result<()> {
println!("\n========== GraphicsCore Reinitialization Test ==========\n");
human_panic::setup_panic!();
let mgr = WinApp::new()?;
let world = mgr.world();
let (tx, rx) = channel::<WorldCommand>();
let rx = Mutex::new(rx);
thread::spawn(move || {
println!("[Timer] 0s: Creating two windows");
let _ = tx.send(Box::new(|world: &mut World| {
world.spawn((
Window {
title: "Test Window 1 (Red)".to_string(),
..Default::default()
},
WindowPos {
position: Some(Point { x: 100, y: 100 }),
size: Some(SizeI {
width: 600,
height: 400,
}),
..Default::default()
},
Rectangle::new(),
Brushes::with_foreground(Brush::RED.as_color().unwrap()),
));
world.spawn((
Window {
title: "Test Window 2 (Blue)".to_string(),
..Default::default()
},
WindowPos {
position: Some(Point { x: 750, y: 100 }),
size: Some(SizeI {
width: 600,
height: 400,
}),
..Default::default()
},
Rectangle::new(),
Brushes::with_foreground(Brush::BLUE.as_color().unwrap()),
));
println!("[Test] Two windows spawned");
}));
thread::sleep(Duration::from_secs(3));
println!("\n[Timer] 3s: Simulating device loss (invalidating GraphicsCore)");
let _ = tx.send(Box::new(|world: &mut World| {
if let Some(mut graphics) = world.get_resource_mut::<GraphicsCore>() {
println!("\n========================================");
println!("[Test] ===== デバイスロスト シミュレーション開始 =====");
println!("[Test] GraphicsCore.invalidate() を呼び出します");
graphics.invalidate();
println!("[Test] GraphicsCore無効化完了。次フレームで自動再初期化されます。");
println!("========================================\n");
} else {
println!(" ❌ [FAIL] GraphicsCore resource not found");
}
}));
thread::sleep(Duration::from_secs(1));
println!("\n[Timer] 4s: Verifying reinitialization");
let _ = tx.send(Box::new(|world: &mut World| {
let graphics_valid = world
.get_resource::<GraphicsCore>()
.map(|g| g.is_valid())
.unwrap_or(false);
let mut query = world.query::<(
Entity,
&WindowHandle,
&WindowGraphics,
&VisualGraphics,
&SurfaceGraphics,
)>();
println!("\n========================================");
println!("[Test] ===== 再初期化検証 =====");
println!("[Test] GraphicsCore.is_valid() = {}", graphics_valid);
let mut all_success = true;
for (entity, handle, wg, v, s) in query.iter(world) {
let wg_valid = wg.is_valid();
let v_valid = v.is_valid();
let s_valid = s.is_valid();
let generation = wg.generation();
println!("[Test] Entity {:?} (HWND {:?}):", entity, handle.hwnd);
println!(
" - WindowGraphics: valid={}, generation={}",
wg_valid, generation
);
println!(" - Visual.is_valid() = {}", v_valid);
println!(" - Surface.is_valid() = {}", s_valid);
if generation > 0 && wg_valid && v_valid && s_valid {
println!(
" ✅ [SUCCESS] 再初期化されました！（generation={} > 0）",
generation
);
} else if generation == 0 && wg_valid && v_valid && s_valid {
println!(" ⏳ [WAIT] 初回作成状態（generation=0）");
} else {
println!(" ❌ [FAIL] コンポーネントが無効または未初期化");
all_success = false;
}
}
if all_success && graphics_valid {
println!("\n 🎉🎉🎉 [TEST SUCCESS] 全コンポーネントが正常に再初期化されました！");
}
println!("========================================\n");
}));
thread::sleep(Duration::from_secs(3));
println!("\n[Timer] 7s: Removing Rectangle from first window (visual effect)");
let _ = tx.send(Box::new(|world: &mut World| {
let mut query = world.query::<(Entity, &WindowHandle)>();
if let Some((entity, handle)) = query.iter(world).next() {
println!(
"[Test] Removing Rectangle from entity {:?} (hwnd {:?})",
entity, handle.hwnd
);
println!(" 赤い四角形が消えます...");
world.entity_mut(entity).remove::<Rectangle>();
}
}));
thread::sleep(Duration::from_secs(3));
println!("\n[Timer] 10s: Closing one window");
let _ = tx.send(Box::new(|world: &mut World| {
let mut query = world.query::<(Entity, &WindowHandle)>();
let entities: Vec<_> = query.iter(world).map(|(e, h)| (e, h.hwnd)).collect();
if let Some((entity, hwnd)) = entities.first() {
println!(
"[Test] Closing window: Entity {:?}, HWND {:?}",
entity, hwnd
);
world.despawn(*entity);
}
}));
thread::sleep(Duration::from_secs(3));
println!("\n[Timer] 13s: Closing last window");
let _ = tx.send(Box::new(|world: &mut World| {
let mut query = world.query::<(Entity, &WindowHandle)>();
if let Some((entity, handle)) = query.iter(world).next() {
println!(
"[Test] Closing last window: Entity {:?}, HWND {:?}",
entity, handle.hwnd
);
world.despawn(entity);
}
}));
});
println!("[Test] Test scenario started");
println!("\nTest Phases:");
println!(" Phase 1 (0s): Create 2 windows with red & blue rectangles");
println!(" Phase 2 (3s): Simulate device loss (invalidate GraphicsCore)");
println!(" Phase 3 (4s): Verify automatic reinitialization (generation++)");
println!(" Phase 4 (7s): Remove Rectangle for visual effect");
println!(" Phase 5 (10s): Close one window");
println!(" Phase 6 (13s): Close last window and exit");
println!("\n========================================\n");
world
.borrow_mut()
.add_systems(wintf::ecs::world::Update, move |world: &mut World| {
let Ok(rx_guard) = rx.lock() else {
return;
};
for command in rx_guard.try_iter() {
command(world);
}
});
mgr.run()?;
Ok(())
}
