add arrow on stub
This commit is contained in:
		
							
								
								
									
										108
									
								
								src/wire.typ
									
									
									
									
									
								
							
							
						
						
									
										108
									
								
								src/wire.typ
									
									
									
									
									
								
							| @@ -19,7 +19,7 @@ | ||||
| #let get-direct-wire(pts) = { | ||||
|   let anchors = ( | ||||
|     "start": pts.first(), | ||||
|     "end": pts.last() | ||||
|     "end": pts.last(), | ||||
|   ) | ||||
|   return (pts, anchors) | ||||
| } | ||||
| @@ -38,21 +38,21 @@ | ||||
|       start, | ||||
|       (horizontal: mid, vertical: ()), | ||||
|       (horizontal: (), vertical: end), | ||||
|       end | ||||
|       end, | ||||
|     ) | ||||
|   } else { | ||||
|     ( | ||||
|       start, | ||||
|       (horizontal: (), vertical: mid), | ||||
|       (horizontal: end, vertical: ()), | ||||
|       end | ||||
|       end, | ||||
|     ) | ||||
|   } | ||||
|   let anchors = ( | ||||
|     "start": start, | ||||
|     "zig": points.at(1), | ||||
|     "zag": points.at(2), | ||||
|     "end": end | ||||
|     "end": end, | ||||
|   ) | ||||
|   return (points, anchors) | ||||
| } | ||||
| @@ -65,7 +65,7 @@ | ||||
|  | ||||
|   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()) | ||||
| @@ -73,7 +73,7 @@ | ||||
|  | ||||
|   let dx1 = margin-start | ||||
|   let dx2 = margin-end | ||||
|    | ||||
|  | ||||
|   if type(margin-start) == ratio { | ||||
|     dx1 = calc.abs(p3.x - p0.x) * margin-start / 100% | ||||
|   } | ||||
| @@ -95,7 +95,7 @@ | ||||
|     (horizontal: (), vertical: (0, dodge-y)), | ||||
|     (horizontal: p2, vertical: ()), | ||||
|     (horizontal: (), vertical: end), | ||||
|     end | ||||
|     end, | ||||
|   ) | ||||
|   let anchors = ( | ||||
|     "start": start, | ||||
| @@ -103,7 +103,7 @@ | ||||
|     "dodge-start": points.at(2), | ||||
|     "dodge-end": points.at(3), | ||||
|     "end2": points.at(4), | ||||
|     "end": end | ||||
|     "end": end, | ||||
|   ) | ||||
|  | ||||
|   return (points, anchors) | ||||
| @@ -128,7 +128,8 @@ | ||||
| /// - 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, | ||||
|   id, | ||||
|   pts, | ||||
|   bus: false, | ||||
|   name: none, | ||||
|   name-pos: "middle", | ||||
| @@ -143,7 +144,7 @@ | ||||
|   zigzag-dir: "vertical", | ||||
|   dodge-y: 0, | ||||
|   dodge-sides: ("east", "west"), | ||||
|   dodge-margins: (5%, 5%) | ||||
|   dodge-margins: (5%, 5%), | ||||
| ) = draw.get-ctx(ctx => { | ||||
|   if not style in wire-styles { | ||||
|     panic("Invalid wire style '" + style + "'") | ||||
| @@ -152,10 +153,14 @@ | ||||
|   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} | ||||
|     thickness: if bus { | ||||
|       bus-width | ||||
|     } else { | ||||
|       signal-width | ||||
|     }, | ||||
|   ) | ||||
|   if dashed { | ||||
|     stroke.insert("dash", "dashed") | ||||
| @@ -166,17 +171,17 @@ | ||||
|  | ||||
|   if style == "direct" { | ||||
|     (points, anchors) = get-direct-wire(pts) | ||||
|      | ||||
|  | ||||
|   } else if style == "zigzag" { | ||||
|     (points, anchors) = get-zigzag-wire(pts, zigzag-ratio, zigzag-dir) | ||||
|      | ||||
|  | ||||
|   } else if style == "dodge" { | ||||
|     (points, anchors) = get-dodge-wire( | ||||
|       pts, | ||||
|       dodge-y, | ||||
|       dodge-margins, | ||||
|       dodge-sides, | ||||
|       ctx | ||||
|       ctx, | ||||
|     ) | ||||
|   } | ||||
|  | ||||
| @@ -184,12 +189,15 @@ | ||||
|   if directed { | ||||
|     mark = (end: ">", fill: color) | ||||
|   } | ||||
|   draw.group(name: id, { | ||||
|     draw.line(..points, stroke: stroke, mark: mark) | ||||
|     for (anchor-name, anchor-pos) in anchors { | ||||
|       draw.anchor(anchor-name, anchor-pos) | ||||
|     } | ||||
|   }) | ||||
|   draw.group( | ||||
|     name: id, | ||||
|     { | ||||
|       draw.line(..points, stroke: stroke, mark: mark) | ||||
|       for (anchor-name, anchor-pos) in anchors { | ||||
|         draw.anchor(anchor-name, anchor-pos) | ||||
|       } | ||||
|     }, | ||||
|   ) | ||||
|  | ||||
|   let first-pt = id + ".start" | ||||
|   let last-pt = id + ".end" | ||||
| @@ -198,12 +206,12 @@ | ||||
|   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) | ||||
|     } | ||||
| @@ -211,24 +219,24 @@ | ||||
|     let (x2, y2, ..) = second-pos | ||||
|     angle = calc.atan2(x2 - x1, y2 - y1) | ||||
|   } | ||||
|    | ||||
|  | ||||
|   if name != none { | ||||
|     let names = () | ||||
|      | ||||
|     if type(name) == str { | ||||
|  | ||||
|     if type(name) == str or type(name) == content { | ||||
|       names = ((name, name-pos),) | ||||
|        | ||||
|  | ||||
|     } else if type(name) == array { | ||||
|       names = ( | ||||
|         (name.at(0), "start"), | ||||
|         (name.at(1), "end") | ||||
|         (name.at(1), "end"), | ||||
|       ) | ||||
|     } | ||||
|  | ||||
|     for (name, pos) in names { | ||||
|       let point | ||||
|       let anchor | ||||
|        | ||||
|  | ||||
|       if pos == "middle" { | ||||
|         point = (first-pt, 50%, last-pt) | ||||
|         anchor = "south" | ||||
| @@ -248,12 +256,12 @@ | ||||
|  | ||||
|   if slice != none { | ||||
|     let slice-txt = "[" + slice.map(b => str(b)).join(":") + "]" | ||||
|      | ||||
|  | ||||
|     draw.content( | ||||
|       first-pt, | ||||
|       anchor: "south-west", | ||||
|       padding: 3pt, | ||||
|       text(slice-txt, size: 0.75em) | ||||
|       text(slice-txt, size: 0.75em), | ||||
|     ) | ||||
|   } | ||||
| }) | ||||
| @@ -267,24 +275,38 @@ | ||||
| /// - 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(port-id, side, name: none, vertical: false, length: 1em, name-offset: 0) = { | ||||
| #let stub( | ||||
|   port-id, | ||||
|   side, | ||||
|   name: none, | ||||
|   vertical: false, | ||||
|   length: 1em, | ||||
|   name-offset: 0, | ||||
|   color: black, | ||||
|   directed: none, | ||||
| ) = { | ||||
|   let end-offset = ( | ||||
|     north: (0, length), | ||||
|     east: (length, 0), | ||||
|     south: (0, -length), | ||||
|     west: (-length, 0) | ||||
|     west: (-length, 0), | ||||
|   ).at(side) | ||||
|    | ||||
|  | ||||
|   let name-offset = ( | ||||
|     north: (name-offset, length), | ||||
|     east: (length, name-offset), | ||||
|     south: (name-offset, -length), | ||||
|     west: (-length, name-offset) | ||||
|     west: (-length, name-offset), | ||||
|   ).at(side) | ||||
|  | ||||
|   let mark = (fill: color) | ||||
|   if directed != none { | ||||
|     mark = ((directed): ">", fill: color) | ||||
|   } | ||||
|   draw.line( | ||||
|     port-id, | ||||
|     (rel: end-offset, to: port-id) | ||||
|     (rel: end-offset, to: port-id), | ||||
|     mark: mark, | ||||
|   ) | ||||
|   if name != none { | ||||
|     let text-anchor = if vertical { | ||||
| @@ -292,15 +314,21 @@ | ||||
|         "north": "west", | ||||
|         "south": "east", | ||||
|         "west": "south", | ||||
|         "east": "north" | ||||
|         "east": "north", | ||||
|       ).at(side) | ||||
|     } else { opposite-anchor(side) } | ||||
|     } else { | ||||
|       opposite-anchor(side) | ||||
|     } | ||||
|     draw.content( | ||||
|       anchor: text-anchor, | ||||
|       padding: 0.2em, | ||||
|       angle: if vertical {90deg} else {0deg}, | ||||
|       angle: if vertical { | ||||
|         90deg | ||||
|       } else { | ||||
|         0deg | ||||
|       }, | ||||
|       (rel: name-offset, to: port-id), | ||||
|       name | ||||
|       name, | ||||
|     ) | ||||
|   } | ||||
| } | ||||
		Reference in New Issue
	
	Block a user