Timestamp queries in render/compute pass descriptor

src/conv.rs

@@ -1132,6 +1132,14 @@ pub fn to_native_composite_alpha_mode(
     }
 }
 
+#[inline]
+pub fn map_query_set_index(index: u32) -> Option<u32> {
+    match index {
+        native::WGPU_QUERY_SET_INDEX_UNDEFINED => None,
+        _ => Some(index),
+    }
+}
+
 #[inline]
 pub fn map_query_set_descriptor<'a>(
     desc: &native::WGPUQuerySetDescriptor,

src/lib.rs

@@ -1,6 +1,7 @@
 use conv::{
     map_device_descriptor, map_instance_backend_flags, map_instance_descriptor,
     map_pipeline_layout_descriptor, map_primitive_state, map_shader_module, map_surface,
+    map_query_set_index,
     CreateSurfaceParams,
 };
 use parking_lot::{Mutex, RwLock};
@@ -996,10 +997,24 @@ pub unsafe extern "C" fn wgpuCommandEncoderBeginComputePass(
         )
     };
 
+    let timestamp_writes = descriptor.map(|descriptor| {
+        descriptor.timestampWrites.as_ref().map(|timestamp_write| {
+            wgc::command::ComputePassTimestampWrites {
+                query_set:
+                    timestamp_write.querySet
+                    .as_ref()
+                    .expect("invalid query set in timestamp writes")
+                    .id,
+                beginning_of_pass_write_index: map_query_set_index(timestamp_write.beginningOfPassWriteIndex),
+                end_of_pass_write_index: map_query_set_index(timestamp_write.endOfPassWriteIndex),
+            }
+        })
+    }).flatten();
+
     let desc = match descriptor {
         Some(descriptor) => wgc::command::ComputePassDescriptor {
             label: ptr_into_label(descriptor.label),
-            timestamp_writes: None,
+            timestamp_writes: timestamp_writes.as_ref(),
         },
         None => wgc::command::ComputePassDescriptor::default(),
     };
@@ -1048,6 +1063,18 @@ pub unsafe extern "C" fn wgpuCommandEncoderBeginRenderPass(
         }
     });
 
+    let timestamp_writes = descriptor.timestampWrites.as_ref().map(|timestamp_write| {
+        wgc::command::RenderPassTimestampWrites {
+            query_set:
+                timestamp_write.querySet
+                .as_ref()
+                .expect("invalid query set in timestamp writes")
+                .id,
+            beginning_of_pass_write_index: map_query_set_index(timestamp_write.beginningOfPassWriteIndex),
+            end_of_pass_write_index: map_query_set_index(timestamp_write.endOfPassWriteIndex),
+        }
+    });
+
     let desc = wgc::command::RenderPassDescriptor {
         label: ptr_into_label(descriptor.label),
         color_attachments: Cow::Owned(
@@ -1070,8 +1097,8 @@ pub unsafe extern "C" fn wgpuCommandEncoderBeginRenderPass(
                 .collect(),
         ),
         depth_stencil_attachment: depth_stencil_attachment.as_ref(),
-        timestamp_writes: None,
-        occlusion_query_set: None,
+        timestamp_writes: timestamp_writes.as_ref(),
+        occlusion_query_set: None, // TODO:
     };
 
     Arc::into_raw(Arc::new(WGPURenderPassEncoderImpl {