From 22b02279a218ca41b8e262528bffc5a996f8af8a Mon Sep 17 00:00:00 2001 From: Caio Date: Mon, 28 Sep 2020 18:00:58 -0300 Subject: [PATCH] A new stack-based vector --- text/2978-stack_based_vec.md | 293 +++++++++++++++++++++++++++++++++++ 1 file changed, 293 insertions(+) create mode 100644 text/2978-stack_based_vec.md diff --git a/text/2978-stack_based_vec.md b/text/2978-stack_based_vec.md new file mode 100644 index 00000000000..349f9167e82 --- /dev/null +++ b/text/2978-stack_based_vec.md @@ -0,0 +1,293 @@ +- Feature Name: `stack_based_vec` +- Start Date: 2020-09-27 +- RFC PR: [rust-lang/rfcs#0000](https://github.com/rust-lang/rfcs/pull/0000) +- Rust Issue: [rust-lang/rust#0000](https://github.com/rust-lang/rust/issues/0000) + +# Summary +[summary]: #summary + +This RFC, which depends and takes advantage of the upcoming stabilization of constant generics (min_const_generics), tries to propose the creation of a new "growable" vector named `ArrayVec` that manages stack memory and can be seen as an alternative for the built-in structure that handles heap-allocated memory, aka `alloc::vec::Vec`. + +# Motivation +[motivation]: #motivation + +`core::collections::ArrayVec` has several use-cases and should be conveniently added into the standard library due to its importance. + +### Unification + +There are a lot of different crates about the subject that tries to do roughly the same thing, a centralized implementation would stop the current fragmentation. + +### Optimization + +Stack-based allocation is generally faster than heap-based allocation and can be used as an optimization in places that otherwise would have to call an allocator. Some resource-constrained embedded devices can also benefit from it. + +### Building block + +Just like `Vec`, `ArrayVec` is also a primitive vector where high-level structures can use it as a building block. For example, a stack-based matrix or binary heap. + +### Useful in the real world + +`arrayvec` is one of the most downloaded project of `crates.io` and is used by thousand of projects, including Rustc itself. + + +# Guide-level explanation +[guide-level-explanation]: #guide-level-explanation + +`ArrayVec` is a container that encapsulates fixed size buffers. + +```rust +let mut v: ArrayVec = ArrayVec::new(); +let _ = v.push(1); +let _ = v.push(2); + +assert_eq!(v.len(), 2); +assert_eq!(v[0], 1); + +assert_eq!(v.pop(), Some(2)); +assert_eq!(v.len(), 1); + +v[0] = 7; +assert_eq!(v[0], 7); + +v.extend([1, 2, 3].iter().copied()); + +for element in &v { + println!("{}", element); +} +assert_eq!(v, [7, 1, 2, 3]); +``` + +Instead of relying on a heap-allocator, stack-based memory area is added and removed on-demand in a last-in-first-out (LIFO) order according to the calling workflow of a program. `ArrayVec` takes advantage of this predictable behavior to reserve an exactly amount of uninitialized bytes up-front and these bytes form a buffer where elements can be included dynamically. + +```rust +// `array_vec` can store up to 64 elements +let mut array_vec: ArrayVec = ArrayVec::new(); +``` + +Of course, fixed buffers lead to inflexibility because unlike `Vec`, the underlying capacity can not expand at run-time and there will never be more than 64 elements in the above example. + +```rust +// This vector can store up to 0 elements, therefore, nothing at all +let mut array_vec: ArrayVec = ArrayVec::new(); +let push_result = array_vec.push(1); +// Ooppss... Our push operation wasn't successful +assert!(push_result.is_err()); +``` + +A good question is: Should I use `core::collections::ArrayVec` or `alloc::collections::Vec`? Well, `Vec` is already good enough for most situations while stack allocation usually shines for small sizes. + +* Do you have a known upper bound? + +* How much memory are you going to allocate for your program? The default values of `RUST_MIN_STACK` or `ulimit -s` might not be enough. + +* Are you using nested `Vec`s? `Vec>` might be better than `Vec>`. + +Each use-case is different and should be pondered individually. In case of doubt, stick with `Vec`. + +For a more technical overview, take a look at the following operations: + +```rust +// `array_vec` has a pre-allocated memory of 2048 bits (32 * 64) that can store up +// to 64 decimals. +let mut array_vec: ArrayVec = ArrayVec::new(); + +// Although reserved, there isn't anything explicitly stored yet +assert_eq!(array_vec.len(), 0); + +// Initializes the first 32 bits with a simple '1' decimal or +// 00000000 00000000 00000000 00000001 bits +array_vec.push(1); + +// Our vector memory is now split into a 32/2016 pair of initialized and +// uninitialized memory respectively +assert_eq!(array_vec.len(), 1); +``` + +# Reference-level explanation +[reference-level-explanation]: #reference-level-explanation + +`ArrayVec` is a contiguous memory block where elements can be collected, therefore, a collection by definition and even though `core::collections` doesn't exist, it is the most natural module placement. + +The API basically mimics most of the current `Vec` surface with some tweaks to manage capacity. + +Notably, these tweaked methods are fallible versions of some well-known functions like `push` that will return `Result` instead of panicking at run-time. Since the upper capacity bound is known at compile-time, the compiler is likely going to remove most of the conditional bounding checking. + +```rust +// Please, bare in mind that these methods are simply suggestions. Discussions about the +// API should probably take place elsewhere. + +pub struct ArrayVec { + data: MaybeUninit<[T; N]>, + len: usize, +} + +impl ArrayVec { + // Constructors + + #[inline] + pub fn from_array(array: [T; N]) -> Self; + + #[inline] + pub fn from_array_and_len(array: [T; N], len: usize) -> Self; + + #[inline] + pub const fn new() -> Self; + + // Methods + + #[inline] + pub fn as_mut_ptr(&mut self) -> *mut T; + + #[inline] + pub fn as_mut_slice(&mut self) -> &mut [T]; + + #[inline] + pub fn as_ptr(&self) -> *const T; + + #[inline] + pub fn as_slice(&self) -> &[T]; + + #[inline] + pub const fn capacity(&self) -> usize; + + pub fn clear(&mut self); + + pub fn dedup(&mut self); + + pub fn dedup_by(&mut self, same_bucket: F) + where + F: FnMut(&mut T, &mut T) -> bool; + + pub fn dedup_by_key(&mut self, key: F) + where + F: FnMut(&mut T) -> K, + K: PartialEq; + + pub fn drain(&mut self, range: R) + where + R: RangeBounds; + + pub fn extend_from_cloneable_slice(&mut self, other: &[T]) -> Result<(), &[T]> + where + T: Clone; + + pub fn extend_from_copyable_slice(&mut self, other: &[T]) -> Result<(), &[T]> + where + T: Copy; + + pub fn insert(&mut self, _idx: usize, element: T) -> Result<(), T>; + + #[inline] + pub const fn is_empty(&self) -> bool; + + #[inline] + pub const fn len(&self) -> usize; + + pub fn pop(&mut self) -> Option; + + #[inline] + pub fn push(&mut self, element: T) -> Result<(), T>; + + pub fn remove(&mut self, idx: usize) -> Option; + + pub fn retain(&mut self, f: F) + where + F: FnMut(&mut T) -> bool; + + #[inline] + pub unsafe fn set_len(&mut self, len: usize); + + pub fn splice(&mut self, range: R, replace_with: I) + where + I: IntoIterator, + R: RangeBounds; + + pub fn split_off(&mut self, at: usize) -> Self; + + pub fn swap_remove(&mut self, idx: usize) -> Option; + + pub fn truncate(&mut self, len: usize); +} +``` + +Meaningless, unstable and deprecated methods like `reserve` or `drain_filter` weren't considered. A concrete implementation is available at https://github.com/c410-f3r/stack-based-vec. + +# Drawbacks +[drawbacks]: #drawbacks + +### Additional complexity + +New and existing users are likely to find it difficult to differentiate the purpose of each vector type, especially people that don't have a theoretical background in memory management. + +### The current ecosystem is fine + +`ArrayVec` might be an overkill in certain situations. If someone wants to use stack memory in a specific application, then it is just a matter of grabbing the appropriated crate. + +# Prior art +[prior-art]: #prior-art + +These are the most known structures: + + * `arrayvec::ArrayVec`: Uses declarative macros and an `Array` trait for implementations but lacks support for arbitrary sizes. + * `heapless::Vec`: With the usage of `typenum`, can support arbitrary sizes without a nightly compiler. + * `staticvec::StaticVec`: Uses unstable constant generics for arrays of arbitrary sizes. + * `tinyvec::ArrayVec`: Supports fixed and arbitrary (unstable feature) sizes but requires `T: Default` for security reasons. + +As seen, there isn't an implementation that stands out among the others because all of them roughly share the same purpose and functionality. Noteworthy is the usage of constant generics that makes it possible to create an efficient and unified approach for arbitrary array sizes. + +# Unresolved questions +[unresolved-questions]: #unresolved-questions + +### Nomenclature + +`ArrayVec` will conflict with `arrayvec::ArrayVec` and `tinyvec::ArrayVec`. + +### Prelude + +Should it be included in the prelude? + +### Macros + +```rust +// Instance with 1i32, 2i32 and 3i32 +let _: ArrayVec = array_vec![1, 2, 3]; + +// Instance with 1i32 and 1i32 +let _: ArrayVec = array_vec![1; 2]; +``` + +# Future possibilities +[future-possibilities]: #future-possibilities + +### Dynamic array + +An hydric approach between heap and stack memory could also be provided natively in the future. + +```rust +pub struct DynVec { + // Hides internal implementation + data: DynVecData, +} + +impl DynVec { + // Much of the `Vec` API goes here +} + +// This is just an example. `Vec` could be `Box` and `enum` an `union`. +enum DynVecData { + Heap(Vec), + Inline(ArrayVec), +} +``` + +The above description is very similar to what `smallvec` already does. + +### Generic collections and generic strings + +Many structures that use `alloc::vec::Vec` as the underlying storage can also use stack or hybrid memory, for example, an hypothetical `GenericString`, where `S` is the storage, could be split into: + +```rust +type DynString = GenericString>; +type HeapString = GenericString>; +type StackString = GenericString>; +```