From 74369b0282bc775d9d9750644041e5db403acc2c Mon Sep 17 00:00:00 2001
From: "U. Bruhin" <urbibruhin@bluewin.ch>
Date: Sun, 8 Oct 2023 22:51:39 +0200
Subject: [PATCH] Add script to generate 3D models for TO-92

---
 .gitignore       |  1 +
 generate_to92.py | 84 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 85 insertions(+)
 create mode 100644 generate_to92.py

diff --git a/.gitignore b/.gitignore
index ed88933..7f04892 100644
--- a/.gitignore
+++ b/.gitignore
@@ -4,4 +4,5 @@ __pycache__/
 .mypy_cache/
 .pytest_cache/
 out/
+out_3d/
 *.pdf
diff --git a/generate_to92.py b/generate_to92.py
new file mode 100644
index 0000000..6f8620d
--- /dev/null
+++ b/generate_to92.py
@@ -0,0 +1,84 @@
+"""
+Generate only the 3D models for TO-92
+"""
+from os import path
+
+import cadquery as cq
+
+from cadquery_helpers import StepAssembly, StepColor, StepConstants
+
+
+def generate_leg_path(plane: str, length: float, delta: float) -> cq.Workplane:
+    straight_length = 1
+    return cq.Workplane(plane) \
+        .vLine(length - straight_length - abs(delta)) \
+        .line(-delta, abs(delta)) \
+        .vLine(straight_length)
+
+
+def generate(
+    name: str,
+    pitch_x: float,
+    pitch_y: float,
+) -> None:
+    print(f'Generating pkg 3D model "{name}"')
+
+    body_standoff = 3.0
+    body_height = 5.0
+    body_diameter = 5.1
+    body_edge = -1.65
+    leg_width = 0.45
+    leg_z = -StepConstants.THT_LEAD_SOLDER_LENGTH
+    leg_length = StepConstants.THT_LEAD_SOLDER_LENGTH + body_standoff + 0.1
+
+    body = cq.Workplane('XY', origin=(0, 0, body_standoff)) \
+        .cylinder(body_height, body_diameter / 2, centered=(True, True, False)) \
+        .fillet(0.3) \
+        .moveTo(0, body_edge - 10) \
+        .box(20, 10, 50, centered=(True, False, False), combine='cut')
+
+    leg_straight = cq.Workplane('XY', origin=(0, 0, leg_z)) \
+        .box(leg_width, leg_width, leg_length, centered=(True, True, False))
+    if pitch_x == 1.27:
+        leg_1 = leg_straight
+        leg_3 = leg_straight
+    else:
+        leg_1 = cq.Workplane('XY') \
+            .rect(leg_width, leg_width) \
+            .sweep(generate_leg_path('XZ', leg_length, -pitch_x + 1.27)) \
+            .translate((-pitch_x + 1.27, 0, leg_z))
+        leg_3 = cq.Workplane('XY') \
+            .rect(leg_width, leg_width) \
+            .sweep(generate_leg_path('XZ', leg_length, pitch_x - 1.27)) \
+            .translate((pitch_x - 1.27, 0, leg_z))
+    if pitch_y == 0:
+        leg_2 = leg_straight
+    else:
+        leg_2 = cq.Workplane('XY') \
+            .rect(leg_width, leg_width) \
+            .sweep(generate_leg_path('YZ', leg_length, pitch_y)) \
+            .translate((0, pitch_y, leg_z))
+
+    assembly = StepAssembly(name)
+    assembly.add_body(body, 'body', StepColor.IC_BODY)
+    assembly.add_body(
+        leg_1, 'leg-1', StepColor.LEAD_SMT,
+        location=cq.Location((-1.27, 0, 0))
+    )
+    assembly.add_body(
+        leg_2, 'leg-2', StepColor.LEAD_SMT,
+        location=cq.Location((0, 0, 0))
+    )
+    assembly.add_body(
+        leg_3, 'leg-3', StepColor.LEAD_SMT,
+        location=cq.Location((1.27, 0, 0))
+    )
+
+    out_path = path.join('out_3d', 'to92', f'{name}.step')
+    assembly.save(out_path, fused=True)
+
+
+if __name__ == '__main__':
+    generate(name='Straight', pitch_x=1.27, pitch_y=0)
+    generate(name='Zigzag', pitch_x=1.27, pitch_y=1.27)
+    generate(name='Wide', pitch_x=2.54, pitch_y=0)