polars-utils 0.54.0

Private utils for the Polars DataFrame library
Documentation 
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

use crate::UnitVec;

/// Vec container with a getter that clears the vec.
#[derive(Default)]
pub struct ScratchVec<T>(Vec<T>);

impl<T> ScratchVec<T> {
    pub fn with_capacity(capacity: usize) -> Self {
        Self(Vec::with_capacity(capacity))
    }

    /// Clear the vec and return a mutable reference to it.
    pub fn get(&mut self) -> &mut Vec<T> {
        self.0.clear();
        &mut self.0
    }
}

/// UnitVec container with a getter that clears the unitvec.
#[derive(Default)]
pub struct ScratchUnitVec<T>(UnitVec<T>);

impl<T> ScratchUnitVec<T> {
    pub fn with_capacity(capacity: usize) -> Self {
        Self(UnitVec::with_capacity(capacity))
    }

    /// Clear the unitvec and return a mutable reference to it.
    pub fn get(&mut self) -> &mut UnitVec<T> {
        self.0.clear();
        &mut self.0
    }
}