zahirscan 0.3.3

Token-efficient content compression for AI analysis using probabilistic template mining
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

//! Format-specific image metadata extraction

use crate::parsers::FileType;
use crate::utils::filetypes::get_extensions_for_file_type;

use super::{bmp, gif, jpeg, png, tiff, webp};

/// Image format types for metadata extraction
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ImageFormat {
    Jpeg,
    Png,
    Gif,
    WebP,
    Bmp,
    Tiff,
}

/// Trait for extracting format-specific metadata
pub trait FormatMetadata {
    /// Extract compression information from image data
    fn extract_compression(&self, data_ref: &[u8]) -> Option<String>;

    /// Extract chroma subsampling information (JPEG only)
    fn extract_chroma_subsampling(&self, _data_ref: &[u8]) -> Option<String> {
        None // Most formats don't use chroma subsampling
    }

    /// Extract bit depth (bits per sample/channel)
    fn extract_bit_depth(&self, _data_ref: &[u8]) -> Option<u32> {
        None // Default: no bit depth extraction
    }

    /// Extract color type (pixel format, e.g., "Rgb8", "Rgba8", "L8")
    fn extract_color_type(&self, _data_ref: &[u8]) -> Option<String> {
        None // Default: no color type extraction
    }
}

impl FormatMetadata for ImageFormat {
    fn extract_compression(&self, data_ref: &[u8]) -> Option<String> {
        match self {
            ImageFormat::Jpeg => Some(jpeg::extract_compression(data_ref)),
            ImageFormat::Png => png::extract_compression(data_ref),
            ImageFormat::Gif => gif::extract_compression(data_ref),
            ImageFormat::WebP => webp::extract_compression(data_ref),
            ImageFormat::Bmp => bmp::extract_compression(data_ref),
            ImageFormat::Tiff => tiff::extract_compression(data_ref),
        }
    }

    fn extract_chroma_subsampling(&self, data_ref: &[u8]) -> Option<String> {
        match self {
            ImageFormat::Jpeg => jpeg::extract_subsampling(data_ref),
            _ => None, // Only JPEG uses chroma subsampling
        }
    }

    fn extract_bit_depth(&self, data_ref: &[u8]) -> Option<u32> {
        match self {
            ImageFormat::Jpeg => jpeg::extract_bit_depth(data_ref),
            ImageFormat::Png => png::extract_bit_depth(data_ref),
            ImageFormat::Bmp => bmp::extract_bit_depth(data_ref),
            ImageFormat::Tiff => tiff::extract_bit_depth(data_ref),
            // GIF and WebP are 8-bit (WebP is typically 8-bit)
            ImageFormat::Gif | ImageFormat::WebP => Some(8),
        }
    }

    fn extract_color_type(&self, data_ref: &[u8]) -> Option<String> {
        match self {
            ImageFormat::Jpeg => jpeg::extract_color_type(data_ref),
            ImageFormat::Png => png::extract_color_type(data_ref),
            ImageFormat::Gif => gif::extract_color_type(data_ref),
            ImageFormat::Bmp => bmp::extract_color_type(data_ref),
            ImageFormat::WebP => webp::extract_color_type(data_ref),
            ImageFormat::Tiff => Some(tiff::extract_color_type(data_ref)),
        }
    }
}

/// Extension to `ImageFormat` mapping
/// Only includes formats that have `ImageFormat` enum variants
const IMAGE_FORMAT_MAP: &[(&str, ImageFormat)] = &[
    ("jpeg", ImageFormat::Jpeg),
    ("jpg", ImageFormat::Jpeg),
    ("png", ImageFormat::Png),
    ("gif", ImageFormat::Gif),
    ("webp", ImageFormat::WebP),
    ("bmp", ImageFormat::Bmp),
    ("tiff", ImageFormat::Tiff),
    ("tif", ImageFormat::Tiff),
];

/// Convert format string to `ImageFormat` enum
/// Verifies format against extension map from tools.rs for consistency
#[must_use]
pub fn format_from_string(format_str_ref: &str) -> Option<ImageFormat> {
    let format_lower = format_str_ref.to_lowercase();
    let image_extensions = get_extensions_for_file_type(FileType::Image);

    // Find matching format in our map, but only if it's in the extension map
    IMAGE_FORMAT_MAP
        .iter()
        .find(|(ext, _)| format_lower.contains(ext) && image_extensions.contains(ext))
        .map(|(_, format)| *format)
}