image masked color is now adaptive to the image

khanghugo · Jan 26, 2025 · 4a2fcca · 4a2fcca
1 parent 833ee8e
commit 4a2fcca
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 13 deletions.
diff --git a/gchimp/src/utils/constants.rs b/gchimp/src/utils/constants.rs
@@ -17,10 +17,6 @@ pub const MAX_SMD_VERTEX: usize = 2048;
 pub const STUDIOMDL_ERROR_PATTERN: &str = "************ ERROR ************";
 pub const MAX_GOLDSRC_MODEL_TEXTURE_COUNT: usize = 64;
 
-pub const PALETTE_PAD_COLOR: [u8; 3] = [0, 0, 0];
-pub const PALETTE_TRANSPARENT_COLOR: [u8; 3] = [0, 255, 0];
-pub const PALETTE_TRANSPARENT_COLOR2: [u8; 3] = [0, 0, 255];
-
 pub const ORIGIN_TEXTURE: &str = "ORIGIN";
 pub const CLIP_TEXTURE: &str = "CLIP";
 pub const CONTENTWATER_TEXTURE: &str = "CONTENTWATER";

diff --git a/gchimp/src/utils/img_stuffs.rs b/gchimp/src/utils/img_stuffs.rs
@@ -1,5 +1,5 @@
 use std::{
-    collections::HashMap,
+    collections::{HashMap, HashSet},
     fs::OpenOptions,
     io::{BufReader, BufWriter, Cursor, Write},
     path::{Path, PathBuf},
@@ -15,8 +15,6 @@ use vtf::Vtf;
 
 use crate::utils::constants::MAX_GOLDSRC_TEXTURE_SIZE;
 
-use super::constants::{PALETTE_PAD_COLOR, PALETTE_TRANSPARENT_COLOR, PALETTE_TRANSPARENT_COLOR2};
-
 use crate::err;
 
 type Palette = Vec<quantette::palette::rgb::Rgb<quantette::palette::encoding::Srgb, u8>>;
@@ -77,20 +75,34 @@ fn maybe_resize_due_to_exceeding_max_goldsrc_texture_size(img: RgbaImage) -> Rgb
     }
 }
 
+fn get_palette_from_rgbaimage(img: &RgbaImage) -> Vec<[u8; 3]> {
+    let mut hash_set = HashSet::<[u8; 3]>::new();
+
+    img.pixels().for_each(|pixel| {
+        hash_set.insert([pixel.0[0], pixel.0[1], pixel.0[2]]);
+    });
+
+    hash_set.into_iter().collect()
+}
+
 // if alpha is 0, then replace it with transparent color
 // otherwise, linearly blend the pixel
 fn rgba8_to_rgb8(img: RgbaImage) -> eyre::Result<RgbImage> {
     let (width, height) = img.dimensions();
+
+    let palette = get_palette_from_rgbaimage(&img);
+    let transparent_color = get_a_color_that_does_not_exist(&palette, 10);
+
     let buf = img
         .par_chunks_exact(4)
         .flat_map(|p| {
             let should_replace = p[3] == 0;
 
             if should_replace {
                 [
-                    PALETTE_TRANSPARENT_COLOR2[0],
-                    PALETTE_TRANSPARENT_COLOR2[1],
-                    PALETTE_TRANSPARENT_COLOR2[2],
+                    transparent_color[0],
+                    transparent_color[1],
+                    transparent_color[2],
                 ]
             } else {
                 let opacity = p[3] as f32 / 255.;
@@ -299,6 +311,28 @@ pub fn tile_and_resize(img: &RgbaImage, scalar: u32) -> RgbaImage {
     imageops::resize(&res, width, height, imageops::FilterType::Lanczos3)
 }
 
+fn get_a_color_that_does_not_exist(palette: &[[u8; 3]], count: usize) -> [u8; 3] {
+    // count to help with getting multiple colors
+    let mut local_count = 0;
+
+    for r in 0..=255u8 {
+        for g in 0..=255u8 {
+            for b in 0..=255u8 {
+                if !palette.contains(&[r, g, b]) {
+                    if local_count == count {
+                        return [r, g, b];
+                    }
+
+                    local_count += 1;
+                }
+            }
+        }
+    }
+
+    // it is not possible for this to happen
+    unreachable!()
+}
+
 // threshold is between 0 and 1
 // if the percentage of the most used color is over the threshold, mark the color transparent
 pub fn eight_bpp_transparent_img(
@@ -337,10 +371,12 @@ pub fn eight_bpp_transparent_img(
     let mut new_img = img.to_vec();
 
     // pad palette
-    new_palette.resize(256, PALETTE_PAD_COLOR);
+    let pad_color = get_a_color_that_does_not_exist(palette, 0);
+    new_palette.resize(256, pad_color);
 
-    // change the final color of the palette to a rare color
-    new_palette[255] = PALETTE_TRANSPARENT_COLOR;
+    // change the final color of the palette to a color that is not in the palette
+    let transparent_color = get_a_color_that_does_not_exist(palette, 1);
+    new_palette[255] = transparent_color;
 
     // swap the most used color (index) with 255
     for pixel in new_img.iter_mut() {