WIP: preserve aspect ratio of images

Antonio Scandurra created

Change summary

crates/call2/src/shared_screen.rs   |   2 
crates/gpui2/src/elements/img.rs    | 134 ++++++++++++++++++------------
crates/gpui2/src/geometry.rs        |   6 +
crates/ui2/src/components/avatar.rs |   5 
4 files changed, 88 insertions(+), 59 deletions(-)

Detailed changes

crates/call2/src/shared_screen.rs 🔗

@@ -72,7 +72,7 @@ impl Render for SharedScreen {
         // self.current_frame_id = self.current_frame_id.wrapping_add(1);
         div()
             .size_full()
-            .children(frame.map(|frame| img().size_full().surface(frame.image())))
+            .children(frame.map(|frame| img(frame.image()).w_full()))
     }
 }
 // impl View for SharedScreen {

crates/gpui2/src/elements/img.rs 🔗

@@ -1,8 +1,8 @@
 use std::sync::Arc;
 
 use crate::{
-    Bounds, Element, ImageData, InteractiveElement, InteractiveElementState, Interactivity,
-    IntoElement, LayoutId, Pixels, SharedString, StyleRefinement, Styled, WindowContext,
+    size, Bounds, Element, ImageData, InteractiveElement, InteractiveElementState, Interactivity,
+    IntoElement, LayoutId, Pixels, SharedString, Size, StyleRefinement, Styled, WindowContext,
 };
 use futures::FutureExt;
 use media::core_video::CVImageBuffer;
@@ -36,38 +36,19 @@ impl From<CVImageBuffer> for ImageSource {
 
 pub struct Img {
     interactivity: Interactivity,
-    source: Option<ImageSource>,
+    source: ImageSource,
     grayscale: bool,
 }
 
-pub fn img() -> Img {
+pub fn img(source: impl Into<ImageSource>) -> Img {
     Img {
         interactivity: Interactivity::default(),
-        source: None,
+        source: source.into(),
         grayscale: false,
     }
 }
 
 impl Img {
-    pub fn uri(mut self, uri: impl Into<SharedString>) -> Self {
-        self.source = Some(ImageSource::from(uri.into()));
-        self
-    }
-
-    pub fn data(mut self, data: Arc<ImageData>) -> Self {
-        self.source = Some(ImageSource::from(data));
-        self
-    }
-
-    pub fn surface(mut self, data: CVImageBuffer) -> Self {
-        self.source = Some(ImageSource::from(data));
-        self
-    }
-
-    pub fn source(mut self, source: impl Into<ImageSource>) -> Self {
-        self.source = Some(source.into());
-        self
-    }
     pub fn grayscale(mut self, grayscale: bool) -> Self {
         self.grayscale = grayscale;
         self
@@ -83,7 +64,52 @@ impl Element for Img {
         cx: &mut WindowContext,
     ) -> (LayoutId, Self::State) {
         self.interactivity.layout(element_state, cx, |style, cx| {
-            cx.request_layout(&style, None)
+            let image_size = match &self.source {
+                ImageSource::Uri(uri) => {
+                    let image_future = cx.image_cache.get(uri.clone());
+                    if let Some(data) = image_future
+                        .clone()
+                        .now_or_never()
+                        .and_then(|result| result.ok())
+                    {
+                        data.size().map(|pixels| Pixels::from(u32::from(pixels)))
+                    } else {
+                        Size::default()
+                    }
+                }
+
+                ImageSource::Data(data) => {
+                    data.size().map(|pixels| Pixels::from(u32::from(pixels)))
+                }
+
+                ImageSource::Surface(surface) => {
+                    size(surface.width().into(), surface.height().into())
+                }
+            };
+            dbg!(image_size);
+
+            cx.request_measured_layout(
+                style,
+                cx.rem_size(),
+                move |known_dimensions, available_space| match dbg!(
+                    known_dimensions.width,
+                    known_dimensions.height,
+                ) {
+                    (None, None) => image_size,
+
+                    (None, Some(height)) => {
+                        let aspect_ratio = height / image_size.height;
+                        size(image_size.width * aspect_ratio, height)
+                    }
+
+                    (Some(width), None) => {
+                        let aspect_ratio = width / image_size.width;
+                        size(width, image_size.height * aspect_ratio)
+                    }
+
+                    (Some(width), Some(height)) => size(width, height),
+                },
+            )
         })
     }
 
@@ -101,38 +127,36 @@ impl Element for Img {
             |style, _scroll_offset, cx| {
                 let corner_radii = style.corner_radii.to_pixels(bounds.size, cx.rem_size());
                 cx.with_z_index(1, |cx| {
-                    if let Some(source) = self.source {
-                        match source {
-                            ImageSource::Uri(uri) => {
-                                let image_future = cx.image_cache.get(uri.clone());
-                                if let Some(data) = image_future
-                                    .clone()
-                                    .now_or_never()
-                                    .and_then(|result| result.ok())
-                                {
-                                    cx.paint_image(bounds, corner_radii, data, self.grayscale)
-                                        .log_err();
-                                } else {
-                                    cx.spawn(|mut cx| async move {
-                                        if image_future.await.ok().is_some() {
-                                            cx.on_next_frame(|cx| cx.notify());
-                                        }
-                                    })
-                                    .detach();
-                                }
-                            }
-
-                            ImageSource::Data(image) => {
-                                cx.paint_image(bounds, corner_radii, image, self.grayscale)
+                    match self.source {
+                        ImageSource::Uri(uri) => {
+                            let image_future = cx.image_cache.get(uri.clone());
+                            if let Some(data) = image_future
+                                .clone()
+                                .now_or_never()
+                                .and_then(|result| result.ok())
+                            {
+                                cx.paint_image(bounds, corner_radii, data, self.grayscale)
                                     .log_err();
+                            } else {
+                                cx.spawn(|mut cx| async move {
+                                    if image_future.await.ok().is_some() {
+                                        cx.on_next_frame(|cx| cx.notify());
+                                    }
+                                })
+                                .detach();
                             }
-
-                            ImageSource::Surface(surface) => {
-                                // TODO: Add support for corner_radii and grayscale.
-                                cx.paint_surface(bounds, surface);
-                            }
-                        };
-                    }
+                        }
+
+                        ImageSource::Data(image) => {
+                            cx.paint_image(bounds, corner_radii, image, self.grayscale)
+                                .log_err();
+                        }
+
+                        ImageSource::Surface(surface) => {
+                            // TODO: Add support for corner_radii and grayscale.
+                            cx.paint_surface(bounds, surface);
+                        }
+                    };
                 });
             },
         )

crates/gpui2/src/geometry.rs 🔗

@@ -905,6 +905,12 @@ impl From<Pixels> for usize {
     }
 }
 
+impl From<usize> for Pixels {
+    fn from(pixels: usize) -> Self {
+        Pixels(pixels as f32)
+    }
+}
+
 #[derive(
     Add, AddAssign, Clone, Copy, Default, Div, Eq, Hash, Ord, PartialEq, PartialOrd, Sub, SubAssign,
 )]

crates/ui2/src/components/avatar.rs 🔗

@@ -20,7 +20,7 @@ impl RenderOnce for Avatar {
     type Rendered = Img;
 
     fn render(self, _: &mut WindowContext) -> Self::Rendered {
-        let mut img = img();
+        let mut img = img(self.src);
 
         if self.shape == Shape::Circle {
             img = img.rounded_full();
@@ -28,8 +28,7 @@ impl RenderOnce for Avatar {
             img = img.rounded_md();
         }
 
-        img.source(self.src.clone())
-            .size_4()
+        img.size_4()
             // todo!(Pull the avatar fallback background from the theme.)
             .bg(gpui::red())
     }