tiny.zen.DiagramDocument

1pub const Document = struct {2    allocator: std.mem.Allocator,3    frame: Frame = .{},4    scales: Scales = .{},5    transforms: std.ArrayList(Transform) = .empty,6    data: std.ArrayList(DataRow) = .empty,7    marks: std.ArrayList(Mark) = .empty,8    scratch_high_water: usize = 0,910    pub fn parse(allocator: std.mem.Allocator, line_scratch: []u8, jsonl: []const u8) !Document {11        var document = Document{ .allocator = allocator };12        errdefer document.deinit();1314        var scratch = std.heap.FixedBufferAllocator.init(line_scratch);15        var lines = LineIterator.init(jsonl);16        while (lines.next()) |line| {17            scratch.reset();18            var parsed = std.json.parseFromSlice(std.json.Value, scratch.allocator(), line, .{}) catch |err| switch (err) {19                error.OutOfMemory => return error.OutOfMemory,20                else => return error.InvalidJsonl,21            };22            defer parsed.deinit();23            try document.applyRecord(parsed.value);24            document.scratch_high_water = @max(document.scratch_high_water, scratch.end_index);25        }26        try validateDocument(&document);27        return document;28    }2930    pub fn deinit(self: *Document) void {31        self.frame.deinit(self.allocator);32        self.scales.deinit(self.allocator);33        for (self.transforms.items) |*transform| transform.deinit(self.allocator);34        self.transforms.deinit(self.allocator);35        for (self.data.items) |*row| row.deinit(self.allocator);36        self.data.deinit(self.allocator);37        for (self.marks.items) |*mark| mark.deinit(self.allocator);38        self.marks.deinit(self.allocator);39    }4041    fn applyRecord(self: *Document, value: std.json.Value) !void {42        const object = switch (value) {43            .object => |object| object,44            else => return error.InvalidRecord,45        };46        switch (try recordKindOf(value)) {47            .frame => try self.applyFrame(object),48            .scale => try self.applyScale(object),49            .transform => try self.applyTransform(object),50            .data => try self.applyData(object),51            .mark => try self.applyChannelMark(object),52            .bar => try self.marks.append(self.allocator, .{ .bar = try parseBar(self.allocator, object) }),53            .point => try self.marks.append(self.allocator, .{ .point = try parsePoint(self.allocator, object) }),54            .rule => try self.marks.append(self.allocator, .{ .rule = try parseRule(self.allocator, object) }),55            .text => try self.marks.append(self.allocator, .{ .text = try parseText(self.allocator, object) }),56            .box => try self.marks.append(self.allocator, .{ .box = try parseBox(self.allocator, object) }),57            .edge => try self.marks.append(self.allocator, .{ .edge = try parseEdge(self.allocator, object) }),58        }59    }6061    fn applyFrame(self: *Document, object: std.json.ObjectMap) !void {62        if (objectNumber(object, "width")) |value| self.frame.width = value;63        if (objectNumber(object, "height")) |value| self.frame.height = value;64        if (objectNumber(object, "inset")) |value| self.frame.inset = value;65        if (objectBool(object, "axes")) |value| self.frame.axes = value;66        if (objectNumber(object, "y_min")) |value| self.frame.y_min = value;67        if (objectNumber(object, "y_max")) |value| self.frame.y_max = value;68        if (objectNumber(object, "x_min")) |value| self.frame.x_min = value;69        if (objectNumber(object, "x_max")) |value| self.frame.x_max = value;70        try replaceString(self.allocator, &self.frame.title, objectString(object, "title"));71        try replaceString(self.allocator, &self.frame.x_label, objectString(object, "x_label"));72        try replaceString(self.allocator, &self.frame.y_label, objectString(object, "y_label"));73        if (objectString(object, "background")) |background| {74            if (!validPaint(background)) return error.InvalidColor;75            try replaceString(self.allocator, &self.frame.background, background);76        }77    }7879    fn applyScale(self: *Document, object: std.json.ObjectMap) !void {80        const axis = try parseAxis(object);81        const scale = switch (axis) {82            .x => &self.scales.x,83            .y => &self.scales.y,84        };85        if (objectString(object, "type")) |value| scale.kind = try parseScaleKind(value);86        if (objectNumber(object, "min")) |value| scale.min = value;87        if (objectNumber(object, "max")) |value| scale.max = value;88        if (objectNumber(object, "base")) |value| scale.base = value;89        try replaceString(self.allocator, &scale.label, objectString(object, "label"));90    }9192    fn applyTransform(self: *Document, object: std.json.ObjectMap) !void {93        var transform = try parseTransform(self.allocator, object);94        errdefer transform.deinit(self.allocator);95        try self.transforms.append(self.allocator, transform);96    }9798    fn applyData(self: *Document, object: std.json.ObjectMap) !void {99        var row = try parseDataRow(self.allocator, object);100        errdefer row.deinit(self.allocator);101        try self.data.append(self.allocator, row);102    }103104    fn applyChannelMark(self: *Document, object: std.json.ObjectMap) !void {105        var channel = try parseChannelMark(self.allocator, object);106        defer channel.deinit(self.allocator);107        var matched = false;108        for (self.data.items) |*row| {109            if (!std.mem.eql(u8, row.name, channel.data)) continue;110            matched = true;111            var mark = try markFromChannel(self.allocator, channel, row);112            var appended = false;113            errdefer if (!appended) mark.deinit(self.allocator);114            try self.marks.append(self.allocator, mark);115            appended = true;116        }117        if (!matched) return error.UnknownData;118    }119};

1/// Owned parsed diagram model; release it through its `deinit` method.2pub const DiagramDocument = diagram.Document;