cmx 0.1.0

Rust Spectral Color Management Library
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

// SPDX-License-Identifier: Apache-2.0 OR MIT
// Copyright (c) 2021-2025, Harbers Bik LLC

use serde::Serialize;
use zerocopy::I32;
use zerocopy::{BigEndian, Immutable, KnownLayout, TryFromBytes, Unaligned};

use crate::tag::tagdata::Lut8Data;

#[derive(TryFromBytes, KnownLayout, Unaligned, Immutable)]
#[repr(C, packed)]
struct Lut8HeaderLayout {
    signature: [u8; 4], // "LUT8"
    _reserved: [u8; 4], // reserved, must be 0
    n: u8,              // input channels
    m: u8,              // output channels
    g: u8,
    _padding: u8,               // padding byte, required to be 0
    e_mat: [I32<BigEndian>; 9], // s15Fixed16Number array
}

#[derive(Serialize)]
pub struct Lut8Type {
    g: usize,                  // number of grid points
    e_mat: [f64; 9],           // s15Fixed16Number array
    input_luts: Vec<Vec<u8>>,  // input LUT
    output_luts: Vec<Vec<u8>>, // output LUT
    multi_lut: Vec<Vec<u8>>,   // multi-dimensional LUT
}

// Organize by grid points (most intuitive)
// [[r,g,b at point 0], [r,g,b at point 1], ...]
fn reshape_by_grid_points(multi_lut: &[u8], m: usize) -> Vec<Vec<u8>> {
    multi_lut.chunks(m).map(|chunk| chunk.to_vec()).collect()
}

impl From<&Lut8Data> for Lut8Type {
    fn from(lut8: &Lut8Data) -> Self {
        let (layout, _) = Lut8HeaderLayout::try_ref_from_prefix(&lut8.0)
            .expect("lut8Type: data too short to contain header");
        let n = layout.n as usize;
        let m = layout.m as usize;
        let g = layout.g as usize;

        // Convert e_mat from s15Fixed16Number to f64
        let e_mat: [f64; 9] = layout
            .e_mat
            .iter()
            .map(|&v| v.get() as f64 / 65536.0)
            .collect::<Vec<f64>>()
            .try_into()
            .unwrap();

        // Calculate sizes and offsets
        let header_size = std::mem::size_of::<Lut8HeaderLayout>();
        let input_luts_size = n * 256;
        let clut_size = g
            .checked_pow(n as u32)
            .and_then(|v| v.checked_mul(m))
            .expect("lut8Type: CLUT size overflow");
        let output_luts_size = m * 256;

        // Bounds check: ensure the payload is large enough before any slicing
        let total_size = header_size
            .checked_add(input_luts_size)
            .and_then(|v| v.checked_add(clut_size))
            .and_then(|v| v.checked_add(output_luts_size))
            .expect("lut8Type: total size overflow");
        assert!(
            lut8.0.len() >= total_size,
            "lut8Type: truncated data (have {}, need >= {})",
            lut8.0.len(),
            total_size
        );

        // Calculate offsets
        let input_luts_offset = header_size;
        let clut_offset = input_luts_offset + input_luts_size;
        let output_luts_offset = clut_offset + clut_size;

        // Read input LUTs
        let input_luts: Vec<Vec<u8>> = lut8.0
            [input_luts_offset..input_luts_offset + input_luts_size]
            .chunks(256)
            .map(|chunk| chunk.to_vec())
            .collect();

        // Read CLUT (multi-dimensional LUT)
        let multi_lut = lut8.0[clut_offset..clut_offset + clut_size].to_vec();

        // Read output LUTs
        let output_luts: Vec<Vec<u8>> = lut8.0
            [output_luts_offset..output_luts_offset + output_luts_size]
            .chunks(256)
            .map(|chunk| chunk.to_vec())
            .collect();

        Lut8Type {
            g,
            e_mat,
            input_luts,
            output_luts,
            multi_lut: reshape_by_grid_points(&multi_lut, m),
        }
    }
}