Qihuanye
130 lines
4.4 KiB
Python
130 lines
4.4 KiB
Python
import os
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
os.environ.setdefault("MPLCONFIGDIR", "/tmp/matplotlib")
|
|
|
|
import schemdraw
|
|
import schemdraw.elements as elm
|
|
from schemdraw.segments import SegmentText
|
|
|
|
FONT_NAME = "STIXGeneral"
|
|
BG_COLOR = "white"
|
|
DEFAULT_OUTPUT = Path(__file__).resolve().parent.parent / "image" / "target_circuit_measure.svg"
|
|
|
|
|
|
def add_text(
|
|
drawing,
|
|
position,
|
|
text,
|
|
*,
|
|
fontsize=18,
|
|
rotation=0,
|
|
align=("center", "center"),
|
|
):
|
|
element = elm.Element().at(position).anchor("center").hold()
|
|
element.anchors["center"] = (0, 0)
|
|
element.segments.append(
|
|
SegmentText(
|
|
(0, 0),
|
|
text,
|
|
rotation=rotation,
|
|
align=align,
|
|
fontsize=fontsize,
|
|
font=FONT_NAME,
|
|
)
|
|
)
|
|
drawing.add(element)
|
|
|
|
|
|
def draw_circuit(save_path=DEFAULT_OUTPUT):
|
|
save_path = Path(save_path)
|
|
save_path.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
with schemdraw.Drawing(show=False) as d:
|
|
d.config(
|
|
unit=1.0,
|
|
fontsize=18,
|
|
font=FONT_NAME,
|
|
lw=2.2,
|
|
margin=0.2,
|
|
bgcolor=BG_COLOR,
|
|
)
|
|
|
|
# 左侧原电路
|
|
left_node = (1.4, 4.0)
|
|
top_node = (5.2, 6.6)
|
|
bottom_node = (5.2, 1.9)
|
|
right_node = (9.3, 4.0)
|
|
bottom_left = (1.4, 0.8)
|
|
|
|
source_start = (4.8, 0.8)
|
|
source_len = 1.6
|
|
source_end = (source_start[0] + source_len, source_start[1])
|
|
|
|
d.add(elm.ResistorIEC().at(left_node).to(top_node).hold())
|
|
d.add(elm.ResistorIEC().at(left_node).to(bottom_node).hold())
|
|
d.add(elm.ResistorIEC().at(top_node).to(right_node).hold())
|
|
d.add(elm.Source().at(bottom_node).up(top_node[1] - bottom_node[1]).hold())
|
|
d.add(elm.ResistorIEC().at(bottom_left).up(left_node[1] - bottom_left[1]).hold())
|
|
|
|
d.add(elm.Line().at(bottom_left).right(source_start[0] - bottom_left[0]).hold())
|
|
d.add(elm.Source().at(source_start).right(source_len).hold())
|
|
d.add(elm.Line().at(source_end).right(right_node[0] - source_end[0]).hold())
|
|
|
|
d.add(elm.Line().at(right_node).down(right_node[1] - source_start[1]).hold())
|
|
|
|
# 左侧源符号细节
|
|
source_center_y = (top_node[1] + bottom_node[1]) / 2
|
|
d.add(elm.Line().at((4.74, source_center_y)).right(0.92).hold())
|
|
d.add(elm.Arrow().at((5.2, 4.9)).up(1.0).hold())
|
|
d.add(elm.Line().at(source_start).right(source_len).hold())
|
|
|
|
# 右侧测量电路，尽量使用 schemdraw 自带电表与可变电阻元件。
|
|
meter_top = (10.75, 4.0)
|
|
meter_bottom_y = bottom_node[1]
|
|
b_node = (14.2, bottom_node[1])
|
|
a_left = (12.15, meter_top[1])
|
|
a_node = (14.2, meter_top[1])
|
|
load_x = 15.8
|
|
|
|
d.add(elm.Line().at(right_node).right(meter_top[0] - right_node[0]).hold())
|
|
d.add(elm.MeterV().at(meter_top).down(meter_top[1] - meter_bottom_y).hold())
|
|
d.add(elm.Line().at(bottom_node).right(b_node[0] - bottom_node[0]).hold())
|
|
|
|
d.add(elm.MeterA().at(meter_top).right(a_left[0] - meter_top[0]).hold())
|
|
d.add(elm.Line().at(a_left).right(a_node[0] - a_left[0]).hold())
|
|
|
|
d.add(elm.Dot(open=True, radius=0.1).at(a_node).hold())
|
|
d.add(elm.Dot(open=True, radius=0.1).at(b_node).hold())
|
|
|
|
d.add(elm.Line().at(a_node).right(load_x - a_node[0]).hold())
|
|
d.add(elm.Line().at(b_node).right(load_x - b_node[0]).hold())
|
|
d.add(elm.ResistorVarIEC().at((load_x, b_node[1])).up(a_node[1] - b_node[1]).hold())
|
|
|
|
# 关键节点
|
|
for point in [left_node, top_node, bottom_node, right_node, (right_node[0], bottom_node[1])]:
|
|
d.add(elm.Dot(radius=0.085).at(point).hold())
|
|
|
|
# 文本标注
|
|
add_text(d, (2.35, 5.18), r"$R_1$", fontsize=19)
|
|
add_text(d, (8.78, 5.12), r"$R_3$", fontsize=19)
|
|
add_text(d, (2.42, 2.62), r"$R_2$", fontsize=19)
|
|
add_text(d, (0.55, 1.28), r"$R_4$", fontsize=19)
|
|
add_text(d, (6.42, 4.82), r"$I_S$", fontsize=19)
|
|
|
|
add_text(d, (4.3, 0.36), "-", fontsize=21)
|
|
add_text(d, (6.98, 0.36), "+", fontsize=21)
|
|
add_text(d, (4.78, -0.05), r"$U_S$", fontsize=19)
|
|
|
|
add_text(d, (13.35, 1.35), "B", fontsize=16)
|
|
add_text(d, (16.55, 3.0), r"$R_L$", fontsize=19)
|
|
d.save(str(save_path), transparent=False)
|
|
|
|
print(f"saved to: {save_path}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
output_path = sys.argv[1] if len(sys.argv) > 1 else DEFAULT_OUTPUT
|
|
draw_circuit(output_path)
|