use crate::*;
impl AssetCache {
pub fn get_state(&self, url: &str) -> Option<AssetState> {
self.get_entries()
.get(url)
.map(|entry: &AssetEntry| entry.get_state())
}
pub fn get_image(&self, url: &str) -> Option<HtmlImageElement> {
let entry: &AssetEntry = self.get_entries().get(url)?;
if entry.get_state() != AssetState::Loaded {
return None;
}
entry.get_image()
}
pub fn is_all_loaded(&self) -> bool {
self.get_entries()
.values()
.all(|entry: &AssetEntry| entry.get_state() != AssetState::Loading)
}
pub fn loaded_count(&self) -> usize {
self.get_entries()
.values()
.filter(|entry: &&AssetEntry| entry.get_state() == AssetState::Loaded)
.count()
}
pub fn clear(&mut self) {
self.get_mut_entries().clear();
}
}
impl Default for AssetCache {
fn default() -> AssetCache {
AssetCache::new()
}
}
impl AssetLoader {
pub fn load_image(&mut self, url: String) {
let image: HtmlImageElement = HtmlImageElement::new().expect("should create image element");
let entry: AssetEntry = AssetEntry::new(
AssetType::Image,
AssetState::Loading,
Some(image.clone()),
url.clone(),
);
self.get_cache()
.borrow_mut()
.get_mut_entries()
.insert(url.clone(), entry);
*self.get_mut_pending_count() += 1;
let cache_clone: Rc<RefCell<AssetCache>> = self.get_cache().clone();
let url_for_onload: String = url.clone();
let url_for_onerror: String = url.clone();
let onload_closure: Closure<dyn FnMut()> = Closure::wrap(Box::new(move || {
let mut cache_ref = cache_clone.borrow_mut();
if let Some(mut entry) = cache_ref.get_entries().get(&url_for_onload).cloned() {
entry.set_state(AssetState::Loaded);
cache_ref
.get_mut_entries()
.insert(url_for_onload.clone(), entry);
}
}) as Box<dyn FnMut()>);
let cache_clone_err: Rc<RefCell<AssetCache>> = self.get_cache().clone();
let onerror_closure: Closure<dyn FnMut()> = Closure::wrap(Box::new(move || {
let mut cache_ref = cache_clone_err.borrow_mut();
if let Some(mut entry) = cache_ref.get_entries().get(&url_for_onerror).cloned() {
entry.set_state(AssetState::Error);
cache_ref
.get_mut_entries()
.insert(url_for_onerror.clone(), entry);
}
}) as Box<dyn FnMut()>);
image.set_onload(Some(onload_closure.as_ref().unchecked_ref()));
image.set_onerror(Some(onerror_closure.as_ref().unchecked_ref()));
image.set_src(&url);
self.get_closures().borrow_mut().push(onload_closure);
self.get_closures().borrow_mut().push(onerror_closure);
}
pub fn is_all_loaded(&self) -> bool {
self.get_cache().borrow().is_all_loaded()
}
pub fn get_image(&self, url: &str) -> Option<HtmlImageElement> {
self.get_cache().borrow().get_image(url)
}
pub fn progress(&self) -> f64 {
let cache_ref = self.get_cache().borrow();
let total: usize = cache_ref.get_entries().len();
if total == 0 {
return 1.0;
}
cache_ref.loaded_count() as f64 / total as f64
}
}
impl Default for AssetLoader {
fn default() -> AssetLoader {
AssetLoader::new()
}
}
impl AssetLoader {
pub fn create_image_element(url: &str) -> Option<HtmlImageElement> {
let image: HtmlImageElement = HtmlImageElement::new().ok()?;
image.set_src(url);
Some(image)
}
}