forked from HEL/circuiteria
		
	split up sources into manageable files
This commit is contained in:
		
							
								
								
									
										230
									
								
								src/wire.typ
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										230
									
								
								src/wire.typ
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,230 @@ | ||||
| #import "@preview/cetz:0.2.2": draw, coordinate | ||||
| #import "util.typ": opposite-anchor | ||||
|  | ||||
| #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) | ||||
| } | ||||
|  | ||||
| #let get-direct-wire(pts) = { | ||||
|   let anchors = ( | ||||
|     "start": pts.first(), | ||||
|     "end": pts.last() | ||||
|   ) | ||||
|   return (pts, anchors) | ||||
| } | ||||
|  | ||||
| #let get-zigzag-wire(pts, ratio) = { | ||||
|   let start = pts.first() | ||||
|   let end = pts.last() | ||||
|   let mid = (start, ratio, end) | ||||
|  | ||||
|   let points = ( | ||||
|     start, | ||||
|     (horizontal: mid, vertical: ()), | ||||
|     (horizontal: (), vertical: end), | ||||
|     end | ||||
|   ) | ||||
|   let anchors = ( | ||||
|     "start": start, | ||||
|     "zig": points.at(1), | ||||
|     "zag": points.at(2), | ||||
|     "end": end | ||||
|   ) | ||||
|   return (points, anchors) | ||||
| } | ||||
|  | ||||
| #let get-dodge-wire(pts, dodge-y, margins, sides, ctx) = { | ||||
|   let start = pts.first() | ||||
|   let end = pts.last() | ||||
|   let (margin-start, margin-end) = margins | ||||
|   let (side-start, side-end) = sides | ||||
|  | ||||
|   let p1 = (start, margin-start, end) | ||||
|   let p2 = (end, margin-end, start) | ||||
|    | ||||
|   let (ctx, p0) = coordinate.resolve(ctx, start) | ||||
|   let (ctx, p3) = coordinate.resolve(ctx, end) | ||||
|   p0 = (x: p0.first(), y: p0.last()) | ||||
|   p3 = (x: p3.first(), y: p3.last()) | ||||
|  | ||||
|   let dx1 = margin-start | ||||
|   let dx2 = margin-end | ||||
|    | ||||
|   if type(margin-start) == ratio { | ||||
|     dx1 = calc.abs(p3.x - p0.x) * margin-start / 100% | ||||
|   } | ||||
|   if type(margin-end) == ratio { | ||||
|     dx2 = calc.abs(p3.x - p0.x) * margin-end / 100% | ||||
|   } | ||||
|   if side-start == "west" { | ||||
|     dx1 *= -1 | ||||
|   } | ||||
|   if side-end == "east" { | ||||
|     dx2 *= -1 | ||||
|   } | ||||
|   p1 = (p0.x + dx1, p0.y) | ||||
|   p2 = (p3.x - dx2, p0.y) | ||||
|  | ||||
|   let points = ( | ||||
|     start, | ||||
|     (horizontal: p1, vertical: ()), | ||||
|     (horizontal: (), vertical: (0, dodge-y)), | ||||
|     (horizontal: p2, vertical: ()), | ||||
|     (horizontal: (), vertical: end), | ||||
|     end | ||||
|   ) | ||||
|   let anchors = ( | ||||
|     "start": start, | ||||
|     "start2": points.at(1), | ||||
|     "dodge-start": points.at(2), | ||||
|     "dodge-end": points.at(3), | ||||
|     "end2": points.at(4), | ||||
|     "end": end | ||||
|   ) | ||||
|  | ||||
|   return (points, anchors) | ||||
| } | ||||
|  | ||||
| #let wire( | ||||
|   id, pts, | ||||
|   bus: false, | ||||
|   name: none, | ||||
|   name-pos: "middle", | ||||
|   slice: none, | ||||
|   color: black, | ||||
|   dashed: false, | ||||
|   style: "direct", | ||||
|   reverse: false, | ||||
|   zigzag-ratio: 50%, | ||||
|   dodge-y: 0, | ||||
|   dodge-sides: ("east", "west"), | ||||
|   dodge-margins: (5%, 5%) | ||||
| ) = 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 stroke = ( | ||||
|     paint: color, | ||||
|     thickness: if bus {bus-width} else {signal-width} | ||||
|   ) | ||||
|   if dashed { | ||||
|     stroke.insert("dash", "dashed") | ||||
|   } | ||||
|  | ||||
|   let points = () | ||||
|   let anchors = () | ||||
|  | ||||
|   if style == "direct" { | ||||
|     (points, anchors) = get-direct-wire(pts) | ||||
|      | ||||
|   } else if style == "zigzag" { | ||||
|     (points, anchors) = get-zigzag-wire(pts, zigzag-ratio) | ||||
|      | ||||
|   } else if style == "dodge" { | ||||
|     (points, anchors) = get-dodge-wire( | ||||
|       pts, | ||||
|       dodge-y, | ||||
|       dodge-margins, | ||||
|       dodge-sides, | ||||
|       ctx | ||||
|     ) | ||||
|   } | ||||
|  | ||||
|   draw.group(name: id, { | ||||
|     draw.line(..points, stroke: stroke) | ||||
|     for (anchor-name, anchor-pos) in anchors { | ||||
|       draw.anchor(anchor-name, anchor-pos) | ||||
|     } | ||||
|   }) | ||||
|  | ||||
|   let first-pt = id + ".start" | ||||
|   let last-pt = id + ".end" | ||||
|   if reverse { | ||||
|     (first-pt, last-pt) = (last-pt, first-pt) | ||||
|   } | ||||
|    | ||||
|   if name != none { | ||||
|     let names = () | ||||
|      | ||||
|     if type(name) == str { | ||||
|       names = ((name, name-pos),) | ||||
|        | ||||
|     } else if type(name) == array { | ||||
|       names = ( | ||||
|         (name.at(0), "start"), | ||||
|         (name.at(1), "end") | ||||
|       ) | ||||
|     } | ||||
|  | ||||
|     for (name, pos) in names { | ||||
|       let point | ||||
|       let anchor | ||||
|        | ||||
|       if pos == "middle" { | ||||
|         point = (first-pt, 50%, last-pt) | ||||
|         anchor = "south" | ||||
|  | ||||
|       } else if pos == "start" { | ||||
|         point = first-pt | ||||
|         anchor = "south-west" | ||||
|  | ||||
|       } else if pos == "end" { | ||||
|         point = last-pt | ||||
|         anchor = "south-east" | ||||
|       } | ||||
|  | ||||
|       draw.content(point, anchor: anchor, padding: 3pt, name) | ||||
|     } | ||||
|   } | ||||
|  | ||||
|   if slice != none { | ||||
|     let (start, end) = slice | ||||
|     let slice-txt = "[" + str(start) + ":" + str(end) + "]" | ||||
|      | ||||
|     draw.content( | ||||
|       first-pt, | ||||
|       anchor: "south-west", | ||||
|       padding: 3pt, | ||||
|       text(slice-txt, size: 0.75em) | ||||
|     ) | ||||
|   } | ||||
| }) | ||||
|  | ||||
| #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 | ||||
|     ) | ||||
|   } | ||||
| } | ||||
		Reference in New Issue
	
	Block a user