extracted cetz import for easier updating

Reviewed-on: #5

README.md

@@ -1,3 +1,64 @@
 # circuiteria
 
-Drawing block circuits with Typst made easy, using CeTZ
+Circuiteria is a [Typst](https://typst.app) package for drawing block circuit diagrams using the [CeTZ](https://typst.app/universe/package/cetz) package.
+
+<p align="center">
+  <img src="./gallery/platypus.png" alt="Perry the platypus">
+</p>
+
+## Examples
+<table>
+  <tr>
+    <td colspan="2">
+      <a href="./gallery/test.typ">
+        <img src="./gallery/test.png" width="500px">
+      </a>
+    </td>
+  </tr>
+  <tr>
+    <td colspan="2">A bit of eveything</td>
+  </tr>
+  <tr>
+    <td colspan="2">
+      <a href="./gallery/test5.typ">
+        <img src="./gallery/test5.png" width="500px">
+      </a>
+    </td>
+  </tr>
+  <tr>
+    <td colspan="2">Wires everywhere</td>
+  </tr>
+  <tr>
+    <td>
+      <a href="./gallery/test4.typ">
+        <img src="./gallery/test4.png" width="250px">
+      </a>
+    </td>
+    <td>
+      <a href="./gallery/test6.typ">
+        <img src="./gallery/test6.png" width="250px">
+      </a>
+    </td>
+  </tr>
+  <tr>
+    <td>Groups</td>
+    <td>Rotated</td>
+  </tr>
+</table>
+
+> **Note**\
+> These circuit layouts were copied from a digital design course given by prof. S. Zahno and recreated using this package
+
+*Click on the example image to jump to the code.*
+
+## Usage
+For more information, see the [manual](manual.pdf)
+
+To use this package, simply import [circuiteria](https://typst.app/universe/package/circuiteria) and call the `circuit` function:
+```typ
+#import "@preview/circuiteria:0.2.0"
+#circuiteria.circuit({
+  import circuiteria: *
+  ...
+})
+```

doc/example.typ

@@ -0,0 +1,41 @@
+#import "/src/cetz.typ": draw
+#import "../src/circuit.typ": circuit
+#import "../src/util.typ"
+
+#let example-preamble = "import \"../src/lib.typ\": *;"
+#let example-scope = (
+  draw: draw
+)
+
+#let example(src, show-src: true, vertical: false, fill: true) = {
+  src = src.text
+  let full-src = example-preamble + src
+  let body = eval(full-src, scope: example-scope)
+  let img = circuit(length: 2em, body)
+
+  block(width: 100%,
+    align(center,
+      box(
+        stroke: black + 1pt,
+        radius: .5em,
+        fill: if fill {util.colors.yellow.lighten(80%)} else {none},
+        if show-src {
+          let src-block = align(left, raw(src, lang: "typc"))
+          table(
+            columns: if vertical {1} else {2},
+            inset: 1em,
+            align: horizon + center,
+            stroke: none,
+            img,
+            if vertical {table.hline()} else {table.vline()}, src-block
+          )
+        } else {
+          table(
+            inset: 1em,
+            img
+          )
+        }
+      )
+    )
+  )
+}

doc/examples.typ

@@ -0,0 +1,150 @@
+#import "example.typ": example
+
+#let alu = example(```
+element.alu(x: 0, y: 0, w: 1, h: 2, id: "alu")
+wire.stub("alu-port-in1", "west")
+wire.stub("alu-port-in2", "west")
+wire.stub("alu-port-out", "east")
+```)
+
+#let block = example(```
+element.block(
+  x: 0, y: 0, w: 2, h: 2, id: "block",
+  ports: (
+    north: ((id: "clk", clock: true),),
+    west: ((id: "in1", name: "A"),
+           (id: "in2", name: "B")),
+    east: ((id: "out", name: "C"),)
+  )
+)
+wire.stub("block-port-clk", "north")
+wire.stub("block-port-in1", "west")
+wire.stub("block-port-in2", "west")
+wire.stub("block-port-out", "east")
+```)
+
+#let extender = example(```
+element.extender(
+  x: 0, y: 0, w: 3, h: 1,
+  id: "extender"
+)
+wire.stub("extender-port-in", "west")
+wire.stub("extender-port-out", "east")
+```)
+
+#let multiplexer = example(```
+element.multiplexer(
+  x: 0, y: 0, w: 1, h: 3,
+  id: "multiplexer",
+  entries: 3
+)
+wire.stub("multiplexer.north", "north")
+wire.stub("multiplexer-port-out", "east")
+
+element.multiplexer(
+  x: 0, y: -4, w: 1, h: 3,
+  id: "multiplexer2",
+  entries: ("A", "B", "C")
+)
+wire.stub("multiplexer2.south", "south")
+wire.stub("multiplexer2-port-out", "east")
+
+for i in range(3) {
+  wire.stub("multiplexer-port-in" + str(i), "west")
+  wire.stub("multiplexer2-port-in" + str(i), "west")
+}
+```)
+
+#let wires = example(```
+for i in range(3) {
+  draw.circle((i * 3, 0), radius: .1, name: "p" + str(i * 2))
+  draw.circle((i * 3 + 2, 1), radius: .1, name: "p" + str(i * 2 + 1))
+  draw.content((i * 3 + 1, -1), raw(wire.wire-styles.at(i)))
+}
+wire.wire("w1", ("p0", "p1"), style: "direct")
+wire.wire("w2", ("p2", "p3"), style: "zigzag")
+wire.wire("w3", ("p4", "p5"), style: "dodge",
+          dodge-y: -0.5, dodge-margins: (0.5, 0.5))
+```, vertical: true)
+
+#let stub = example(```
+draw.circle((0, 0), radius: .1, name: "p")
+wire.stub("p", "north", name: "north", length: 1)
+wire.stub("p", "east", name: "east", vertical: true)
+wire.stub("p", "south", name: "south", length: 15pt)
+wire.stub("p", "west", name: "west", length: 3em)
+```)
+
+#let gate-and = example(```
+gates.gate-and(x: 0, y: 0, w: 1.5, h: 1.5)
+gates.gate-and(x: 3, y: 0, w: 1.5, h: 1.5, inverted: "all")
+```, vertical: true)
+
+#let gate-nand = example(```
+gates.gate-nand(x: 0, y: 0, w: 1.5, h: 1.5)
+gates.gate-nand(x: 3, y: 0, w: 1.5, h: 1.5, inverted: "all")
+```, vertical: true)
+
+#let gate-buf = example(```
+gates.gate-buf(x: 0, y: 0, w: 1.5, h: 1.5)
+gates.gate-buf(x: 3, y: 0, w: 1.5, h: 1.5, inverted: "all")
+```, vertical: true)
+
+#let gate-not = example(```
+gates.gate-not(x: 0, y: 0, w: 1.5, h: 1.5)
+gates.gate-not(x: 3, y: 0, w: 1.5, h: 1.5, inverted: "all")
+```, vertical: true)
+
+#let gate-or = example(```
+gates.gate-or(x: 0, y: 0, w: 1.5, h: 1.5)
+gates.gate-or(x: 3, y: 0, w: 1.5, h: 1.5, inverted: "all")
+```, vertical: true)
+
+#let gate-nor = example(```
+gates.gate-nor(x: 0, y: 0, w: 1.5, h: 1.5)
+gates.gate-nor(x: 3, y: 0, w: 1.5, h: 1.5, inverted: "all")
+```, vertical: true)
+
+#let gate-xor = example(```
+gates.gate-xor(x: 0, y: 0, w: 1.5, h: 1.5)
+gates.gate-xor(x: 3, y: 0, w: 1.5, h: 1.5, inverted: "all")
+```, vertical: true)
+
+#let gate-xnor = example(```
+gates.gate-xnor(x: 0, y: 0, w: 1.5, h: 1.5)
+gates.gate-xnor(x: 3, y: 0, w: 1.5, h: 1.5, inverted: "all")
+```, vertical: true)
+
+#let group = example(```
+element.group(
+  id: "g1", name: "Group 1", stroke: (dash: "dashed"),
+  {
+    element.block(id: "b1", w: 2, h: 2,
+      x: 0, y: 1.5,
+      ports: (east: ((id: "out"),)),
+      fill: util.colors.green
+    )
+    element.block(id: "b2", w: 2, h: 1,
+      x: 0, y: 0,
+      ports: (east: ((id: "out"),)),
+      fill: util.colors.orange
+    )
+  }
+)
+element.block(id: "b3", w: 2, h: 3,
+  x: (rel: 1, to: "g1.east"),
+  y: (from: "b1-port-out", to: "in1"),
+  ports: (west: ((id: "in1"), (id: "in2"))),
+  fill: util.colors.blue
+)
+wire.wire("w1", ("b1-port-out", "b3-port-in1"))
+wire.wire("w2", ("b2-port-out", "b3-port-in2"),
+          style: "zigzag")
+```)
+
+#let intersection = example(```
+wire.wire("w1", ((0, 0), (1, 1)), style: "zigzag")
+wire.wire("w2", ((0, 0), (1, -.5)),
+          style: "zigzag", zigzag-ratio: 80%)
+wire.intersection("w1.zig")
+```)

gallery/platypus.typ

@@ -0,0 +1,77 @@
+#import "../src/lib.typ": *
+
+#set page(width: auto, height: auto, margin: .5cm)
+
+#let teal = rgb(37, 155, 166)
+#let orange = rgb(254, 160, 93)
+#let brown = rgb(97, 54, 60)
+
+#circuit({
+  element.group(id: "platypus", name: "A platypus", {
+    element.block(
+      x: 0, y: 0, w: 2, h: 3, id: "body",
+      fill: teal,
+      ports: (
+        east: (
+          (id: "out"),
+        )
+      ),
+      ports-margins: (
+        east: (50%, 10%)
+      )
+    )
+
+    element.block(
+      x: 2.5, y: 1.5, w: 1.5, h: 1, id: "beak",
+      fill: orange,
+      ports: (
+        south: (
+          (id: "in"),
+        )
+      )
+    )
+
+    wire.wire("w1", ("body-port-out", "beak-port-in"), style: "zigzag", zigzag-ratio: 100%)
+  })
+
+  let O = (rel: (2, 0), to: "platypus.south-east")
+  
+  element.group(id: "perry", name: "Perry the platypus", {
+    element.block(
+      x: (rel: 0, to: O), y: 0, w: 2, h: 3, id: "body",
+      fill: teal,
+      ports: (
+        east: (
+          (id: "out"),
+        )
+      ),
+      ports-margins: (
+        east: (50%, 10%)
+      )
+    )
+
+    element.block(
+      x: (rel: 2.5, to: O), y: 1.5, w: 1.5, h: 1, id: "beak",
+      fill: orange,
+      ports: (
+        south: (
+          (id: "in"),
+        )
+      )
+    )
+
+    element.block(
+      x: (rel: 0.25, to: O), y: 3.2, w: 1.5, h: 0.5, id: "hat",
+      fill: brown
+    )
+
+    wire.wire("w2", ("body-port-out", "beak-port-in"), style: "zigzag", zigzag-ratio: 100%)
+  })
+
+  wire.wire(
+    "w3",
+    ("platypus.east", (horizontal: "perry.west", vertical: ())),
+    directed: true,
+    bus: true
+  )
+})

gallery/target_api.typ

@@ -0,0 +1,286 @@
+#import "../src/lib.typ": *
+#import "/src/cetz.typ": draw
+
+#set text(font: "Source Sans 3")
+#set page(width: auto, height: auto, margin: .5cm)
+
+#circuit({
+  element.block(
+    size: (1.5, 2.2),
+    id: "PCBuf",
+    fill: util.colors.orange,
+    ports: (
+      west: "PCNext",
+      north: (id: "CLK", clock: true),
+      east: "PC",
+      south: (("EN", "EN"),)
+    )
+  )
+  
+  wire.stub("PCBuf.CLK", name: "CLK")
+  wire.stub("PCBuf.EN", name: "PCWrite")
+  
+  element.multiplexer(
+    pos: (
+      3, (align: "in0", with: "PCBuf.PC")
+    ),
+    size: (1, 2),
+    id: "AdrSrc-MP",
+    fill: util.colors.orange,
+    entries: 2
+  )
+  wire.wire(
+    "PCBuf.PC",
+    "AdrSrc-MP.in0",
+    id: "wPCBuf-InstDataMgr",
+    name: "PC",
+    bus: true
+  )
+  wire.stub("AdrSrc-MP.north", side: "north", name: "AdrSrc")
+  
+  element.block(
+    pos: (
+      6, (align: "A", with: "AdrSrc-MP.out")
+    ),
+    size: (3, 4),
+    id: "InstDataMgr",
+    fill: util.colors.yellow,
+    ports: (
+      west: (
+        ("A", "A"),
+        ("WD", "WD")
+      ),
+      north: (
+        (id: "CLK", clock: true),
+        (id: "WE", name: "WE", vertical: true),
+        (id: "IRWrite", name: "IRWrite", vertical: true)
+      ),
+      east: (
+        ("Instr", "Instr."),
+        ("RD", "RD")
+      )
+    ),
+    ports-margins: (
+      west: (30%, 0%),
+      east: (40%, 0%)
+    )
+  )
+  wire.wire(
+    "AdrSrc-MP.out",
+    "InstDataMgr.A",
+    id: "wAdrSrcMP-InstDataMgr",
+    name: (none, "Adr"),
+    bus: true
+  )
+  
+  wire.stub("InstDataMgr.CLK", name: "CLK")
+  wire.stub("InstDataMgr.WE")
+  wire.stub("InstDataMgr.IRWrite")
+  wire.stub("InstDataMgr.WD")
+
+  element.block(
+    pos: (
+      15, (align: "WD3", with: "InstDataMgr.RD")
+    ),
+    size: (3, 4),
+    id: "RegFile",
+    fill: util.colors.pink,
+    ports: (
+      west: (
+        ("A1", "A1"),
+        ("A2", "A2"),
+        ("A3", "A3"),
+        ("WD3", "WD3"),
+      ),
+      north: (
+        (id: "CLK", clock: true),
+        (id: "WE3", name: "WE3", vertical: true)
+      ),
+      east: (
+        ("RD1", "RD1"),
+        ("RD2", "RD2"),
+      )
+    ),
+    ports-margins: (
+      east: (20%, 20%)
+    )
+  )
+  wire.stub("RegFile.CLK", name: "CLK")
+  wire.stub("RegFile.WE3", name: "Regwrite", name-offset: 0.6)
+  wire.stub("RegFile.A2")
+  wire.stub("RegFile.RD2")
+
+  element.extender(
+    pos: (15, -3.5),
+    size: (3, 1),
+    id: "Extender",
+    fill: util.colors.green
+  )
+  wire.wire(
+    "Extender.north",
+    (18, -2),
+    id: "wExtender-ImmSrc",
+    style: "zigzag",
+    zigzag-ratio: 0%,
+    name: (none, "ImmSrc"),
+    bus: true
+  )
+
+  
+  let mid = ("InstDataMgr.east", 50%, "RegFile.west")
+  wire.wire(
+    "InstDataMgr.Instr",
+    (vertical: (), horizontal: mid),
+    id: "wInstDataMgr-Bus",
+    name: ("Instr", none),
+    bus: true
+  )
+  wire.wire(
+    (v => (v.at(0), -3.5), mid),
+    (horizontal: (), vertical: (0, 3.5)),
+    id: "wBus",
+    bus: true
+  )
+  wire.wire(
+    "RegFile.A1",
+    (horizontal: mid, vertical: ()),
+    id: "wBus-RegFile-A1",
+    name: (none, "RS1"),
+    slice: (19, 15),
+    reverse: true,
+    bus: true
+  )
+  wire.wire(
+    "RegFile.A3",
+    (horizontal: mid, vertical: ()),
+    id: "wBus-RegFile-A3",
+    name: (none, "RD"),
+    slice: (11, 7),
+    reverse: true,
+    bus: true
+  )
+  wire.wire(
+    "Extender.in",
+    (horizontal: mid, vertical: ()),
+    id: "wBus-Extender",
+    slice: (31, 7),
+    reverse: true,
+    bus: true
+  )
+  
+  element.alu(
+    pos: (
+      22, (align: "in1", with: "RegFile.RD1")
+    ),
+    size: (1, 2),
+    id: "ALU",
+    fill: util.colors.purple
+  )
+  wire.wire(
+    "RegFile.RD1",
+    "ALU.in1",
+    id: "wRegFile-ALU",
+    name: ("A", "SrcA"),
+    bus: true
+  )
+  wire.stub("ALU.north", side: "north")
+
+  element.block(
+    pos: (
+      26, (align: "in", with: "ALU.out")
+    ),
+    size: (1.5, 2),
+    id: "OutBuf",
+    fill: util.colors.orange,
+    ports: (
+      west: "in",
+      north: (id: "CLK", clock: true),
+      east: "out"
+    )
+  )
+  wire.stub("OutBuf.CLK", name: "CLK")
+  wire.wire(
+    "ALU.out",
+    "OutBuf.in",
+    id: "wALU-OutBuf",
+    name: "ALUResult",
+    bus: true
+  )
+
+  element.multiplexer(
+    pos: (
+      30, (align: "in0", with: "OutBuf.out")
+    ),
+    size: (1, 2.5),
+    id: "Res-MP",
+    fill: util.colors.orange,
+    entries: 3
+  )
+  wire.stub("Res-MP.north", side: "north", name: "ResultSrc")
+  wire.stub("Res-MP.in2")
+  wire.wire(
+    "OutBuf.out",
+    "Res-MP.in0",
+    id: "wOutBuf-ResMP",
+    name: "ALUOut",
+    bus: true
+  )
+
+  wire.wire(
+    "Extender.out",
+    "ALU.in2",
+    id: "wExt-ALU",
+    name: ("ImmExt", "SrcB"),
+    bus: true,
+    style: "zigzag",
+    zigzag-ratio: 60%
+  )
+
+  wire.wire(
+    "InstDataMgr.RD",
+    "Res-MP.in1",
+    id: "wInstDataMgr-ResMP",
+    style: "dodge",
+    dodge-y: -4,
+    dodge-sides: ("east", "west"),
+    name: ("Data", none),
+    bus: true
+  )
+
+  wire.wire(
+    "Res-MP.out",
+    "AdrSrc-MP.in1",
+    id: "wResMP-AdrSrc",
+    style: "dodge",
+    dodge-y: -5,
+    dodge-sides: ("east", "west"),
+    dodge-margins: (0.5, 1),
+    bus: true
+  )
+
+  wire.wire(
+    "Res-MP.out",
+    "RegFile.WD3",
+    id: "wResMP-RegFile",
+    style: "dodge",
+    dodge-y: -5,
+    dodge-sides: ("east", "west"),
+    dodge-margins: (0.5, 1),
+    bus: true
+  )
+
+  wire.wire(
+    "Res-MP.out",
+    "PCBuf.PCNext",
+    id: "wResMP-PCBuf",
+    style: "dodge",
+    dodge-y: -5,
+    dodge-sides: ("east", "west"),
+    dodge-margins: (0.5, 1.5),
+    name: (none, "PCNext"),
+    bus: true
+  )
+
+  wire.intersection("wResMP-RegFile.dodge-end", radius: .2)
+  wire.intersection("wResMP-AdrSrc.dodge-end", radius: .2)
+})

gallery/test.typ

@@ -1,11 +1,11 @@
 #import "../src/lib.typ": circuit, element, util, wire
 
-#set page(flipped: true)
+#set page(width: auto, height: auto, margin: .5cm)
 
 #circuit({
   element.block(
     x: 0, y: 0, w: 1.5, h: 2.2,
-    id: "PC-buf",
+    id: "PCBuf",
     fill: util.colors.orange,
     ports: (
       west: (
@@ -22,18 +22,18 @@
       )
     )
   )
-  wire.stub("PC-buf-port-CLK", "north", name: "CLK")
-  wire.stub("PC-buf-port-EN", "south", name: "PCWrite")
+  wire.stub("PCBuf-port-CLK", "north", name: "CLK")
+  wire.stub("PCBuf-port-EN", "south", name: "PCWrite")
   
   element.multiplexer(
-    x: 3, y: (from: "PC-buf-port-PC", to: "in0"), w: 1, h: 2,
+    x: 3, y: (from: "PCBuf-port-PC", to: "in0"), w: 1, h: 2,
     id: "AdrSrc-MP",
     fill: util.colors.orange,
     entries: 2
   )
   wire.wire(
     "wPCBuf-InstDataMgr", (
-      "PC-buf-port-PC",
+      "PCBuf-port-PC",
       "AdrSrc-MP-port-in0"
     ),
     name: "PC",
@@ -105,7 +105,7 @@
     )
   )
   wire.stub("RegFile-port-CLK", "north", name: "CLK")
-  wire.stub("RegFile-port-WE3", "north", name: "Regwrite", vertical: true)
+  wire.stub("RegFile-port-WE3", "north", name: "Regwrite", name-offset: 0.6)
   wire.stub("RegFile-port-A2", "west")
   wire.stub("RegFile-port-RD2", "east")
 
@@ -283,7 +283,7 @@
   wire.wire(
     "wResMP-PCBuf", (
       "Res-MP-port-out",
-      "PC-buf-port-PCNext"
+      "PCBuf-port-PCNext"
     ),
     style: "dodge",
     dodge-y: -5,
@@ -294,7 +294,6 @@
     bus: true
   )
 
-  
-  wire.intersection("wResMP-RegFile.dodge-end")
-  wire.intersection("wResMP-AdrSrc.dodge-end")
+  wire.intersection("wResMP-RegFile.dodge-end", radius: .2)
+  wire.intersection("wResMP-AdrSrc.dodge-end", radius: .2)
 })

gallery/test2.typ

@@ -0,0 +1,312 @@
+#import "../src/lib.typ": circuit, element, util, wire
+
+#set page(width: auto, height: auto, margin: .5cm)
+
+#circuit({
+  element.block(
+    x: 0, y: 0, w: 1.5, h: 2.2,
+    id: "PCBuf",
+    fill: util.colors.orange,
+    ports: (
+      west: (
+        (id: "PCNext"),
+      ),
+      north: (
+        (id: "CLK", clock: true),
+      ),
+      east: (
+        (id: "PC"),
+      ),
+      south: (
+        (id: "EN", name: "EN"),
+      )
+    )
+  )
+  wire.stub("PCBuf-port-CLK", "north", name: "CLK")
+  wire.stub("PCBuf-port-EN", "south", name: "PCWrite")
+  
+  element.multiplexer(
+    x: (rel: 1.5, to: "PCBuf.east"),
+    y: (from: "PCBuf-port-PC", to: "in0"),
+    w: 1, h: 2,
+    id: "AdrSrc-MP",
+    fill: util.colors.orange,
+    entries: 2
+  )
+  wire.wire(
+    "wPCBuf-InstDataMgr", (
+      "PCBuf-port-PC",
+      "AdrSrc-MP-port-in0"
+    ),
+    name: "PC",
+    bus: true
+  )
+  wire.stub("AdrSrc-MP.north", "north", name: "AdrSrc")
+  
+  element.block(
+    x: (rel: 2, to: "AdrSrc-MP.east"),
+    y: (from: "AdrSrc-MP-port-out", to: "A"),
+    w: 3, h: 4,
+    id: "InstDataMgr",
+    fill: util.colors.yellow,
+    ports: (
+      west: (
+        (id: "A", name: "A"),
+        (id: "WD", name: "WD")
+      ),
+      north: (
+        (id: "CLK", clock: true),
+        (id: "WE", name: "WE", vertical: true),
+        (id: "IRWrite", name: "IRWrite", vertical: true)
+      ),
+      east: (
+        (id: "Instr", name: "Instr."),
+        (id: "RD", name: "RD")
+      )
+    ),
+    ports-margins: (
+      west: (30%, 0%),
+      east: (40%, 0%)
+    )
+  )
+  wire.wire(
+    "wAdrSrcMP-InstDataMgr", (
+      "AdrSrc-MP-port-out",
+      "InstDataMgr-port-A"
+    ),
+    name: "Adr",
+    name-pos: "end",
+    bus: true
+  )
+  
+  wire.stub("InstDataMgr-port-CLK", "north", name: "CLK")
+  wire.stub("InstDataMgr-port-WE", "north")
+  wire.stub("InstDataMgr-port-IRWrite", "north")
+  wire.stub("InstDataMgr-port-WD", "west")
+
+  element.block(
+    x: (rel: 6, to: "InstDataMgr.east"),
+    y: (from: "InstDataMgr-port-RD", to: "WD3"), w: 3, h: 4,
+    id: "RegFile",
+    fill: util.colors.pink,
+    ports: (
+      west: (
+        (id: "A1", name: "A1"),
+        (id: "A2", name: "A2"),
+        (id: "A3", name: "A3"),
+        (id: "WD3", name: "WD3"),
+      ),
+      north: (
+        (id: "CLK", clock: true),
+        (id: "WE3", name: "WE3", vertical: true)
+      ),
+      east: (
+        (id: "RD1", name: "RD1"),
+        (id: "RD2", name: "RD2"),
+      )
+    ),
+    ports-margins: (
+      east: (20%, 20%)
+    )
+  )
+  wire.stub("RegFile-port-CLK", "north", name: "CLK")
+  wire.stub("RegFile-port-WE3", "north", name: "Regwrite", name-offset: 0.6)
+  wire.stub("RegFile-port-A2", "west")
+  wire.stub("RegFile-port-RD2", "east")
+
+  element.extender(
+    x: (rel: 0, to: "RegFile.west"),
+    y: -3.5,
+    w: 3, h: 1,
+    id: "Extender",
+    fill: util.colors.green
+  )
+  wire.wire(
+    "wExtender-ImmSrc", (
+      "Extender.north",
+      (18, -2)
+    ),
+    style: "zigzag",
+    zigzag-ratio: 0%,
+    name: "ImmSrc",
+    name-pos: "end",
+    bus: true
+  )
+
+  let mid = ("InstDataMgr.east", 50%, "RegFile.west")
+  wire.wire(
+    "wInstDataMgr-Bus", (
+      "InstDataMgr-port-Instr",
+      (vertical: (), horizontal: mid)
+    ),
+    name: "Instr",
+    name-pos: "start",
+    bus: true
+  )
+  wire.wire(
+    "wBus", (
+      (v => (v.at(0), -3.5), mid),
+      (horizontal: (), vertical: (0, 3.5)),
+    ),
+    bus: true
+  )
+  wire.wire(
+    "wBus-RegFile-A1", (
+      "RegFile-port-A1",
+      (horizontal: mid, vertical: ()),
+    ),
+    name: "RS1",
+    name-pos: "end",
+    slice: (19, 15),
+    reverse: true,
+    bus: true
+  )
+  wire.wire(
+    "wBus-RegFile-A3", (
+      "RegFile-port-A3",
+      (horizontal: mid, vertical: ()),
+    ),
+    name: "RD",
+    name-pos: "end",
+    slice: (11, 7),
+    reverse: true,
+    bus: true
+  )
+  wire.wire(
+    "wBus-Extender", (
+      "Extender-port-in",
+      (horizontal: mid, vertical: ()),
+    ),
+    slice: (31, 7),
+    reverse: true,
+    bus: true
+  )
+
+  element.alu(
+    x: (rel: 4, to: "RegFile.east"),
+    y: (from: "RegFile-port-RD1", to: "in1"),
+    w: 1, h: 2,
+    id: "ALU",
+    fill: util.colors.purple
+  )
+  wire.wire(
+    "wRegFile-ALU", (
+      "RegFile-port-RD1",
+      "ALU-port-in1"
+    ),
+    name: ("A", "SrcA"),
+    bus: true
+  )
+
+  element.block(
+    x: (rel: 3, to: "ALU.east"),
+    y: (from: "ALU-port-out", to: "in"),
+    w: 1.5, h: 2,
+    id: "OutBuf",
+    fill: util.colors.orange,
+    ports: (
+      west: (
+        (id: "in"),
+      ),
+      north: (
+        (id: "CLK", clock: true),
+      ),
+      east: (
+        (id: "out"),
+      )
+    )
+  )
+  wire.stub("OutBuf-port-CLK", "north", name: "CLK")
+  wire.wire(
+    "wALU-OutBuf", (
+      "ALU-port-out",
+      "OutBuf-port-in"
+    ),
+    name: "ALUResult",
+    bus: true
+  )
+
+  element.multiplexer(
+    x: (rel: 2.5, to: "OutBuf.east"),
+    y: (from: "OutBuf-port-out", to: "in0"),
+    w: 1, h: 2.5,
+    id: "Res-MP",
+    fill: util.colors.orange,
+    entries: 3
+  )
+  wire.stub("Res-MP.north", "north", name: "ResultSrc")
+  wire.stub("Res-MP-port-in2", "west")
+  wire.wire(
+    "wOutBuf-ResMP", (
+      "OutBuf-port-out",
+      "Res-MP-port-in0"
+    ),
+    name: "ALUOut",
+    bus: true
+  )
+
+  wire.wire(
+    "wExt-ALU", (
+      "Extender-port-out",
+      "ALU-port-in2",
+    ),
+    name: ("ImmExt", "SrcB"),
+    bus: true,
+    style: "zigzag",
+    zigzag-ratio: 60%
+  )
+
+  wire.wire(
+    "wInstDataMgr-ResMP", (
+      "InstDataMgr-port-RD",
+      "Res-MP-port-in1"
+    ),
+    style: "dodge",
+    dodge-y: -4,
+    dodge-sides: ("east", "west"),
+    name: "Data",
+    name-pos: "start",
+    bus: true
+  )
+
+  wire.wire(
+    "wResMP-AdrSrc", (
+      "Res-MP-port-out",
+      "AdrSrc-MP-port-in1"
+    ),
+    style: "dodge",
+    dodge-y: -5,
+    dodge-sides: ("east", "west"),
+    dodge-margins: (0.5, 1),
+    bus: true
+  )
+
+  wire.wire(
+    "wResMP-RegFile", (
+      "Res-MP-port-out",
+      "RegFile-port-WD3"
+    ),
+    style: "dodge",
+    dodge-y: -5,
+    dodge-sides: ("east", "west"),
+    dodge-margins: (0.5, 1),
+    bus: true
+  )
+
+  wire.wire(
+    "wResMP-PCBuf", (
+      "Res-MP-port-out",
+      "PCBuf-port-PCNext"
+    ),
+    style: "dodge",
+    dodge-y: -5,
+    dodge-sides: ("east", "west"),
+    dodge-margins: (0.5, 1.5),
+    name: "PCNext",
+    name-pos: "end",
+    bus: true
+  )
+  
+  wire.intersection("wResMP-RegFile.dodge-end", radius: .2)
+  wire.intersection("wResMP-AdrSrc.dodge-end", radius: .2)
+})

gallery/test3.typ

@@ -0,0 +1,84 @@
+#import "/src/cetz.typ": draw
+#import "../src/lib.typ": circuit, element, util, wire
+
+#set page(width: auto, height: auto, margin: .5cm)
+
+#circuit({
+  element.block(
+    x: 0, y: 0, w: 2, h: 3, id: "block",
+    name: "Test",
+    ports: (
+      east: (
+        (id: "out0"),
+        (id: "out1"),
+        (id: "out2"),
+      )
+    )
+  )
+  element.gate-and(
+    x: 4, y: 0, w: 2, h: 2, id: "and1",
+    inverted: ("in1")
+  )
+  element.gate-or(
+    x: 7, y: 0, w: 2, h: 2, id: "or1",
+    inverted: ("in0", "out")
+  )
+
+  wire.wire(
+    "w1",
+    ("block-port-out0", "and1-port-in0"),
+    style: "dodge",
+    dodge-y: 3,
+    dodge-margins: (20%, 20%)
+  )
+  wire.wire(
+    "w2",
+    ("block-port-out1", "and1-port-in1"),
+    style: "zigzag"
+  )
+  wire.wire(
+    "w3",
+    ("and1-port-out", "or1-port-in0")
+  )
+
+  element.gate-and(
+    x: 11, y: 0, w: 2, h: 2, id: "and2", inputs: 3,
+    inverted: ("in0", "in2")
+  )
+  for i in range(3) {
+    wire.stub("and2-port-in"+str(i), "west")
+  }
+
+  element.gate-xor(
+    x: 14, y: 0, w: 2, h: 2, id: "xor",
+    inverted: ("in1")
+  )
+  
+  element.gate-buf(
+    x: 0, y: -3, w: 2, h: 2, id: "buf"
+  )
+  element.gate-not(
+    x: 0, y: -6, w: 2, h: 2, id: "not"
+  )
+  
+  element.gate-and(
+    x: 3, y: -3, w: 2, h: 2, id: "and"
+  )
+  element.gate-nand(
+    x: 3, y: -6, w: 2, h: 2, id: "nand"
+  )
+  
+  element.gate-or(
+    x: 6, y: -3, w: 2, h: 2, id: "or"
+  )
+  element.gate-nor(
+    x: 6, y: -6, w: 2, h: 2, id: "nor"
+  )
+  
+  element.gate-xor(
+    x: 9, y: -3, w: 2, h: 2, id: "xor"
+  )
+  element.gate-xnor(
+    x: 9, y: -6, w: 2, h: 2, id: "xnor"
+  )
+})

gallery/test4.typ

@@ -0,0 +1,221 @@
+#import "/src/cetz.typ": draw
+#import "../src/lib.typ": *
+
+#set page(width: auto, height: auto, margin: .5cm)
+
+#circuit({
+  element.group(id: "toplvl", name: "Toplevel", {
+    element.group(
+      id: "proc",
+      name: "Processor",
+      padding: 1.5em,
+      stroke: (dash: "dashed"),
+      {
+      element.block(
+        x: 0, y: 0, w: 8, h: 4,
+        id: "dp",
+        fill: util.colors.pink,
+        name: "Datapath",
+        ports: (
+          north: (
+            (id: "clk", clock: true, small: true),
+            (id: "Zero"),
+            (id: "Regsrc"),
+            (id: "PCSrc"),
+            (id: "ResultSrc"),
+            (id: "ALUControl"),
+            (id: "ImmSrc"),
+            (id: "RegWrite"),
+            (id: "dummy")
+          ),
+          east: (
+            (id: "PC", name: "PC"),
+            (id: "Instr", name: "Instr"),
+            (id: "ALUResult", name: "ALUResult"),
+            (id: "dummy"),
+            (id: "WriteData", name: "WriteData"),
+            (id: "ReadData", name: "ReadData"),
+          ),
+          west: (
+            (id: "rst"),
+          )
+        ),
+        ports-margins: (
+          north: (0%, 0%),
+          west: (0%, 70%)
+        )
+      )
+      
+      element.block(
+        x: 0, y: 7, w: 8, h: 3,
+        id: "ctrl",
+        fill: util.colors.orange,
+        name: "Controller",
+        ports: (
+          east: (
+            (id: "Instr", name: "Instr"),
+          ),
+          south: (
+            (id: "dummy"),
+            (id: "Zero"),
+            (id: "Regsrc"),
+            (id: "PCSrc"),
+            (id: "ResultSrc"),
+            (id: "ALUControl"),
+            (id: "ImmSrc"),
+            (id: "RegWrite"),
+            (id: "MemWrite")
+          )
+        ),
+        ports-margins: (
+          south: (0%, 0%)
+        )
+      )
+      wire.wire(
+        "w-Zero",
+        ("dp-port-Zero", "ctrl-port-Zero"),
+        name: "Zero",
+        name-pos: "start",
+        directed: true
+      )
+      for p in ("Regsrc", "PCSrc", "ResultSrc", "ALUControl", "ImmSrc", "RegWrite") {
+        wire.wire(
+          "w-" + p,
+          ("ctrl-port-"+p, "dp-port-"+p),
+          name: p,
+          name-pos: "start",
+          directed: true
+        )
+      }
+
+      draw.content(
+        (rel: (0, 1em), to: "ctrl.north"),
+        [*RISCV single*],
+        anchor: "south"
+      )
+    })
+    
+    element.block(
+      x: (rel: 3.5, to: "dp.east"),
+      y: (from: "dp-port-ReadData", to: "RD"),
+      w: 3, h: 4,
+      id: "dmem",
+      fill: util.colors.green,
+      name: "Data\n Memory",
+      ports: (
+        north: (
+          (id: "clk", clock: true, small: true),
+          (id: "WE", name: "WE")
+        ),
+        west: (
+          (id: "dummy"),
+          (id: "dummy"),
+          (id: "A", name: "A"),
+          (id: "dummy"),
+          (id: "WD", name: "WD"),
+          (id: "RD", name: "RD"),
+        )
+      ),
+      ports-margins: (
+        north: (0%, 10%)
+      )
+    )
+    wire.wire(
+      "w-DataAddr",
+      ("dp-port-ALUResult", "dmem-port-A"),
+      name: "DataAddr",
+      name-pos: "end",
+      directed: true
+    )
+    wire.wire(
+      "w-WriteData",
+      ("dp-port-WriteData", "dmem-port-WD"),
+      name: "WriteData",
+      name-pos: "end",
+      directed: true
+    )
+    wire.wire(
+      "w-ReadData",
+      ("dmem-port-RD", "dp-port-ReadData"),
+      name: "ReadData",
+      name-pos: "end",
+      reverse: true,
+      directed: true
+    )
+    wire.wire(
+      "w-MemWrite",
+      ("ctrl-port-MemWrite", "dmem-port-WE"),
+      style: "zigzag",
+      name: "MemWrite",
+      name-pos: "start",
+      zigzag-dir: "horizontal",
+      zigzag-ratio: 80%,
+      directed: true
+    )
+    wire.stub(
+      "dmem-port-clk", "north",
+      name: "clk", length: 3pt
+    )
+
+    element.block(
+      x: (rel: 3.5, to: "dp.east"),
+      y: (from: "ctrl-port-Instr", to: "dummy"),
+      w: 3, h: 4,
+      id: "imem",
+      fill: util.colors.green,
+      name: "Instruction\n Memory",
+      ports: (
+        west: (
+          (id: "A", name: "A"),
+          (id: "dummy"),
+          (id: "dummy2"),
+          (id: "RD", name: "RD"),
+        )
+      )
+    )
+    wire.wire(
+      "w-PC",
+      ("dp-port-PC", "imem-port-A"),
+      style: "zigzag",
+      directed: true
+    )
+    wire.wire(
+      "w-Instr1",
+      ("imem-port-RD", "dp-port-Instr"),
+      style: "zigzag",
+      zigzag-ratio: 30%,
+      directed: true
+    )
+    wire.wire(
+      "w-Instr2",
+      ("imem-port-RD", "ctrl-port-Instr"),
+      style: "zigzag",
+      zigzag-ratio: 30%,
+      directed: true
+    )
+    wire.intersection("w-Instr1.zig", radius: 2pt)
+    draw.content("w-Instr1.zig", "Instr", anchor: "south", padding: 4pt)
+    draw.content("w-PC.zig", "PC", anchor: "south-east", padding: 2pt)
+
+    draw.content("dmem.south-west", [*External Memories*], anchor: "north", padding: 10pt)
+  })
+
+  draw.line(name: "w-dp-clk",
+    "dp-port-clk",
+    (rel: (0, .5), to: ()),
+    (
+      rel: (-.5, 0),
+      to: (horizontal: "toplvl.west", vertical: ())
+    )
+  )
+  draw.content("w-dp-clk.end", "clk", anchor: "east", padding: 3pt)
+  
+  draw.line(name: "w-dp-rst",
+    "dp-port-rst",
+    (
+      rel: (-.5, 0),
+      to: (horizontal: "toplvl.west", vertical: ())
+    )
+  )
+  draw.content("w-dp-rst.end", "rst", anchor: "east", padding: 3pt)
+})

gallery/test5.typ

@@ -0,0 +1,435 @@
+#import "/src/cetz.typ": draw
+#import "../src/lib.typ": *
+
+#set page(width: auto, height: auto, margin: .5cm)
+
+#circuit({
+  element.multiplexer(
+    x: 0, y: 0, w: .5, h: 1.5, id: "PCMux",
+    entries: 2,
+    fill: util.colors.blue,
+    h-ratio: 80%
+  )
+  element.block(
+    x: (rel: 2, to: "PCMux.east"),
+    y: (from: "PCMux-port-out", to: "in"),
+    w: 1, h: 1.5, id: "PCBuf",
+    ports: (
+      north: ((id: "clk", clock: true),),
+      west: ((id: "in"),),
+      east: ((id: "out"),)
+    ),
+    fill: util.colors.green
+  )
+
+  element.block(
+    x: (rel: 2, to: "PCBuf.east"),
+    y: (from: "PCBuf-port-out", to: "A"),
+    w: 3, h: 4, id: "IMem",
+    ports: (
+      west: (
+        (id: "A", name: "A"),
+      ),
+      east: (
+        (id: "RD", name: "RD"),
+      )
+    ),
+    ports-margins: (
+      west: (0%, 50%),
+      east: (0%, 50%)
+    ),
+    fill: util.colors.green,
+    name: "Instruction\nMemory"
+  )
+  element.block(
+    x: (rel: 3, to: "IMem.east"),
+    y: (from: "IMem-port-RD", to: "A1"),
+    w: 4.5, h: 4, id: "RegFile",
+    ports: (
+      north: (
+        (id: "clk", clock: true, small: true),
+        (id: "WE3", name: "WE3"),
+        (id: "dummy1")
+      ),
+      west: (
+        (id: "dummy2"),
+        (id: "A1", name: "A1"),
+        (id: "dummy3"),
+        (id: "A2", name: "A2"),
+        (id: "A3", name: "A3"),
+        (id: "dummy4"),
+        (id: "WD3", name: "WD3"),
+      ),
+      east: (
+        (id: "RD1", name: "RD1"),
+        (id: "RD2", name: "RD2"),
+      )
+    ),
+    ports-margins: (
+      north: (-20%, -20%),
+      east: (0%, 10%)
+    ),
+    fill: util.colors.green,
+    name: "Register\nFile"
+  )
+
+  element.alu(
+    x: (rel: -.7, to: "IMem.center"),
+    y: -7,
+    w: 1.4, h: 2.8, id: "PCAdd",
+    name: text("+", size: 1.5em),
+    name-anchor: "name",
+    fill: util.colors.pink
+  )
+  element.extender(
+    x: (rel: 0, to: "RegFile.west"),
+    y: (from: "PCAdd-port-out", to: "in"),
+    w: 4, h: 1.5, id: "Ext",
+    h-ratio: 50%,
+    name: "Extend",
+    name-anchor: "south",
+    align-out: false,
+    fill: util.colors.green
+  )
+
+  element.multiplexer(
+    x: (rel: 3, to: "RegFile.east"),
+    y: (from: "RegFile-port-RD2", to: "in0"),
+    w: .5, h: 1.5, id: "SrcBMux",
+    fill: util.colors.blue,
+    h-ratio: 80%
+  )
+
+  element.alu(
+    x: (rel: 2, to: "SrcBMux.east"),
+    y: (from: "SrcBMux-port-out", to: "in2"),
+    w: 1.4, h: 2.8, id: "ALU",
+    name: rotate("ALU", -90deg),
+    name-anchor: "name",
+    fill: util.colors.pink
+  )
+  element.alu(
+    x: (rel: 2, to: "SrcBMux.east"),
+    y: (from: "Ext-port-out", to: "in2"),
+    w: 1.4, h: 2.8, id: "JumpAdd",
+    name: text("+", size: 1.5em),
+    name-anchor: "name",
+    fill: util.colors.pink
+  )
+
+  element.block(
+    x: (rel: 4, to: "ALU.east"),
+    y: (from: "ALU-port-out", to: "A"),
+    w: 3, h: 4, id: "DMem",
+    name: "Data\nMemory",
+    ports: (
+      north: (
+        (id: "clk", clock: true, small: true),
+        (id: "dummy1"),
+        (id: "WE", name: "WE")
+      ),
+      west: (
+        (id: "A", name: "A"),
+        (id: "WD", name: "WD")
+      ),
+      east: (
+        (id: "RD", name: "RD"),
+        (id: "dummy2")
+      )
+    ),
+    ports-margins: (
+      north: (-10%, -10%),
+      west: (-20%, -30%),
+      east: (-10%, -20%)
+    ),
+    fill: util.colors.green
+  )
+
+  element.multiplexer(
+    x: (rel: 3, to: "DMem.east"),
+    y: (from: "DMem-port-RD", to: "in1"),
+    w: .5, h: 1.5, id: "ResMux",
+    entries: 2,
+    fill: util.colors.blue,
+    h-ratio: 80%
+  )
+
+  element.block(
+    x: (rel: 0, to: "RegFile.west"),
+    y: 3.5, w: 2.5, h: 5, id: "Ctrl",
+    name: "Control\nUnit",
+    name-anchor: "north",
+    ports: (
+      west: (
+        (id: "op", name: "op"),
+        (id: "funct3", name: "funct3"),
+        (id: "funct7", name: [funct7#sub("[5]")]),
+        (id: "zero", name: "Zero"),
+      ),
+      east: (
+        (id: "PCSrc"),
+        (id: "ResSrc"),
+        (id: "MemWrite"),
+        (id: "ALUCtrl"),
+        (id: "ALUSrc"),
+        (id: "ImmSrc"),
+        (id: "RegWrite"),
+      )
+    ),
+    ports-margins: (
+      west: (40%, 0%)
+    ),
+    fill: util.colors.orange
+  )
+
+  // Wires
+  wire.wire(
+    "wPCNext", ("PCMux-port-out", "PCBuf-port-in"),
+    name: "PCNext"
+  )
+  wire.stub("PCBuf-port-clk", "north", name: "clk", length: 0.25)
+  wire.wire(
+    "wPC1", ("PCBuf-port-out", "IMem-port-A"),
+    name: "PC"
+  )
+  wire.wire(
+    "wPC2", ("PCBuf-port-out", "JumpAdd-port-in1"),
+    style: "zigzag",
+    zigzag-ratio: 1
+  )
+  wire.wire(
+    "wPC3", ("PCBuf-port-out", "PCAdd-port-in1"),
+    style: "zigzag",
+    zigzag-ratio: 1
+  )
+  wire.intersection("wPC2.zig")
+  wire.intersection("wPC2.zag")
+  wire.stub("PCAdd-port-in2", "west", name: "4", length: 1.5)
+  wire.wire(
+    "wPC+4", ("PCAdd-port-out", "PCMux-port-in0"),
+    style: "dodge",
+    dodge-sides: ("east", "west"),
+    dodge-y: -7.5,
+    dodge-margins: (1.2, .5),
+    name: "PC+4",
+    name-pos: "start"
+  )
+  
+  let mid = ("IMem-port-RD", 50%, "RegFile-port-A1")
+  wire.wire(
+    "wInstr", ("IMem-port-RD", mid),
+    bus: true,
+    name: "Instr",
+    name-pos: "start"
+  )
+  draw.hide({
+    draw.line(name: "bus-top",
+      mid,
+      (horizontal: (), vertical: "Ctrl-port-op")
+    )
+    draw.line(name: "bus-bot",
+      mid,
+      (horizontal: (), vertical: "Ext-port-in")
+    )
+  })
+  wire.wire(
+    "wInstrBus", ("bus-top.end", "bus-bot.end"),
+    bus: true
+  )
+  wire.wire(
+    "wOp", ("Ctrl-port-op", (horizontal: mid, vertical: ())),
+    bus: true,
+    reverse: true,
+    slice: (6, 0)
+  )
+  wire.wire(
+    "wF3", ("Ctrl-port-funct3", (horizontal: mid, vertical: ())),
+    bus: true,
+    reverse: true,
+    slice: (14, 12)
+  )
+  wire.wire(
+    "wF7", ("Ctrl-port-funct7", (horizontal: mid, vertical: ())),
+    bus: true,
+    reverse: true,
+    slice: (30,)
+  )
+  wire.wire(
+    "wA1", ("RegFile-port-A1", (horizontal: mid, vertical: ())),
+    bus: true,
+    reverse: true,
+    slice: (19, 15)
+  )
+  wire.wire(
+    "wA2", ("RegFile-port-A2", (horizontal: mid, vertical: ())),
+    bus: true,
+    reverse: true,
+    slice: (24, 20)
+  )
+  wire.wire(
+    "wA3", ("RegFile-port-A3", (horizontal: mid, vertical: ())),
+    bus: true,
+    reverse: true,
+    slice: (11, 7)
+  )
+  wire.wire(
+    "wExt", ("Ext-port-in", (horizontal: mid, vertical: ())),
+    bus: true,
+    reverse: true,
+    slice: (31, 7)
+  )
+  wire.intersection("wF3.end")
+  wire.intersection("wF7.end")
+  wire.intersection("wA1.end")
+  wire.intersection("wA2.end")
+  wire.intersection("wA3.end")
+
+  wire.stub("RegFile-port-clk", "north", name: "clk", length: 0.25)
+  wire.wire("wRD2", ("RegFile-port-RD2", "SrcBMux-port-in0"))
+  wire.wire(
+    "wWD", ("RegFile-port-RD2", "DMem-port-WD"),
+    style: "zigzag",
+    zigzag-ratio: 1.5,
+    name: "WriteData",
+    name-pos: "end"
+  )
+  wire.intersection("wWD.zig")
+
+  wire.wire(
+    "wImmALU", ("Ext-port-out", "SrcBMux-port-in1"),
+    style: "zigzag",
+    zigzag-ratio: 2.5,
+    name: "ImmExt",
+    name-pos: "start"
+  )
+  wire.wire(
+    "wImmJump", ("Ext-port-out", "JumpAdd-port-in2")
+  )
+  wire.intersection("wImmALU.zig")
+  wire.wire(
+    "wJumpPC", ("JumpAdd-port-out", "PCMux-port-in1"),
+    style: "dodge",
+    dodge-sides: ("east", "west"),
+    dodge-y: -8,
+    dodge-margins: (1, 1),
+    name: "PCTarget",
+    name-pos: "start"
+  )
+
+  wire.wire(
+    "wSrcA", ("RegFile-port-RD1", "ALU-port-in1"),
+    name: "SrcA",
+    name-pos: "end"
+  )
+  wire.wire(
+    "wSrcB", ("SrcBMux-port-out", "ALU-port-in2"),
+    name: "SrcB",
+    name-pos: "end"
+  )
+
+  wire.wire(
+    "wZero", (
+      ("ALU.north-east", 50%, "ALU-port-out"),
+      "Ctrl-port-zero"
+    ),
+    style: "dodge",
+    dodge-sides: ("east", "west"),
+    dodge-y: 3,
+    dodge-margins: (1.5, 1),
+    name: "Zero",
+    name-pos: "start"
+  )
+  wire.wire(
+    "wALURes1", ("ALU-port-out", "DMem-port-A"),
+    name: "ALUResult",
+    name-pos: "start"
+  )
+  wire.wire(
+    "wALURes2", ("ALU-port-out", "ResMux-port-in0"),
+    style: "dodge",
+    dodge-sides: ("east", "west"),
+    dodge-y: 2,
+    dodge-margins: (3, 2)
+  )
+  wire.intersection("wALURes2.start2")
+
+  wire.stub("DMem-port-clk", "north", name: "clk", length: 0.25)
+  wire.wire(
+    "wRD", ("DMem-port-RD", "ResMux-port-in1"),
+    name: "ReadData",
+    name-pos: "start"
+  )
+
+  wire.wire(
+    "wRes", ("ResMux-port-out", "RegFile-port-WD3"),
+    style: "dodge",
+    dodge-sides: ("east", "west"),
+    dodge-y: -7.5,
+    dodge-margins: (1, 2)
+  )
+  draw.content(
+    "wRes.dodge-start",
+    "Result",
+    anchor: "south-east",
+    padding: 5pt
+  )
+
+  // Other wires
+  draw.group({
+    draw.stroke(util.colors.blue)
+    draw.line(name: "wPCSrc", 
+      "Ctrl-port-PCSrc",
+      (horizontal: "RegFile.east", vertical: ()),
+      (horizontal: (), vertical: (rel: (0, 0.5), to: "Ctrl.north")),
+      (horizontal: "PCMux.north", vertical: ()),
+      "PCMux.north"
+    )
+    draw.line(name: "wResSrc",
+      "Ctrl-port-ResSrc",
+      (horizontal: "ResMux.north", vertical: ()),
+      "ResMux.north"
+    )
+    draw.line(name: "wMemWrite",
+      "Ctrl-port-MemWrite",
+      (horizontal: "DMem-port-WE", vertical: ()),
+      "DMem-port-WE"
+    )
+    draw.line(name: "wALUCtrl",
+      "Ctrl-port-ALUCtrl",
+      (horizontal: "ALU.north", vertical: ()),
+      "ALU.north"
+    )
+    draw.line(name: "wALUSrc",
+      "Ctrl-port-ALUSrc",
+      (horizontal: "SrcBMux.north", vertical: ()),
+      "SrcBMux.north"
+    )
+    draw.line(name: "wImmSrc",
+      "Ctrl-port-ImmSrc",
+      (rel: (1, 0), to: (horizontal: "RegFile.east", vertical: ())),
+      (horizontal: (), vertical: (rel: (0, -.5), to: "RegFile.south")),
+      (horizontal: "Ext.north", vertical: ()),
+      "Ext.north"
+    )
+    draw.line(name: "wRegWrite",
+      "Ctrl-port-RegWrite",
+      (rel: (.5, 0), to: (horizontal: "RegFile.east", vertical: ())),
+      (horizontal: (), vertical: ("Ctrl.south", 50%, "RegFile.north")),
+      (horizontal: "RegFile-port-WE3", vertical: ()),
+      "RegFile-port-WE3"
+    )
+
+    let names = (
+      "PCSrc": "PCSrc",
+      "ResSrc": "ResultSrc",
+      "MemWrite": "MemWrite",
+      "ALUCtrl": [ALUControl#sub("[2:0]")],
+      "ALUSrc": "ALUSrc",
+      "ImmSrc": [ImmSrc#sub("[1:0]")],
+      "RegWrite": "RegWrite"
+    )
+    for (port, name) in names {
+      draw.content("Ctrl-port-"+port, name, anchor: "south-west", padding: 3pt)
+    }
+  })
+})

gallery/test6.typ

@@ -0,0 +1,194 @@
+#import "/src/cetz.typ": draw, vector
+#import "../src/lib.typ": *
+
+#set page(width: auto, height: auto, margin: .5cm)
+
+#circuit({
+  element.multiplexer(
+    x: 10, y: 0, w: 1, h: 6, id: "ResMux",
+    entries: ("000", "001", "010", "011", "101"),
+    h-ratio: 90%,
+    fill: util.colors.blue
+  )
+  element.extender(
+    x: (rel: -3, to: "ResMux.west"),
+    y: (from: "ResMux-port-in4", to: "out"),
+    w: 2, h: 1, id: "Ext",
+    name: "Zero Ext",
+    name-anchor: "south",
+    fill: util.colors.green
+  )
+  gates.gate-or(
+    x: (rel: -2, to: "ResMux.west"),
+    y: (from: "ResMux-port-in3", to: "out"),
+    w: 1, h: 1, id: "Or"
+  )
+  gates.gate-and(
+    x: (rel: -2, to: "ResMux.west"),
+    y: (from: "ResMux-port-in2", to: "out"),
+    w: 1, h: 1, id: "And"
+  )
+  element.alu(
+    x: (rel: -2.5, to: "Ext.west"),
+    y: (from: "ResMux-port-in0", to: "out"),
+    w: 1.5, h: 3, id: "Add",
+    name: text("+", size: 1.5em),
+    name-anchor: "name",
+    fill: util.colors.pink
+  )
+  element.multiplexer(
+    x: (rel: -1.5, to: "Add.west"),
+    y: (from: "Add-port-in1", to: "out"),
+    w: 0.5, h: 1.5, id: "NotMux",
+    h-ratio: 80%,
+    fill: util.colors.blue
+  )
+  gates.gate-not(
+    x: (rel: -2, to: "NotMux.west"),
+    y: (from: "NotMux-port-in1", to: "out"),
+    w: 1, h: 1, id: "Not"
+  )
+  
+  draw.hide(
+    draw.line(name: "l1",
+      "Not-port-in0",
+      (rel: (-2, 0), to: ()),
+      (horizontal: (), vertical: "NotMux-port-in0")
+    )
+  )
+  let b = "l1.end"
+  draw.hide(
+    draw.line(name: "l2",
+      b,
+      (horizontal: (), vertical: "Add-port-in2")
+    )
+  )
+  let a = "l2.end"
+
+  wire.wire("wB0", (b, "NotMux-port-in0"), bus: true)
+  wire.wire(
+    "wB1", (b, "Not-port-in0"),
+    style: "zigzag",
+    zigzag-ratio: 1.5,
+    bus: true
+  )
+  wire.wire(
+    "wB2", (b, "And-port-in0"),
+    style: "zigzag",
+    zigzag-ratio: 1,
+    bus: true
+  )
+  wire.wire(
+    "wB3", (b, "Or-port-in0"),
+    style: "zigzag",
+    zigzag-ratio: 1,
+    bus: true
+  )
+  wire.intersection("wB1.zig")
+  wire.intersection("wB2.zig")
+  wire.intersection("wB2.zag")
+
+  wire.wire("wNot", ("Not-port-out", "NotMux-port-in1"), bus: true)
+  wire.wire("wAddA", ("NotMux-port-out", "Add-port-in1"), bus: true)
+
+  wire.wire("wA0", (a, "Add-port-in2"), bus: true)
+  wire.wire(
+    "wA1", (a, "And-port-in1"),
+    style: "zigzag",
+    zigzag-ratio: 0.5,
+    bus: true
+  )
+  wire.wire(
+    "wA2", (a, "Or-port-in1"),
+    style: "zigzag",
+    zigzag-ratio: 0.5,
+    bus: true
+  )
+  wire.intersection("wA1.zig")
+  wire.intersection("wA1.zag")
+
+  wire.wire("wMux0", ("Add-port-out", "ResMux-port-in0"), bus: true)
+  wire.wire(
+    "wMux1", ("Add-port-out", "ResMux-port-in1"),
+    style: "zigzag",
+    zigzag-ratio: 2,
+    bus: true
+  )
+  wire.wire("wMux2", ("And-port-out", "ResMux-port-in2"), bus: true)
+  wire.wire("wMux3", ("Or-port-out", "ResMux-port-in3"), bus: true)
+  wire.wire("wMux4", ("Ext-port-out", "ResMux-port-in4"), bus: true)
+
+  wire.wire(
+    "wAdd", ("Add-port-out", "Ext-port-in"),
+    style: "zigzag",
+    zigzag-ratio: 0.5,
+    bus: true
+  )
+
+  wire.intersection("wMux1.zig")
+  wire.intersection("wAdd.zig")
+
+  let c = (rel: (0, 2), to: "ResMux.north")
+  wire.wire("wResCtrl", (c, "ResMux.north"), bus: true)
+  wire.wire(
+    "wAddCtrl", (c, "Add.north"),
+    style: "zigzag",
+    zigzag-dir: "horizontal"
+  )
+
+  let d = (rel: (1, 0), to: "ResMux-port-out")
+  wire.wire("wRes", ("ResMux-port-out", d), bus: true)
+
+  draw.content(
+    "wAddCtrl.zag",
+    [ALUControl#sub("[1]")],
+    anchor: "south-west",
+    padding: 3pt
+  )
+  
+  wire.wire(
+    "wCout", ("Add.south", (horizontal: (), vertical: "Ext.north-east"))
+  )
+  draw.content(
+    "wCout.end",
+    [C#sub("out")],
+    angle: 90deg,
+    anchor: "east",
+    padding: 3pt
+  )
+  draw.content(
+    a,
+    [A],
+    angle: 90deg,
+    anchor: "south",
+    padding: 3pt
+  )
+  draw.content(
+    b,
+    [B],
+    angle: 90deg,
+    anchor: "south",
+    padding: 3pt
+  )
+  draw.content(
+    c,
+    [ALUControl#sub("[2:0]")],
+    angle: 90deg,
+    anchor: "west",
+    padding: 3pt
+  )
+  draw.content(
+    d,
+    [Result],
+    angle: 90deg,
+    anchor: "north",
+    padding: 3pt
+  )
+  draw.content(
+    ("wAdd.zig", 0.2, "wAdd.zag"),
+    text("[N-1]", size: 0.8em),
+    angle: 90deg,
+    anchor: "north-east",
+    padding: 3pt
+  )
+})

justfile

@@ -0,0 +1,11 @@
+# Local Variables:
+# mode: makefile
+# End:
+gallery_dir := "./gallery"
+set shell := ["bash", "-uc"]
+
+manual:
+  typst c manual.typ manual.pdf
+
+gallery:
+  for f in "{{gallery_dir}}"/*.typ; do typst c --root . "$f" "${f%typ}png"; done

manual.typ

@@ -0,0 +1,213 @@
+#import "@preview/tidy:0.4.1"
+#import "/src/cetz.typ": draw, canvas
+#import "src/lib.typ"
+#import "doc/examples.typ"
+#import "src/circuit.typ": circuit
+#import "src/element.typ"
+#import "src/gates.typ"
+#import "src/util.typ"
+#import "src/wire.typ"
+
+#set heading(numbering: (..num) => if num.pos().len() < 4 {
+  numbering("1.1", ..num)
+})
+#{
+  outline(indent: auto, depth: 3)
+}
+
+#show link: set text(blue)
+#show heading.where(level: 3): it => context {
+  let cnt = counter(heading)
+  let i = cnt.get().at(it.depth) - 1
+  let color = util.colors.values().at(i)
+  block(width: 100%)[
+    #grid(
+      columns: (auto, 1fr),
+      column-gutter: 1em,
+      align: horizon,
+      it,
+      {
+        place(horizon)[
+          #line(
+            start: (0%, 0%),
+            end: (100%, 0%),
+            stroke: color + 1pt
+          )
+        ]
+        place(horizon)[
+          #circle(radius: 3pt, stroke: none, fill: color)
+        ]
+        place(horizon+right)[
+          #circle(radius: 3pt, stroke: none, fill: color)
+        ]
+      }
+    )
+  ]
+}
+
+#set page(numbering: "1/1", header: align(right)[circuiteria #sym.dash.em v#lib.version])
+#set page(
+  header: context {
+    let txt = [circuiteria #sym.dash.em v#lib.version]
+    let cnt = counter(heading)
+    let cnt-val = cnt.get()
+    if cnt-val.len() < 2 {
+      align(left, txt)
+      return
+    }
+    let i = cnt-val.at(1) - 1
+    grid(
+      columns: (auto, 1fr),
+      column-gutter: 1em,
+      align: horizon,
+      txt,
+      place(horizon + left)[
+        #rect(width: 100%, height: .5em, radius: .25em, stroke: none, fill: util.colors.values().at(i))
+      ]
+    )
+  },
+  footer: context {
+    let cnt = counter(heading)
+    let cnt-val = cnt.get()
+    if cnt-val.len() < 2 { return }
+    let i = cnt-val.at(1) - 1
+    grid(
+      columns: (1fr, auto),
+      column-gutter: 1em,
+      align: horizon,
+      place(horizon + left)[
+        #rect(width: 100%, height: .5em, radius: .25em, stroke: none, fill: util.colors.values().at(i))
+      ],
+      counter(page).display("1/1", both: true)
+    )
+  }
+)
+
+#let doc-ref(target, full: false, var: false) = {
+  let (module, func) = target.split(".")
+  let label-name = module + "-" + func
+  let display-name = func
+  if full {
+    display-name = target
+  }
+  if not var {
+    label-name += "()"
+    display-name += "()"
+  }
+  link(label(label-name), raw(display-name))
+}
+
+= Introduction
+
+This package provides a way to make beautiful block circuit diagrams using the CeTZ package.
+
+= Usage
+
+Simply import Circuiteria and call the `circuit` function:
+#pad(left: 1em)[```typ
+#import "@preview/circuiteria:0.2.0"
+#circuiteria.circuit({
+  import circuiteria: *
+  ...
+})
+```]
+
+== Project installation
+If you have installed Circuiteria directly in your project, import #link("src/lib.typ") and call the `circuit` function:
+#pad(left: 1em)[```typ
+#import "src/lib.typ" as circuiteria
+#circuiteria.circuit({
+  import circuiteria: *
+  ...
+})
+```]
+
+= Reference
+
+#let circuit-docs = tidy.parse-module(
+  read("src/circuit.typ"),
+  name: "circuit",
+  old-syntax: true,
+  require-all-parameters: true
+)
+#tidy.show-module(circuit-docs)
+
+#pagebreak()
+
+#let util-docs = tidy.parse-module(
+  read("src/util.typ"),
+  name: "util",
+  old-syntax: true,
+  require-all-parameters: true,
+  scope: (
+    util: util,
+    canvas: canvas,
+    draw: draw
+  )
+)
+#tidy.show-module(util-docs)
+
+#pagebreak()
+
+#let wire-docs = tidy.parse-module(
+  read("src/wire.typ"),
+  name: "wire",
+  old-syntax: true,
+  require-all-parameters: true,
+  scope: (
+    wire: wire,
+    circuit: circuit,
+    draw: draw,
+    examples: examples,
+    doc-ref: doc-ref
+  )
+)
+#tidy.show-module(wire-docs)
+
+#pagebreak()
+
+#let element-docs = tidy.parse-module(
+  read("src/elements/element.typ") + "\n" +
+  read("src/elements/alu.typ") + "\n" +
+  read("src/elements/block.typ") + "\n" +
+  read("src/elements/extender.typ") + "\n" +
+  read("src/elements/multiplexer.typ") + "\n" +
+  read("src/elements/group.typ"),
+  name: "element",
+  old-syntax: true,
+  scope: (
+    element: element,
+    circuit: circuit,
+    draw: draw,
+    wire: wire,
+    tidy: tidy,
+    examples: examples,
+    doc-ref: doc-ref
+  )
+)
+
+#tidy.show-module(element-docs, sort-functions: false)
+
+#pagebreak()
+
+#let gates-docs = tidy.parse-module(
+  read("src/elements/logic/gate.typ") + "\n" +
+  read("src/elements/logic/and.typ") + "\n" +
+  read("src/elements/logic/buf.typ") + "\n" +
+  read("src/elements/logic/or.typ") + "\n" +
+  read("src/elements/logic/xor.typ"),
+  name: "gates",
+  old-syntax: true,
+  scope: (
+    element: element,
+    circuit: circuit,
+    gates: gates,
+    draw: draw,
+    wire: wire,
+    tidy: tidy,
+    examples: examples,
+    doc-ref: doc-ref
+  )
+)
+
+#tidy.show-module(gates-docs, sort-functions: false)

src/cetz.typ

@@ -0,0 +1 @@
+#import "@preview/cetz:0.3.4": *

src/circuit.typ

@@ -1,6 +1,60 @@
-#import "@preview/cetz:0.2.2": canvas
+#import "/src/cetz.typ": canvas
+#import "@preview/tidy:0.3.0"
 
+/// Draws a block circuit diagram
+///
+/// This function is also available at the package root
+///
+/// - body (none, array, element): A code block in which draw functions have been called
+/// - length (length, ratio): Optional base unit
+/// -> none
 #let circuit(body, length: 2em) = {
-  set text(font: "Source Sans 3")
-  canvas(length: length, body)
+  let next-id = 0
+  let elements = (:)
+  let ids = ()
+  for e in body {
+    if type(e) == dictionary and "id" in e {
+      ids.push(e.id)
+    }
+  }
+
+  for element in body {
+    let internal = type(element) == dictionary and "id" in element
+    let eid = if internal {element.id} else {auto}
+    if eid == auto {
+      while str(next-id) in ids {
+        next-id += 1
+      }
+      eid = str(next-id)
+      ids.push(eid)
+      if internal {
+        element.id = eid
+      }
+      next-id += 1
+    }
+    if eid in elements {
+      panic("An element with the id '" + eid + "' already exists. Please use a different id")
+    }
+    elements.insert(eid, element)
+  }
+
+  for element in elements.values() {
+    if type(element) == dictionary and "pre-process" in element {
+      elements = (element.pre-process)(elements, element)
+      assert(
+        type(elements) == dictionary,
+        message: "The `pre-process` method of element '" + element.id + "' did not return the elements dictionary"
+      )
+    }
+  }
+
+  canvas(length: length, {
+    for element in elements.values() {
+      if type(element) == dictionary and "draw" in element {
+        (element.draw)(element)
+      } else {
+        (element,)
+      }
+    }
+  })
 }

src/element.typ

@@ -4,4 +4,12 @@
 #import "elements/alu.typ": alu
 #import "elements/block.typ": block
 #import "elements/extender.typ": extender
-#import "elements/multiplexer.typ": multiplexer
+#import "elements/multiplexer.typ": multiplexer
+
+#import "elements/logic/gate.typ": gate
+#import "elements/logic/and.typ": gate-and, gate-nand
+#import "elements/logic/or.typ": gate-or, gate-nor
+#import "elements/logic/xor.typ": gate-xor, gate-xnor
+#import "elements/logic/buf.typ": gate-buf, gate-not
+
+#import "elements/group.typ": group

src/elements/alu.typ

@@ -1,56 +1,48 @@
-#import "@preview/cetz:0.2.2": draw
+#import "/src/cetz.typ": draw
 #import "element.typ"
 #import "ports.typ": add-port
+#import "../util.typ"
+
+#let draw-shape(elmt, bounds) = {
+  let tl = bounds.tl
+  let tr = bounds.tr
+  let bl = bounds.bl
+  let br = bounds.br
 
-#let draw-shape(id, tl, tr, br, bl, fill, stroke) = {
   let p0 = tl
-  let p1 = (tr, 10%, br)
-  let p2 = (tr, 90%, br)
+  let p1 = util.lerp(tr, 10%, br)
+  let p2 = util.lerp(tr, 90%, br)
   let p3 = bl
-  let p4 = (tl, 55%, bl)
-  let p5 = (tl, 50%, br)
-  let p6 = (tl, 45%, bl)
+  let p4 = util.lerp(tl, 55%, bl)
+  let p5 = util.lerp(tl, 50%, br)
+  let p6 = util.lerp(tl, 45%, bl)
+
+  let bounds2 = element.complete-bounds(elmt, (
+    tl: p0,
+    tr: p1,
+    br: p2,
+    bl: p3
+  ))
   
-  let f1 = draw.group(name: id, {
-    
-    draw.merge-path(
-      inset: 0.5em,
-      fill: fill,
-      stroke: stroke,
-      close: true,
-      draw.line(p0, p1, p2, p3, p4, p5, p6)
-    )
-    draw.anchor("north", (p0, 50%, p1))
-    draw.anchor("south", (p2, 50%, p3))
-    draw.anchor("west", (p0, 50%, p3))
-    draw.anchor("east", (p1, 50%, p2))
-  })
-
-  let f2 = add-port(id, "west", (id: "in1"), (p0, 50%, p6))
-  let f3 = add-port(id, "west", (id: "in2"), (p3, 50%, p4))
-  let f4 = add-port(id, "east", (id: "out"), (p1, 50%, p2))
-
   let f = {
-    f1; f2; f3; f4
+    draw.line(
+      p0, p1, p2, p3, p4, p5, p6,
+      fill: elmt.fill,
+      stroke: elmt.stroke,
+      close: true
+    )
+    draw.anchor("name", (p5, 50%, (p1, 50%, p2)))
   }
 
-  return (f, tl, tr, br, bl)
+  return (f, bounds2)
 }
 
+/// Draws an ALU with two inputs
+///
+/// #examples.alu
+/// For parameters description, see #doc-ref("element.elmt")
 #let alu(
-  x: none,
-  y: none,
-  w: none,
-  h: none,
-  name: none,
-  name-anchor: "center",
-  fill: none,
-  stroke: black + 1pt,
-  id: "",
-  debug: (
-    grid: false,
-    ports: false
-  )
+  ..args
 ) = {
   let ports = (
     west: (
@@ -62,23 +54,18 @@
     )
   )
   
-  element.elmt(
+  return element.elmt(
     draw-shape: draw-shape,
-    x: x,
-    y: y,
-    w: w,
-    h: h,
-    name: name,
-    name-anchor: name-anchor,
     ports: ports,
-    fill: fill,
-    stroke: stroke,
-    id: id,
-    auto-ports: false,
-    ports-y: (
-      in1: (h) => {h * 0.225},
-      in2: (h) => {h * 0.775},
+    ports-pos: (
+      west: (
+        in1: l => l * 0.225,
+        in2: l => l * 0.775,
+      ),
+      east: (
+        out: l => l * 0.5
+      )
     ),
-    debug: debug
+    ..args
   )
 }

src/elements/block.typ

@@ -1,46 +1,25 @@
-#import "@preview/cetz:0.2.2": draw
+#import "/src/cetz.typ": draw
 #import "element.typ"
 
-#let draw-shape(id, tl, tr, br, bl, fill, stroke) = {
+#let draw-shape(elmt, bounds) = {
   let f = draw.rect(
     radius: 0.5em,
     inset: 0.5em,
-    fill: fill,
-    stroke: stroke,
-    name: id,
-    bl, tr
+    fill: elmt.fill,
+    stroke: elmt.stroke,
+    bounds.bl, bounds.tr
   )
-  return (f, tl, tr, br, bl)
+  return (f, bounds)
 }
 
+/// Draws a block element
+///
+/// #examples.block
+/// For parameters description, see #doc-ref("element.elmt")
 #let block(
-  x: none,
-  y: none,
-  w: none,
-  h: none,
-  name: none,
-  name-anchor: "center",
-  ports: (),
-  ports-margins: (),
-  fill: none,
-  stroke: black + 1pt,
-  id: "",
-  debug: (
-    grid: false,
-    ports: false
-  )
+  ..args
 ) = element.elmt(
+  cls: "block",
   draw-shape: draw-shape,
-  x: x,
-  y: y,
-  w: w,
-  h: h,
-  name: name,
-  name-anchor: name-anchor,
-  ports: ports,
-  ports-margins: ports-margins,
-  fill: fill,
-  stroke: stroke,
-  id: id,
-  debug: debug
+  ..args
 )

src/elements/element.typ

@@ -1,5 +1,6 @@
-#import "@preview/cetz:0.2.2": draw, coordinate
-#import "ports.typ": add-ports, add-port
+#import "/src/cetz.typ": draw, coordinate, matrix, vector
+#import "ports.typ": add-ports, add-port, get-port-pos, get-port-idx
+#import "../util.typ"
 
 #let find-port(ports, id) = {
   for (side, side-ports) in ports {
@@ -9,91 +10,315 @@
       }
     }
   }
-  panic("Could not find port with id " + str(id))
+  panic("Could not find port with id '" + str(id) + "' in ports " + repr(ports))
 }
 
-#let default-draw-shape(id, tl, tr, br, bl, fill, stroke) = {
-  return ({}, tl, tr, br, bl)
+#let local-to-global(origin, u, v, points) = {
+  return points-real = points.map(p => {
+    let (pu, pv) = p
+    return vector.add(
+      origin,
+      vector.add(
+        vector.scale(u, pu),
+        vector.scale(v, pv)
+      )
+    )
+  })
 }
 
-#let elmt(
-  draw-shape: default-draw-shape,
-  x: none,
-  y: none,
-  w: none,
-  h: none,
-  name: none,
-  name-anchor: "center",
-  ports: (),
-  ports-margins: (),
-  fill: none,
-  stroke: black + 1pt,
-  id: "",
-  auto-ports: true,
-  ports-y: (),
-  debug: (
-    grid: false,
-    ports: false
+#let default-draw-shape(elmt, bounds) = {
+  return ({}, bounds)
+}
+
+#let default-pre-process(elements, element) = {
+  return elements
+}
+
+#let resolve-offset(ctx, offset, from, axis) = {
+  let (ctx, pos) = coordinate.resolve(
+    ctx,
+    (rel: offset, to: from)
   )
-) = draw.get-ctx(ctx => {
-  let width = w
-  let height = h
+  return pos.at(axis)
+}
 
-  let y = y
-  if x == none { panic("Parameter x must be set") }
-  if y == none { panic("Parameter y must be set") }
-  if w == none { panic("Parameter w must be set") }
-  if h == none { panic("Parameter h must be set") }
-
-  if (type(y) == dictionary) {
-    let from = y.from
-    let to = y.to
-    let (to-side, i) = find-port(ports, to)
-    let margins = (0%, 0%)
-    if to-side in ports-margins {
-      margins = ports-margins.at(to-side)
-    }
-    let used-pct = 100% - margins.at(0) - margins.at(1)
-    let used-height = height * used-pct / 100%
-    let top-margin = height * margins.at(0) / 100%
-    
-    let dy = used-height * (i + 1) / (ports.at(to-side).len() + 1)
-
-    if not auto-ports {
-      top-margin = 0
-      dy = ports-y.at(to)(height)
-    }
-    
-    let (ctx, from-pos) = coordinate.resolve(ctx, from)
-    y = from-pos.at(1) + dy - height + top-margin
+#let resolve-align(ctx, elmt, bounds, align, with, axis) = {
+  let (align-side, i) = find-port(elmt.ports, align)
+  let margins = (0%, 0%)
+  if align-side in elmt.ports-margins {
+    margins = elmt.ports-margins.at(align-side)
   }
 
-  let tl = (x, y + height)
-  let tr = (x + width, y + height)
-  let br = (x + width, y)
-  let bl = (x, y)
+  let parallel-sides = (
+    ("north", "south"),
+    ("west", "east")
+  ).at(axis)
+
+  let ortho-sides = (
+    ("west", "east"),
+    ("north", "south")
+  ).at(axis)
+
+  let dl
+  let start-margin
+  let len = elmt.size.at(axis)
+  if align-side in parallel-sides {
+    let used-pct = 100% - margins.at(0) - margins.at(1)
+    let used-len = len * used-pct / 100%
+    start-margin = len * margins.at(0) / 100%
+    
+    //dl = used-len * (i + 1) / (elmt.ports.at(align-side).len() + 1)
+    dl = get-port-pos(elmt, bounds, align-side, align, get-port-idx(elmt, align, side: align-side))
+    /*if not elmt.auto-ports {
+      start-margin = 0
+      dl = elmt.ports-pos.at(align)(len)
+    }*/
+  } else if align-side == ortho-sides.first() {
+    dl = 0
+    start-margin = 0
+  } else {
+    dl = len
+    start-margin = 0
+  }
+
+  if axis == 1 {
+    dl = len - dl
+  }
+  
+  let (ctx, with-pos) = coordinate.resolve(ctx, with)
+  return with-pos.at(axis) - dl + start-margin
+}
+
+#let resolve-coordinate(ctx, elmt, bounds, coord, axis) = {
+  if type(coord) == dictionary {
+    let offset = coord.at("offset", default: none)
+    let from = coord.at("from", default: none)
+    let align = coord.at("align", default: none)
+    let with = coord.at("with", default: none)
+
+    if none not in (offset, from) {
+      if type(offset) != array {
+        let a = (0, 0)
+        a.at(axis) = offset
+        offset = a
+      }
+      return resolve-offset(ctx, offset, from, axis)
+      
+    } else if none not in (align, with) {
+      return resolve-align(ctx, elmt, bounds, align, with, axis)
+    } else {
+      panic("Dictionnary must either provide both 'offset' and 'from', or 'align' and 'with'")
+    }
+  }
+  if type(coord) not in (int, float, length) {
+    panic("Invalid " + "xy".at(axis) + " coordinate: " + repr(coord))
+  }
+  return coord
+}
+
+#let complete-bounds(elmt, bounds) = {
+  let b = bounds
+  bounds += (
+    center: (
+      (b.br.at(0) + b.tl.at(0))/2,
+      (b.br.at(1) + b.tl.at(1))/2
+    ),
+    b: (
+      (b.br.at(0) + b.bl.at(0))/2,
+      (b.br.at(1) + b.bl.at(1))/2
+    ),
+    t: (
+      (b.tr.at(0) + b.tl.at(0))/2,
+      (b.tr.at(1) + b.tl.at(1))/2
+    ),
+    l: (
+      (b.bl.at(0) + b.tl.at(0))/2,
+      (b.bl.at(1) + b.tl.at(1))/2
+    ),
+    r: (
+      (b.br.at(0) + b.tr.at(0))/2,
+      (b.br.at(1) + b.tr.at(1))/2
+    ),
+    sides: (
+      north: (bounds.tl, bounds.tr),
+      south: (bounds.bl, bounds.br),
+      west: (bounds.tl, bounds.bl),
+      east: (bounds.tr, bounds.br),
+    ),
+    lengths: (
+      north: (bounds.tr.at(0) - bounds.tl.at(0)),
+      south: (bounds.br.at(0) - bounds.bl.at(0)),
+      west: (bounds.tl.at(1) - bounds.bl.at(1)),
+      east: (bounds.tr.at(1) - bounds.br.at(1)),
+    ),
+    ports: (:)
+  )
+  for (side, props) in bounds.sides.pairs() {
+    let props2 = props
+    if side in elmt.ports-margins {
+      let (pt0, pt1) = props
+      let margins = elmt.ports-margins.at(side)
+      let a = util.lerp(pt0, margins.at(0), pt1)
+      let b = util.lerp(pt0, 100% - margins.at(1), pt1)
+      props2 = (a, b)
+    }
+    bounds.ports.insert(side, props2)
+  }
+  return bounds
+}
+
+#let make-bounds(elmt, x, y, w, h) = {
+  let w2 = w / 2
+  let h2 = h / 2
+  
+  let bounds = (
+    bl: (x, y),
+    tl: (x, y + h),
+    tr: (x + w, y + h),
+    br: (x + w, y),
+  )
+  return complete-bounds(elmt, bounds)
+}
+
+#let render(draw-shape, elmt) = draw.group(name: elmt.id, ctx => {
+  let width = elmt.size.first()
+  let height = elmt.size.last()
+
+  let x = elmt.pos.first()
+  let y = elmt.pos.last()
+
+  let bounds = make-bounds(elmt, 0, 0, width, height)
+  x = resolve-coordinate(ctx, elmt, bounds, x, 0)
+  y = resolve-coordinate(ctx, elmt, bounds, y, 1)
+  bounds = make-bounds(elmt, x, y, width, height)
 
   // Workaround because CeTZ needs to have all draw functions in the body
   let func = {}
-  (func, tl, tr, br, bl) = draw-shape(id, tl, tr, br, bl, fill, stroke)
+  let res = draw-shape(elmt, bounds)
+  assert(
+    type(res) == array and res.len() == 2,
+    message: "The drawing function of element '" + elmt.id + "' did not return a function and new bounds"
+  )
+  (func, bounds) = res
+  if type(func) == function {
+    func = (func,)
+  }
+  assert(
+    type(bounds) == dictionary,
+    message: "The drawing function of element '" + elmt.id + "' did not return the correct bounds dictionary"
+  )
   func
 
-  if (name != none) {
+  draw.anchor("north", bounds.t)
+  draw.anchor("south", bounds.b)
+  draw.anchor("west", bounds.l)
+  draw.anchor("east", bounds.r)
+  draw.anchor("north-west", bounds.tl)
+  draw.anchor("north-east", bounds.tr)
+  draw.anchor("south-east", bounds.br)
+  draw.anchor("south-west", bounds.bl)
+
+  if elmt.name != none {
     draw.content(
-      (name: id, anchor: name-anchor),
-      anchor: name-anchor,
+      (name: elmt.id, anchor: elmt.name-anchor),
+      anchor: if elmt.name-anchor in util.valid-anchors {elmt.name-anchor} else {"center"},
       padding: 0.5em,
-      align(center)[*#name*]
+      align(center)[*#elmt.name*]
     )
   }
 
-  if auto-ports {
-    add-ports(
-      id,
-      tl, tr, br, bl,
-      ports,
-      ports-margins,
-      debug: debug.ports
+  add-ports(elmt, bounds)
+
+  if elmt.debug.bounds {
+    draw.line(
+      bounds.tl, bounds.tr, bounds.br, bounds.bl,
+      stroke: red,
+      close: true
     )
   }
-})
+})
+
+/// Draws an element
+/// - draw-shape (function): Draw function
+/// - x (number, dictionary): The x position (bottom-left corner).
+///
+///   If it is a dictionary, it should be in the format `(rel: number, to: str)`, where `rel` is the offset and `to` the base anchor
+/// - y (number, dictionary): The y position (bottom-left corner).
+///
+///   If it is a dictionary, it should be in the format `(from: str, to: str)`, where `from` is the base anchor and `to` is the id of the port to align with the anchor
+/// - w (number): Width of the element
+/// - h (number): Height of the element
+/// - name (none, str): Optional name of the block
+/// - name-anchor (str): Anchor for the optional name
+/// - ports (dictionary): Dictionary of ports. The keys are cardinal directions ("north", "east", "south" and/or "west"). The values are arrays of ports (dictionaries) with the following fields:
+///   - `id` (`str`): (Required) Port id
+///   - `name` (`str`): Optional name displayed *in* the block
+///   - `clock` (`bool`): Whether it is a clock port (triangle symbol)
+///   - `vertical` (`bool`): Whether the name should be drawn vertically
+/// - ports-margins (dictionary): Dictionary of ports margins (used with automatic port placement). They keys are cardinal directions ("north", "east", "south", "west"). The values are tuples of (`<start>`, `<end>`) margins (numbers)
+/// - fill (none, color): Fill color
+/// - stroke (stroke): Border stroke
+/// - id (str): The block id (for future reference)
+/// - auto-ports (bool): Whether to use auto port placements or not. If false, `draw-shape` is responsible for adding the appropiate ports
+/// - ports-y (dictionary): Dictionary of the ports y offsets (used with `auto-ports: false`)
+/// - debug (dictionary): Dictionary of debug options.
+///
+///   Supported fields include:
+///     - `ports`: if true, shows dots on all ports of the element
+#let elmt(
+  cls: "element",
+  draw-shape: default-draw-shape,
+  pre-process: default-pre-process,
+  pos: (0, 0),
+  size: (1, 1),
+  name: none,
+  name-anchor: "center",
+  ports: (:),
+  ports-margins: (:),
+  fill: none,
+  stroke: black + 1pt,
+  id: auto,
+  ports-pos: auto,
+  debug: (
+    bounds: false,
+    ports: false
+  ),
+  extra: (:)
+) = {
+  for (key, side-ports) in ports.pairs() {
+    if type(side-ports) == str {
+      side-ports = ((id: side-ports),)
+    } else if type(side-ports) == dictionary {
+      side-ports = (side-ports,)
+    }
+
+    for (i, port) in side-ports.enumerate() {
+      if type(port) == array {
+        side-ports.at(i) = (
+          id: port.at(0, default: ""),
+          name: port.at(1, default: "")
+        )
+      } else if type(port) == str {
+        side-ports.at(i) = (id: port)
+      }
+    }
+    ports.at(key) = side-ports
+  }
+
+  return ((
+    cls: cls,
+    id: id,
+    draw: render.with(draw-shape),
+    pre-process: pre-process,
+    pos: pos,
+    size: size,
+    name: name,
+    name-anchor: name-anchor,
+    ports: ports,
+    ports-margins: ports-margins,
+    fill: fill,
+    stroke: stroke,
+    ports-pos: ports-pos,
+    debug: debug
+  ) + extra,)
+}

src/elements/extender.typ

@@ -1,66 +1,56 @@
-#import "@preview/cetz:0.2.2": draw
+#import "/src/cetz.typ": draw
 #import "element.typ"
+#import "ports.typ": add-port
+#import "../util.typ"
 
-#let draw-shape(id, tl, tr, br, bl, fill, stroke) = {
-  let (x, y) = bl
-  let (width, height) = (tr.at(0) - x, tr.at(1) - y)
+#let draw-shape(elmt, bounds) = {
+  let ratio = elmt.l-ratio / 100%
+  let tl2 = util.lerp(bounds.bl, elmt.l-ratio, bounds.tl)
+  let bounds2 = element.complete-bounds(elmt, (
+    tl: tl2,
+    tr: bounds.tr,
+    br: bounds.br,
+    bl: bounds.bl,
+  ))
 
-  tl = (x, y + height * 0.75)
-  let tr2 = (x + width, y + height * 0.75)
-  let br = (x + width, y)
-  (tr, tr2) = (tr2, tr)
-  let f = draw.group(name: id, {
-    draw.merge-path(
-      inset: 0.5em,
-      fill: fill,
-      stroke: stroke,
-      close: true,
-      draw.line(tl, tr2, br, bl)
-    )
-    draw.anchor("north", (tl, 50%, tr2))
-    draw.anchor("south", (bl, 50%, br))
-    draw.anchor("west", (tl, 50%, bl))
-    draw.anchor("east", (tr2, 50%, br))
-  })
-  return (f, tl, tr, br, bl)
+  let f = draw.line(
+    bounds2.tl, bounds2.tr, bounds2.br, bounds2.bl,
+    fill: elmt.fill,
+    stroke: elmt.stroke,
+    close: true
+  )
+  return (f, bounds2)
 }
 
+/// Draws a bit extender
+///
+/// #examples.extender
+/// For other parameters description, see #doc-ref("element.elmt")
+/// - h-ratio (ratio): The height ratio of the left side relative to the full height
+/// - align-out (bool): If true, the output and input ports are aligned, otherwise, the output port is centered on the right side
 #let extender(
-  x: none,
-  y: none,
-  w: none,
-  h: none,
-  name: none,
-  name-anchor: "center",
-  fill: none,
-  stroke: black + 1pt,
-  id: "",
-  debug: (
-    grid: false,
-    ports: false
-  )
+  l-ratio: 75%,
+  align-out: true,
+  ..args
 ) = {
   let ports = (
-    west: (
-      (id: "in"),
-    ),
-    east: (
-      (id: "out"),
-    )
+    west: ((id: "in"),),
+    east: ((id: "out"),)
+  )
+  let out-pct = if align-out {l-ratio / 2} else {50%}
+  let ports-pos = (
+    "east": (l, i) => {l - l * (out-pct / 100%)}
   )
   
-  element.elmt(
+  return element.elmt(
+    cls: "extender",
     draw-shape: draw-shape,
-    x: x,
-    y: y,
-    w: w,
-    h: h,
-    name: name,
-    name-anchor: name-anchor,
     ports: ports,
-    fill: fill,
-    stroke: stroke,
-    id: id,
-    debug: debug
+    ports-pos: ports-pos,
+    extra: (
+      l-ratio: l-ratio,
+      align-out: align-out
+    ),
+    ..args
   )
 }

src/elements/group.typ

@@ -0,0 +1,80 @@
+#import "/src/cetz.typ": draw, coordinate
+#import "../util.typ"
+
+/// Draws a group of elements
+///
+/// #examples.group
+/// - body (elements, function): Elements to group
+/// - id (str): see #doc-ref("element.elmt")
+/// - name (str): The group's name
+/// - name-anchor (str): The anchor for the name. \
+///     Note: the name will be placed on the *outside* of the group
+/// - fill (color): see #doc-ref("element.elmt")
+/// - stroke (stroke): see #doc-ref("element.elmt")
+/// - padding (float,length,array,dictionary): The inside padding:
+///    - float / length: same for all sides
+///    - array: either (`<all>`,), (`<vertical>`, `<horizontal>`) or (`<top>`, `<right>`, `<bottom>`, `<left>`)
+///    - dictionary: valid keys are "top", "right", "bottom" and "left"
+/// - radius (number): The corner radius
+#let group(
+  body,
+  id: "",
+  name: none,
+  name-anchor: "south",
+  fill: none,
+  stroke: black + 1pt,
+  padding: 0.5em,
+  radius: 0.5em
+) = {
+  let min-x = none
+  let max-x = none
+  let min-y = none
+  let max-y = none
+
+  let pad-top = padding
+  let pad-bottom = padding
+  let pad-left = padding
+  let pad-right = padding
+
+  if type(padding) == array {
+    if padding.len() == 0 {
+      panic("Padding array must contain at least one value")
+    } else if padding.len() == 1 {
+      pad-top = padding.first()
+      pad-bottom = padding.first()
+      pad-left = padding.first()
+      pad-right = padding.first()
+    } else if padding.len() == 2 {
+      pad-top = padding.first()
+      pad-bottom = padding.first()
+      pad-left = padding.last()
+      pad-right = padding.last()
+    } else if padding.len() == 4 {
+      (pad-top, pad-right, pad-bottom, pad-left) = padding
+    }
+  } else if type(padding) == dictionary {
+    pad-top = padding.at("top", default: 0.5em)
+    pad-right = padding.at("right", default: 0.5em)
+    pad-bottom = padding.at("bottom", default: 0.5em)
+    pad-left = padding.at("left", default: 0.5em)
+  }
+
+  draw.hide(draw.group(name: id+"-inner", body))
+  draw.rect(
+    (rel: (-pad-left, -pad-bottom), to: id+"-inner.south-west"),
+    (rel: (pad-right, pad-top), to: id+"-inner.north-east"),
+    name: id,
+    radius: radius,
+    stroke: stroke,
+    fill: fill
+  )
+  if name != none {
+    draw.content(
+      id + "." + name-anchor,
+      anchor: util.opposite-anchor(name-anchor),
+      padding: 5pt,
+      [*#name*]
+    )
+  }
+  body
+}

src/elements/logic/and.typ

@@ -0,0 +1,93 @@
+#import "/src/cetz.typ": draw
+#import "gate.typ"
+
+#let draw-shape(id, tl, tr, br, bl, fill, stroke) = {
+  let (x, y) = bl
+  let (width, height) = (tr.at(0) - x, tr.at(1) - y)
+
+  let t = (x + width / 2, y + height)
+  let b = (x + width / 2, y)
+
+  let f = draw.group(name: id, {
+    draw.merge-path(
+      inset: 0.5em,
+      fill: fill,
+      stroke: stroke,
+      name: id + "-path",
+      close: true, {
+        draw.line(bl, tl, t)
+        draw.arc-through((), (tr , 50%, br), b)
+        draw.line((), b)
+      }
+    )
+    
+    draw.anchor("north", t)
+    draw.anchor("south", b)
+  })
+  return (f, tl, tr, br, bl)
+}
+
+/// Draws an AND gate. This function is also available as `element.gate-and()`
+/// 
+/// For parameters, see #doc-ref("gates.gate")
+/// #examples.gate-and
+#let gate-and(
+  x: none,
+  y: none,
+  w: none,
+  h: none,
+  inputs: 2,
+  fill: none,
+  stroke: black + 1pt,
+  id: "",
+  inverted: (),
+  debug: (
+    ports: false
+  )
+) = {
+  gate.gate(
+    draw-shape: draw-shape,
+    x: x,
+    y: y,
+    w: w,
+    h: h,
+    inputs: inputs,
+    fill: fill,
+    stroke: stroke,
+    id: id,
+    inverted: inverted,
+    debug: debug
+  )
+}
+
+/// Draws an NAND gate. This function is also available as `element.gate-nand()`
+/// 
+/// For parameters, see #doc-ref("gates.gate")
+/// #examples.gate-nand
+#let gate-nand(
+  x: none,
+  y: none,
+  w: none,
+  h: none,
+  inputs: 2,
+  fill: none,
+  stroke: black + 1pt,
+  id: "",
+  inverted: (),
+  debug: (
+    ports: false
+  )
+) = {
+  gate-and(
+    x: x,
+    y: y,
+    w: w,
+    h: h,
+    inputs: inputs,
+    fill: fill,
+    stroke: stroke,
+    id: id,
+    inverted: if inverted != "all" {inverted + ("out",)} else {inverted},
+    debug: debug
+  )
+}

src/elements/logic/buf.typ

@@ -0,0 +1,89 @@
+#import "/src/cetz.typ": draw
+#import "gate.typ"
+
+#let draw-shape(id, tl, tr, br, bl, fill, stroke) = {
+  let (x, y) = bl
+  let (width, height) = (tr.at(0) - x, tr.at(1) - y)
+
+  let r = (x + width, y + height / 2)
+
+  let f = draw.group(name: id, {
+    draw.merge-path(
+      inset: 0.5em,
+      fill: fill,
+      stroke: stroke,
+      name: id + "-path",
+      close: true, {
+        draw.line(bl, tl, r)
+      }
+    )
+    
+    draw.anchor("north", (tl, 50%, r))
+    draw.anchor("south", (bl, 50%, r))
+  })
+  return (f, tl, tr, br, bl)
+}
+
+/// Draws a buffer gate. This function is also available as `element.gate-buf()`
+/// 
+/// For parameters, see #doc-ref("gates.gate")
+/// #examples.gate-buf
+#let gate-buf(
+  x: none,
+  y: none,
+  w: none,
+  h: none,
+  inputs: 1,
+  fill: none,
+  stroke: black + 1pt,
+  id: "",
+  inverted: (),
+  debug: (
+    ports: false
+  )
+) = {
+  gate.gate(
+    draw-shape: draw-shape,
+    x: x,
+    y: y,
+    w: w,
+    h: h,
+    inputs: inputs,
+    fill: fill,
+    stroke: stroke,
+    id: id,
+    inverted: inverted,
+    debug: debug
+  )
+}
+
+/// Draws a NOT gate. This function is also available as `element.gate-not()`
+/// 
+/// For parameters, see #doc-ref("gates.gate")
+/// #examples.gate-not
+#let gate-not(x: none,
+  y: none,
+  w: none,
+  h: none,
+  inputs: 1,
+  fill: none,
+  stroke: black + 1pt,
+  id: "",
+  inverted: (),
+  debug: (
+    ports: false
+  )
+) = {
+  gate-buf(
+    x: x,
+    y: y,
+    w: w,
+    h: h,
+    inputs: inputs,
+    fill: fill,
+    stroke: stroke,
+    id: id,
+    inverted: if inverted != "all" {inverted + ("out",)} else {inverted},
+    debug: debug
+  )
+}

src/elements/logic/gate.typ

@@ -0,0 +1,117 @@
+#import "/src/cetz.typ": draw, coordinate
+#import "../ports.typ": add-ports, add-port
+#import "../element.typ"
+
+#let default-draw-shape(id, tl, tr, br, bl, fill, stroke) = {
+  let f = {draw.rect(tl, br, fill: fill, stroke: stroke)}
+  return (f, tl, tr, br, bl)
+}
+
+
+/// Draws a logic gate. This function is also available as `element.gate()`
+///
+/// - draw-shape (function): see #doc-ref("element.elmt")
+/// - x (number, dictionary): see #doc-ref("element.elmt")
+/// - y (number, dictionary): see #doc-ref("element.elmt")
+/// - w (number): see #doc-ref("element.elmt")
+/// - h (number): see #doc-ref("element.elmt")
+/// - inputs (int): The number of inputs
+/// - fill (none, color): see #doc-ref("element.elmt")
+/// - stroke (stroke): see #doc-ref("element.elmt")
+/// - id (str): see #doc-ref("element.elmt")
+/// - inverted (str, array): Either "all" or an array of port ids to display as inverted
+/// - inverted-radius (number): The radius of inverted ports dot
+/// - debug (dictionary): see #doc-ref("element.elmt")
+#let gate(
+  draw-shape: default-draw-shape,
+  x: none,
+  y: none,
+  w: none,
+  h: none,
+  inputs: 2,
+  fill: none,
+  stroke: black + 1pt,
+  id: "",
+  inverted: (),
+  inverted-radius: 0.1,
+  debug: (
+    ports: false
+  )
+) = draw.get-ctx(ctx => {
+  let width = w
+  let height = h
+
+  let x = x
+  let y = y
+  if x == none { panic("Parameter x must be set") }
+  if y == none { panic("Parameter y must be set") }
+  if w == none { panic("Parameter w must be set") }
+  if h == none { panic("Parameter h must be set") }
+
+  if (type(x) == dictionary) {
+    let offset = x.rel
+    let to = x.to
+    let (ctx, to-pos) = coordinate.resolve(ctx, (rel: (offset, 0), to: to))
+    x = to-pos.at(0)
+  }
+  
+  if (type(y) == dictionary) {
+    let from = y.from
+    let to = y.to
+    
+    let dy
+    if to == "out" {
+      dy = height / 2
+    } else {
+      dy = height * (i + 0.5) / inputs
+    }
+    
+    let (ctx, from-pos) = coordinate.resolve(ctx, from)
+    y = from-pos.at(1) + dy - height
+  }
+
+  let tl = (x, y + height)
+  let tr = (x + width, y + height)
+  let br = (x + width, y)
+  let bl = (x, y)
+
+  // Workaround because CeTZ needs to have all draw functions in the body
+  let func = {}
+  (func, tl, tr, br, bl) = draw-shape(id, tl, tr, br, bl, fill, stroke)
+  func
+
+  let space = 100% / inputs
+  for i in range(inputs) {
+    let pct = (i + 0.5) * space
+    let a = (tl, pct, bl)
+    let b = (tr, pct, br)
+    let int-name = id + "i" + str(i)
+    draw.intersections(
+      int-name,
+      func,
+      draw.hide(draw.line(a, b))
+    )
+    let port-name = "in" + str(i)
+    let port-pos = int-name + ".0"
+    if inverted == "all" or port-name in inverted {
+      draw.circle(port-pos, radius: inverted-radius, anchor: "east", stroke: stroke)
+      port-pos = (rel: (-2 * inverted-radius, 0), to: port-pos)
+    }
+    add-port(
+      id, "west",
+      (id: port-name), port-pos,
+      debug: debug.ports
+    )
+  }
+
+  let out-pos = id + ".east"
+  if inverted == "all" or "out" in inverted {
+    draw.circle(out-pos, radius: inverted-radius, anchor: "west", stroke: stroke)
+    out-pos = (rel: (2 * inverted-radius, 0), to: out-pos)
+  }
+  add-port(
+    id, "east",
+    (id: "out"), out-pos,
+    debug: debug.ports
+  )
+})

src/elements/logic/or.typ

@@ -0,0 +1,104 @@
+#import "/src/cetz.typ": draw
+#import "gate.typ"
+
+#let draw-shape(id, tl, tr, br, bl, fill, stroke) = {
+  let (x, y) = bl
+  let (width, height) = (tr.at(0) - x, tr.at(1) - y)
+
+  let t = (x + width / 2, y + height)
+  let b = (x + width / 2, y)
+
+  let ctrl-bl = (x + width / 2, y)
+  let ctrl-br = (x + width * 0.8, y + height * 0.1)
+  let ctrl-tl = (x + width / 2, y + height)
+  let ctrl-tr = (x + width * 0.8, y + height * 0.9)
+
+  let l = (x + width * 0.2, y + height / 2)
+  let r = (x + width, y + height / 2)
+
+  let f = draw.group(name: id, {
+    draw.merge-path(
+      inset: 0.5em,
+      fill: fill,
+      stroke: stroke,
+      name: id + "-path", {
+        draw.bezier-through(bl, l, tl)
+        draw.bezier((), r, ctrl-tl, ctrl-tr)
+        draw.bezier((), bl, ctrl-br, ctrl-bl)
+      }
+    )
+    
+    draw.intersections("i",
+      id + "-path",
+      draw.hide(draw.line(t, b))
+    )
+    draw.anchor("north", "i.0")
+    draw.anchor("south", "i.1")
+  })
+  return (f, tl, tr, br, bl)
+}
+
+/// Draws an OR gate. This function is also available as `element.gate-or()`
+/// 
+/// For parameters, see #doc-ref("gates.gate")
+/// #examples.gate-or
+#let gate-or(
+  x: none,
+  y: none,
+  w: none,
+  h: none,
+  inputs: 2,
+  fill: none,
+  stroke: black + 1pt,
+  id: "",
+  inverted: (),
+  debug: (
+    ports: false
+  )
+) = {
+  gate.gate(
+    draw-shape: draw-shape,
+    x: x,
+    y: y,
+    w: w,
+    h: h,
+    inputs: inputs,
+    fill: fill,
+    stroke: stroke,
+    id: id,
+    inverted: inverted,
+    debug: debug
+  )
+}
+
+/// Draws a NOR gate. This function is also available as `element.gate-nor()`
+/// 
+/// For parameters, see #doc-ref("gates.gate")
+/// #examples.gate-nor
+#let gate-nor(
+  x: none,
+  y: none,
+  w: none,
+  h: none,
+  inputs: 2,
+  fill: none,
+  stroke: black + 1pt,
+  id: "",
+  inverted: (),
+  debug: (
+    ports: false
+  )
+) = {
+  gate-or(
+    x: x,
+    y: y,
+    w: w,
+    h: h,
+    inputs: inputs,
+    fill: fill,
+    stroke: stroke,
+    id: id,
+    inverted: if inverted != "all" {inverted + ("out",)} else {inverted},
+    debug: debug
+  )
+}

src/elements/logic/xor.typ

@@ -0,0 +1,111 @@
+#import "/src/cetz.typ": draw
+#import "gate.typ"
+
+#let space = 10%
+
+#let draw-shape(id, tl, tr, br, bl, fill, stroke) = {
+  let (x, y) = bl
+  let (width, height) = (tr.at(0) - x, tr.at(1) - y)
+
+  let tl2 = (tl, space, tr)
+  let bl2 = (bl, space, br)
+
+  let t = (x + width / 2, y + height)
+  let b = (x + width / 2, y)
+
+  let ctrl-bl = (x + width / 2, y)
+  let ctrl-br = (x + width * 0.8, y + height * 0.1)
+  let ctrl-tl = (x + width / 2, y + height)
+  let ctrl-tr = (x + width * 0.8, y + height * 0.9)
+
+  let l = (x + width * 0.2, y + height / 2)
+  let l2 = (x + width * (0.2 + space / 100%), y + height / 2)
+  let r = (x + width, y + height / 2)
+
+  let f = draw.group(name: id, {
+    draw.bezier-through(bl, l, tl, stroke: stroke)
+    draw.merge-path(
+      inset: 0.5em,
+      fill: fill,
+      stroke: stroke,
+      name: id + "-path", {
+        draw.bezier-through(bl2, l2, tl2)
+        draw.bezier((), r, ctrl-tl, ctrl-tr)
+        draw.bezier((), bl2, ctrl-br, ctrl-bl)
+      }
+    )
+    
+    draw.intersections("i",
+      id + "-path",
+      draw.hide(draw.line(t, b))
+    )
+    draw.anchor("north", "i.0")
+    draw.anchor("south", "i.1")
+  })
+  return (f, tl, tr, br, bl)
+}
+
+/// Draws a XOR gate. This function is also available as `element.gate-xor()`
+/// 
+/// For parameters, see #doc-ref("gates.gate")
+/// #examples.gate-xor
+#let gate-xor(
+  x: none,
+  y: none,
+  w: none,
+  h: none,
+  inputs: 2,
+  fill: none,
+  stroke: black + 1pt,
+  id: "",
+  inverted: (),
+  debug: (
+    ports: false
+  )
+) = {
+  gate.gate(
+    draw-shape: draw-shape,
+    x: x,
+    y: y,
+    w: w,
+    h: h,
+    inputs: inputs,
+    fill: fill,
+    stroke: stroke,
+    id: id,
+    inverted: inverted,
+    debug: debug
+  )
+}
+
+/// Draws a XNOR gate. This function is also available as `element.gate-xnor()`
+/// 
+/// For parameters, see #doc-ref("gates.gate")
+/// #examples.gate-xnor
+#let gate-xnor(
+  x: none,
+  y: none,
+  w: none,
+  h: none,
+  inputs: 2,
+  fill: none,
+  stroke: black + 1pt,
+  id: "",
+  inverted: (),
+  debug: (
+    ports: false
+  )
+) = {
+  gate-xor(
+    x: x,
+    y: y,
+    w: w,
+    h: h,
+    inputs: inputs,
+    fill: fill,
+    stroke: stroke,
+    id: id,
+    inverted: if inverted != "all" {inverted + ("out",)} else {inverted},
+    debug: debug
+  )
+}

src/elements/multiplexer.typ

@@ -1,64 +1,73 @@
-#import "@preview/cetz:0.2.2": draw
+#import "/src/cetz.typ": draw
 #import "../util.typ"
 #import "element.typ"
+#import "ports.typ": add-port
 
-#let draw-shape(id, tl, tr, br, bl, fill, stroke) = {
-  let tr2 = (tr, 20%, br)
-  let br2 = (tr, 80%, br)
-  let f = draw.group(name: id, {
-    draw.merge-path(
-      inset: 0.5em,
-      fill: fill,
-      stroke: stroke,
-      close: true,
-      draw.line(tl, tr2, br2, bl)
-    )
-    draw.anchor("north", (tl, 50%, tr2))
-    draw.anchor("south", (bl, 50%, br2))
-    draw.anchor("west", (tl, 50%, bl))
-    draw.anchor("east", (tr2, 50%, br2))
-  })
-  return (f, tl, tr, br, bl)
+#let draw-shape(elmt, bounds) = {
+  let margin = (100% - elmt.l-ratio) / 2
+  let tr2 = util.lerp(bounds.tr, margin, bounds.br)
+  let br2 = util.lerp(bounds.br, margin, bounds.tr)
+  let bounds2 = element.complete-bounds(elmt, (
+    tl: bounds.tl,
+    bl: bounds.bl,
+    tr: tr2,
+    br: br2,
+  ))
+  let f = draw.line(
+    bounds2.tl, bounds2.tr, bounds2.br, bounds2.bl,
+    close: true,
+    fill: elmt.fill,
+    stroke: elmt.stroke
+  )
+
+  return (f, bounds2)
 }
 
+/// Draws a multiplexer
+///
+/// #examples.multiplexer
+/// For other parameters description, see #doc-ref("element.elmt")
+/// - entries (int, array): If it is an integer, it defines the number of input ports (automatically named with their binary index). If it is an array of strings, it defines the name of each input.
+/// - h-ratio (ratio): The height ratio of the right side relative to the full height
 #let multiplexer(
-  x: none,
-  y: none,
-  w: none,
-  h: none,
-  name: none,
-  name-anchor: "center",
   entries: 2,
-  fill: none,
-  stroke: black + 1pt,
-  id: "",
-  debug: (
-    grid: false,
-    ports: false
-  )
+  l-ratio: 60%,
+  ..args
 ) = {
-  let ports = ()
+  let in-ports = ()
+  let ports-pos = (
+    "east": auto,
+  )
 
   if (type(entries) == int) {
     let nbits = calc.ceil(calc.log(entries, base: 2))
     for i in range(entries) {
       let bits = util.lpad(str(i, base: 2), nbits)
-      ports.push((id: "in" + str(i), name: bits))
+      in-ports.push((id: "in" + str(i), name: bits))
+    }
+  } else {
+    for (i, port) in entries.enumerate() {
+      in-ports.push((id: "in" + str(i), name: port))
     }
   }
+
+  let n = in-ports.len()
+  ports-pos.insert("west", (l, i) => {l * (i + 0.5) / n})
   
   element.elmt(
+    cls: "multiplexer",
     draw-shape: draw-shape,
-    x: x,
-    y: y,
-    w: w,
-    h: h,
-    name: name,
-    name-anchor: name-anchor,
-    ports: (west: ports, east: ((id: "out"),)),
-    fill: fill,
-    stroke: stroke,
-    id: id,
-    debug: debug
+    ports: (west: in-ports, east: ((id: "out"),)),
+    ports-pos: ports-pos,
+    extra: (l-ratio: l-ratio),
+    ..args
   )
+
+  /*
+  for (i, port) in ports.enumerate() {
+    let pct = (i + 0.5) * space
+    add-port(id, "west", port, (id+".north-west", pct, id+".south-west"))
+  }
+  add-port(id, "east", (id: "out"), (id+".north-east", 50%, id+".south-east"))
+  */
 }

src/elements/ports.typ

@@ -1,6 +1,76 @@
-#import "@preview/cetz:0.2.2": draw
+#import "/src/cetz.typ": draw
 #import "../util.typ": rotate-anchor
 
+#let get-port-side(elmt, port) = {
+  for (side, ports) in elmt.ports {
+    for p in ports {
+      if p.id == port {
+        return side
+      }
+    }
+  }
+  panic(
+    "Unknown port '" + port + "' on element '" + elmt.id + "', could not automatically determine side"
+  )
+}
+
+#let get-port-idx(elmt, port, side: auto) = {
+  if side == auto {
+    side = get-port-side(elmt, port)
+  }
+  assert(
+    side in elmt.ports,
+    message: "No ports on side '" + side + "' of element '" + elmt.id + "'"
+  )
+  let i = elmt.ports.at(side).position(p => p.id == port)
+  assert(
+    i != none,
+    message: "Could not find port '" + port + "' on side '" + side + "' of element '" + elmt.id + "'"
+  )
+  return i
+}
+
+#let get-port-pos(elmt, bounds, side, port, port-i) = {
+  let (pt0, pt1) = bounds.ports.at(side)
+  let side-len = if side in ("north", "south") {
+    pt1.at(0) - pt0.at(0)
+  } else {
+    pt0.at(1) - pt1.at(1)
+  }
+
+  let offset = if (
+    elmt.ports-pos == auto or
+    elmt.ports-pos.at(side, default: auto) == auto
+  ) {
+    let space = 100% / (elmt.ports.at(side, default: ()).len() + 1)
+    (port-i + 1) * space
+  } else {
+    assert(
+      side in elmt.ports-pos,
+      message: "Could not reliably compute port position (missing side)"
+    )
+    let side-pos = elmt.ports-pos.at(side)
+    if type(side-pos) == function {
+      (side-pos)(side-len, port-i)
+    } else if type(side-pos) == array {
+      (side-pos.at(i))(side-len)
+    } else if type(side-pos) == dictionary {
+      assert(
+        port in side-pos,
+        message: "Could not reliably compute port position (missing port)"
+      )
+      (side-pos.at(port))(side-len)
+    } else {
+      panic("Could not reliably compute port position (invalid type)")
+    }
+  }
+
+  if type(offset) == ratio {
+    offset = offset * side-len / 100%
+  }
+  return offset
+}
+
 #let add-port(
   elmt-id, side, port, pos,
   prev: none,
@@ -15,11 +85,12 @@
       panic("Clock port must have previous and next positions")
     }
     
+    let size = if port.at("small", default: false) {8pt} else {1em}
     let offset
-    if      (side == "north") { offset = (   0, -1em) }
-    else if (side == "east")  { offset = (-1em,    0) }
-    else if (side == "south") { offset = (   0,  1em) }
-    else if (side == "west")  { offset = ( 1em,    0) }
+    if      (side == "north") { offset = (    0, -size) }
+    else if (side == "east")  { offset = (-size,     0) }
+    else if (side == "south") { offset = (    0,  size) }
+    else if (side == "west")  { offset = ( size,     0) }
 
     let pos1 = (rel: offset, to: pos)
 
@@ -33,12 +104,10 @@
     angle: if name-rotate {90deg} else {0deg},
     name
   )
-  let id = elmt-id + "-port-" + port.at("id")
 
   if debug {
     draw.circle(
       pos,
-      name: id,
       radius: .1,
       stroke: none,
       fill: red
@@ -48,52 +117,56 @@
     draw.hide(draw.circle(
       pos,
       radius: 0,
-      stroke: none,
-      name: id
+      stroke: none
     ))
   }
+  draw.anchor(port.id, pos)
 }
 
 #let add-ports(
-  elmt-id,
-  tl, tr, br, bl,
-  ports,
-  ports-margins,
-  debug: false
+  elmt,
+  bounds
 ) = {
-  let sides = (
-    "north": (tl, tr),
-    "east": (tr, br),
-    "south": (bl, br),
-    "west": (tl, bl)
-  )
+  let debug = elmt.debug.at("ports", default: false)
 
-  for (side, props) in sides {
-    let side-ports = ports.at(side, default: ())
+  if type(elmt.ports) != dictionary {
+    return
+  }
+
+  for (side, props) in bounds.ports {
+    let side-ports = elmt.ports.at(side, default: ())
     let space = 100% / (side-ports.len() + 1)
-
+    
+    let (pt0, pt1) = props
+    let side-len = if side in ("north", "south") {
+      pt1.at(0) - pt0.at(0)
+    } else {
+      pt0.at(1) - pt1.at(1)
+    }
     for (i, port) in side-ports.enumerate() {
-      let pct = (i + 1) * space
-      let pt0 = props.at(0)
-      let pt1 = props.at(1)
+      let offset = get-port-pos(elmt, bounds, side, port.id, i)
 
-      if side in ports-margins {
-        let (a, b) = (pt0, pt1)
-        let margins = ports-margins.at(side)
-        a = (pt0, margins.at(0), pt1)
-        b = (pt0, 100% - margins.at(1), pt1)
-        pt0 = a
-        pt1 = b
+      let pos = (pt0, offset, pt1)
+      let offset-prev = if type(offset) == ratio {
+        offset - space / 2
+      } else {
+        offset - space * side-len / 200%
+      }
+      let offset-next = if type(offset) == ratio {
+        offset + space / 2
+      } else {
+        offset + space * side-len / 200%
+      }
+      let pos-prev = (pt0, offset-prev, pt1)
+      let pos-next = (pt0, offset-next, pt1)
+
+      if port.at("small", default: false) {
+        pos-prev = (pos, 4pt, pt0)
+        pos-next = (pos, 4pt, pt1)
       }
-      
-      let pos = (pt0, pct, pt1)
-      let pct-prev = (i + 0.5) * space
-      let pct-next = (i + 1.5) * space
-      let pos-prev = (pt0, pct-prev, pt1)
-      let pos-next = (pt0, pct-next, pt1)
 
       add-port(
-        elmt-id,
+        elmt.id,
         side,
         port,
         pos,

src/gates.typ

@@ -0,0 +1,5 @@
+#import "elements/logic/gate.typ": gate
+#import "elements/logic/and.typ": gate-and, gate-nand
+#import "elements/logic/or.typ": gate-or, gate-nor
+#import "elements/logic/xor.typ": gate-xor, gate-xnor
+#import "elements/logic/buf.typ": gate-buf, gate-not

src/lib.typ

@@ -1,6 +1,7 @@
-#let version = version((0,0,1))
+#let version = version(0, 2, 0)
 
 #import "circuit.typ": circuit
 #import "element.typ"
+#import "gates.typ"
 #import "util.typ"
 #import "wire.typ"

src/util.typ

@@ -1,16 +1,42 @@
+/// Predefined color palette
+/// #let w = 4
+/// #box(width: 100%, align(center)[
+///   #canvas(length: 2em, {
+///     for (i, color) in util.colors.pairs().enumerate() {
+///       let x = calc.rem(i, w) * 2
+///       let y = calc.floor(i / w) * 2
+///       draw.rect((x, -y), (x + 1, -y - 1), fill: color.last(), stroke: none)
+///       draw.content((x + 0.5, -y - 1.25), color.first())
+///     }
+///   })
+/// ])
 #let colors = (
   orange: rgb(245, 180, 147),
   yellow: rgb(250, 225, 127),
   green: rgb(127, 200, 172),
   pink: rgb(236, 127, 178),
-  purple: rgb(189, 151, 255)
+  purple: rgb(189, 151, 255),
+  blue: rgb(127, 203, 235)
 )
 
-#let lpad(s, len) = {
-  let res = "0" * len + s
+/// Pads a string on the left with 0s to the given length
+///
+/// #example(`#util.lpad("0100", 8)`, mode: "markup")
+///
+/// - string (str): The string to pad
+/// - len (int): The target length
+/// -> str
+#let lpad(string, len) = {
+  let res = "0" * len + string
   return res.slice(-len)
 }
 
+/// Returns the anchor on the opposite side of the given one
+///
+/// #example(`#util.opposite-anchor("west")`, mode: "markup")
+///
+/// - anchor (str): The input anchor
+/// -> str
 #let opposite-anchor(anchor) = {
   return (
     north: "south",
@@ -25,6 +51,11 @@
   ).at(anchor)
 }
 
+/// Returns the anchor rotated 90 degrees clockwise relative to the given one
+///
+/// #example(`#util.rotate-anchor("west")`, mode: "markup")
+/// - anchor (str): The anchor to rotate
+/// -> str
 #let rotate-anchor(anchor) = {
   return (
     north: "east",
@@ -37,4 +68,16 @@
     south-east: "south-west",
     south-west: "north-west"
   ).at(anchor)
+}
+
+#let valid-anchors = (
+  "center", "north", "east", "west", "south",
+  "north-east", "north-west", "south-east", "south-west"
+)
+
+#let lerp(pt0, ratio, pt1) = {
+  return (
+    (pt1.at(0) - pt0.at(0)) * ratio / 100% + pt0.at(0),
+    (pt1.at(1) - pt0.at(1)) * ratio / 100% + pt0.at(1)
+  )
 }

src/wire.typ

@@ -1,12 +1,21 @@
-#import "@preview/cetz:0.2.2": draw, coordinate
+#import "/src/cetz.typ": draw, coordinate
 #import "util.typ": opposite-anchor
+#import "elements/ports.typ": get-port-side
+#import "elements/element.typ"
 
+/// List of valid wire styles
+/// #examples.wires
 #let wire-styles = ("direct", "zigzag", "dodge")
 #let signal-width = 1pt
 #let bus-width = 1.5pt
 
-#let intersection(pt) = {
-  draw.circle(pt, radius: .2, stroke: none, fill: black)
+/// Draws a wire intersection at the given anchor
+/// #examples.intersection
+/// - pt (point): A CeTZ compatible point / anchor
+/// - radius (number): The radius of the intersection
+/// - fill (color): The fill color
+#let intersection(pt, radius: .1, fill: black) = {
+  draw.circle(pt, radius: radius, stroke: none, fill: fill)
 }
 
 #let get-direct-wire(pts) = {
@@ -17,17 +26,30 @@
   return (pts, anchors)
 }
 
-#let get-zigzag-wire(pts, ratio) = {
+#let get-zigzag-wire(pts, ratio, dir) = {
   let start = pts.first()
   let end = pts.last()
-  let mid = (start, ratio, end)
+  let mid = if dir == "vertical" {
+    (start, ratio, (horizontal: end, vertical: ()))
+  } else {
+    (start, ratio, (horizontal: (), vertical: end))
+  }
 
-  let points = (
-    start,
-    (horizontal: mid, vertical: ()),
-    (horizontal: (), vertical: end),
-    end
-  )
+  let points = if dir == "vertical" {
+    (
+      start,
+      (horizontal: mid, vertical: ()),
+      (horizontal: (), vertical: end),
+      end
+    )
+  } else {
+    (
+      start,
+      (horizontal: (), vertical: mid),
+      (horizontal: end, vertical: ()),
+      end
+    )
+  }
   let anchors = (
     "start": start,
     "zig": points.at(1),
@@ -89,28 +111,49 @@
   return (points, anchors)
 }
 
+/// Draws a wire between two points
+/// - id (str): The wire's id, for future reference (anchors)
+/// - pts (array): The two points (as CeTZ compatible coordinates, i.e. XY, relative positions, ids, etc.)
+/// - bus (bool): Whether the wire is a bus (multiple bits) or a simple signal (single bit)
+/// - name (none, str, array): Optional name of the wire. If it is an array, the first name will be put at the start of the wire, and the second at the end
+/// - name-pos (str): Position of the name. One of: "middle", "start" or "end"
+/// - slice (none, array): Optional bits slice (start and end bit indices). If set, it will be displayed at the start of the wire
+/// - color (color): The stroke color
+/// - dashed (bool): Whether the stroke is dashed or not
+/// - style (str): The wire's style (see #doc-ref("wire.wire-styles", var: true) for possible values)
+/// - reverse (bool): If true, the start and end points will be swapped (useful in cases where the start point depends on the end point, for example with perpendiculars)
+/// - directed (bool): If true, the wire will be directed, meaning an arrow will be drawn at the endpoint
+/// - rotate-name (bool): If true, the name will be rotated according to the wire's slope
+/// - zigzag-ratio (ratio): Position of the zigzag vertical relative to the horizontal span (only with style "zigzag")
+/// - zigzag-dir (str): The zigzag's direction. As either "vertical" or "horizontal" (only with dstyle "zigzag")
+/// - dodge-y (number): Y position to dodge the wire to (only with style "dodge")
+/// - dodge-sides (array): The start and end sides (going out of the connected element) of the wire (only with style "dodge")
+/// - dodge-margins (array): The start and end margins (i.e. space before dodging) of the wire (only with style "dodge")
 #let wire(
-  id, pts,
+  pt0,
+  pt1,
+  id: none,
   bus: false,
   name: none,
-  name-pos: "middle",
   slice: none,
   color: black,
   dashed: false,
   style: "direct",
   reverse: false,
+  directed: false,
+  rotate-name: true,
   zigzag-ratio: 50%,
+  zigzag-dir: "vertical",
   dodge-y: 0,
   dodge-sides: ("east", "west"),
-  dodge-margins: (5%, 5%)
+  dodge-margins: (5%, 5%),
+  ..args
 ) = draw.get-ctx(ctx => {
   if not style in wire-styles {
     panic("Invalid wire style '" + style + "'")
   }
 
-  if pts.len() != 2 {
-    panic("Wrong number of points (got " + str(pts.len()) + " instead of 2)")
-  }
+  let pts = (pt0, pt1)
   
   let stroke = (
     paint: color,
@@ -127,7 +170,7 @@
     (points, anchors) = get-direct-wire(pts)
     
   } else if style == "zigzag" {
-    (points, anchors) = get-zigzag-wire(pts, zigzag-ratio)
+    (points, anchors) = get-zigzag-wire(pts, zigzag-ratio, zigzag-dir)
     
   } else if style == "dodge" {
     (points, anchors) = get-dodge-wire(
@@ -139,8 +182,12 @@
     )
   }
 
+  let mark = (fill: color)
+  if directed {
+    mark = (end: ">", fill: color)
+  }
   draw.group(name: id, {
-    draw.line(..points, stroke: stroke)
+    draw.line(..points, stroke: stroke, mark: mark)
     for (anchor-name, anchor-pos) in anchors {
       draw.anchor(anchor-name, anchor-pos)
     }
@@ -148,15 +195,30 @@
 
   let first-pt = id + ".start"
   let last-pt = id + ".end"
+  let first-pos = points.first()
+  let second-pos = points.at(1)
   if reverse {
     (first-pt, last-pt) = (last-pt, first-pt)
   }
   
+  let angle = 0deg
+  if rotate-name {
+    (ctx, first-pos) = coordinate.resolve(ctx, first-pos)
+    (ctx, second-pos) = coordinate.resolve(ctx, second-pos)
+    
+    if reverse {
+      (first-pos, second-pos) = (second-pos, first-pos)
+    }
+    let (x1, y1, ..) = first-pos
+    let (x2, y2, ..) = second-pos
+    angle = calc.atan2(x2 - x1, y2 - y1)
+  }
+  
   if name != none {
     let names = ()
     
     if type(name) == str {
-      names = ((name, name-pos),)
+      names = ((name, "middle"),)
       
     } else if type(name) == array {
       names = (
@@ -166,6 +228,9 @@
     }
 
     for (name, pos) in names {
+      if name == none {
+        continue
+      }
       let point
       let anchor
       
@@ -182,13 +247,12 @@
         anchor = "south-east"
       }
 
-      draw.content(point, anchor: anchor, padding: 3pt, name)
+      draw.content(point, anchor: anchor, padding: 3pt, angle: angle, name)
     }
   }
 
   if slice != none {
-    let (start, end) = slice
-    let slice-txt = "[" + str(start) + ":" + str(end) + "]"
+    let slice-txt = "[" + slice.map(b => str(b)).join(":") + "]"
     
     draw.content(
       first-pt,
@@ -199,32 +263,82 @@
   }
 })
 
-#let stub(port-id, side, name: none, vertical: false, length: 1em) = {
-  let offset = (
-    north: (0, length),
-    east: (length, 0),
-    south: (0, -length),
-    west: (-length, 0)
-  ).at(side)
-  draw.line(
-    port-id,
-    (rel: offset, to: port-id)
-  )
-  if name != none {
-    let text-anchor = if vertical {
-      (
-        "north": "west",
-        "south": "east",
-        "west": "south",
-        "east": "north"
-      ).at(side)
-    } else { opposite-anchor(side) }
-    draw.content(
-      anchor: text-anchor,
-      padding: 0.2em,
-      angle: if vertical {90deg} else {0deg},
-      (rel: offset, to: port-id),
-      name
-    )
+/// Draws a wire stub (useful for unlinked ports)
+///
+/// #examples.stub
+/// - port-id (str): The port anchor
+/// - side (str): The side on which the port is (one of "north", "east", "south", "west")
+/// - name (none, str): Optional name displayed at the end of the stub
+/// - vertical (bool): Whether the name should be displayed vertically
+/// - length (number): The length of the stub
+/// - name-offset (number): The name offset, perpendicular to the stub
+#let stub(anchor, side: auto, name: none, vertical: false, length: 1em, name-offset: 0) = {
+  if "." not in anchor {
+    panic("`anchor` must be a valid anchor of an element")
   }
+  let parts = anchor.split(".")
+  let port-id = parts.last()
+  let port-elmt-id = parts.slice(0, -1).join(".")
+
+  let pre-process = (elements, elmt) => {
+    let eid = elmt.id
+
+    let side = side
+    if side == auto {
+      if port-elmt-id not in elements {
+        panic("Unknown element " + port-elmt-id)
+      }
+      let port-elmt = elements.at(port-elmt-id)
+      side = get-port-side(port-elmt, port-id)
+    }
+    elements.at(eid).insert("side", side)
+    return elements
+  }
+
+  let draw-func(elmt, bounds) = {
+    let side = elmt.side
+    let end-offset = (
+      north: (0, length),
+      east: (length, 0),
+      south: (0, -length),
+      west: (-length, 0)
+    ).at(side)
+    
+    let name-offset = (
+      north: (name-offset, length),
+      east: (length, name-offset),
+      south: (name-offset, -length),
+      west: (-length, name-offset)
+    ).at(side)
+
+    let shapes = ()
+    shapes += draw.line(
+      anchor,
+      (rel: end-offset, to: anchor)
+    )
+    if name != none {
+      let text-anchor = if vertical {
+        (
+          "north": "west",
+          "south": "east",
+          "west": "south",
+          "east": "north"
+        ).at(side)
+      } else { opposite-anchor(side) }
+      shapes += draw.content(
+        anchor: text-anchor,
+        padding: 0.2em,
+        angle: if vertical {90deg} else {0deg},
+        (rel: name-offset, to: anchor),
+        name
+      )
+    }
+
+    return (shapes, bounds)
+  }
+
+  return element.elmt(
+    draw-shape: draw-func,
+    pre-process: pre-process
+  )
 }

typst.toml

@@ -1,7 +1,7 @@
 [package]
 name = "circuiteria"
-version = "0.0.1"
-compiler = "0.11.0"
+version = "0.2.0"
+compiler = "0.13.0"
 repository = "https://git.kb28.ch/HEL/circuiteria"
 entrypoint = "src/lib.typ"
 authors = [
@@ -11,4 +11,4 @@ categories = [ "visualization" ]
 license = "Apache-2.0"
 description = "Drawing block circuits with Typst made easy, using CeTZ"
 keywords = [ "circuit", "block", "draw" ]
-exclude = [ "/gallery/*" ]
+exclude = [ "gallery", "justfile", "doc" ]