2020-05-06 13:45:17 +02:00
|
|
|
#![cfg(not(feature = "no_object"))]
|
|
|
|
|
2020-07-31 16:30:23 +02:00
|
|
|
use crate::any::Dynamic;
|
2020-04-21 17:25:12 +02:00
|
|
|
use crate::def_package;
|
2020-08-16 17:41:59 +02:00
|
|
|
use crate::engine::Map;
|
2020-05-25 07:44:28 +02:00
|
|
|
use crate::parser::{ImmutableString, INT};
|
2020-08-14 18:04:10 +02:00
|
|
|
use crate::plugin::*;
|
2020-04-20 18:11:25 +02:00
|
|
|
|
2020-06-21 10:37:05 +02:00
|
|
|
use crate::stdlib::vec::Vec;
|
2020-04-24 06:39:24 +02:00
|
|
|
|
2020-04-22 08:55:40 +02:00
|
|
|
def_package!(crate:BasicMapPackage:"Basic object map utilities.", lib, {
|
2020-08-21 15:48:45 +02:00
|
|
|
lib.combine_flatten(exported_module!(map_functions));
|
2020-08-14 18:04:10 +02:00
|
|
|
});
|
|
|
|
|
|
|
|
#[export_module]
|
|
|
|
mod map_functions {
|
2020-08-20 16:11:41 +02:00
|
|
|
#[inline(always)]
|
2020-08-14 18:04:10 +02:00
|
|
|
pub fn has(map: &mut Map, prop: ImmutableString) -> bool {
|
|
|
|
map.contains_key(&prop)
|
|
|
|
}
|
2020-08-20 16:11:41 +02:00
|
|
|
#[inline(always)]
|
2020-08-14 18:04:10 +02:00
|
|
|
pub fn len(map: &mut Map) -> INT {
|
2020-08-16 17:41:59 +02:00
|
|
|
map.len() as INT
|
|
|
|
}
|
2020-08-16 18:13:52 +02:00
|
|
|
#[rhai_fn(get = "len")]
|
2020-08-20 16:11:41 +02:00
|
|
|
#[inline(always)]
|
2020-08-16 17:41:59 +02:00
|
|
|
pub fn len_prop(map: &mut Map) -> INT {
|
|
|
|
len(map)
|
2020-08-14 18:04:10 +02:00
|
|
|
}
|
2020-08-20 16:11:41 +02:00
|
|
|
#[inline(always)]
|
2020-08-14 18:04:10 +02:00
|
|
|
pub fn clear(map: &mut Map) {
|
2020-05-13 13:21:42 +02:00
|
|
|
map.clear();
|
2020-08-14 18:04:10 +02:00
|
|
|
}
|
2020-08-20 16:11:41 +02:00
|
|
|
#[inline]
|
2020-08-14 18:04:10 +02:00
|
|
|
pub fn remove(x: &mut Map, name: ImmutableString) -> Dynamic {
|
|
|
|
x.remove(&name).unwrap_or_else(|| ().into())
|
|
|
|
}
|
|
|
|
pub fn mixin(map1: &mut Map, map2: Map) {
|
|
|
|
map2.into_iter().for_each(|(key, value)| {
|
2020-08-16 17:41:59 +02:00
|
|
|
map1.insert(key, value);
|
2020-08-14 18:04:10 +02:00
|
|
|
});
|
|
|
|
}
|
2020-08-16 17:41:59 +02:00
|
|
|
#[rhai_fn(name = "+=")]
|
2020-08-20 16:11:41 +02:00
|
|
|
#[inline(always)]
|
2020-08-16 17:41:59 +02:00
|
|
|
pub fn mixin_operator(map1: &mut Map, map2: Map) {
|
|
|
|
mixin(map1, map2)
|
2020-07-31 16:30:23 +02:00
|
|
|
}
|
2020-08-16 17:41:59 +02:00
|
|
|
#[rhai_fn(name = "+")]
|
|
|
|
pub fn merge(mut map1: Map, map2: Map) -> Map {
|
2020-08-14 18:04:10 +02:00
|
|
|
map2.into_iter().for_each(|(key, value)| {
|
|
|
|
map1.insert(key, value);
|
|
|
|
});
|
2020-08-16 17:41:59 +02:00
|
|
|
map1
|
2020-08-14 18:04:10 +02:00
|
|
|
}
|
2020-08-16 17:41:59 +02:00
|
|
|
pub fn fill_with(map1: &mut Map, map2: Map) {
|
2020-08-14 18:04:10 +02:00
|
|
|
map2.into_iter().for_each(|(key, value)| {
|
2020-08-16 17:41:59 +02:00
|
|
|
if !map1.contains_key(&key) {
|
|
|
|
map1.insert(key, value);
|
|
|
|
}
|
2020-08-14 18:04:10 +02:00
|
|
|
});
|
|
|
|
}
|
2020-04-21 17:01:10 +02:00
|
|
|
|
2020-08-22 16:26:49 +02:00
|
|
|
#[cfg(not(feature = "no_index"))]
|
|
|
|
pub mod indexing {
|
|
|
|
pub fn keys(map: &mut Map) -> Vec<Dynamic> {
|
|
|
|
map.iter().map(|(k, _)| k.clone().into()).collect()
|
|
|
|
}
|
|
|
|
pub fn values(map: &mut Map) -> Vec<Dynamic> {
|
|
|
|
map.iter().map(|(_, v)| v.clone()).collect()
|
|
|
|
}
|
2020-08-14 18:04:10 +02:00
|
|
|
}
|
|
|
|
}
|