wgpu_core/track/
query_set.rs

1use crate::{
2    resource::{QuerySet, Trackable},
3    track::metadata::ResourceMetadata,
4};
5use alloc::sync::Arc;
6
7/// A tracker that holds tracks QuerySets.
8///
9/// This is mostly a safe shell around [`ResourceMetadata`]
10#[derive(Debug)]
11pub(crate) struct QuerySetTracker {
12    metadata: ResourceMetadata<Arc<QuerySet>>,
13    size: usize,
14}
15
16impl QuerySetTracker {
17    pub fn new() -> Self {
18        Self {
19            metadata: ResourceMetadata::new(),
20            size: 0,
21        }
22    }
23
24    /// Inserts a single resource into the resource tracker.
25    ///
26    /// Returns a reference to the newly inserted resource.
27    /// (This allows avoiding a clone/reference count increase in many cases.)
28    pub fn insert_single(&mut self, resource: Arc<QuerySet>) -> &Arc<QuerySet> {
29        let index = resource.tracker_index().as_usize();
30        self.allow_index(index);
31
32        unsafe {
33            // # SAFETY: we just allowed this resource, which makes the metadata object larger if
34            // it's not in bounds
35            self.metadata.insert(index, resource)
36        }
37    }
38
39    /// Sets the size of all the vectors inside the tracker.
40    ///
41    /// Must be called with the highest possible Texture ID before
42    /// all unsafe functions are called.
43    pub fn set_size(&mut self, size: usize) {
44        self.size = size;
45        self.metadata.set_size(size)
46    }
47
48    /// Extend the vectors to let the given index be valid.
49    fn allow_index(&mut self, index: usize) {
50        if index >= self.size {
51            self.set_size(index + 1);
52        }
53    }
54
55    /// Returns true if the tracker owns the given query set.
56    pub fn contains(&self, query_set: &QuerySet) -> bool {
57        self.metadata.contains(query_set.tracker_index().as_usize())
58    }
59
60    pub fn used_resources(&self) -> impl Iterator<Item = &Arc<QuerySet>> {
61        self.metadata.owned_resources()
62    }
63}
wgpu_core/track/query_set.rs

wgpu_core/track/
query_set.rs
