add jst lib generator
This commit is contained in:
parent
48717f0d28
commit
667882f48e
148
generate_jst.py
Normal file
148
generate_jst.py
Normal file
|
|
@ -0,0 +1,148 @@
|
|||
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, 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(lib)
|
||||
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()
|
||||
Loading…
Reference in New Issue
Block a user