aboutsummaryrefslogtreecommitdiff
path: root/stockton-render/src/texture/resolver.rs
blob: f66b724e9cd3385b00c30b3d7a4b720f23925370 (plain)
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
//! Resolves a texture in a BSP File to an image

use crate::texture::image::LoadableImage;
use stockton_levels::{parts::IsTexture, prelude::HasTextures};

use std::{
    path::Path,
    sync::{Arc, RwLock},
};

use image::{io::Reader, RgbaImage};

/// An object that can be used to resolve a texture from a BSP File
pub trait TextureResolver {
    type Image: LoadableImage;

    /// Get the given texture, or None if it's corrupt/not there.
    fn resolve(&mut self, texture_id: u32) -> Option<Self::Image>;
}

/// A basic filesystem resolver which gets the texture name from any HasTextures Object.
pub struct FsResolver<'a, T: HasTextures> {
    path: &'a Path,
    map_lock: Arc<RwLock<T>>,
}

impl<'a, T: HasTextures> FsResolver<'a, T> {
    pub fn new(path: &'a Path, map_lock: Arc<RwLock<T>>) -> Self {
        FsResolver { path, map_lock }
    }
}

impl<'a, T: HasTextures> TextureResolver for FsResolver<'a, T> {
    type Image = RgbaImage;

    fn resolve(&mut self, tex: u32) -> Option<Self::Image> {
        let map = self.map_lock.read().unwrap();
        let tex = map.get_texture(tex)?;
        let path = self.path.join(&tex.name());

        // drop(tex);
        // drop(map);

        if let Ok(file) = Reader::open(path) {
            if let Ok(guessed) = file.with_guessed_format() {
                if let Ok(decoded) = guessed.decode() {
                    return Some(decoded.into_rgba8());
                }
            }
        }

        log::warn!("Couldn't resolve texture {:?}", tex.name());
        None
    }
}