ruffle/core/src/library.rs

use crate::avm1::globals::SystemPrototypes;
use crate::avm1::Object;
use crate::backend::audio::SoundHandle;
use crate::character::Character;
use crate::display_object::TDisplayObject;
use crate::font::Font;
use crate::prelude::*;
use crate::tag_utils::SwfMovie;
use gc_arena::MutationContext;
use std::collections::HashMap;
use std::sync::{Arc, Weak};
use swf::CharacterId;
use weak_table::PtrWeakKeyHashMap;

/// Symbol library for a single given SWF.
pub struct MovieLibrary<'gc> {
    characters: HashMap<CharacterId, Character<'gc>>,
    export_characters: HashMap<String, Character<'gc>>,
    jpeg_tables: Option<Vec<u8>>,
    device_font: Option<Font<'gc>>,
}

impl<'gc> MovieLibrary<'gc> {
    pub fn new() -> Self {
        MovieLibrary {
            characters: HashMap::new(),
            export_characters: HashMap::new(),
            jpeg_tables: None,
            device_font: None,
        }
    }

    pub fn register_character(&mut self, id: CharacterId, character: Character<'gc>) {
        // TODO(Herschel): What is the behavior if id already exists?
        if !self.contains_character(id) {
            self.characters.insert(id, character);
        } else {
            log::error!("Character ID collision: Tried to register ID {} twice", id);
        }
    }

    /// Registers an export name for a given character ID.
    /// This character will then be instantiable from AVM1.
    pub fn register_export(&mut self, id: CharacterId, export_name: &str) {
        use std::collections::hash_map::Entry;
        if let Some(character) = self.characters.get(&id) {
            match self.export_characters.entry(export_name.to_string()) {
                Entry::Vacant(e) => {
                    e.insert(character.clone());
                }
                Entry::Occupied(_) => {
                    log::warn!(
                        "Can't register export {}: Export already exists",
                        export_name
                    );
                }
            }
        } else {
            log::warn!(
                "Can't register export {}: Character ID {} doesn't exist",
                export_name,
                id
            )
        }
    }

    pub fn contains_character(&self, id: CharacterId) -> bool {
        self.characters.contains_key(&id)
    }

    #[allow(dead_code)]
    pub fn get_character_by_id(&self, id: CharacterId) -> Option<&Character<'gc>> {
        self.characters.get(&id)
    }

    #[allow(dead_code)]
    pub fn get_character_by_export_name(&self, name: &str) -> Option<&Character<'gc>> {
        self.export_characters.get(name)
    }

    /// Instantiates the library item with the given character ID into a display object.
    pub fn instantiate_by_id(
        &self,
        id: CharacterId,
        gc_context: MutationContext<'gc, '_>,
        prototypes: &SystemPrototypes<'gc>,
    ) -> Result<DisplayObject<'gc>, Box<dyn std::error::Error>> {
        if let Some(character) = self.characters.get(&id) {
            self.instantiate_display_object(character, gc_context, prototypes)
        } else {
            log::error!("Tried to instantiate non-registered character ID {}", id);
            Err("Character id doesn't exist".into())
        }
    }

    /// Instantiates the library item with the given export name into a display object.
    pub fn instantiate_by_export_name(
        &self,
        export_name: &str,
        gc_context: MutationContext<'gc, '_>,
        prototypes: &SystemPrototypes<'gc>,
    ) -> Result<DisplayObject<'gc>, Box<dyn std::error::Error>> {
        if let Some(character) = self.export_characters.get(export_name) {
            self.instantiate_display_object(character, gc_context, prototypes)
        } else {
            log::error!(
                "Tried to instantiate non-registered character {}",
                export_name
            );
            Err("Character id doesn't exist".into())
        }
    }

    /// Instantiates the given character into a display object.
    fn instantiate_display_object(
        &self,
        character: &Character<'gc>,
        gc_context: MutationContext<'gc, '_>,
        prototypes: &SystemPrototypes<'gc>,
    ) -> Result<DisplayObject<'gc>, Box<dyn std::error::Error>> {
        let (mut obj, proto): (DisplayObject<'gc>, Object<'gc>) = match character {
            Character::Bitmap(bitmap) => (bitmap.instantiate(gc_context), prototypes.object),
            Character::EditText(edit_text) => {
                (edit_text.instantiate(gc_context), prototypes.text_field)
            }
            Character::Graphic(graphic) => (graphic.instantiate(gc_context), prototypes.object),
            Character::MorphShape(morph_shape) => {
                (morph_shape.instantiate(gc_context), prototypes.object)
            }
            Character::MovieClip(movie_clip) => {
                (movie_clip.instantiate(gc_context), prototypes.movie_clip)
            }
            Character::Button(button) => (button.instantiate(gc_context), prototypes.object),
            Character::Text(text) => (text.instantiate(gc_context), prototypes.object),
            _ => return Err("Not a DisplayObject".into()),
        };
        obj.post_instantiation(gc_context, obj, proto);
        Ok(obj)
    }

    pub fn get_font(&self, id: CharacterId) -> Option<Font<'gc>> {
        if let Some(&Character::Font(font)) = self.characters.get(&id) {
            Some(font)
        } else {
            None
        }
    }

    pub fn get_sound(&self, id: CharacterId) -> Option<SoundHandle> {
        if let Some(Character::Sound(sound)) = self.characters.get(&id) {
            Some(*sound)
        } else {
            None
        }
    }

    pub fn set_jpeg_tables(&mut self, data: Vec<u8>) {
        if self.jpeg_tables.is_some() {
            // SWF spec says there should only be one JPEGTables tag.
            // TODO: What is the behavior when there are multiples?
            log::warn!("SWF contains multiple JPEGTables tags");
            return;
        }
        // Some SWFs have a JPEGTables tag with 0 length; ignore these.
        // (Does this happen when there is only a single DefineBits tag?)
        self.jpeg_tables = if data.is_empty() {
            None
        } else {
            Some(crate::backend::render::remove_invalid_jpeg_data(&data[..]).to_vec())
        }
    }

    pub fn jpeg_tables(&self) -> Option<&[u8]> {
        self.jpeg_tables.as_ref().map(|data| &data[..])
    }

    /// Returns the device font for use when a font is unavailable.
    pub fn device_font(&self) -> Option<Font<'gc>> {
        self.device_font
    }

    /// Sets the device font.
    pub fn set_device_font(&mut self, font: Option<Font<'gc>>) {
        self.device_font = font;
    }
}

unsafe impl<'gc> gc_arena::Collect for MovieLibrary<'gc> {
    #[inline]
    fn trace(&self, cc: gc_arena::CollectionContext) {
        for character in self.characters.values() {
            character.trace(cc);
        }
        self.device_font.trace(cc);
    }
}

impl Default for MovieLibrary<'_> {
    fn default() -> Self {
        Self::new()
    }
}

/// Symbol library for multiple movies.
pub struct Library<'gc> {
    /// All the movie libraries.
    movie_libraries: PtrWeakKeyHashMap<Weak<SwfMovie>, MovieLibrary<'gc>>,
}

unsafe impl<'gc> gc_arena::Collect for Library<'gc> {
    #[inline]
    fn trace(&self, cc: gc_arena::CollectionContext) {
        for (_, val) in self.movie_libraries.iter() {
            val.trace(cc);
        }
    }
}

impl<'gc> Library<'gc> {
    pub fn library_for_movie(&self, movie: Arc<SwfMovie>) -> Option<&MovieLibrary<'gc>> {
        self.movie_libraries.get(&movie)
    }

    pub fn library_for_movie_mut(&mut self, movie: Arc<SwfMovie>) -> &mut MovieLibrary<'gc> {
        if !self.movie_libraries.contains_key(&movie) {
            self.movie_libraries
                .insert(movie.clone(), MovieLibrary::default());
        };

        self.movie_libraries.get_mut(&movie).unwrap()
    }
}

impl<'gc> Default for Library<'gc> {
    fn default() -> Self {
        Self {
            movie_libraries: PtrWeakKeyHashMap::new(),
        }
    }
}