gd-icon-renderer/src/assets.rs

use plist;

use std::collections::HashMap;

use image::GenericImageView;
use image::DynamicImage;

/// "{1,2}" -> `(1, 2)`
fn parse_vec(str: &str) -> (i32, i32) {
    let parts: Vec<&str> = str[1..str.len()-1].split(",").collect();
    let a: Vec<i32> = parts
        .iter()
        .map(|s| s.trim().parse::<i32>().unwrap())
        .collect();

    return (a[0], a[1])
}
/// parse_vec, but for float64
fn parse_vec_f32(str: &str) -> (f32, f32) {
    let parts: Vec<&str> = str[1..str.len()-1].split(",").collect();
    let a: Vec<f32> = parts
        .iter()
        .map(|s| s.trim().parse::<f32>().unwrap())
        .collect();

    return (a[0], a[1])
}
/// `"{{1,2},{3,4}}"` -> `{{1, 2}, {3, 4}}`
fn parse_rect_vecs(str: &str) -> ((i32, i32), (i32, i32)) {
    let cleaned_str = str.replace("{", "").replace("}", "");
    let parts: Vec<&str> = cleaned_str.split(",").collect();
    let a: Vec<i32> = parts
        .iter()
        .map(|s| s.trim().parse::<i32>().unwrap())
        .collect();

    return ((a[0], a[1]), (a[2], a[3]))
}

/// Represents a sprite along with its texture data in a spritesheet.
#[derive(Clone, Copy, Debug)]
pub struct Sprite {
    /// Whenever rendering the sprite, offset it by this much.
    pub offset: (f32, f32),
    /// {left, top}, {width, height}. Controls the cropping.
    pub rect: ((i32, i32), (i32, i32)),
    /// Whether the texture needs to be counter-rotated 90 degrees counter-clockwise.
    pub rotated: bool,
    /// Size of the sprite.
    pub size: (i32, i32),
    /// Difference between this and `size` is unknown to me.
    pub source_size: (i32, i32)
}

impl Sprite {
    /// Shorthand for initializing a sprite with its .plist representation.
    fn initialize(obj: plist::Value) -> Sprite {
        let hash = obj.as_dictionary().expect("object must be a dict");

        let hash_keys = vec!["spriteOffset", "spriteSize", "spriteSourceSize", "textureRect", "textureRotated"];

        let isolated: Vec<(&&str, Option<&plist::Value>)> = hash_keys
            .iter()
            .map(|s| (s, hash.get(s)))
            .collect();

        let missing: Vec<&(&&str, Option<&plist::Value>)> = isolated
            .iter()
            .filter(|&&(_, value)| value.is_none())
            .collect();

        if !missing.is_empty() {
            let missing_entries: Vec<&str> = missing.iter().map(|(&key, _)| key).collect();
            panic!("missing entries: {:?}", missing_entries);
        }

        let isolated_hash: HashMap<String, plist::Value> = isolated
            .iter()
            .map(|&(key, value)| (key.to_string(), value.expect("value is none after checking").clone()))
            .collect();

        return Sprite {
            offset: parse_vec_f32(isolated_hash.get("spriteOffset").expect("missing spriteOffset").as_string().expect("spriteOffset is not a string")),
            rect: parse_rect_vecs(isolated_hash.get("textureRect").expect("missing textureRect").as_string().expect("textureRect is not a string")),
            rotated: isolated_hash.get("textureRotated").unwrap_or(&plist::Value::from(false)).as_boolean().expect("textureRotated is not a boolean").clone(),
            size: parse_vec(isolated_hash.get("spriteSize").expect("missing spriteSize").as_string().expect("spriteSize is not a string")),
            source_size: parse_vec(isolated_hash.get("spriteSourceSize").expect("missing spriteSourceSize").as_string().expect("spriteSourceSize is not a string"))
        }
    }
}

/// Represents a spritesheet along with its sprites.
#[derive(Clone)]
pub struct Spritesheet {
    pub sprites: HashMap<String, Sprite>,

    pub texture_file_name: String,
    pub size: (i32, i32)
}

impl Spritesheet {
    /// Shorthand for initializing a spritesheet with its .plist representation.
    fn initialize(obj: plist::Value) -> Spritesheet {
        let hash = obj.as_dictionary().expect("object must be a dict");

        let sprites = hash.get("frames").expect("object must have a `frames` object").as_dictionary().expect("`frames` must be a dict");
        let metadata = hash.get("metadata").expect("object must have a `metadata` object").as_dictionary().expect("`metadata` must be a dict");

        return Spritesheet {
            sprites: sprites.iter().map(|(key, value)| (key.clone(), Sprite::initialize(value.clone()))).collect(),
            texture_file_name: metadata.get("textureFileName").expect("metadata must have a `textureFileName` object").as_string().expect("`textureFileName` must be a string").to_string(),
            size: parse_vec(metadata.get("size").expect("metadata must have a `size` object").as_string().expect("`size` must be a string"))
        }
    }
}

/// Stores both a spritesheet and its associated `DynamicImage` for easy access.
#[derive(Clone)]
pub struct LoadedSpritesheet {
    pub spritesheet: Spritesheet,
    pub texture: DynamicImage
}

/// Loads the spritesheet and readies the associated image.
pub fn load_spritesheet(path: &str) -> LoadedSpritesheet {
    return LoadedSpritesheet {
        spritesheet: Spritesheet::initialize(plist::from_file(path).expect("could not load plist")),
        texture: image::open(path.replace(".plist", ".png")).expect("could not load texture")
    }
}

/// Represents the metadata of an animation frame's sprite
#[derive(Clone, Debug)]
pub struct AnimationSprite {
    pub texture: String,
    pub position: (f32, f32),
    pub scale: (f32, f32),
    pub rotation: f64,
    pub flipped: (bool, bool),
    pub z: i32
}

impl AnimationSprite {
    fn initialize(obj: plist::Value) -> AnimationSprite {
        let hash = obj.as_dictionary().expect("object must be a dict");

        let hash_keys = vec!["texture", "position", "scale", "rotation", "flipped", "zValue"];

        let isolated: Vec<(&&str, Option<&plist::Value>)> = hash_keys
            .iter()
            .map(|s| (s, hash.get(s)))
            .collect();

        let missing: Vec<&(&&str, Option<&plist::Value>)> = isolated
            .iter()
            .filter(|&&(_, value)| value.is_none())
            .collect();

        if !missing.is_empty() {
            let missing_entries: Vec<&str> = missing.iter().map(|(&key, _)| key).collect();
            panic!("missing entries: {:?}", missing_entries);
        }

        let isolated_hash: HashMap<String, plist::Value> = isolated
            .iter()
            .map(|&(key, value)| (key.to_string(), value.expect("value is none after checking").clone()))
            .collect();

        return AnimationSprite {
            texture: isolated_hash.get("texture").expect("missing texture").as_string().expect("texture is not a string").to_string(),
            position: parse_vec_f32(isolated_hash.get("position").expect("missing position").as_string().expect("position is not a string")),
            scale: parse_vec_f32(isolated_hash.get("scale").expect("missing scale").as_string().expect("scale is not a string")),
            rotation: isolated_hash.get("rotation").expect("missing rotation").as_string().expect("rotation is not a string").parse::<f64>().expect("couldnt parse rotation as f64"),
            flipped: {
                let flipped_numbers = parse_vec(isolated_hash.get("flipped").expect("missing flipped").as_string().expect("flipped is not a string"));
                (flipped_numbers.0 > 0, flipped_numbers.1 > 0)
            },
            z: isolated_hash.get("zValue").expect("missing zValue").as_string().expect("zValue is not a string").parse::<i32>().expect("couldnt parse zValue as i32")
        }
    }
}

pub type Animations = HashMap<String, Vec<AnimationSprite>>;

pub fn load_animations(path: &str) -> Animations {
    let loaded_plist: plist::Value = plist::from_file(path).expect("could not load plist");
    let animations = loaded_plist.as_dictionary().expect("object must be a dict").get("animationContainer").expect("key `animationContainer` doesnt exist").as_dictionary().expect("`animationContainer` must be a dict");
    let mut parsed_animations: Animations = HashMap::new();
    for (k, v) in animations.iter() {
        parsed_animations.insert(k.clone(), vec![] as Vec<AnimationSprite>);
        parsed_animations.get_mut(k.as_str()).expect("this should exist..")
            .extend(v.as_dictionary().expect("animation must be a dict")
            .iter().map(|(_, v)| AnimationSprite::initialize(v.clone())));
    }
    return parsed_animations;
}

/// Trims out a sprite from an image according to a .plist spritesheet.
pub fn get_sprite(spritesheet: Spritesheet, img: &DynamicImage, key: String) -> Option<(DynamicImage, Sprite)> {
    let sprite = spritesheet.sprites.get(&key);

    if sprite.is_none() {
        return None;
    }

    if let Some(sprite) = sprite {
        let rect = sprite.rect;
        
        let (mut left, mut top, mut width, mut height) = (rect.0.0, rect.0.1, rect.1.0, rect.1.1);
        if sprite.rotated {
            (left, top, width, height) = (left, top, height, width);
        }
        
        let mut canvas: DynamicImage = image::DynamicImage::ImageRgba8(img.view(left as u32, top as u32, width as u32, height as u32).to_image());

        if sprite.rotated {
            canvas = canvas.rotate270();
        }

        return Some((canvas, sprite.clone()));
    }

    unreachable!("The sprite should have been found in the spritesheet or not found at all")
}

/// Trims out a sprite from an image according to a LoadedSpritesheet object.
pub fn get_sprite_from_loaded(spritesheet: &LoadedSpritesheet, key: String) -> Option<(DynamicImage, Sprite)> {
    let sprite = get_sprite(spritesheet.spritesheet.clone(), &spritesheet.texture, key);
    return sprite;
}