use super::dynamic::{AccessMode, Variant};
use crate::{Dynamic, Identifier, ImmutableString, StaticVec, ThinVec};
#[cfg(feature = "no_std")]
use std::prelude::v1::*;
use std::{
fmt, iter,
iter::{Extend, FromIterator},
marker::PhantomData,
};
pub const MIN_SCOPE_ENTRIES: usize = 8;
#[derive(Debug, Hash, Default)]
pub struct Scope<'a> {
values: ThinVec<Dynamic>,
names: ThinVec<ImmutableString>,
aliases: ThinVec<StaticVec<ImmutableString>>,
dummy: PhantomData<&'a ()>,
}
impl fmt::Display for Scope<'_> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
for (i, (name, constant, value)) in self.iter_inner().enumerate() {
writeln!(
f,
"[{}] {}{} = {:?}",
i + 1,
if constant { "const " } else { "" },
name,
value,
)?;
}
Ok(())
}
}
impl Clone for Scope<'_> {
#[inline]
fn clone(&self) -> Self {
Self {
values: self
.values
.iter()
.map(|v| {
let mut v2 = v.clone();
v2.set_access_mode(v.access_mode());
v2
})
.collect(),
names: self.names.clone(),
aliases: self.aliases.clone(),
dummy: self.dummy,
}
}
}
impl IntoIterator for Scope<'_> {
type Item = (String, Dynamic, Vec<ImmutableString>);
type IntoIter = Box<dyn Iterator<Item = Self::Item>>;
fn into_iter(self) -> Self::IntoIter {
Box::new(
self.values
.into_iter()
.zip(
self.names.into_iter().zip(
self.aliases
.into_iter()
.map(|a| a.to_vec())
.chain(iter::repeat(Vec::new())),
),
)
.map(|(value, (name, alias))| (name.to_string(), value, alias)),
)
}
}
impl<'a> IntoIterator for &'a Scope<'_> {
type Item = (&'a str, &'a Dynamic, &'a [ImmutableString]);
type IntoIter = Box<dyn Iterator<Item = Self::Item> + 'a>;
fn into_iter(self) -> Self::IntoIter {
Box::new(
self.values
.iter()
.zip(
self.names.iter().zip(
self.aliases
.iter()
.map(<_>::as_ref)
.chain(iter::repeat(&[][..])),
),
)
.map(|(value, (name, alias))| (name.as_str(), value, alias)),
)
}
}
impl Scope<'_> {
#[inline(always)]
#[must_use]
pub fn new() -> Self {
Self {
values: ThinVec::new(),
names: ThinVec::new(),
aliases: ThinVec::new(),
dummy: PhantomData,
}
}
#[inline(always)]
#[must_use]
pub fn with_capacity(capacity: usize) -> Self {
Self {
values: ThinVec::with_capacity(capacity),
names: ThinVec::with_capacity(capacity),
aliases: ThinVec::new(),
dummy: PhantomData,
}
}
#[inline(always)]
pub fn clear(&mut self) -> &mut Self {
self.names.clear();
self.values.clear();
self.aliases.clear();
self
}
#[inline(always)]
#[must_use]
pub fn len(&self) -> usize {
self.values.len()
}
#[inline(always)]
#[must_use]
pub fn is_empty(&self) -> bool {
self.values.is_empty()
}
#[inline(always)]
pub fn push(&mut self, name: impl Into<Identifier>, value: impl Variant + Clone) -> &mut Self {
self.push_entry(
name.into().into(),
AccessMode::ReadWrite,
Dynamic::from(value),
)
}
#[inline(always)]
pub fn push_dynamic(&mut self, name: impl Into<Identifier>, value: Dynamic) -> &mut Self {
self.push_entry(name.into().into(), value.access_mode(), value)
}
#[inline(always)]
pub fn push_constant(
&mut self,
name: impl Into<Identifier>,
value: impl Variant + Clone,
) -> &mut Self {
self.push_entry(
name.into().into(),
AccessMode::ReadOnly,
Dynamic::from(value),
)
}
#[inline(always)]
pub fn push_constant_dynamic(
&mut self,
name: impl Into<Identifier>,
value: Dynamic,
) -> &mut Self {
self.push_entry(name.into().into(), AccessMode::ReadOnly, value)
}
#[inline]
pub(crate) fn push_entry(
&mut self,
name: ImmutableString,
access: AccessMode,
mut value: Dynamic,
) -> &mut Self {
if self.is_empty() {
self.names.reserve(MIN_SCOPE_ENTRIES);
self.values.reserve(MIN_SCOPE_ENTRIES);
}
self.names.push(name);
value.set_access_mode(access);
self.values.push(value);
self
}
#[inline(always)]
pub fn pop(&mut self) -> &mut Self {
self.names
.pop()
.unwrap_or_else(|| panic!("`Scope` is empty"));
self.values.truncate(self.names.len());
self.aliases.truncate(self.names.len());
self
}
#[inline(always)]
#[allow(dead_code)]
pub(crate) fn pop_entry(&mut self) -> Option<(ImmutableString, Dynamic, Vec<ImmutableString>)> {
self.values.pop().map(|value| {
(
self.names.pop().unwrap(),
value,
if self.aliases.len() > self.values.len() {
self.aliases.pop().unwrap().to_vec()
} else {
Vec::new()
},
)
})
}
#[inline(always)]
pub fn rewind(&mut self, size: usize) -> &mut Self {
self.names.truncate(size);
self.values.truncate(size);
self.aliases.truncate(size);
self
}
#[inline]
#[must_use]
pub fn contains(&self, name: &str) -> bool {
self.names.iter().any(|key| name == key)
}
#[inline]
#[must_use]
pub(crate) fn search(&self, name: &str) -> Option<usize> {
self.names
.iter()
.rev() .position(|key| name == key)
.map(|i| self.len() - 1 - i)
}
#[inline]
#[must_use]
pub fn get_value<T: Variant + Clone>(&self, name: &str) -> Option<T> {
self.names
.iter()
.rev()
.position(|key| name == key)
.and_then(|i| self.values[self.len() - 1 - i].flatten_clone().try_cast())
}
#[inline]
#[must_use]
pub fn get_value_ref<T: Variant + Clone>(&self, name: &str) -> Option<&T> {
self.names
.iter()
.rev()
.position(|key| name == key)
.and_then(|i| {
let v = &self.values[self.len() - 1 - i];
#[cfg(not(feature = "no_closure"))]
assert!(!v.is_shared());
v.downcast_ref()
})
}
#[inline]
#[must_use]
pub fn get_value_mut<T: Variant + Clone>(&mut self, name: &str) -> Option<&mut T> {
let len = self.len();
self.names
.iter_mut()
.rev()
.position(|key| name == key)
.and_then(move |i| {
let v = &mut self.values[len - 1 - i];
#[cfg(not(feature = "no_closure"))]
assert!(!v.is_shared());
v.downcast_mut()
})
}
#[inline]
#[must_use]
pub fn is_constant(&self, name: &str) -> Option<bool> {
self.search(name)
.map(|n| match self.values[n].access_mode() {
AccessMode::ReadWrite => false,
AccessMode::ReadOnly => true,
})
}
#[inline]
pub fn set_or_push(
&mut self,
name: impl AsRef<str> + Into<Identifier>,
value: impl Variant + Clone,
) -> &mut Self {
match self
.search(name.as_ref())
.map(|n| (n, self.values[n].access_mode()))
{
None | Some((.., AccessMode::ReadOnly)) => {
self.push(name, value);
}
Some((index, AccessMode::ReadWrite)) => {
let value_ref = self.values.get_mut(index).unwrap();
*value_ref = Dynamic::from(value);
}
}
self
}
#[inline]
pub fn set_value(
&mut self,
name: impl AsRef<str> + Into<Identifier>,
value: impl Variant + Clone,
) -> &mut Self {
match self
.search(name.as_ref())
.map(|n| (n, self.values[n].access_mode()))
{
None => {
self.push(name, value);
}
Some((.., AccessMode::ReadOnly)) => panic!("variable {} is constant", name.as_ref()),
Some((index, AccessMode::ReadWrite)) => {
let value_ref = self.values.get_mut(index).unwrap();
*value_ref = Dynamic::from(value);
}
}
self
}
#[inline(always)]
#[must_use]
pub fn get(&self, name: &str) -> Option<&Dynamic> {
self.search(name).map(|index| &self.values[index])
}
#[inline(always)]
#[allow(dead_code)]
pub(crate) fn get_entry_by_index(&self, index: usize) -> (&str, &Dynamic, &[ImmutableString]) {
(
&self.names[index],
&self.values[index],
if self.aliases.len() > index {
&self.aliases[index]
} else {
&[]
},
)
}
#[inline(always)]
#[must_use]
pub fn remove<T: Variant + Clone>(&mut self, name: &str) -> Option<T> {
self.search(name).and_then(|index| {
self.names.remove(index);
if self.aliases.len() > index {
self.aliases.remove(index);
}
self.values.remove(index).try_cast()
})
}
#[inline]
#[must_use]
pub fn get_mut(&mut self, name: &str) -> Option<&mut Dynamic> {
self.search(name)
.and_then(move |n| match self.values[n].access_mode() {
AccessMode::ReadWrite => Some(self.get_mut_by_index(n)),
AccessMode::ReadOnly => None,
})
}
#[inline(always)]
pub(crate) fn get_mut_by_index(&mut self, index: usize) -> &mut Dynamic {
&mut self.values[index]
}
#[cfg(not(feature = "no_module"))]
#[inline]
pub(crate) fn add_alias_by_index(&mut self, index: usize, alias: ImmutableString) -> &mut Self {
if self.aliases.len() <= index {
self.aliases.resize(index + 1, <_>::default());
}
let aliases = self.aliases.get_mut(index).unwrap();
if !aliases.contains(&alias) {
aliases.push(alias);
}
self
}
#[cfg(not(feature = "no_module"))]
#[inline]
pub fn set_alias(
&mut self,
name: impl AsRef<str> + Into<Identifier>,
alias: impl Into<ImmutableString>,
) {
if let Some(index) = self.search(name.as_ref()) {
let alias = match alias.into() {
x if x.is_empty() => name.into().into(),
x => x,
};
self.add_alias_by_index(index, alias);
}
}
#[inline]
#[must_use]
pub fn clone_visible(&self) -> Self {
let len = self.len();
let mut scope = Self::new();
self.names.iter().rev().enumerate().for_each(|(i, name)| {
if scope.names.contains(name) {
return;
}
let index = len - 1 - i;
let v1 = &self.values[index];
scope.push_entry(name.clone(), v1.access_mode(), v1.clone());
if self.aliases.len() > index {
scope.aliases.resize(scope.len() - 1, <_>::default());
scope.aliases.push(self.aliases[index].clone());
}
});
scope
}
#[inline(always)]
pub fn iter(&self) -> impl Iterator<Item = (&str, bool, Dynamic)> {
self.iter_inner()
.map(|(name, constant, value)| (name.as_str(), constant, value.flatten_clone()))
}
#[inline(always)]
pub fn iter_raw(&self) -> impl Iterator<Item = (&str, bool, &Dynamic)> {
self.iter_rev_inner()
.map(|(name, constant, value)| (name.as_str(), constant, value))
}
#[inline]
pub(crate) fn iter_inner(&self) -> impl Iterator<Item = (&ImmutableString, bool, &Dynamic)> {
self.names
.iter()
.zip(self.values.iter())
.map(|(name, value)| (name, value.is_read_only(), value))
}
#[inline]
pub(crate) fn iter_rev_inner(
&self,
) -> impl Iterator<Item = (&ImmutableString, bool, &Dynamic)> {
self.names
.iter()
.rev()
.zip(self.values.iter().rev())
.map(|(name, value)| (name, value.is_read_only(), value))
}
#[inline]
#[allow(dead_code)]
pub(crate) fn remove_range(&mut self, start: usize, len: usize) {
self.values.drain(start..start + len).for_each(|_| {});
self.names.drain(start..start + len).for_each(|_| {});
if self.aliases.len() > start {
if self.aliases.len() <= start + len {
self.aliases.truncate(start);
} else {
self.aliases.drain(start..start + len).for_each(|_| {});
}
}
}
}
impl<K: Into<Identifier>> Extend<(K, Dynamic)> for Scope<'_> {
#[inline]
fn extend<T: IntoIterator<Item = (K, Dynamic)>>(&mut self, iter: T) {
for (name, value) in iter {
self.push_entry(name.into().into(), AccessMode::ReadWrite, value);
}
}
}
impl<K: Into<Identifier>> FromIterator<(K, Dynamic)> for Scope<'_> {
#[inline]
fn from_iter<T: IntoIterator<Item = (K, Dynamic)>>(iter: T) -> Self {
let mut scope = Self::new();
scope.extend(iter);
scope
}
}
impl<K: Into<Identifier>> Extend<(K, bool, Dynamic)> for Scope<'_> {
#[inline]
fn extend<T: IntoIterator<Item = (K, bool, Dynamic)>>(&mut self, iter: T) {
for (name, is_constant, value) in iter {
self.push_entry(
name.into().into(),
if is_constant {
AccessMode::ReadOnly
} else {
AccessMode::ReadWrite
},
value,
);
}
}
}
impl<K: Into<Identifier>> FromIterator<(K, bool, Dynamic)> for Scope<'_> {
#[inline]
fn from_iter<T: IntoIterator<Item = (K, bool, Dynamic)>>(iter: T) -> Self {
let mut scope = Self::new();
scope.extend(iter);
scope
}
}