summaryrefslogtreecommitdiff
path: root/src/jpg/quantization-table.zig
blob: c4c22d526859e6c02562cd596481a640256015db (plain)
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
const std = @import("std");

const Self = @This();

pub const Precision = enum(u1) {
    b8 = 0,
    b16 = 1,
};

const Header = packed struct(u8) {
    identifier: u4,
    precision: u4,
};

precision: Precision,
identifier: u2,
elements: union(Precision) {
    b8: [64]u8,
    b16: [64]u16,
},

pub fn read(reader: *std.Io.Reader) !Self {
    const header: Header = @bitCast(try reader.takeByte());
    var self: Self = undefined;
    self.identifier = @truncate(header.identifier);
    self.precision = @enumFromInt(@as(u1, @truncate(header.precision)));

    switch (self.precision) {
        .b8 => {
            self.elements = .{ .b8 = undefined };
            for (&self.elements.b8) |*element| {
                element.* = try reader.takeByte();
            }
        },
        .b16 => {
            self.elements = .{ .b16 = undefined };
            for (&self.elements.b16) |*element| {
                element.* = try reader.takeInt(u16, .big);
            }
        },
    }

    return self;
}

pub fn size_in_file(self: Self) usize {
    return 65 + 64 * @as(usize, @intFromEnum(self.precision));
}