Skip to content

Commit

Permalink
wayland-server: inline a few more common functions
Browse files Browse the repository at this point in the history
these have shown up in a lot of traces in smithay
  • Loading branch information
cmeissl committed May 8, 2024
1 parent d768d5e commit 85f69b8
Show file tree
Hide file tree
Showing 4 changed files with 32 additions and 0 deletions.
2 changes: 2 additions & 0 deletions wayland-backend/src/server_api.rs
Original file line number Diff line number Diff line change
Expand Up @@ -249,13 +249,15 @@ impl WeakHandle {
/// Try to upgrade this weak handle to a [`Handle`]
///
/// Returns `None` if the associated backend was already dropped.
#[inline]
pub fn upgrade(&self) -> Option<Handle> {
self.handle.upgrade().map(|handle| Handle { handle })
}
}

impl Handle {
/// Get a [`WeakHandle`] from this handle
#[inline]
pub fn downgrade(&self) -> WeakHandle {
WeakHandle { handle: self.handle.downgrade() }
}
Expand Down
4 changes: 4 additions & 0 deletions wayland-scanner/src/server_gen.rs
Original file line number Diff line number Diff line change
Expand Up @@ -83,6 +83,7 @@ fn generate_objects_for(interface: &Interface) -> TokenStream {
}

impl std::cmp::PartialEq for #iface_name {
#[inline]
fn eq(&self, other: &#iface_name) -> bool {
self.id == other.id
}
Expand All @@ -91,18 +92,21 @@ fn generate_objects_for(interface: &Interface) -> TokenStream {
impl std::cmp::Eq for #iface_name {}

impl PartialEq<Weak<#iface_name>> for #iface_name {
#[inline]
fn eq(&self, other: &Weak<#iface_name>) -> bool {
self.id == other.id()
}
}

impl std::borrow::Borrow<ObjectId> for #iface_name {
#[inline]
fn borrow(&self) -> &ObjectId {
&self.id
}
}

impl std::hash::Hash for #iface_name {
#[inline]
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
self.id.hash(state)
}
Expand Down
20 changes: 20 additions & 0 deletions wayland-scanner/tests/scanner_assets/test-server-code.rs
Original file line number Diff line number Diff line change
Expand Up @@ -54,22 +54,26 @@ pub mod wl_callback {
handle: WeakHandle,
}
impl std::cmp::PartialEq for WlCallback {
#[inline]
fn eq(&self, other: &WlCallback) -> bool {
self.id == other.id
}
}
impl std::cmp::Eq for WlCallback {}
impl PartialEq<Weak<WlCallback>> for WlCallback {
#[inline]
fn eq(&self, other: &Weak<WlCallback>) -> bool {
self.id == other.id()
}
}
impl std::borrow::Borrow<ObjectId> for WlCallback {
#[inline]
fn borrow(&self) -> &ObjectId {
&self.id
}
}
impl std::hash::Hash for WlCallback {
#[inline]
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
self.id.hash(state)
}
Expand Down Expand Up @@ -316,22 +320,26 @@ pub mod test_global {
handle: WeakHandle,
}
impl std::cmp::PartialEq for TestGlobal {
#[inline]
fn eq(&self, other: &TestGlobal) -> bool {
self.id == other.id
}
}
impl std::cmp::Eq for TestGlobal {}
impl PartialEq<Weak<TestGlobal>> for TestGlobal {
#[inline]
fn eq(&self, other: &Weak<TestGlobal>) -> bool {
self.id == other.id()
}
}
impl std::borrow::Borrow<ObjectId> for TestGlobal {
#[inline]
fn borrow(&self) -> &ObjectId {
&self.id
}
}
impl std::hash::Hash for TestGlobal {
#[inline]
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
self.id.hash(state)
}
Expand Down Expand Up @@ -822,22 +830,26 @@ pub mod secondary {
handle: WeakHandle,
}
impl std::cmp::PartialEq for Secondary {
#[inline]
fn eq(&self, other: &Secondary) -> bool {
self.id == other.id
}
}
impl std::cmp::Eq for Secondary {}
impl PartialEq<Weak<Secondary>> for Secondary {
#[inline]
fn eq(&self, other: &Weak<Secondary>) -> bool {
self.id == other.id()
}
}
impl std::borrow::Borrow<ObjectId> for Secondary {
#[inline]
fn borrow(&self) -> &ObjectId {
&self.id
}
}
impl std::hash::Hash for Secondary {
#[inline]
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
self.id.hash(state)
}
Expand Down Expand Up @@ -981,22 +993,26 @@ pub mod tertiary {
handle: WeakHandle,
}
impl std::cmp::PartialEq for Tertiary {
#[inline]
fn eq(&self, other: &Tertiary) -> bool {
self.id == other.id
}
}
impl std::cmp::Eq for Tertiary {}
impl PartialEq<Weak<Tertiary>> for Tertiary {
#[inline]
fn eq(&self, other: &Weak<Tertiary>) -> bool {
self.id == other.id()
}
}
impl std::borrow::Borrow<ObjectId> for Tertiary {
#[inline]
fn borrow(&self) -> &ObjectId {
&self.id
}
}
impl std::hash::Hash for Tertiary {
#[inline]
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
self.id.hash(state)
}
Expand Down Expand Up @@ -1140,22 +1156,26 @@ pub mod quad {
handle: WeakHandle,
}
impl std::cmp::PartialEq for Quad {
#[inline]
fn eq(&self, other: &Quad) -> bool {
self.id == other.id
}
}
impl std::cmp::Eq for Quad {}
impl PartialEq<Weak<Quad>> for Quad {
#[inline]
fn eq(&self, other: &Weak<Quad>) -> bool {
self.id == other.id()
}
}
impl std::borrow::Borrow<ObjectId> for Quad {
#[inline]
fn borrow(&self) -> &ObjectId {
&self.id
}
}
impl std::hash::Hash for Quad {
#[inline]
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
self.id.hash(state)
}
Expand Down
6 changes: 6 additions & 0 deletions wayland-server/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -160,6 +160,7 @@ pub trait Resource: Clone + std::fmt::Debug + Sized {
fn version(&self) -> u32;

/// Checks if the Wayland object associated with this proxy is still alive
#[inline]
fn is_alive(&self) -> bool {
if let Some(handle) = self.handle().upgrade() {
handle.object_info(self.id()).is_ok()
Expand Down Expand Up @@ -231,6 +232,7 @@ pub trait Resource: Clone + std::fmt::Debug + Sized {
///
/// This can be of use if you need to store resources in the used data of other objects and want
/// to be sure to avoid reference cycles that would cause memory leaks.
#[inline]
fn downgrade(&self) -> Weak<Self> {
Weak { handle: self.handle().clone(), id: self.id(), _iface: std::marker::PhantomData }
}
Expand Down Expand Up @@ -285,6 +287,7 @@ impl<I: Resource> Weak<I> {
/// This will fail if either:
/// - the object represented by this handle has already been destroyed at the protocol level
/// - the Wayland connection has already been closed
#[inline]
pub fn upgrade(&self) -> Result<I, InvalidId> {
let handle = self.handle.upgrade().ok_or(InvalidId)?;
// Check if the object has been destroyed
Expand Down Expand Up @@ -313,6 +316,7 @@ impl<I: Resource> Weak<I> {
}

impl<I> PartialEq for Weak<I> {
#[inline]
fn eq(&self, other: &Self) -> bool {
self.id == other.id
}
Expand All @@ -321,12 +325,14 @@ impl<I> PartialEq for Weak<I> {
impl<I> Eq for Weak<I> {}

impl<I> Hash for Weak<I> {
#[inline]
fn hash<H: Hasher>(&self, state: &mut H) {
self.id.hash(state);
}
}

impl<I: Resource> PartialEq<I> for Weak<I> {
#[inline]
fn eq(&self, other: &I) -> bool {
self.id == other.id()
}
Expand Down

0 comments on commit 85f69b8

Please sign in to comment.