oak-gpui/crates/gpui2/src/color.rs

#![allow(dead_code)]

use serde::de::{self, Deserialize, Deserializer, Visitor};
use smallvec::SmallVec;
use std::fmt;
use std::{num::ParseIntError, ops::Range};

pub fn rgb<C: From<Rgba>>(hex: u32) -> C {
    let r = ((hex >> 16) & 0xFF) as f32 / 255.0;
    let g = ((hex >> 8) & 0xFF) as f32 / 255.0;
    let b = (hex & 0xFF) as f32 / 255.0;
    Rgba { r, g, b, a: 1.0 }.into()
}

#[derive(Clone, Copy, Default, Debug)]
pub struct Rgba {
    pub r: f32,
    pub g: f32,
    pub b: f32,
    pub a: f32,
}

struct RgbaVisitor;

impl<'de> Visitor<'de> for RgbaVisitor {
    type Value = Rgba;

    fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
        formatter.write_str("a string in the format #rrggbb or #rrggbbaa")
    }

    fn visit_str<E: de::Error>(self, value: &str) -> Result<Rgba, E> {
        if value.len() == 7 || value.len() == 9 {
            let r = u8::from_str_radix(&value[1..3], 16).unwrap() as f32 / 255.0;
            let g = u8::from_str_radix(&value[3..5], 16).unwrap() as f32 / 255.0;
            let b = u8::from_str_radix(&value[5..7], 16).unwrap() as f32 / 255.0;
            let a = if value.len() == 9 {
                u8::from_str_radix(&value[7..9], 16).unwrap() as f32 / 255.0
            } else {
                1.0
            };
            Ok(Rgba { r, g, b, a })
        } else {
            Err(E::custom(
                "Bad format for RGBA. Expected #rrggbb or #rrggbbaa.",
            ))
        }
    }
}

impl<'de> Deserialize<'de> for Rgba {
    fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
        deserializer.deserialize_str(RgbaVisitor)
    }
}

pub trait Lerp {
    fn lerp(&self, level: f32) -> Hsla;
}

impl Lerp for Range<Hsla> {
    fn lerp(&self, level: f32) -> Hsla {
        let level = level.clamp(0., 1.);
        Hsla {
            h: self.start.h + (level * (self.end.h - self.start.h)),
            s: self.start.s + (level * (self.end.s - self.start.s)),
            l: self.start.l + (level * (self.end.l - self.start.l)),
            a: self.start.a + (level * (self.end.a - self.start.a)),
        }
    }
}

impl From<gpui::color::Color> for Rgba {
    fn from(value: gpui::color::Color) -> Self {
        Self {
            r: value.0.r as f32 / 255.0,
            g: value.0.g as f32 / 255.0,
            b: value.0.b as f32 / 255.0,
            a: value.0.a as f32 / 255.0,
        }
    }
}

impl From<Hsla> for Rgba {
    fn from(color: Hsla) -> Self {
        let h = color.h;
        let s = color.s;
        let l = color.l;

        let c = (1.0 - (2.0 * l - 1.0).abs()) * s;
        let x = c * (1.0 - ((h * 6.0) % 2.0 - 1.0).abs());
        let m = l - c / 2.0;
        let cm = c + m;
        let xm = x + m;

        let (r, g, b) = match (h * 6.0).floor() as i32 {
            0 | 6 => (cm, xm, m),
            1 => (xm, cm, m),
            2 => (m, cm, xm),
            3 => (m, xm, cm),
            4 => (xm, m, cm),
            _ => (cm, m, xm),
        };

        Rgba {
            r,
            g,
            b,
            a: color.a,
        }
    }
}

impl TryFrom<&'_ str> for Rgba {
    type Error = ParseIntError;

    fn try_from(value: &'_ str) -> Result<Self, Self::Error> {
        let r = u8::from_str_radix(&value[1..3], 16)? as f32 / 255.0;
        let g = u8::from_str_radix(&value[3..5], 16)? as f32 / 255.0;
        let b = u8::from_str_radix(&value[5..7], 16)? as f32 / 255.0;
        let a = if value.len() > 7 {
            u8::from_str_radix(&value[7..9], 16)? as f32 / 255.0
        } else {
            1.0
        };

        Ok(Rgba { r, g, b, a })
    }
}

impl Into<gpui::color::Color> for Rgba {
    fn into(self) -> gpui::color::Color {
        gpui::color::rgba(self.r, self.g, self.b, self.a)
    }
}

#[derive(Default, Copy, Clone, Debug, PartialEq)]
pub struct Hsla {
    pub h: f32,
    pub s: f32,
    pub l: f32,
    pub a: f32,
}

pub fn hsla(h: f32, s: f32, l: f32, a: f32) -> Hsla {
    Hsla {
        h: h.clamp(0., 1.),
        s: s.clamp(0., 1.),
        l: l.clamp(0., 1.),
        a: a.clamp(0., 1.),
    }
}

pub fn black() -> Hsla {
    Hsla {
        h: 0.,
        s: 0.,
        l: 0.,
        a: 1.,
    }
}

impl From<Rgba> for Hsla {
    fn from(color: Rgba) -> Self {
        let r = color.r;
        let g = color.g;
        let b = color.b;

        let max = r.max(g.max(b));
        let min = r.min(g.min(b));
        let delta = max - min;

        let l = (max + min) / 2.0;
        let s = if l == 0.0 || l == 1.0 {
            0.0
        } else if l < 0.5 {
            delta / (2.0 * l)
        } else {
            delta / (2.0 - 2.0 * l)
        };

        let h = if delta == 0.0 {
            0.0
        } else if max == r {
            ((g - b) / delta).rem_euclid(6.0) / 6.0
        } else if max == g {
            ((b - r) / delta + 2.0) / 6.0
        } else {
            ((r - g) / delta + 4.0) / 6.0
        };

        Hsla {
            h,
            s,
            l,
            a: color.a,
        }
    }
}

impl Hsla {
    /// Scales the saturation and lightness by the given values, clamping at 1.0.
    pub fn scale_sl(mut self, s: f32, l: f32) -> Self {
        self.s = (self.s * s).clamp(0., 1.);
        self.l = (self.l * l).clamp(0., 1.);
        self
    }

    /// Increases the saturation of the color by a certain amount, with a max
    /// value of 1.0.
    pub fn saturate(mut self, amount: f32) -> Self {
        self.s += amount;
        self.s = self.s.clamp(0.0, 1.0);
        self
    }

    /// Decreases the saturation of the color by a certain amount, with a min
    /// value of 0.0.
    pub fn desaturate(mut self, amount: f32) -> Self {
        self.s -= amount;
        self.s = self.s.max(0.0);
        if self.s < 0.0 {
            self.s = 0.0;
        }
        self
    }

    /// Brightens the color by increasing the lightness by a certain amount,
    /// with a max value of 1.0.
    pub fn brighten(mut self, amount: f32) -> Self {
        self.l += amount;
        self.l = self.l.clamp(0.0, 1.0);
        self
    }

    /// Darkens the color by decreasing the lightness by a certain amount,
    /// with a max value of 0.0.
    pub fn darken(mut self, amount: f32) -> Self {
        self.l -= amount;
        self.l = self.l.clamp(0.0, 1.0);
        self
    }
}

impl From<gpui::color::Color> for Hsla {
    fn from(value: gpui::color::Color) -> Self {
        Rgba::from(value).into()
    }
}

impl Into<gpui::color::Color> for Hsla {
    fn into(self) -> gpui::color::Color {
        Rgba::from(self).into()
    }
}

impl<'de> Deserialize<'de> for Hsla {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        // First, deserialize it into Rgba
        let rgba = Rgba::deserialize(deserializer)?;

        // Then, use the From<Rgba> for Hsla implementation to convert it
        Ok(Hsla::from(rgba))
    }
}

pub struct ColorScale {
    colors: SmallVec<[Hsla; 2]>,
    positions: SmallVec<[f32; 2]>,
}

pub fn scale<I, C>(colors: I) -> ColorScale
where
    I: IntoIterator<Item = C>,
    C: Into<Hsla>,
{
    let mut scale = ColorScale {
        colors: colors.into_iter().map(Into::into).collect(),
        positions: SmallVec::new(),
    };
    let num_colors: f32 = scale.colors.len() as f32 - 1.0;
    scale.positions = (0..scale.colors.len())
        .map(|i| i as f32 / num_colors)
        .collect();
    scale
}

impl ColorScale {
    fn at(&self, t: f32) -> Hsla {
        // Ensure that the input is within [0.0, 1.0]
        debug_assert!(
            0.0 <= t && t <= 1.0,
            "t value {} is out of range. Expected value in range 0.0 to 1.0",
            t
        );

        let position = match self
            .positions
            .binary_search_by(|a| a.partial_cmp(&t).unwrap())
        {
            Ok(index) | Err(index) => index,
        };
        let lower_bound = position.saturating_sub(1);
        let upper_bound = position.min(self.colors.len() - 1);
        let lower_color = &self.colors[lower_bound];
        let upper_color = &self.colors[upper_bound];

        match upper_bound.checked_sub(lower_bound) {
            Some(0) | None => *lower_color,
            Some(_) => {
                let interval_t = (t - self.positions[lower_bound])
                    / (self.positions[upper_bound] - self.positions[lower_bound]);
                let h = lower_color.h + interval_t * (upper_color.h - lower_color.h);
                let s = lower_color.s + interval_t * (upper_color.s - lower_color.s);
                let l = lower_color.l + interval_t * (upper_color.l - lower_color.l);
                let a = lower_color.a + interval_t * (upper_color.a - lower_color.a);
                Hsla { h, s, l, a }
            }
        }
    }
}