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
use {
    crate::escape::{Escape, Handle, Terminal},
    smallvec::SmallVec,
    std::collections::VecDeque,
};

/// Resource usage epochs.
#[doc(hidden)]
#[derive(Clone, Debug)]
pub struct Epochs {
    /// 2D array of epochs.
    pub values: SmallVec<[SmallVec<[u64; 8]>; 4]>,
}

impl Epochs {
    fn is_before(&self, other: &Self) -> bool {
        debug_assert_eq!(self.values.len(), other.values.len());
        self.values.iter().zip(other.values.iter()).all(|(a, b)| {
            debug_assert_eq!(a.len(), b.len());
            a.iter().zip(b.iter()).all(|(a, b)| a < b)
        })
    }
}

/// Resource handler.
#[derive(Debug, derivative::Derivative)]
#[derivative(Default(bound = ""))]
pub struct ResourceTracker<T> {
    terminal: Terminal<T>,
    dropped: VecDeque<(Epochs, T)>,
}

impl<T> ResourceTracker<T> {
    /// Create new resource manager.
    pub fn new() -> Self {
        Self::default()
    }

    /// Wrap resource instance into `Escape`.
    pub fn escape(&self, resource: T) -> Escape<T>
    where
        T: Sized,
    {
        Escape::escape(resource, &self.terminal)
    }

    /// Wrap resource instance into `Handle`.
    pub fn handle(&self, resource: T) -> Handle<T>
    where
        T: Sized,
    {
        self.escape(resource).into()
    }

    /// Cleanup dropped resources.
    ///
    /// # Safety
    ///
    /// `next` epochs must contain epoch indices that aren't started yet
    /// `complete` epochs must contain epoch indices that are complete.
    /// Can be guaranteed with fence wait.
    ///
    pub fn cleanup(&mut self, mut dispose: impl FnMut(T), next: &Epochs, complete: &Epochs) {
        while let Some((epoch, resource)) = self.dropped.pop_front() {
            if !epoch.is_before(complete) {
                self.dropped.push_front((epoch, resource));
                break;
            }

            dispose(resource);
        }

        self.dropped
            .extend(self.terminal.drain().map(|res| (next.clone(), res)));
    }

    /// Cleanup all dropped resources.
    ///
    /// # Safety
    ///
    /// All dropped resources must be unused.
    /// Can be guaranteed with device idle wait.
    ///
    pub fn dispose(&mut self, dispose: impl FnMut(T)) {
        self.dropped
            .drain(..)
            .map(|(_, res)| res)
            .chain(self.terminal.drain())
            .for_each(dispose);
    }
}