use anyhow::{bail, Context}; use serde::de::{self, Deserialize, Deserializer, Visitor}; use std::{ fmt::{self, Display, Formatter}, hash::{Hash, Hasher}, }; /// Convert an RGB hex color code number to a color type pub fn rgb(hex: u32) -> Rgba { 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 } } /// Convert an RGBA hex color code number to [`Rgba`] pub fn rgba(hex: u32) -> Rgba { let r = ((hex >> 24) & 0xFF) as f32 / 255.0; let g = ((hex >> 16) & 0xFF) as f32 / 255.0; let b = ((hex >> 8) & 0xFF) as f32 / 255.0; let a = (hex & 0xFF) as f32 / 255.0; Rgba { r, g, b, a } } /// Swap from RGBA with premultiplied alpha to BGRA pub(crate) fn swap_rgba_pa_to_bgra(color: &mut [u8]) { color.swap(0, 2); if color[3] > 0 { let a = color[3] as f32 / 255.; color[0] = (color[0] as f32 / a) as u8; color[1] = (color[1] as f32 / a) as u8; color[2] = (color[2] as f32 / a) as u8; } } /// An RGBA color #[derive(PartialEq, Clone, Copy, Default)] pub struct Rgba { /// The red component of the color, in the range 0.0 to 1.0 pub r: f32, /// The green component of the color, in the range 0.0 to 1.0 pub g: f32, /// The blue component of the color, in the range 0.0 to 1.0 pub b: f32, /// The alpha component of the color, in the range 0.0 to 1.0 pub a: f32, } impl fmt::Debug for Rgba { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { write!(f, "rgba({:#010x})", u32::from(*self)) } } impl Rgba { /// Create a new [`Rgba`] color by blending this and another color together pub fn blend(&self, other: Rgba) -> Self { if other.a >= 1.0 { other } else if other.a <= 0.0 { return *self; } else { return Rgba { r: (self.r * (1.0 - other.a)) + (other.r * other.a), g: (self.g * (1.0 - other.a)) + (other.g * other.a), b: (self.b * (1.0 - other.a)) + (other.b * other.a), a: self.a, }; } } } impl From for u32 { fn from(rgba: Rgba) -> Self { let r = (rgba.r * 255.0) as u32; let g = (rgba.g * 255.0) as u32; let b = (rgba.b * 255.0) as u32; let a = (rgba.a * 255.0) as u32; (r << 24) | (g << 16) | (b << 8) | a } } 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(self, value: &str) -> Result { Rgba::try_from(value).map_err(E::custom) } } impl<'de> Deserialize<'de> for Rgba { fn deserialize>(deserializer: D) -> Result { deserializer.deserialize_str(RgbaVisitor) } } impl From 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 = anyhow::Error; fn try_from(value: &'_ str) -> Result { const RGB: usize = "rgb".len(); const RGBA: usize = "rgba".len(); const RRGGBB: usize = "rrggbb".len(); const RRGGBBAA: usize = "rrggbbaa".len(); const EXPECTED_FORMATS: &str = "Expected #rgb, #rgba, #rrggbb, or #rrggbbaa"; const INVALID_UNICODE: &str = "invalid unicode characters in color"; let Some(("", hex)) = value.trim().split_once('#') else { bail!("invalid RGBA hex color: '{value}'. {EXPECTED_FORMATS}"); }; let (r, g, b, a) = match hex.len() { RGB | RGBA => { let r = u8::from_str_radix( hex.get(0..1).with_context(|| { format!("{INVALID_UNICODE}: r component of #rgb/#rgba for value: '{value}'") })?, 16, )?; let g = u8::from_str_radix( hex.get(1..2).with_context(|| { format!("{INVALID_UNICODE}: g component of #rgb/#rgba for value: '{value}'") })?, 16, )?; let b = u8::from_str_radix( hex.get(2..3).with_context(|| { format!("{INVALID_UNICODE}: b component of #rgb/#rgba for value: '{value}'") })?, 16, )?; let a = if hex.len() == RGBA { u8::from_str_radix( hex.get(3..4).with_context(|| { format!("{INVALID_UNICODE}: a component of #rgba for value: '{value}'") })?, 16, )? } else { 0xf }; /// Duplicates a given hex digit. /// E.g., `0xf` -> `0xff`. const fn duplicate(value: u8) -> u8 { value << 4 | value } (duplicate(r), duplicate(g), duplicate(b), duplicate(a)) } RRGGBB | RRGGBBAA => { let r = u8::from_str_radix( hex.get(0..2).with_context(|| { format!( "{}: r component of #rrggbb/#rrggbbaa for value: '{}'", INVALID_UNICODE, value ) })?, 16, )?; let g = u8::from_str_radix( hex.get(2..4).with_context(|| { format!( "{INVALID_UNICODE}: g component of #rrggbb/#rrggbbaa for value: '{value}'" ) })?, 16, )?; let b = u8::from_str_radix( hex.get(4..6).with_context(|| { format!( "{INVALID_UNICODE}: b component of #rrggbb/#rrggbbaa for value: '{value}'" ) })?, 16, )?; let a = if hex.len() == RRGGBBAA { u8::from_str_radix( hex.get(6..8).with_context(|| { format!( "{INVALID_UNICODE}: a component of #rrggbbaa for value: '{value}'" ) })?, 16, )? } else { 0xff }; (r, g, b, a) } _ => bail!("invalid RGBA hex color: '{value}'. {EXPECTED_FORMATS}"), }; Ok(Rgba { r: r as f32 / 255., g: g as f32 / 255., b: b as f32 / 255., a: a as f32 / 255., }) } } /// An HSLA color #[derive(Default, Copy, Clone, Debug)] #[repr(C)] pub struct Hsla { /// Hue, in a range from 0 to 1 pub h: f32, /// Saturation, in a range from 0 to 1 pub s: f32, /// Lightness, in a range from 0 to 1 pub l: f32, /// Alpha, in a range from 0 to 1 pub a: f32, } impl PartialEq for Hsla { fn eq(&self, other: &Self) -> bool { self.h .total_cmp(&other.h) .then(self.s.total_cmp(&other.s)) .then(self.l.total_cmp(&other.l).then(self.a.total_cmp(&other.a))) .is_eq() } } impl PartialOrd for Hsla { fn partial_cmp(&self, other: &Self) -> Option { Some(self.cmp(other)) } } impl Ord for Hsla { fn cmp(&self, other: &Self) -> std::cmp::Ordering { self.h .total_cmp(&other.h) .then(self.s.total_cmp(&other.s)) .then(self.l.total_cmp(&other.l).then(self.a.total_cmp(&other.a))) } } impl Eq for Hsla {} impl Hash for Hsla { fn hash(&self, state: &mut H) { state.write_u32(u32::from_be_bytes(self.h.to_be_bytes())); state.write_u32(u32::from_be_bytes(self.s.to_be_bytes())); state.write_u32(u32::from_be_bytes(self.l.to_be_bytes())); state.write_u32(u32::from_be_bytes(self.a.to_be_bytes())); } } impl Display for Hsla { fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result { write!( f, "hsla({:.2}, {:.2}%, {:.2}%, {:.2})", self.h * 360., self.s * 100., self.l * 100., self.a ) } } /// Construct an [`Hsla`] object from plain values 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.), } } /// Pure black in [`Hsla`] pub const fn black() -> Hsla { Hsla { h: 0., s: 0., l: 0., a: 1., } } /// Transparent black in [`Hsla`] pub const fn transparent_black() -> Hsla { Hsla { h: 0., s: 0., l: 0., a: 0., } } /// Transparent black in [`Hsla`] pub const fn transparent_white() -> Hsla { Hsla { h: 0., s: 0., l: 1., a: 0., } } /// Opaque grey in [`Hsla`], values will be clamped to the range [0, 1] pub fn opaque_grey(lightness: f32, opacity: f32) -> Hsla { Hsla { h: 0., s: 0., l: lightness.clamp(0., 1.), a: opacity.clamp(0., 1.), } } /// Pure white in [`Hsla`] pub const fn white() -> Hsla { Hsla { h: 0., s: 0., l: 1., a: 1., } } /// The color red in [`Hsla`] pub const fn red() -> Hsla { Hsla { h: 0., s: 1., l: 0.5, a: 1., } } /// The color blue in [`Hsla`] pub const fn blue() -> Hsla { Hsla { h: 0.6666666667, s: 1., l: 0.5, a: 1., } } /// The color green in [`Hsla`] pub const fn green() -> Hsla { Hsla { h: 0.3333333333, s: 1., l: 0.25, a: 1., } } /// The color yellow in [`Hsla`] pub const fn yellow() -> Hsla { Hsla { h: 0.1666666667, s: 1., l: 0.5, a: 1., } } impl Hsla { /// Converts this HSLA color to an RGBA color. pub fn to_rgb(self) -> Rgba { self.into() } /// The color red pub const fn red() -> Self { red() } /// The color green pub const fn green() -> Self { green() } /// The color blue pub const fn blue() -> Self { blue() } /// The color black pub const fn black() -> Self { black() } /// The color white pub const fn white() -> Self { white() } /// The color transparent black pub const fn transparent_black() -> Self { transparent_black() } /// Returns true if the HSLA color is fully transparent, false otherwise. pub fn is_transparent(&self) -> bool { self.a == 0.0 } /// Blends `other` on top of `self` based on `other`'s alpha value. The resulting color is a combination of `self`'s and `other`'s colors. /// /// If `other`'s alpha value is 1.0 or greater, `other` color is fully opaque, thus `other` is returned as the output color. /// If `other`'s alpha value is 0.0 or less, `other` color is fully transparent, thus `self` is returned as the output color. /// Else, the output color is calculated as a blend of `self` and `other` based on their weighted alpha values. /// /// Assumptions: /// - Alpha values are contained in the range [0, 1], with 1 as fully opaque and 0 as fully transparent. /// - The relative contributions of `self` and `other` is based on `self`'s alpha value (`self.a`) and `other`'s alpha value (`other.a`), `self` contributing `self.a * (1.0 - other.a)` and `other` contributing its own alpha value. /// - RGB color components are contained in the range [0, 1]. /// - If `self` and `other` colors are out of the valid range, the blend operation's output and behavior is undefined. pub fn blend(self, other: Hsla) -> Hsla { let alpha = other.a; if alpha >= 1.0 { other } else if alpha <= 0.0 { return self; } else { let converted_self = Rgba::from(self); let converted_other = Rgba::from(other); let blended_rgb = converted_self.blend(converted_other); return Hsla::from(blended_rgb); } } /// Returns a new HSLA color with the same hue, and lightness, but with no saturation. pub fn grayscale(&self) -> Self { Hsla { h: self.h, s: 0., l: self.l, a: self.a, } } /// Fade out the color by a given factor. This factor should be between 0.0 and 1.0. /// Where 0.0 will leave the color unchanged, and 1.0 will completely fade out the color. pub fn fade_out(&mut self, factor: f32) { self.a *= 1.0 - factor.clamp(0., 1.); } /// Returns a new HSLA color with the same hue, saturation, and lightness, but with a modified alpha value. pub fn opacity(&self, factor: f32) -> Self { Hsla { h: self.h, s: self.s, l: self.l, a: self.a * factor.clamp(0., 1.), } } } impl From 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<'de> Deserialize<'de> for Hsla { fn deserialize(deserializer: D) -> Result where D: Deserializer<'de>, { // First, deserialize it into Rgba let rgba = Rgba::deserialize(deserializer)?; // Then, use the From for Hsla implementation to convert it Ok(Hsla::from(rgba)) } } /// The orientation of a background. #[derive(Default, Debug, Clone, Copy, PartialEq, Eq)] #[repr(C)] pub enum BackgroundOrientation { /// The background is oriented horizontally. #[default] Horizontal = 0, /// The background is oriented vertically. Vertical = 1, } impl Display for BackgroundOrientation { fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result { match self { BackgroundOrientation::Horizontal => write!(f, "Horizontal"), BackgroundOrientation::Vertical => write!(f, "Vertical"), } } } #[derive(Debug, Clone, Copy, PartialEq)] #[repr(C)] pub(crate) enum BackgroundTag { Solid = 0, LinearGradient = 1, PatternSlash = 2, PatternDash = 3, } /// A color space for color interpolation. /// /// References: /// - https://developer.mozilla.org/en-US/docs/Web/CSS/color-interpolation-method /// - https://www.w3.org/TR/css-color-4/#typedef-color-space #[derive(Debug, Clone, Copy, PartialEq, Default)] #[repr(C)] pub enum ColorSpace { #[default] /// The sRGB color space. Srgb = 0, /// The Oklab color space. Oklab = 1, } impl Display for ColorSpace { fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result { match self { ColorSpace::Srgb => write!(f, "sRGB"), ColorSpace::Oklab => write!(f, "Oklab"), } } } /// A background color, which can be either a solid color or a linear gradient. #[derive(Debug, Clone, Copy, PartialEq)] #[repr(C)] pub struct Background { pub(crate) tag: BackgroundTag, pub(crate) color_space: ColorSpace, pub(crate) solid: Hsla, pub(crate) angle: f32, pub(crate) colors: [LinearColorStop; 2], pub(crate) orientation: BackgroundOrientation, /// Padding for alignment for repr(C) layout. pad: u32, } impl Eq for Background {} impl Default for Background { fn default() -> Self { Self { tag: BackgroundTag::Solid, solid: Hsla::default(), color_space: ColorSpace::default(), angle: 0.0, colors: [LinearColorStop::default(), LinearColorStop::default()], orientation: BackgroundOrientation::default(), pad: 0, } } } /// Creates a hash pattern background pub fn pattern_slash(color: Hsla) -> Background { Background { tag: BackgroundTag::PatternSlash, solid: color, ..Default::default() } } /// Creates a dash pattern background pub fn pattern_horizontal_dash(color: Hsla) -> Background { Background { tag: BackgroundTag::PatternDash, orientation: BackgroundOrientation::Horizontal, solid: color, ..Default::default() } } /// Creates a vertical dash pattern background pub fn pattern_vertical_dash(color: Hsla) -> Background { Background { tag: BackgroundTag::PatternDash, solid: color, orientation: BackgroundOrientation::Vertical, ..Default::default() } } /// Creates a LinearGradient background color. /// /// The gradient line's angle of direction. A value of `0.` is equivalent to to top; increasing values rotate clockwise from there. /// /// The `angle` is in degrees value in the range 0.0 to 360.0. /// /// https://developer.mozilla.org/en-US/docs/Web/CSS/gradient/linear-gradient pub fn linear_gradient( angle: f32, from: impl Into, to: impl Into, ) -> Background { Background { tag: BackgroundTag::LinearGradient, angle, colors: [from.into(), to.into()], ..Default::default() } } /// A color stop in a linear gradient. /// /// https://developer.mozilla.org/en-US/docs/Web/CSS/gradient/linear-gradient#linear-color-stop #[derive(Debug, Clone, Copy, Default, PartialEq)] #[repr(C)] pub struct LinearColorStop { /// The color of the color stop. pub color: Hsla, /// The percentage of the gradient, in the range 0.0 to 1.0. pub percentage: f32, } /// Creates a new linear color stop. /// /// The percentage of the gradient, in the range 0.0 to 1.0. pub fn linear_color_stop(color: impl Into, percentage: f32) -> LinearColorStop { LinearColorStop { color: color.into(), percentage, } } impl LinearColorStop { /// Returns a new color stop with the same color, but with a modified alpha value. pub fn opacity(&self, factor: f32) -> Self { Self { percentage: self.percentage, color: self.color.opacity(factor), } } } impl Background { /// Use specified color space for color interpolation. /// /// https://developer.mozilla.org/en-US/docs/Web/CSS/color-interpolation-method pub fn color_space(mut self, color_space: ColorSpace) -> Self { self.color_space = color_space; self } /// Returns a new background color with the same hue, saturation, and lightness, but with a modified alpha value. pub fn opacity(&self, factor: f32) -> Self { let mut background = *self; background.solid = background.solid.opacity(factor); background.colors = [ self.colors[0].opacity(factor), self.colors[1].opacity(factor), ]; background } /// Returns whether the background color is transparent. pub fn is_transparent(&self) -> bool { match self.tag { BackgroundTag::Solid => self.solid.is_transparent(), BackgroundTag::LinearGradient => self.colors.iter().all(|c| c.color.is_transparent()), BackgroundTag::PatternSlash => self.solid.is_transparent(), BackgroundTag::PatternDash => self.solid.is_transparent(), } } } impl From for Background { fn from(value: Hsla) -> Self { Background { tag: BackgroundTag::Solid, solid: value, ..Default::default() } } } impl From for Background { fn from(value: Rgba) -> Self { Background { tag: BackgroundTag::Solid, solid: Hsla::from(value), ..Default::default() } } } #[cfg(test)] mod tests { use serde_json::json; use super::*; #[test] fn test_deserialize_three_value_hex_to_rgba() { let actual: Rgba = serde_json::from_value(json!("#f09")).unwrap(); assert_eq!(actual, rgba(0xff0099ff)) } #[test] fn test_deserialize_four_value_hex_to_rgba() { let actual: Rgba = serde_json::from_value(json!("#f09f")).unwrap(); assert_eq!(actual, rgba(0xff0099ff)) } #[test] fn test_deserialize_six_value_hex_to_rgba() { let actual: Rgba = serde_json::from_value(json!("#ff0099")).unwrap(); assert_eq!(actual, rgba(0xff0099ff)) } #[test] fn test_deserialize_eight_value_hex_to_rgba() { let actual: Rgba = serde_json::from_value(json!("#ff0099ff")).unwrap(); assert_eq!(actual, rgba(0xff0099ff)) } #[test] fn test_deserialize_eight_value_hex_with_padding_to_rgba() { let actual: Rgba = serde_json::from_value(json!(" #f5f5f5ff ")).unwrap(); assert_eq!(actual, rgba(0xf5f5f5ff)) } #[test] fn test_deserialize_eight_value_hex_with_mixed_case_to_rgba() { let actual: Rgba = serde_json::from_value(json!("#DeAdbEeF")).unwrap(); assert_eq!(actual, rgba(0xdeadbeef)) } #[test] fn test_background_solid() { let color = Hsla::from(rgba(0xff0099ff)); let mut background = Background::from(color); assert_eq!(background.tag, BackgroundTag::Solid); assert_eq!(background.solid, color); assert_eq!(background.opacity(0.5).solid, color.opacity(0.5)); assert_eq!(background.is_transparent(), false); background.solid = hsla(0.0, 0.0, 0.0, 0.0); assert_eq!(background.is_transparent(), true); } #[test] fn test_background_linear_gradient() { let from = linear_color_stop(rgba(0xff0099ff), 0.0); let to = linear_color_stop(rgba(0x00ff99ff), 1.0); let background = linear_gradient(90.0, from, to); assert_eq!(background.tag, BackgroundTag::LinearGradient); assert_eq!(background.colors[0], from); assert_eq!(background.colors[1], to); assert_eq!(background.opacity(0.5).colors[0], from.opacity(0.5)); assert_eq!(background.opacity(0.5).colors[1], to.opacity(0.5)); assert_eq!(background.is_transparent(), false); assert_eq!(background.opacity(0.0).is_transparent(), true); } }