refactor: destructure timestamp writes in `command_encoder_create_{co…

…mpute,render}_pass`
gfx-rs · Nov 21, 2024 · d0b9929 · d0b9929
1 parent 64114b4
commit d0b9929
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/wgpu-core/src/command/compute.rs b/wgpu-core/src/command/compute.rs
@@ -28,7 +28,9 @@ use crate::{
 use thiserror::Error;
 use wgt::{BufferAddress, DynamicOffset};
 
-use super::{bind::BinderError, memory_init::CommandBufferTextureMemoryActions};
+use super::{
+    bind::BinderError, memory_init::CommandBufferTextureMemoryActions, SimplifiedQueryType,
+};
 use crate::ray_tracing::TlasAction;
 use std::sync::Arc;
 use std::{fmt, mem::size_of, str};
@@ -309,6 +311,12 @@ impl Global {
         };
 
         arc_desc.timestamp_writes = if let Some(tw) = desc.timestamp_writes {
+            let &PassTimestampWrites {
+                query_set,
+                beginning_of_pass_write_index: _,
+                end_of_pass_write_index: _,
+            } = tw;
+
             match cmd_buf
                 .device
                 .require_features(wgt::Features::TIMESTAMP_QUERY)
@@ -317,7 +325,7 @@ impl Global {
                 Err(e) => return make_err(e.into(), arc_desc),
             }
 
-            let query_set = match hub.query_sets.get(tw.query_set).get() {
+            let query_set = match hub.query_sets.get(query_set).get() {
                 Ok(query_set) => query_set,
                 Err(e) => return make_err(e.into(), arc_desc),
             };

diff --git a/wgpu-core/src/command/render.rs b/wgpu-core/src/command/render.rs
@@ -1392,7 +1392,13 @@ impl Global {
                 };
 
             arc_desc.timestamp_writes = if let Some(tw) = desc.timestamp_writes {
-                let query_set = query_sets.get(tw.query_set).get()?;
+                let &PassTimestampWrites {
+                    query_set,
+                    beginning_of_pass_write_index: _,
+                    end_of_pass_write_index: _,
+                } = tw;
+
+                let query_set = query_sets.get(query_set).get()?;
 
                 device.require_features(wgt::Features::TIMESTAMP_QUERY)?;