Delete FeatureTest.rs
This commit is contained in:
parent
010f2b4fcd
commit
2f228ff2cf
|
@ -1,69 +0,0 @@
|
|||
use std::collections::HashMap;
|
||||
|
||||
fn extract_color_distribution(image_data: &[u8]) -> HashMap<u32, f32> {
|
||||
let mut color_distribution: HashMap<u32, f32> = HashMap::new();
|
||||
let total_pixels = image_data.len() as f32 / 4.0;//für 4 Werte
|
||||
|
||||
for pixel in image_data.chunks_exact(4) {
|
||||
let r = pixel[0] as u32;
|
||||
let g = pixel[1] as u32;
|
||||
let b = pixel[2] as u32;
|
||||
let a = pixel[3] as u32;
|
||||
let rgba = (r << 24) | (g << 16) | (b << 8) | a;
|
||||
|
||||
*color_distribution.entry(rgba).or_insert(0.0) += 1.0;
|
||||
}
|
||||
|
||||
for (_, count) in &mut color_distribution {
|
||||
*count /= total_pixels;
|
||||
}
|
||||
|
||||
color_distribution
|
||||
}
|
||||
|
||||
fn extract_average_brightness(image: &[u8]) -> u8 {
|
||||
let mut sum: u32 = 0;
|
||||
let mut count: u32 = 0;
|
||||
|
||||
for i in (0..image.len()).step_by(4) {
|
||||
let r = image[i] as u32;
|
||||
let g = image[i + 1] as u32;
|
||||
let b = image[i + 2] as u32;
|
||||
|
||||
// (0.299 * R) + (0.587 * G) + (0.114 * B)
|
||||
let brightness = ((0.299 * r as f32) + (0.587 * g as f32) + (0.114 * b as f32)).round() as u32;
|
||||
|
||||
sum += brightness;
|
||||
count += 1;
|
||||
}
|
||||
let average_brightness = (sum / count) as u8;
|
||||
average_brightness
|
||||
}
|
||||
|
||||
|
||||
fn main() {
|
||||
|
||||
test2();
|
||||
|
||||
}
|
||||
|
||||
fn test2(){
|
||||
|
||||
let image_data: Vec<(u8, u8, u8, u8)> = vec![
|
||||
(255, 0, 0, 255), // Red
|
||||
(0, 255, 0, 255), // Green
|
||||
(0, 0, 255, 255), // Blue
|
||||
];
|
||||
//convert image data to useable &u8 slice
|
||||
let byte_slice: &[u8] = unsafe {
|
||||
std::slice::from_raw_parts(
|
||||
image_data.as_ptr() as *const u8,
|
||||
image_data.len() * 4,
|
||||
)
|
||||
};
|
||||
let color_distribution = extract_color_distribution(&byte_slice);
|
||||
let color_distribution_vec: Vec<f32> = color_distribution.values().cloned().collect();
|
||||
let average_brightness = extract_average_brightness(&byte_slice);
|
||||
println!("{:?}", average_brightness);
|
||||
println!("{:?}", color_distribution_vec);
|
||||
}
|
Loading…
Reference in New Issue