fixed typo in participant.typ

gallery/notes.typ

@@ -21,10 +21,10 @@
   _par("a", display-name: "Alice")
   _par("b", display-name: "Bob")
 
-  _note("left", [This is displayed\ left of Alice.], pos: "a", color: rgb("#00FFFF"))
+  _note("left", [This is displayed\ left of Alice.], pos: "a", fill: rgb("#00FFFF"))
   _note("right", [This is displayed right of Alice.], pos: "a")
   _note("over", [This is displayed over Alice.], pos: "a")
-  _note("over", [This is displayed\ over Bob and Alice.], pos: ("a", "b"), color: rgb("#FFAAAA"))
+  _note("over", [This is displayed\ over Bob and Alice.], pos: ("a", "b"), fill: rgb("#FFAAAA"))
   _note("over", [This is yet another\ example of\ a long note.], pos: ("a", "b"))
 })
 

src/core/draw/note.typ

@@ -1,14 +1,51 @@
-#import "/src/cetz.typ": draw
+#import "/src/cetz.typ": draw, styles
 
 #import "/src/consts.typ": *
-#import "/src/core/utils.typ": get-ctx, set-ctx, expand-parent-group
+#import "/src/core/utils.typ": get-ctx, is-elmt, set-ctx, expand-parent-group, normalize-measure
+
+#let note-default-style = (
+  shape: "default",
+  fill: rgb("#FEFFDD"),
+  stroke: black + .5pt
+)
+
+#let resolve-style(ctx, note) = {
+  return styles.resolve(
+    ctx.style,
+    merge: note.style,
+    root: "note",
+    base: note-default-style
+  )
+}
+
+#let pre-resolve-styles() = get-ctx(ctx => {
+  let ctx = ctx
+  let notes = ctx.setup.notes
+
+  for (i, elmt) in ctx.setup.elements.enumerate() {
+    if type(elmt) == function {
+      ctx = elmt(ctx).ctx
+    } else if is-elmt(elmt) {
+      if elmt.type == "note" {
+        let style = resolve-style(ctx, elmt)
+        notes.at(elmt.id).insert("resolved-style", style)
+      }
+    }
+  }
+
+  set-ctx(c => {
+    c.setup.notes = notes
+    return c
+  })
+})
 
 #let get-size(note) = {
-  let PAD = if note.shape == "hex" {NOTE-HEX-PAD} else {NOTE-PAD}
+  let style = note.resolved-style
+  let PAD = if style.shape == "hex" {NOTE-HEX-PAD} else {NOTE-PAD}
   let m = measure(box(note.content))
   let w = m.width / 1pt + PAD.last() * 2
   let h = m.height / 1pt + PAD.first() * 2
-  if note.shape == "default" {
+  if style.shape == "default" {
     w += NOTE-CORNER-SIZE
   }
   return (
@@ -31,14 +68,15 @@
 }
 
 #let get-box(note) = {
-  let PAD = if note.shape == "hex" {NOTE-HEX-PAD} else {NOTE-PAD}
+  let style = note.resolved-style
+  let PAD = if style.shape == "hex" {NOTE-HEX-PAD} else {NOTE-PAD}
   let inset = (
     left: PAD.last() * 1pt,
     right: PAD.last() * 1pt,
     top: PAD.first() * 1pt,
     bottom: PAD.first() * 1pt,
   )
-  if note.shape == "default" {
+  if style.shape == "default" {
     inset.right += NOTE-CORNER-SIZE * 1pt
   }
   if note.side == "left" {
@@ -62,18 +100,21 @@
   }
 
   get-ctx(ctx => {
+    let note = ctx.notes.at(note.id)
     let y = y
     if y == auto {
       y = ctx.y
     }
-    
 
-    let PAD = if note.shape == "hex" {NOTE-HEX-PAD} else {NOTE-PAD}
-    let m = measure(box(note.content))
-    let w = m.width / 1pt + PAD.last() * 2
-    let h = m.height / 1pt + PAD.first() * 2
+    let style = note.resolved-style
+    let shape = style.shape
+    
+    let PAD = if shape == "hex" {NOTE-HEX-PAD} else {NOTE-PAD}
+    let m = normalize-measure(box(note.content))
+    let w = m.width + PAD.last() * 2
+    let h = m.height + PAD.first() * 2
     let total-w = w
-    if note.shape == "default" {
+    if shape == "default" {
       total-w += NOTE-CORNER-SIZE
     }
 
@@ -106,31 +147,31 @@
     }
     let y1 = y0 - h
 
-    if note.shape == "default" {
+    if shape == "default" {
       draw.line(
         (x0, y0),
         (x1, y0),
         (x2, y0 - NOTE-CORNER-SIZE),
         (x2, y1),
         (x0, y1),
-        stroke: black + .5pt,
-        fill: note.color,
+        stroke: style.stroke,
+        fill: style.fill,
         close: true
       )
       draw.line(
         (x1, y0),
         (x1, y0 - NOTE-CORNER-SIZE),
         (x2, y0 - NOTE-CORNER-SIZE),
-        stroke: black + .5pt
+        stroke: style.stroke
       )
-    } else if note.shape == "rect" {
+    } else if shape == "rect" {
       draw.rect(
         (x0, y0),
         (x2, y1),
-        stroke: black + .5pt,
-        fill: note.color
+        stroke: style.stroke,
+        fill: style.fill
       )
-    } else if note.shape == "hex" {
+    } else if shape == "hex" {
       let lx = x0 + PAD.last()
       let rx = x2 - PAD.last()
       let my = (y0 + y1) / 2
@@ -141,8 +182,8 @@
         (rx, y1),
         (lx, y1),
         (x0, my),
-        stroke: black + .5pt,
-        fill: note.color,
+        stroke: style.stroke,
+        fill: style.fill,
         close: true
       )
     }

src/core/draw/participant.typ

@@ -3,38 +3,28 @@
 #import "/src/consts.typ": *
 #import "/src/core/utils.typ": get-ctx, get-style, is-elmt, set-ctx
 
-#import "participant/default.typ" as shape-default
-#import "participant/actor.typ" as shape-actor
-#import "participant/boundary.typ" as shape-boundary
-#import "participant/control.typ" as shape-control
-#import "participant/entity.typ" as shape-entity
-#import "participant/database.typ" as shape-database
-#import "participant/collections.typ" as shape-collections
-#import "participant/queue.typ" as shape-queue
-#import "participant/custom.typ" as shape-custom
-
 #let shapes = {
-  let from-module(mod) = {
-    let p = "participant/default.typ"
+  let from-module(path) = {
+    import path as mod
     return (mod.name: (
       get-size: mod.get-size,
       render: mod.render,
       default-style: mod.default-style
     ))
   }
-  from-module(shape-default)
-  from-module(shape-actor)
-  from-module(shape-boundary)
-  from-module(shape-control)
-  from-module(shape-entity)
-  from-module(shape-database)
-  from-module(shape-collections)
-  from-module(shape-queue)
-  from-module(shape-custom)
+  from-module("participant/default.typ")
+  from-module("participant/actor.typ")
+  from-module("participant/boundary.typ")
+  from-module("participant/control.typ")
+  from-module("participant/entity.typ")
+  from-module("participant/database.typ")
+  from-module("participant/collections.typ")
+  from-module("participant/queue.typ")
+  from-module("participant/custom.typ")
 }
 
 #let participant-default-style = (
-  fill: auto,
+  fill: rgb("#E2E2F0"),
   stroke: black + .5pt,
   from-start: true,
   show-bottom: true,
@@ -64,8 +54,10 @@
   return style
 }
 
-#let pre-resolve-styles(ctx, elements, participants) = {
+#let pre-resolve-styles() = get-ctx(ctx => {
   let idx = (:)
+  let elements = ctx.setup.elements
+  let participants = ctx.setup.participants
   for (i, par) in participants.enumerate() {
     par.insert("resolved-style", resolve-style(ctx, par))
     participants.at(i) = par
@@ -75,7 +67,7 @@
     if type(elmt) == function {
       ctx = elmt(ctx).ctx
     } else if is-elmt(elmt) {
-      if elmt.type == par {
+      if elmt.type == "par" {
         let style = resolve-style(ctx, elmt)
         elements.at(i).insert("resolved-style", style)
         let i = idx.at(elmt.name)
@@ -83,12 +75,17 @@
       }
     }
   }
-  return (elements, participants)
-}
+  
+  set-ctx(c => {
+    c.setup.elements = elements
+    c.setup.participants = participants
+    return c
+  })
+})
 
 #let get-size(par) = {
   if par.invisible {
-    return (width: 0pt, height: 0pt)
+    return (width: 0, height: 0)
   }
   let style = par.resolved-style
   let func = shapes.at(style.shape).get-size
@@ -119,7 +116,7 @@
     // Compute lifeline rectangles + destruction positions
     for event in ctx.lifelines.at(p.i).events {
       if event.type == "create" {
-        last-y = line.at(1)
+        last-y = event.y
 
       } else if event.type == "enable" {
         if stack.len() == 0 {

src/core/draw/sequence.typ

@@ -70,7 +70,8 @@
   h = calc.max(
     h,
     ..seq.linked-notes.map(n => {
-      note.get-size(n).height / 2
+      let nt = ctx.notes.at(n.id)
+      note.get-size(nt).height / 2
     })
   )
   ctx.y -= h
@@ -354,7 +355,8 @@
     end-info.y = calc.min(
       end-info.y,
       y0 - calc.max(..seq.linked-notes.map(n => {
-        let m = note.get-size(n)
+        let nt = ctx.notes.at(n.id)
+        let m = note.get-size(nt)
         return m.height / 2
       }))
     )

src/core/renderer.typ

@@ -1,19 +1,28 @@
 #import "/src/cetz.typ": canvas, draw
 
-#import "draw/note.typ": get-box as get-note-box, get-size as get-note-size
+#import "draw/note.typ"
 #import "draw/participant.typ"
 #import "utils.typ": *
 #import "/src/consts.typ": *
 
 #let DEBUG-INVISIBLE = false
 
-#let init-lifelines(participants) = {
-  return participants.map(p => {
+#let init-lifelines() = set-ctx(ctx => {
+  ctx.setup.participants = ctx.setup.participants.map(p => {
     p.insert("lifeline-lvl", 0)
     p.insert("max-lifelines", 0)
     p
   })
-}
+  return ctx
+})
+
+#let set-participants-i() = set-ctx(ctx => {
+  ctx.setup.insert(
+    "pars-i",
+    get-participants-i(ctx.setup.participants)
+  )
+  return ctx
+})
 
 #let unwrap-syncs(elements) = {
   let i = 0
@@ -78,28 +87,29 @@
   return participants
 }
 
-#let note-get-cell(pars-i, note) = {
+#let note-get-cell(notes, pars-i, n) = {
   let (p1, p2) = (none, none)
   let cell = none
-  if note.side == "left" {
-    p1 = note.pos2
-    p2 = note.pos
-    cell = get-note-box(note)
-  } else if note.side == "right" {
-    p1 = note.pos
-    p2 = note.pos2
-    cell = get-note-box(note)
-  } else if note.side == "over" and note.aligned-with != none {
-    let box1 = get-note-box(note)
-    let box2 = get-note-box(note.aligned-with)
+  if n.side == "left" {
+    p1 = n.pos2
+    p2 = n.pos
+    cell = note.get-box(n)
+  } else if n.side == "right" {
+    p1 = n.pos
+    p2 = n.pos2
+    cell = note.get-box(n)
+  } else if n.side == "over" and n.aligned-with != none {
+    let aligned-with = notes.at(n.aligned-with.id)
+    let box1 = note.get-box(n)
+    let box2 = note.get-box(aligned-with)
     let m1 = measure(box1)
     let m2 = measure(box2)
     cell = box(
       width: (m1.width + m2.width) / 2,
       height: calc.max(m1.height, m2.height)
     )
-    p1 = note.pos
-    p2 = note.aligned-with.pos
+    p1 = n.pos
+    p2 = n.aligned-with.pos
   } else {
     return none
   }
@@ -107,7 +117,7 @@
   let i1 = pars-i.at(p1)
   let i2 = pars-i.at(p2)
   cell = (
-    elmt: note,
+    elmt: n,
     i1: calc.min(i1, i2),
     i2: calc.max(i1, i2),
     cell: cell
@@ -116,7 +126,7 @@
   return cell
 }
 
-#let compute-max-lifeline-levels(participants, elements, pars-i) = {
+#let compute-max-lifeline-levels(participants, elements, notes, pars-i) = {
   let cells = ()
   for elmt in elements {
     if elmt.type == "seq" {
@@ -135,14 +145,14 @@
       )
     
     } else if elmt.type == "note" {
-      let cell = note-get-cell(pars-i, elmt)
+      let cell = note-get-cell(notes, pars-i, notes.at(elmt.id))
       if cell != none {
         cells.push(cell)
       }
     }
   }
 
-  return (participants, elements, cells)
+  return (participants, cells)
 }
 
 /// Compute minimum widths for participant names and shapes
@@ -161,13 +171,12 @@
 }
 
 /// Compute minimum width for over notes
-#let notes-min-col-widths(elements, widths, pars-i) = {
+#let notes-min-col-widths(notes, widths, pars-i) = {
   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)) {
     
-    let m = get-note-size(n)
+    let m = note.get-size(n)
     let i = pars-i.at(n.pos)
 
     if i < widths.len() {
@@ -306,62 +315,71 @@
   return widths
 }
 
-#let compute-columns-width(participants, elements, pars-i) = {
-  elements = elements.filter(is-elmt)
+#let compute-columns-width() = set-ctx(ctx => {
+  let elements = ctx.setup.elements.filter(is-elmt)
+  let participants = ctx.setup.participants
+  let notes = ctx.setup.notes
+  let pars-i = ctx.setup.pars-i
   elements = unwrap-syncs(elements)
 
   let cells
-  (participants, elements, cells) = compute-max-lifeline-levels(participants, elements, pars-i)
+  (participants, cells) = compute-max-lifeline-levels(participants, elements, notes, pars-i)
 
   let widths = participants-min-col-widths(participants)
-  widths = notes-min-col-widths(elements, widths, pars-i)
+  widths = notes-min-col-widths(notes, 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(participants, cells, widths)
   widths = col-widths-add-lifelines(participants, widths)
   widths = process-col-elements(elements, widths, pars-i)
-  return widths
-}
 
-#let setup-ctx(participants, elements) = (ctx => {
-  let state = ctx.at("shared-state", default: (:))
+  let ctx = ctx
+  ctx.insert("widths", widths)
 
-  let (elements, participants) = participant.pre-resolve-styles(
-    extract-ctx(ctx, with-style: true),
-    elements,
-    participants
-  )
-
-  let chronos-ctx = (
-    participants: init-lifelines(participants),
-    pars-i: get-participants-i(participants),
-    y: 0,
-    groups: (),
-    lifelines: participants.map(_ => (
-      level: 0,
-      events: ()
-    ))
-  )
-  chronos-ctx.insert(
-    "widths",
-    compute-columns-width(
-      chronos-ctx.participants,
-      elements,
-      chronos-ctx.pars-i
-    )
-  )
-
-  // Compute each column's X position
   let x-pos = (0,)
-  for width in chronos-ctx.widths {
+  for width in widths {
     x-pos.push(x-pos.last() + width)
   }
-  chronos-ctx.insert("x-pos", x-pos)
-  state.insert("chronos", chronos-ctx)
-  ctx.shared-state = state
-  return (
-    ctx: ctx
-  )
+  ctx.insert("x-pos", x-pos)
+
+  return ctx
+})
+
+#let setup-ctx(participants, elements, notes) = draw.get-ctx(_ => {
+  set-ctx(c => {
+    c.insert("participants", participants)
+    c.insert("setup", (
+      elements: elements,
+      participants: participants,
+      notes: notes
+    ))
+    return c
+  })
+
+  participant.pre-resolve-styles()
+
+  init-lifelines()
+  set-participants-i()
+  
+  note.pre-resolve-styles()
+  compute-columns-width()
+
+  set-ctx(c => {
+    let setup = c.setup
+    c += (
+      participants: setup.participants,
+      pars-i: setup.pars-i,
+      y: 0,
+      groups: (),
+      lifelines: setup.participants.map(_ => (
+        level: 0,
+        events: ()
+      )),
+      notes: setup.notes
+    )
+    c.remove("setup")
+    return c
+  })
 },)
 
 #let render-debug() = get-ctx(ctx => {
@@ -382,9 +400,9 @@
   }
 })
 
-#let render(participants, elements) = context canvas(length: 1pt, {
-  setup-ctx(participants, elements)
-  
+#let render(participants, elements, notes) = context canvas(length: 1pt, {
+  setup-ctx(participants, elements, notes)
+
   // Draw participants (start)
   get-ctx(ctx => {
     for p in ctx.participants {

src/core/setup.typ

@@ -126,6 +126,7 @@
     "linked",
     note.pos == none and note.side != "across"
   )
+  note.insert("id", ctx.notes.len())
   let names = ctx.participants.map(p => p.name)
   if note.pos == none and note.side != "across" {
     let i1 = names.position(n => n == ctx.last-seq.p1)
@@ -149,6 +150,7 @@
     let n = ctx.last-note.note
     n.aligned-with = note
     ctx.elmts.at(ctx.last-note.i) = n
+    ctx.notes.at(ctx.last-note.note.id) = n
   }
 
   if note.side in ("left", "right") {
@@ -186,6 +188,8 @@
     }
   }
 
+  ctx.notes.push(note)
+
   ctx.elmts.at(ctx.i) = note
 
   ctx.last-note = (
@@ -226,6 +230,7 @@
     linked: (),
     last-seq: none,
     last-note: none,
+    notes: (),
     participants: (),
     elmts: elmts,
     i: 0
@@ -290,10 +295,10 @@
     ).first())
   }
 
-  return (ctx.elmts, participants)
+  return (ctx.elmts, participants, ctx.notes)
 }
 
-#let finalize-setup(elmts, participants) = {
+#let finalize-setup(elmts, participants, notes) = {
   for (i, p) in participants.enumerate() {
     p.insert("i", i)
     participants.at(i) = p
@@ -323,14 +328,11 @@
     elmts.at(i).insert("max-i", max-i)
   }
 
-  return (elmts, participants)
+  return (elmts, participants, notes)
 }
 
 #let setup(elements) = {
   let (elmts, activation-history) = unwrap-containers(elements)
-  
-  let participants
-  (elmts, participants) = prepare-participants(elmts)
-
-  return finalize-setup(elmts, participants)
+  let (elmts, participants, notes) = prepare-participants(elmts)
+  return finalize-setup(elmts, participants, notes)
 }

src/diagram.typ

@@ -1,16 +1,15 @@
-#import "core/draw/event.typ": render as evt-render
 #import "core/renderer.typ": render
 #import "core/setup.typ": setup
-#import "core/utils.typ": fit-canvas, set-ctx
+#import "core/utils.typ": fit-canvas
 
 #let diagram(elements, width: auto) = {
   if elements == none {
     return
   }
   
-  let (elmts, participants) = setup(elements)
+  let (elmts, participants, notes) = setup(elements)
 
-  let canvas = render(participants, elmts)
+  let canvas = render(participants, elmts, notes)
   fit-canvas(canvas, width: width)
 }
 

src/note.typ

@@ -18,10 +18,9 @@
   side,
   content,
   pos: none,
-  color: COL-NOTE,
-  shape: "default",
   aligned: false,
-  allow-overlap: true
+  allow-overlap: true,
+  ..style
 ) = {
   if side == "over" {
     if pos == none {
@@ -33,19 +32,15 @@
       panic("Aligned notes can only be over a participant (got side '" + side + "')")
     }
   }
-  if color == auto {
-    color = COL-NOTE
-  }
   return ((
     type: "note",
     draw: note.render,
     side: side,
     content: content,
     pos: pos,
-    color: color,
-    shape: shape,
     aligned: aligned,
     aligned-with: none,
-    allow-overlap: allow-overlap
+    allow-overlap: allow-overlap,
+    style: style.named()
   ),)
 }

src/participant.typ

@@ -25,7 +25,7 @@
     draw: participant.render,
     name: name,
     display-name: if display-name == auto {name} else {display-name},
-    invisible: false,
+    invisible: invisible,
     style: style.named()
   ),)
 }

tests/note/color/test.typ

@@ -5,9 +5,9 @@
   _par("a", display-name: "Alice")
   _par("b", display-name: "Bob")
 
-  _note("left", [This is displayed\ left of Alice.], pos: "a", color: rgb("#00FFFF"))
+  _note("left", [This is displayed\ left of Alice.], pos: "a", fill: rgb("#00FFFF"))
   _note("right", [This is displayed right of Alice.], pos: "a")
   _note("over", [This is displayed over Alice.], pos: "a")
-  _note("over", [This is displayed\ over Bob and Alice.], pos: ("a", "b"), color: rgb("#FFAAAA"))
+  _note("over", [This is displayed\ over Bob and Alice.], pos: ("a", "b"), fill: rgb("#FFAAAA"))
   _note("over", [This is yet another\ example of\ a long note.], pos: ("a", "b"))
 })