feat: wayland

src/wayland/core/buffer.rs

@@ -0,0 +1,89 @@
+use std::sync::Arc;
+
+#[cfg(feature = "dmabuf")]
+use crate::wayland::dmabuf::buffer_backing::DmabufBacking;
+use crate::{
+	core::registry::Registry,
+	wayland::{GraphicsInfo, core::shm_buffer_backing::ShmBufferBacking},
+};
+use bevy::{
+	asset::{Assets, Handle},
+	image::Image,
+};
+use mint::Vector2;
+pub use waynest::server::protocol::core::wayland::wl_buffer::*;
+use waynest::{
+	server::{Client, Dispatcher, Result},
+	wire::ObjectId,
+};
+
+pub static BUFFER_REGISTRY: Registry<Buffer> = Registry::new();
+
+#[derive(Debug)]
+pub enum BufferBacking {
+	Shm(ShmBufferBacking),
+	#[cfg(feature = "dmabuf")]
+	Dmabuf(DmabufBacking),
+}
+impl BufferBacking {
+	// Returns true if the buffer can be released immediately after texture update
+	pub fn can_release_after_update(&self) -> bool {
+		matches!(self, BufferBacking::Shm(_))
+	}
+}
+
+#[derive(Debug, Dispatcher)]
+pub struct Buffer {
+	pub id: ObjectId,
+	backing: BufferBacking,
+}
+
+impl Buffer {
+	pub fn new(client: &mut Client, id: ObjectId, backing: BufferBacking) -> Arc<Self> {
+		let buffer = client.insert(id, Self { id, backing });
+		BUFFER_REGISTRY.add_raw(&buffer);
+		buffer
+	}
+
+	#[allow(unused)]
+	pub fn init_tex(self: Arc<Self>, graphics_info: &mut GraphicsInfo) {
+		match &self.backing {
+			BufferBacking::Shm(_) => (),
+			#[cfg(feature = "dmabuf")]
+			BufferBacking::Dmabuf(backing) => backing.init_tex(graphics_info, self.clone()),
+		}
+	}
+
+	/// Returns the tex if it was updated
+	pub fn update_tex(&self, images: &mut Assets<Image>) -> Option<Handle<Image>> {
+		tracing::debug!("Updating texture for buffer {:?}", self.id);
+		match &self.backing {
+			BufferBacking::Shm(backing) => backing.update_tex(images),
+			#[cfg(feature = "dmabuf")]
+			BufferBacking::Dmabuf(backing) => backing
+				.get_tex()
+				.map(|tex| tex.get_id().to_string())
+				.and_then(|tex_id| Tex::find(tex_id).ok()),
+		}
+	}
+
+	pub fn can_release_after_update(&self) -> bool {
+		self.backing.can_release_after_update()
+	}
+
+	pub fn size(&self) -> Vector2<usize> {
+		match &self.backing {
+			BufferBacking::Shm(backing) => backing.size(),
+			#[cfg(feature = "dmabuf")]
+			BufferBacking::Dmabuf(backing) => backing.size(),
+		}
+	}
+}
+
+impl WlBuffer for Buffer {
+	/// https://wayland.app/protocols/wayland#wl_buffer:request:destroy
+	async fn destroy(&self, _client: &mut Client, _sender_id: ObjectId) -> Result<()> {
+		tracing::info!("Destroying buffer {:?}", self.id);
+		Ok(())
+	}
+}