color-management: add more capabilities

src/ifs/wl_surface/wp_color_management_surface_v1.rs

@@ -0,0 +1,91 @@
+use {
+    crate::{
+        client::{Client, ClientError},
+        ifs::{color_management, wl_surface::WlSurface},
+        leaks::Tracker,
+        object::{Object, Version},
+        wire::{
+            WpColorManagementSurfaceV1Id,
+            wp_color_management_surface_v1::{
+                Destroy, SetImageDescription, UnsetImageDescription,
+                WpColorManagementSurfaceV1RequestHandler,
+            },
+        },
+    },
+    std::rc::Rc,
+    thiserror::Error,
+};
+
+pub struct WpColorManagementSurfaceV1 {
+    pub id: WpColorManagementSurfaceV1Id,
+    pub client: Rc<Client>,
+    pub version: Version,
+    pub tracker: Tracker<Self>,
+    pub surface: Rc<WlSurface>,
+}
+
+impl WpColorManagementSurfaceV1 {
+    pub fn install(self: &Rc<Self>) -> Result<(), WpColorManagementSurfaceV1Error> {
+        if self.surface.color_management_surface.is_some() {
+            return Err(WpColorManagementSurfaceV1Error::HasSurface);
+        }
+        self.surface
+            .color_management_surface
+            .set(Some(self.clone()));
+        Ok(())
+    }
+}
+
+impl WpColorManagementSurfaceV1RequestHandler for WpColorManagementSurfaceV1 {
+    type Error = WpColorManagementSurfaceV1Error;
+
+    fn destroy(&self, _req: Destroy, _slf: &Rc<Self>) -> Result<(), Self::Error> {
+        self.surface.color_management_surface.take();
+        self.client.remove_obj(self)?;
+        Ok(())
+    }
+
+    fn set_image_description(
+        &self,
+        req: SetImageDescription,
+        _slf: &Rc<Self>,
+    ) -> Result<(), Self::Error> {
+        if req.render_intent != color_management::RENDER_INTENT_PERCEPTUAL {
+            return Err(WpColorManagementSurfaceV1Error::UnsupportedRenderIntent(
+                req.render_intent,
+            ));
+        }
+        let desc = self.client.lookup(req.image_description)?;
+        self.surface.pending.borrow_mut().color_description = Some(Some(desc.description.clone()));
+        Ok(())
+    }
+
+    fn unset_image_description(
+        &self,
+        _req: UnsetImageDescription,
+        _slf: &Rc<Self>,
+    ) -> Result<(), Self::Error> {
+        self.surface.pending.borrow_mut().color_description = Some(None);
+        Ok(())
+    }
+}
+
+object_base! {
+    self = WpColorManagementSurfaceV1;
+    version = self.version;
+}
+
+impl Object for WpColorManagementSurfaceV1 {}
+
+simple_add_obj!(WpColorManagementSurfaceV1);
+
+#[derive(Debug, Error)]
+pub enum WpColorManagementSurfaceV1Error {
+    #[error(transparent)]
+    ClientError(Box<ClientError>),
+    #[error("{} is not a supported render intent", .0)]
+    UnsupportedRenderIntent(u32),
+    #[error("wl_surface already has a color-management extension")]
+    HasSurface,
+}
+efrom!(WpColorManagementSurfaceV1Error, ClientError);