1#define_import_path bevy_pbr::rgb9e5
2
3const RGB9E5_EXPONENT_BITS        = 5u;
4const RGB9E5_MANTISSA_BITS        = 9;
5const RGB9E5_MANTISSA_BITSU       = 9u;
6const RGB9E5_EXP_BIAS             = 15;
7const RGB9E5_MAX_VALID_BIASED_EXP = 31u;
8
9//#define MAX_RGB9E5_EXP               (RGB9E5_MAX_VALID_BIASED_EXP - RGB9E5_EXP_BIAS)
10//#define RGB9E5_MANTISSA_VALUES       (1<<RGB9E5_MANTISSA_BITS)
11//#define MAX_RGB9E5_MANTISSA          (RGB9E5_MANTISSA_VALUES-1)
12//#define MAX_RGB9E5                   ((f32(MAX_RGB9E5_MANTISSA))/RGB9E5_MANTISSA_VALUES * (1<<MAX_RGB9E5_EXP))
13//#define EPSILON_RGB9E5_              ((1.0/RGB9E5_MANTISSA_VALUES) / (1<<RGB9E5_EXP_BIAS))
14
15const MAX_RGB9E5_EXP              = 16u;
16const RGB9E5_MANTISSA_VALUES      = 512;
17const MAX_RGB9E5_MANTISSA         = 511;
18const MAX_RGB9E5_MANTISSAU        = 511u;
19const MAX_RGB9E5_                 = 65408.0;
20const EPSILON_RGB9E5_             = 0.000000059604645;
21
22fn floor_log2_(x: f32) -> i32 {
23    let f = bitcast<u32>(x);
24    let biasedexponent = (f & 0x7F800000u) >> 23u;
25    return i32(biasedexponent) - 127;
26}
27
28// https://www.khronos.org/registry/OpenGL/extensions/EXT/EXT_texture_shared_exponent.txt
29fn vec3_to_rgb9e5_(rgb_in: vec3<f32>) -> u32 {
30    let rgb = clamp(rgb_in, vec3(0.0), vec3(MAX_RGB9E5_));
31
32    let maxrgb = max(rgb.r, max(rgb.g, rgb.b));
33    var exp_shared = max(-RGB9E5_EXP_BIAS - 1, floor_log2_(maxrgb)) + 1 + RGB9E5_EXP_BIAS;
34    var denom = exp2(f32(exp_shared - RGB9E5_EXP_BIAS - RGB9E5_MANTISSA_BITS));
35
36    let maxm = i32(floor(maxrgb / denom + 0.5));
37    if (maxm == RGB9E5_MANTISSA_VALUES) {
38        denom *= 2.0;
39        exp_shared += 1;
40    }
41
42    let n = vec3<u32>(floor(rgb / denom + 0.5));
43    
44    return (u32(exp_shared) << 27u) | (n.b << 18u) | (n.g << 9u) | (n.r << 0u);
45}
46
47// Builtin extractBits() is not working on WEBGL or DX12
48// DX12: HLSL: Unimplemented("write_expr_math ExtractBits")
49fn extract_bits(value: u32, offset: u32, bits: u32) -> u32 {
50    let mask = (1u << bits) - 1u;
51    return (value >> offset) & mask;
52}
53
54fn rgb9e5_to_vec3_(v: u32) -> vec3<f32> {
55    let exponent = i32(extract_bits(v, 27u, RGB9E5_EXPONENT_BITS)) - RGB9E5_EXP_BIAS - RGB9E5_MANTISSA_BITS;
56    let scale = exp2(f32(exponent));
57
58    return vec3(
59        f32(extract_bits(v, 0u, RGB9E5_MANTISSA_BITSU)),
60        f32(extract_bits(v, 9u, RGB9E5_MANTISSA_BITSU)),
61        f32(extract_bits(v, 18u, RGB9E5_MANTISSA_BITSU))
62    ) * scale;
63}