-
Notifications
You must be signed in to change notification settings - Fork 969
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
b452cab
commit 3bba040
Showing
11 changed files
with
293 additions
and
142 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,167 @@ | ||
| use std::sync::atomic::Ordering; | ||
|
|
||
| use glow::HasContext; | ||
|
|
||
| use crate::AtomicFenceValue; | ||
|
|
||
| #[derive(Debug, Copy, Clone)] | ||
| struct GLFence { | ||
| sync: glow::Fence, | ||
| value: crate::FenceValue, | ||
| } | ||
|
|
||
| #[derive(Debug)] | ||
| pub struct Fence { | ||
| last_completed: AtomicFenceValue, | ||
| pending: Vec<GLFence>, | ||
| fence_mode: wgt::GlShortCircuitFences, | ||
| } | ||
|
|
||
| impl crate::DynFence for Fence {} | ||
|
|
||
| #[cfg(send_sync)] | ||
| unsafe impl Send for Fence {} | ||
| #[cfg(send_sync)] | ||
| unsafe impl Sync for Fence {} | ||
|
|
||
| impl Fence { | ||
| pub fn new(options: &wgt::GlBackendOptions) -> Self { | ||
| Self { | ||
| last_completed: AtomicFenceValue::new(0), | ||
| pending: Vec::new(), | ||
| fence_mode: options.short_circuit_fences, | ||
| } | ||
| } | ||
|
|
||
| pub fn signal( | ||
| &mut self, | ||
| gl: &glow::Context, | ||
| value: crate::FenceValue, | ||
| ) -> Result<(), crate::DeviceError> { | ||
| if self.fence_mode.is_short_circuit() { | ||
| *self.last_completed.get_mut() = value; | ||
| return Ok(()); | ||
| } | ||
|
|
||
| let sync = unsafe { gl.fence_sync(glow::SYNC_GPU_COMMANDS_COMPLETE, 0) } | ||
| .map_err(|_| crate::DeviceError::OutOfMemory)?; | ||
| self.pending.push(GLFence { sync, value }); | ||
|
|
||
| Ok(()) | ||
| } | ||
|
|
||
| pub fn satisfied(&self, value: crate::FenceValue) -> bool { | ||
| self.last_completed.load(Ordering::Relaxed) >= value | ||
| } | ||
|
|
||
| pub fn get_latest(&self, gl: &glow::Context) -> crate::FenceValue { | ||
| let mut max_value = self.last_completed.load(Ordering::Relaxed); | ||
|
|
||
| if self.fence_mode.is_short_circuit() { | ||
| return max_value; | ||
| } | ||
|
|
||
| for gl_fence in self.pending.iter() { | ||
| if gl_fence.value <= max_value { | ||
| // We already know this was good, no need to check again | ||
| continue; | ||
| } | ||
| let status = unsafe { gl.get_sync_status(gl_fence.sync) }; | ||
| if status == glow::SIGNALED { | ||
| max_value = gl_fence.value; | ||
| } else { | ||
| // Anything after the first unsignalled is guaranteed to also be unsignalled | ||
| break; | ||
| } | ||
| } | ||
|
|
||
| // Track the latest value, to save ourselves some querying later | ||
| self.last_completed.fetch_max(max_value, Ordering::Relaxed); | ||
|
|
||
| max_value | ||
| } | ||
|
|
||
| pub fn maintain(&mut self, gl: &glow::Context) { | ||
| if self.fence_mode.is_short_circuit() { | ||
| return; | ||
| } | ||
|
|
||
| let latest = self.get_latest(gl); | ||
| for &gl_fence in self.pending.iter() { | ||
| if gl_fence.value <= latest { | ||
| unsafe { | ||
| gl.delete_sync(gl_fence.sync); | ||
| } | ||
| } | ||
| } | ||
| self.pending.retain(|&gl_fence| gl_fence.value > latest); | ||
| } | ||
|
|
||
| pub fn wait( | ||
| &self, | ||
| gl: &glow::Context, | ||
| wait_value: crate::FenceValue, | ||
| timeout_ns: u64, | ||
| ) -> Result<bool, crate::DeviceError> { | ||
| let last_completed = self.last_completed.load(Ordering::Relaxed); | ||
|
|
||
| if self.fence_mode.is_short_circuit() { | ||
| return Ok(last_completed >= wait_value); | ||
| } | ||
|
|
||
| // We already know this fence has been signalled to that value. Return signalled. | ||
| if last_completed >= wait_value { | ||
| return Ok(true); | ||
| } | ||
|
|
||
| // Find a matching fence | ||
| let gl_fence = self | ||
| .pending | ||
| .iter() | ||
| // Greater or equal as an abundance of caution, but there should be one fence per value | ||
| .find(|gl_fence| gl_fence.value >= wait_value); | ||
|
|
||
| let Some(gl_fence) = gl_fence else { | ||
| log::warn!("Tried to wait for {wait_value} but that value has not been signalled yet"); | ||
| return Ok(false); | ||
| }; | ||
|
|
||
| // We should have found a fence with the exact value. | ||
| debug_assert_eq!(gl_fence.value, wait_value); | ||
|
|
||
| let status = unsafe { | ||
| gl.client_wait_sync( | ||
| gl_fence.sync, | ||
| glow::SYNC_FLUSH_COMMANDS_BIT, | ||
| timeout_ns as i32, | ||
| ) | ||
| }; | ||
|
|
||
| let signalled = match status { | ||
| glow::ALREADY_SIGNALED | glow::CONDITION_SATISFIED => true, | ||
| glow::TIMEOUT_EXPIRED | glow::WAIT_FAILED => false, | ||
| _ => { | ||
| log::warn!("Unexpected result from client_wait_sync: {status}"); | ||
| false | ||
| } | ||
| }; | ||
|
|
||
| if signalled { | ||
| self.last_completed.fetch_max(wait_value, Ordering::Relaxed); | ||
| } | ||
|
|
||
| Ok(signalled) | ||
| } | ||
|
|
||
| pub fn destroy(self, gl: &glow::Context) { | ||
| if self.fence_mode.is_short_circuit() { | ||
| return; | ||
| } | ||
|
|
||
| for gl_fence in self.pending { | ||
| unsafe { | ||
| gl.delete_sync(gl_fence.sync); | ||
| } | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.