1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136

#![cfg_attr(feature = "nightly", feature(allocator_api))]
#![doc = include_str!("../README.md")]

mod raw;
mod shared;
mod vector;
mod drain;
mod splice;

pub use raw::{AtomicRefCount, BufferSize, DefaultRefCount, RefCount};
pub use shared::{AtomicSharedVector, RefCountedVector, SharedVector};
pub use vector::{Vector, RawVector};

pub mod alloc {
    pub use allocator_api2::alloc::{AllocError, Allocator, Global};
}

pub(crate) fn grow_amortized(len: usize, additional: usize) -> usize {
    let required = len.saturating_add(additional);
    let cap = len.saturating_add(len).max(required).max(8);

    const MAX: usize = BufferSize::MAX as usize;

    if cap > MAX {
        if required <= MAX {
            return required;
        }

        panic!("Required allocation size is too large");
    }

    cap
}

#[macro_export]
macro_rules! vector {
    (@one@ $x:expr) => (1usize);
    ($elem:expr; $n:expr) => ({
        $crate::Vector::from_elem($elem, $n)
    });
    ($($x:expr),*$(,)*) => ({
        let count = 0usize $(+ $crate::vector!(@one@ $x))*;
        let mut vec = $crate::Vector::with_capacity(count);
        $(vec.push($x);)*
        vec
    });
    ([$($x:expr),*$(,)*] in $allocator:expr) => ({
        let count = 0usize $(+ $crate::vector!(@one@ $x))*;
        let mut vec = $crate::Vector::try_with_capacity_in(count, $allocator).unwrap();
        $(vec.push($x);)*
        vec
    });
    ([$x:expr;$n:expr] in $allocator:expr) => ({
        let mut vec = $crate::Vector::try_with_capacity_in($n, $allocator).unwrap();
        for _ in 0..$n { vec.push($x.clone()); }
        vec
    });
}

#[macro_export]
macro_rules! rc_vector {
    ($elem:expr; $n:expr) => ({
        let mut vec = $crate::SharedVector::with_capacity($n);
        for _ in 0..$n { vec.push($elem.clone()); }
        vec
    });
    ($($x:expr),*$(,)*) => ({
        let count = 0usize $(+ $crate::vector!(@one@ $x))*;
        let mut vec = $crate::SharedVector::with_capacity(count);
        $(vec.push($x);)*
        vec
    });
    ([$($x:expr),*$(,)*] in $allocator:expr) => ({
        let count = 0usize $(+ $crate::vector!(@one@ $x))*;
        let mut vec = $crate::SharedVector::try_with_capacity_in(count, $allocator).unwrap();
        $(vec.push($x);)*
        vec
    });
    ([$elem:expr;$n:expr] in $allocator:expr) => ({
        let mut vec = $crate::SharedVector::try_with_capacity_in($n, $allocator).unwrap();
        for _ in 0..$n { vec.push($elem.clone()); }
        vec
    });
}

#[macro_export]
macro_rules! arc_vector {
    ($elem:expr; $n:expr) => ({
        let mut vec = $crate::AtomicSharedVector::with_capacity($n);
        for _ in 0..$n { vec.push($elem.clone()); }
        vec
    });
    ($($x:expr),*$(,)*) => ({
        let count = 0usize $(+ $crate::vector!(@one@ $x))*;
        let mut vec = $crate::AtomicSharedVector::with_capacity(count);
        $(vec.push($x);)*
        vec
    });
    ([$($x:expr),*$(,)*] in $allocator:expr) => ({
        let count = 0usize $(+ $crate::vector!(@one@ $x))*;
        let mut vec = $crate::AtomicSharedVector::try_with_capacity_in(count, $allocator).unwrap();
        $(vec.push($x);)*
        vec
    });
    ([$elem:expr;$n:expr] in $allocator:expr) => ({
        let mut vec = $crate::AtomicSharedVector::try_with_capacity_in($n, $allocator).unwrap();
        for _ in 0..$n { vec.push($elem.clone()); }
        vec
    });
}

#[test]
fn vector_macro() {
    pub use allocator_api2::alloc::{Allocator, Global};

    let v1: Vector<u32> = vector![0, 1, 2, 3, 4, 5];
    let v2: Vector<u32> = vector![2; 4];
    let v3: Vector<u32> = vector!([6, 7] in Global);
    assert_eq!(v1.as_slice(), &[0, 1, 2, 3, 4, 5]);
    assert_eq!(v2.as_slice(), &[2, 2, 2, 2]);
    assert_eq!(v3.as_slice(), &[6, 7]);

    let v1: SharedVector<u32> = rc_vector![0, 1, 2, 3, 4, 5];
    let v2: SharedVector<u32> = rc_vector![3; 5];
    let v3: SharedVector<u32> = rc_vector!([4; 3] in Global);
    assert_eq!(v1.as_slice(), &[0, 1, 2, 3, 4, 5]);
    assert_eq!(v2.as_slice(), &[3, 3, 3, 3, 3]);
    assert_eq!(v3.as_slice(), &[4, 4, 4]);

    let v1: AtomicSharedVector<u32> = arc_vector![0, 1, 2, 3, 4, 5];
    let v2: AtomicSharedVector<u32> = arc_vector![1; 4];
    let v3: AtomicSharedVector<u32> = arc_vector![[3, 2, 1] in Global];
    assert_eq!(v1.as_slice(), &[0, 1, 2, 3, 4, 5]);
    assert_eq!(v2.as_slice(), &[1, 1, 1, 1]);
    assert_eq!(v3.as_slice(), &[3, 2, 1]);
}