refactored pre-rendering

This commit is contained in:
2025-07-13 16:06:02 +02:00
parent c1191de661
commit 57a3a371eb
2 changed files with 232 additions and 138 deletions

View File

@ -1,119 +1,157 @@
#import "/src/cetz.typ": canvas, draw #import "/src/cetz.typ" as cetz: canvas, draw
#import "utils.typ": get-participants-i, get-style, normalize-units #import "utils.typ": get-participants-i, get-style, normalize-units, is-elmt
#import "group.typ" #import "../group.typ"
#import "participant.typ" #import "../participant.typ"
#import participant: PAR-SPECIALS #import participant: PAR-SPECIALS
#import "sequence.typ" #import "../sequence.typ"
#import "separator.typ" #import "../separator.typ"
#import "sync.typ" #import "../sync.typ"
#import "consts.typ": * #import "../consts.typ": *
#import "note.typ" as note: get-note-box #import "../note.typ" as note: get-note-box
#import "../styles.typ"
#let DEBUG-INVISIBLE = false #let DEBUG-INVISIBLE = false
#let get-columns-width(participants, elements) = { #let init-lifelines(participants) = {
participants = participants.map(p => { return participants.map(p => {
p.insert("lifeline-lvl", 0) p.insert("lifeline-lvl", 0)
p.insert("max-lifelines", 0) p.insert("max-lifelines", 0)
p p
}) })
let pars-i = get-participants-i(participants) }
let cells = ()
// Unwrap syncs #let unwrap-syncs(elements) = {
let i = 0 let i = 0
while i < elements.len() { while i < elements.len() {
let elmt = elements.at(i) let elmt = elements.at(i)
if elmt.type == "sync" { if elmt.type == "sync" {
elements = elements.slice(0, i + 1) + elmt.elmts + elements.slice(i + 1) elements = (
elements.slice(0, i + 1) +
elmt.elmts +
elements.slice(i + 1)
)
} }
i += 1 i += 1
} }
return elements
}
// Compute max lifeline levels #let seq-update-lifelines(participants, pars-i, seq) = {
for elmt in elements { let participants = participants
if elmt.type == "seq" { let com = if seq.comment == none {""} else {seq.comment}
let com = if elmt.comment == none {""} else {elmt.comment} let i1 = pars-i.at(seq.p1)
let i1 = pars-i.at(elmt.p1) let i2 = pars-i.at(seq.p2)
let i2 = pars-i.at(elmt.p2) let cell = (
cells.push( elmt: seq,
(
elmt: elmt,
i1: calc.min(i1, i2), i1: calc.min(i1, i2),
i2: calc.max(i1, i2), i2: calc.max(i1, i2),
cell: box(com, inset: 3pt) cell: box(com, inset: 3pt)
) )
)
if elmt.disable-src or elmt.destroy-src { if seq.disable-src or seq.destroy-src {
let p = participants.at(i1) let p = participants.at(i1)
p.lifeline-lvl -= 1 p.lifeline-lvl -= 1
participants.at(i1) = p participants.at(i1) = p
} }
if elmt.disable-dst { if seq.disable-dst {
let p = participants.at(i2) let p = participants.at(i2)
p.lifeline-lvl -= 1 p.lifeline-lvl -= 1
participants.at(i2) = p participants.at(i2) = p
} }
if elmt.enable-dst { if seq.enable-dst {
let p = participants.at(i2) let p = participants.at(i2)
p.lifeline-lvl += 1 p.lifeline-lvl += 1
p.max-lifelines = calc.max(p.max-lifelines, p.lifeline-lvl) p.max-lifelines = calc.max(p.max-lifelines, p.lifeline-lvl)
participants.at(i2) = p participants.at(i2) = p
} }
} else if elmt.type == "evt" {
let par-name = elmt.participant return (participants, cell)
}
#let evt-update-lifelines(participants, pars-i, evt) = {
let par-name = evt.participant
let i = pars-i.at(par-name) let i = pars-i.at(par-name)
let par = participants.at(i) let par = participants.at(i)
if elmt.event == "disable" or elmt.event == "destroy" { if evt.event == "disable" or evt.event == "destroy" {
par.lifeline-lvl -= 1 par.lifeline-lvl -= 1
} else if elmt.event == "enable" { } else if evt.event == "enable" {
par.lifeline-lvl += 1 par.lifeline-lvl += 1
par.max-lifelines = calc.max(par.max-lifelines, par.lifeline-lvl) par.max-lifelines = calc.max(par.max-lifelines, par.lifeline-lvl)
} }
participants.at(i) = par participants.at(i) = par
return participants
}
} else if elmt.type == "note" { #let note-get-cell(note) = {
let (p1, p2) = (none, none) let (p1, p2) = (none, none)
let cell = none let cell = none
if elmt.side == "left" { if note.side == "left" {
p1 = "[" p1 = "["
p2 = elmt.pos p2 = note.pos
cell = get-note-box(elmt) cell = get-note-box(note)
} else if elmt.side == "right" { } else if note.side == "right" {
p1 = elmt.pos p1 = note.pos
p2 = "]" p2 = "]"
cell = get-note-box(elmt) cell = get-note-box(note)
} else if elmt.side == "over" { } else if note.side == "over" and note.aligned-with != none {
if elmt.aligned-with != none { let box1 = get-note-box(note)
let box1 = get-note-box(elmt) let box2 = get-note-box(note.aligned-with)
let box2 = get-note-box(elmt.aligned-with)
let m1 = measure(box1) let m1 = measure(box1)
let m2 = measure(box2) let m2 = measure(box2)
cell = box(width: (m1.width + m2.width) / 2, height: calc.max(m1.height, m2.height)) cell = box(
p1 = elmt.pos width: (m1.width + m2.width) / 2,
p2 = elmt.aligned-with.pos height: calc.max(m1.height, m2.height)
} )
p1 = note.pos
p2 = note.aligned-with.pos
} else {
return none
} }
if p1 != none and p2 != none and cell != none {
let i1 = pars-i.at(p1) let i1 = pars-i.at(p1)
let i2 = pars-i.at(p2) let i2 = pars-i.at(p2)
cells.push( cell = (
( elmt: note,
elmt: elmt,
i1: calc.min(i1, i2), i1: calc.min(i1, i2),
i2: calc.max(i1, i2), i2: calc.max(i1, i2),
cell: cell cell: cell
) )
return cell
}
#let compute-max-lifeline-levels(participants, elements, pars-i) = {
let cells = ()
for elmt in elements {
if elmt.type == "seq" {
let cell
(participants, cell) = seq-update-lifelines(
participants,
pars-i,
elmt
) )
cells.push(cell)
} else if elmt.type == "evt" {
participants = evt-update-lifelines(
participants,
pars-i,
elmt
)
} else if elmt.type == "note" {
let cell = note-get-cell(elmt)
if cell != none {
cells.push(cell)
} }
} }
} }
// Compute column widths return (participants, elements, cells)
// Compute minimum widths for participant names and shapes }
/// Compute minimum widths for participant names and shapes
#let participants-min-col-widths(participants) = {
let widths = () let widths = ()
for i in range(participants.len() - 1) { for i in range(participants.len() - 1) {
let p1 = participants.at(i) let p1 = participants.at(i)
@ -124,10 +162,14 @@
let w2 = m2.width let w2 = m2.width
widths.push(w1 / 2pt + w2 / 2pt + PAR-SPACE) widths.push(w1 / 2pt + w2 / 2pt + PAR-SPACE)
} }
return widths
}
// Compute minimum width for over notes /// Compute minimum width for over notes
for n in elements.filter(e => (e.type == "note" and #let notes-min-col-widths(elements, widths, pars-i) = {
e.side == "over" and let widths = widths
let notes = elements.filter(e => e.type == "note")
for n in notes.filter(e => (e.side == "over" and
type(e.pos) == str)) { type(e.pos) == str)) {
let m = note.get-size(n) let m = note.get-size(n)
@ -146,8 +188,12 @@
) )
} }
} }
return widths
}
// Compute minimum width for simple sequences (spanning 1 column) /// Compute minimum width for simple sequences (spanning 1 column)
#let simple-seq-min-col-widths(cells, widths) = {
let widths = widths
for cell in cells.filter(c => c.i2 - c.i1 == 1) { for cell in cells.filter(c => c.i2 - c.i1 == 1) {
let m = measure(cell.cell) let m = measure(cell.cell)
widths.at(cell.i1) = calc.max( widths.at(cell.i1) = calc.max(
@ -155,9 +201,14 @@
m.width / 1pt + COMMENT-PAD m.width / 1pt + COMMENT-PAD
) )
} }
return widths
}
// Compute minimum width for self sequences /// Compute minimum width for self sequences
for cell in cells.filter(c => c.elmt.type == "seq" and c.i1 == c.i2) { #let self-seq-min-col-widths(cells, widths) = {
let widths = widths
for cell in cells.filter(c => (c.elmt.type == "seq" and
c.i1 == c.i2)) {
let m = measure(cell.cell) let m = measure(cell.cell)
let i = cell.i1 let i = cell.i1
if cell.elmt.flip { if cell.elmt.flip {
@ -170,35 +221,49 @@
) )
} }
} }
return widths
}
// Compute remaining widths for longer sequences (spanning multiple columns) /// Compute remaining widths for longer sequences (spanning multiple columns)
#let long-seq-min-col-widths(cells, widths) = {
let widths = widths
let multicol-cells = cells.filter(c => c.i2 - c.i1 > 1) let multicol-cells = cells.filter(c => c.i2 - c.i1 > 1)
multicol-cells = multicol-cells.sorted(key: c => { multicol-cells = multicol-cells.sorted(key: c => {
c.i1 * 1000 + c.i2 c.i1 * 1000 + c.i2
}) })
for cell in multicol-cells { for cell in multicol-cells {
let m = measure(cell.cell) let m = measure(cell.cell)
let width = (
m.width / 1pt +
COMMENT-PAD -
widths.slice(cell.i1, cell.i2 - 1).sum()
)
widths.at(cell.i2 - 1) = calc.max( widths.at(cell.i2 - 1) = calc.max(
widths.at(cell.i2 - 1), widths.at(cell.i2 - 1), width
m.width / 1pt + COMMENT-PAD - widths.slice(cell.i1, cell.i2 - 1).sum()
) )
} }
return widths
}
// Add lifeline widths /// Add lifeline widths
for (i, w) in widths.enumerate() { #let col-widths-add-lifelines(participants, widths) = {
return widths.enumerate().map(((i, w)) => {
let p1 = participants.at(i) let p1 = participants.at(i)
let p2 = participants.at(i + 1) let p2 = participants.at(i + 1)
let w = w + p1.max-lifelines * LIFELINE-W / 2 w += p1.max-lifelines * LIFELINE-W / 2
if p2.max-lifelines != 0 { if p2.max-lifelines != 0 {
w += LIFELINE-W / 2 w += LIFELINE-W / 2
} }
widths.at(i) = w return w
})
} }
for elmt in elements { #let process-col-elements(elements, widths, pars-i) = {
if elmt.type == "col" { let widths = widths
let i1 = pars-i.at(elmt.p1) let cols = elements.filter(e => e.type == "col")
let i2 = pars-i.at(elmt.p2) for col in cols {
let i1 = pars-i.at(col.p1)
let i2 = pars-i.at(col.p2)
if calc.abs(i1 - i2) != 1 { if calc.abs(i1 - i2) != 1 {
let i-min = calc.min(i1, i2) let i-min = calc.min(i1, i2)
let i-max = calc.max(i1, i2) let i-max = calc.max(i1, i2)
@ -215,27 +280,53 @@
} }
let i = calc.min(i1, i2) let i = calc.min(i1, i2)
if elmt.width != auto {
widths.at(i) = normalize-units(elmt.width)
}
let width = widths.at(i) let width = widths.at(i)
width = calc.max(width, normalize-units(elmt.min-width))
if elmt.max-width != none { if col.width != auto {
width = calc.min(width, normalize-units(elmt.max-width)) width = normalize-units(col.width)
}
widths.at(i) = width + normalize-units(elmt.margin)
}
} }
width = calc.max(
width,
normalize-units(col.min-width)
)
if col.max-width != none {
width = calc.min(
width,
normalize-units(col.max-width)
)
}
widths.at(i) = width + normalize-units(col.margin)
}
return widths
}
#let get-columns-width(participants, elements, pars-i) = {
elements = elements.filter(is-elmt)
elements = unwrap-syncs(elements)
let cells
(participants, elements, cells) = compute-max-lifeline-levels(participants, elements, pars-i)
let widths = participants-min-col-widths(participants)
widths = notes-min-col-widths(elements, widths, pars-i)
widths = simple-seq-min-col-widths(cells, widths)
widths = self-seq-min-col-widths(cells, widths)
widths = long-seq-min-col-widths(cells, widths)
widths = col-widths-add-lifelines(participants, widths)
widths = process-col-elements(elements, widths, pars-i)
return widths return widths
} }
#let render(participants, elements) = context canvas(length: 1pt, { #let render(participants, elements) = context canvas(length: 1pt, {
let participants = participants
let elements = elements
let shapes = () let shapes = ()
participants = init-lifelines(participants)
let pars-i = get-participants-i(participants) let pars-i = get-participants-i(participants)
let widths = get-columns-width(participants, elements) let widths = get-columns-width(participants, elements, pars-i)
// Compute each column's X position // Compute each column's X position
let x-pos = (0,) let x-pos = (0,)
@ -267,8 +358,11 @@
// Draw elemnts // Draw elemnts
for elmt in elements { for elmt in elements {
if not is-elmt(elmt) {
shapes.push(elmt)
// Sequences // Sequences
if elmt.type == "seq" { } else if elmt.type == "seq" {
let shps let shps
(y, lifelines, shps) = draw-seq(elmt, y, lifelines) (y, lifelines, shps) = draw-seq(elmt, y, lifelines)
shapes += shps shapes += shps

View File

@ -1,5 +1,5 @@
#import "core/utils.typ": fit-canvas #import "core/utils.typ": fit-canvas
#import "renderer.typ": render #import "core/renderer.typ": render
#import "participant.typ" as participant: _par, PAR-SPECIALS #import "participant.typ" as participant: _par, PAR-SPECIALS
#import "sequence.typ": _seq #import "sequence.typ": _seq