kicad/generate_jst.py

149 lines
5.0 KiB
Python

import time
import zipfile
from pathlib import Path
from tempfile import TemporaryDirectory
from typing import Literal, Optional, Union
import numpy as np
import regex as re
from selenium import webdriver
from selenium.webdriver.support.select import By
from fp_generator import INCH, Footprint
dir_ = Path(__file__).parent
PARTS = [
# JST-SH, Right angle
*[f"SM{i:02d}B-SRSS-TB" for i in range(2, 16)],
# JST-SH, Vertical
# *[f"BM{i:02d}B-SRSS-TB" for i in range(2, 16)],
]
def round2grid(val: float, direction: Optional[Literal["up", "down"]] = None, grid: float = 0.005 * INCH):
if direction == "up":
val = val + grid / 2
elif direction == "down":
val = val - grid / 2
return np.round(val / grid) * grid
def download_step(part: Union[str, int], headless: bool = True) -> None:
print(part)
with TemporaryDirectory(prefix=str(dir_ / "tmp") + "/") as tmp:
tmp = Path(tmp)
chrome_options = webdriver.ChromeOptions()
chrome_options.add_experimental_option(
"prefs",
{"download.default_directory": str(tmp)},
)
if headless:
chrome_options.add_argument("--headless")
with webdriver.Chrome(
options=chrome_options,
) as browser:
browser.get(
f"https://www.jst-mfg.com/product/index.php?type=10&series=231&doc=2&filename={part.upper()}.zip"
)
browser.implicitly_wait(10)
browser.find_element(by=By.ID, value="company").send_keys("N/A")
browser.find_element(by=By.ID, value="section").send_keys("N/A")
browser.find_element(by=By.ID, value="name").send_keys("Brendan Haines")
browser.find_element(by=By.ID, value="address").send_keys("7415 Spy Glass Ct")
browser.find_element(by=By.ID, value="tel").send_keys("(505) 738-6174")
browser.find_element(by=By.ID, value="email").send_keys("brendan.haines@gmail.com")
# browser.find_element(by=By.ID, value="url").send_keys("")
browser.find_element(by=By.XPATH, value="//input[@type='button' and @value='I agree']").click()
# wait for download to finish
while len(list((Path(tmp).glob("*.zip")))) == 0:
time.sleep(0.1)
# print(list((Path(tmp).glob("*.zip"))))
# unzip
with zipfile.ZipFile(tmp / f"{part.upper()}.zip") as zip:
zip.extractall(path=tmp)
# move
(Path(tmp) / f"{part.upper()}.STEP").rename(dir_ / "jst.pretty" / "3dshapes" / f"{part.upper()}.stp")
def footprint(part: str, lib: Optional[Union[Path, str]] = None):
print(part)
if lib is None:
lib = dir_ / "jst.pretty"
lib = Path(lib)
if part.startswith("SM"):
# Right angle
match = re.match(r"SM(\d\d)B-SRSS-TB", part)
pins = int(match[1])
fp = Footprint(part, lib=lib, description=f"connector, JST-SH, 1x{pins:02f}, RA")
fp.add_text("reference", "REF**", (5.55 + (0.025 + 0.005) * INCH), 0, rotation=90, layer="F.SilkS")
fp.add_text("value", "VAL**", 4 * 0.025 * INCH, 0, rotation=90, layer="F.Fab")
fp.add_text("user", r"${REFERENCE}**", 2 * 0.025 * INCH, 0, rotation=90, layer="F.Fab")
a, b = {
2: (1, 4),
3: (2, 5),
4: (3, 6),
5: (4, 7),
6: (5, 8),
7: (6, 9),
8: (7, 10),
9: (8, 11),
10: (9, 12),
11: (10, 13),
12: (11, 14),
13: (12, 15),
14: (13, 16),
15: (14, 17),
20: (19, 22),
}[pins]
# fp.add_line((0.05 * INCH, -0.05 * INCH), (0.05 * INCH, 0.05 * INCH), layer="F.SilkS")
for pad in range(pins):
fp.add_pad(
pad + 1,
4 + (5.55 - 4) / 2,
-1.0 * (pins - 1) / 2 + 1.0 * pad,
size=(5.55 - 4, 0.6),
shape="rect",
)
# mechanical holes
for s in [1, -1]:
fp.add_pad(0, 1.8 / 2, s * (1.0 * (pins - 1) / 2 + 0.7 + 1.2 / 2), size=(1.8, 1.2), shape="rect")
fp.add_model(
lib / "3dshapes" / f"{part}.stp",
(4, 0, -0.3),
rotate=(-90, 0, 90),
)
fp.add_rect((0.2, -b / 2), (0.2 + 4.25, b / 2), layer="F.Fab")
for s in [1, -1]:
fp.add_line(
(0.190 * INCH, s * round2grid((a + 0.6) / 2 + 0.010 * INCH, "up")),
(0.190 * INCH, s * round2grid(b / 2 + 0.010 * INCH, "up")),
layer="F.SilkS",
)
fp.add_line(
(0.190 * INCH, s * round2grid(b / 2 + 0.010 * INCH, "up")),
(round2grid(1.8 + 0.010 * INCH), s * round2grid(b / 2 + 0.010 * INCH, "up")),
layer="F.SilkS",
)
fp.write()
else:
raise NotImplementedError
if __name__ == "__main__":
from multiprocessing import Pool
pool = Pool(10)
# pool.map(download_step, PARTS)
for part in PARTS:
footprint(part)
pool.close()