Add missing profiling scopes in re_renderer (#1567)

crates/re_renderer/src/allocator/cpu_write_gpu_read_belt.rs

@@ -301,6 +301,8 @@ impl CpuWriteGpuReadBelt {
         buffer_pool: &GpuBufferPool,
         num_elements: usize,
     ) -> CpuWriteGpuReadBuffer<T> {
+        crate::profile_function!();
+
         // Potentially overestimate alignment with Self::MIN_ALIGNMENT, see Self::MIN_ALIGNMENT doc string.
         let alignment = (std::mem::align_of::<T>() as wgpu::BufferAddress).max(Self::MIN_ALIGNMENT);
         // Pad out the size of the used buffer to a multiple of Self::MIN_ALIGNMENT.
@@ -377,6 +379,8 @@ impl CpuWriteGpuReadBelt {
     /// further writes) until after [`CpuWriteGpuReadBelt::after_queue_submit`] is called *and* the GPU is done
     /// copying the data from them.
     pub fn before_queue_submit(&mut self) {
+        crate::profile_function!();
+
         // This would be a great usecase for persistent memory mapping, i.e. mapping without the need to unmap
         // https://github.com/gfx-rs/wgpu/issues/1468
         // However, WebGPU does not support this!
@@ -393,6 +397,7 @@ impl CpuWriteGpuReadBelt {
     /// copy operations are submitted. Additional calls are harmless.
     /// Not calling this as soon as possible may result in increased buffer memory usage.
     pub fn after_queue_submit(&mut self) {
+        crate::profile_function!();
         self.receive_chunks();
 
         let sender = &self.sender;

crates/re_renderer/src/context.rs

@@ -108,6 +108,8 @@ impl RenderContext {
         queue: Arc<wgpu::Queue>,
         config: RenderContextConfig,
     ) -> Self {
+        crate::profile_function!();
+
         let mut gpu_resources = WgpuResourcePools::default();
         let global_bindings = GlobalBindings::new(&mut gpu_resources, &device);
 
@@ -310,6 +312,7 @@ impl RenderContext {
         self.cpu_write_gpu_read_belt.lock().before_queue_submit();
 
         if let Some(command_encoder) = self.active_frame.encoder.lock().0.take() {
+            crate::profile_scope!("finish & submit frame-global encoder");
             let command_buffer = command_encoder.finish();
 
             // TODO(andreas): For better performance, we should try to bundle this with the single submit call that is currently happening in eframe.

crates/re_renderer/src/view_builder.rs

@@ -448,6 +448,8 @@ impl ViewBuilder {
         phase: DrawPhase,
         pass: &mut wgpu::RenderPass<'a>,
     ) {
+        crate::profile_function!();
+
         for queued_draw in &self.queued_draws {
             if queued_draw.participated_phases.contains(&phase) {
                 let res = (queued_draw.draw_func)(ctx, phase, pass, queued_draw.draw_data.as_ref())

crates/re_renderer/src/wgpu_resources/dynamic_resource_pool.rs

@@ -89,6 +89,7 @@ where
         desc: &Desc,
         creation_func: F,
     ) -> Arc<DynamicResource<Handle, Desc, Res>> {
+        crate::profile_function!();
         let mut state = self.state.write();
 
         // First check if we can reclaim a resource we have around from a previous frame.
@@ -143,6 +144,7 @@ where
     }
 
     pub fn begin_frame(&mut self, frame_index: u64, mut on_destroy_resource: impl FnMut(&Res)) {
+        crate::profile_function!();
         self.current_frame_index = frame_index;
         let state = self.state.get_mut();
 

crates/re_renderer/src/wgpu_resources/render_pipeline_pool.rs

@@ -161,6 +161,7 @@ impl GpuRenderPipelinePool {
         shader_modules: &mut GpuShaderModulePool,
         pipeline_layouts: &mut GpuPipelineLayoutPool,
     ) {
+        crate::profile_function!();
         self.pool.current_frame_index = frame_index;
 
         // Recompile render pipelines referencing shader modules that have been recompiled this frame.

crates/re_renderer/src/wgpu_resources/static_resource_pool.rs

@@ -51,6 +51,7 @@ where
         creation_func: F,
     ) -> Handle {
         *self.lookup.entry(desc.clone()).or_insert_with(|| {
+            crate::profile_scope!("Creating new static resource", std::any::type_name::<Res>());
             re_log::debug!(?desc, "Created new resource");
             let resource = creation_func(desc);
             self.resources.insert(StoredResource {
@@ -64,6 +65,8 @@ where
     }
 
     pub fn recreate_resources<F: FnMut(&Desc) -> Option<Res>>(&mut self, mut recreation_func: F) {
+        crate::profile_function!();
+
         for (desc, handle) in &self.lookup {
             if let Some(new_resource) = recreation_func(desc) {
                 let resource = self.resources.get_mut(*handle).unwrap();