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

use amethyst::{
	ecs::prelude::*,
	renderer::SpriteRender,
	assets::AssetStorage,
};
use amethyst_imgui::imgui;
use crate::Inspect;
use imgui::im_str;

/// Add this as a resource and insert your handles into it to get a dropdown for SpriteSheetHandle selection
pub type SpriteList = std::collections::HashMap<String, amethyst::renderer::SpriteSheetHandle>;

impl<'a> Inspect<'a> for SpriteRender {
	type SystemData = (
		ReadStorage<'a, Self>,
		ReadExpect<'a, AssetStorage<amethyst::renderer::SpriteSheet>>,
		Read<'a, SpriteList>,
		Read<'a, LazyUpdate>,
	);

	fn inspect((storage, sprites, sprite_list, lazy): &mut Self::SystemData, entity: Entity) {
		amethyst_imgui::with(|ui| {
			let me = if let Some(x) = storage.get(entity) { x } else { return; };
			let mut new_me = me.clone();
			ui.push_id(im_str!("sprite_render"));

			if !sprite_list.is_empty() {
				let mut current = 0;
				let mut items = Vec::<imgui::ImString>::with_capacity(9);
				let list_vec = sprite_list.iter().collect::<Vec<_>>();
				for (i, (key, sprite_sheet)) in list_vec.iter().enumerate() {
					if me.sprite_sheet == **sprite_sheet {
						current = i as i32;
					}
					items.push(im_str!("{}", key).into());
				}

				ui.combo(im_str!("sprite sheet"), &mut current, items.iter().map(std::ops::Deref::deref).collect::<Vec<_>>().as_slice(), 10);
				new_me.sprite_sheet = list_vec[current as usize].1.clone();
				if new_me.sprite_sheet != me.sprite_sheet {
					new_me.sprite_number = 0;
				}
			}

			let mut sprite_number = new_me.sprite_number as i32;
			ui.slider_int(
				im_str!("# sprite"),
				&mut sprite_number,
				0,
				sprites.get(&new_me.sprite_sheet).unwrap_or_else(f!()).sprites.len() as i32 - 1,
			)
			.build();
			new_me.sprite_number = sprite_number as usize;

			if compare_fields!(me, new_me, sprite_number, sprite_sheet) {
				lazy.insert(entity, new_me);
			}
			ui.pop_id();
		});
	}

	fn can_add((_, _, sprite_list, _): &mut Self::SystemData, _: Entity) -> bool {
		!sprite_list.is_empty()
	}

	fn add((_, _, sprite_list, lazy): &mut Self::SystemData, entity: Entity) {
		lazy.insert(entity, SpriteRender { sprite_sheet: sprite_list.values().nth(0).unwrap_or_else(f!()).clone(), sprite_number: 0 });
	}
}