Skip to content

Commit

Permalink
A new stack-based vector
Browse files Browse the repository at this point in the history
  • Loading branch information
c410-f3r committed Sep 28, 2020
1 parent 1aa4590 commit 22b0227
Showing 1 changed file with 293 additions and 0 deletions.
293 changes: 293 additions & 0 deletions text/2978-stack_based_vec.md
Original file line number Diff line number Diff line change
@@ -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<T>`.

# Motivation
[motivation]: #motivation

`core::collections::ArrayVec<T>` 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<i32, 4> = 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<i32, 64> = 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<i32, 0> = 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<T>` or `alloc::collections::Vec<T>`? 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<ArrayVec<T, N>>` might be better than `Vec<Vec<T>>`.

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<i32, 64> = 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<T, const N: usize> {
data: MaybeUninit<[T; N]>,
len: usize,
}

impl<T, const N: usize> ArrayVec<T, N> {
// 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<F>(&mut self, same_bucket: F)
where
F: FnMut(&mut T, &mut T) -> bool;

pub fn dedup_by_key<F, K>(&mut self, key: F)
where
F: FnMut(&mut T) -> K,
K: PartialEq<K>;

pub fn drain<R>(&mut self, range: R)
where
R: RangeBounds<usize>;

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<T>;

#[inline]
pub fn push(&mut self, element: T) -> Result<(), T>;

pub fn remove(&mut self, idx: usize) -> Option<T>;

pub fn retain<F>(&mut self, f: F)
where
F: FnMut(&mut T) -> bool;

#[inline]
pub unsafe fn set_len(&mut self, len: usize);

pub fn splice<R, I>(&mut self, range: R, replace_with: I)
where
I: IntoIterator<Item = T>,
R: RangeBounds<usize>;

pub fn split_off(&mut self, at: usize) -> Self;

pub fn swap_remove(&mut self, idx: usize) -> Option<T>;

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<i32, 33> = array_vec![1, 2, 3];

// Instance with 1i32 and 1i32
let _: ArrayVec<i32, 64> = 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<T, const N: usize> {
// Hides internal implementation
data: DynVecData,
}

impl<T, const N: usize> DynVec<T, N> {
// Much of the `Vec` API goes here
}

// This is just an example. `Vec<T>` could be `Box` and `enum` an `union`.
enum DynVecData<T, const N: usize> {
Heap(Vec<T>),
Inline(ArrayVec<T, N>),
}
```

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<S>`, where `S` is the storage, could be split into:

```rust
type DynString = GenericString<DynVec<u8>>;
type HeapString = GenericString<Vec<u8>>;
type StackString = GenericString<ArrayVec<u8>>;
```

0 comments on commit 22b0227

Please sign in to comment.