ableos/shadeable/src/pixel_format.rs

use core::ops::{BitAnd, BitOr, Shr};

use vga::colors::Color16;
pub type Rgba64 = u64;

pub fn get_r(rgba: Rgba64) -> u8 {
   rgba.bitand(0xff_00_00_00).shr(0o30) as u8
}
pub fn get_g(rgba: Rgba64) -> u8 {
   rgba.bitand(0xff_00_00).shr(0o20) as u8
}

pub fn get_b(rgba: Rgba64) -> u8 {
   rgba.bitand(0xff_00).shr(0o10) as u8
}

pub fn get_a(rgba: Rgba64) -> u8 {
   (rgba & 0xff) as u8
}

pub fn set_r(rgba: Rgba64, r: u8) -> Rgba64 {
   rgba
      .bitand(0xffffffff_00_ff_ff_ff)
      .bitor((r as Rgba64).shr(0o30))
}

pub fn set_g(rgba: Rgba64, g: u8) -> Rgba64 {
   rgba
      .bitand(0xffffffff_ff_00_ff_ff)
      .bitor((g as Rgba64).shr(0o20))
}

pub fn set_b(rgba: Rgba64, b: u8) -> Rgba64 {
   rgba
      .bitand(0xffffffff_ff_ff_00_ff)
      .bitor((b as Rgba64).shr(0o10))
}

pub fn set_a(rgba: Rgba64, a: u8) -> Rgba64 {
   rgba.bitand(0xffffffff_ff_ff_ff_00).bitor(a as Rgba64)
}

pub fn rgba_div(a: Rgba64, b: Rgba64) -> Rgba64 {
   set_r(0, get_r(a) / get_r(b))
      | set_g(0, get_g(a) / get_g(b))
      | set_g(0, get_b(a) / get_b(b))
      | set_g(0, get_a(a) / get_a(b))
}

pub fn new_rgba64(r: u8, g: u8, b: u8, a: u8) -> Rgba64 {
   set_r(0, r) | set_g(0, g) | set_b(0, b) | set_a(0, a)
}

enum ChannelValue {
   Dark,
   Low,
   Mid,
   High,
}

impl From<u8> for ChannelValue {
   fn from(b: u8) -> Self {
      use ChannelValue::*;
      match b {
         0x00..=0x3f => Dark,
         0x40..=0x7f => Low,
         0x80..=0xbf => Mid,
         0xc0..=0xff => High,
      }
   }
}

pub fn into_vga_16(rgba_64: Rgba64) -> Color16 {
   use ChannelValue::*;
   use Color16::*;
   match (
      get_r(rgba_64).into(),
      get_g(rgba_64).into(),
      get_b(rgba_64).into(),
   ) {
      (Dark, Dark, Dark) => Black,
      (Dark, Dark, Low) => Black,
      (Dark, Dark, Mid) => Blue,
      (Dark, Dark, High) => Blue,
      (Dark, Low, Dark) => Black,
      (Dark, Low, Low) => Black,
      (Dark, Low, Mid) => Blue,
      (Dark, Low, High) => Blue,
      (Dark, Mid, Dark) => Green,
      (Dark, Mid, Low) => Green,
      (Dark, Mid, Mid) => Cyan,
      (Dark, Mid, High) => Cyan,
      (Dark, High, Dark) => Green,
      (Dark, High, Low) => Green,
      (Dark, High, Mid) => Green,
      (Dark, High, High) => Cyan,
      (Low, Dark, Dark) => Black,
      (Low, Dark, Low) => Black,
      (Low, Dark, Mid) => Blue,
      (Low, Dark, High) => Blue,
      (Low, Low, Dark) => Black,
      (Low, Low, Low) => DarkGrey,
      (Low, Low, Mid) => LightGrey,
      (Low, Low, High) => Blue,
      (Low, Mid, Dark) => DarkGrey,
      (Low, Mid, Low) => LightGrey,
      (Low, Mid, Mid) => Cyan,
      (Low, Mid, High) => Cyan,
      (Low, High, Dark) => Green,
      (Low, High, Low) => Green,
      (Low, High, Mid) => Cyan,
      (Low, High, High) => Cyan,
      (Mid, Dark, Dark) => Red,
      (Mid, Dark, Low) => Red,
      (Mid, Dark, Mid) => Magenta,
      (Mid, Dark, High) => Magenta,
      (Mid, Low, Dark) => Brown,
      (Mid, Low, Low) => Red,
      (Mid, Low, Mid) => DarkGrey,
      (Mid, Low, High) => LightBlue,
      (Mid, Mid, Dark) => Brown,
      (Mid, Mid, Low) => Brown,
      (Mid, Mid, Mid) => LightGrey,
      (Mid, Mid, High) => LightBlue,
      (Mid, High, Dark) => Green,
      (Mid, High, Low) => Green,
      (Mid, High, Mid) => LightGreen,
      (Mid, High, High) => LightCyan,
      (High, Dark, Dark) => Red,
      (High, Dark, _) => Magenta,
      (High, Low, Dark) => Red,
      (High, Low, Low) => LightRed,
      (High, Low, Mid) => Pink,
      (High, Low, High) => Magenta,
      (High, Mid, Dark) => Yellow,
      (High, Mid, Low) => Yellow,
      (High, Mid, Mid) => LightRed,
      (High, Mid, High) => Pink,
      (High, High, Dark) => Yellow,
      (High, High, Low) => White,
      (High, High, Mid) => White,
      (High, High, High) => White,
   }
}

pub fn from_vga_16(color: Color16) -> Rgba64 {
   use Color16::*;
   match color {
      Black => new_rgba64(0, 0, 0, 0),
      DarkGrey => new_rgba64(105, 105, 105, 0),
      LightGrey => new_rgba64(211, 211, 211, 0),
      //
      Blue => new_rgba64(0, 0, 0xff, 0),
      Green => new_rgba64(0, 0xff, 0, 0),
      Red => new_rgba64(0xff, 0, 0, 0),
      //
      Yellow => new_rgba64(0xff, 0xff, 0, 0),
      Cyan => new_rgba64(0, 0xff, 0xff, 0),
      Magenta => new_rgba64(0xff, 0, 0xff, 0),

      Brown => new_rgba64(165, 42, 42, 0),
      Pink => new_rgba64(0xff, 105, 180, 0),
      White => new_rgba64(0xff, 0xff, 0xff, 0),

      LightBlue => new_rgba64(173, 216, 230, 0),
      LightGreen => new_rgba64(144, 238, 144, 0),
      LightCyan => new_rgba64(88, 100, 100, 0),
      LightRed => new_rgba64(0xff, 204, 203, 0),
   }
}