solidlib/solidLib/primitives.py

84 lines
2.0 KiB
Python

from typing import List, Iterable
import solid as scad
from .globals import SEGMENTS
import numpy as np
import logging as log
log.basicConfig(level=log.INFO)
def round_box(x_dim: float, y_dim: float, z_dim: float, radius: float, segments: int = 32):
"""
creates a round box with radius r
Args:
x_dim (float): .
Important:
deprecated
.. image:: img/round_box.png
"""
box = scad.cube([x_dim - 2 * radius, y_dim - 2 * radius, z_dim - 2 * radius])
wall = scad.sphere(r=radius, segments=segments)
return scad.translate([radius, radius, radius])(
scad.minkowski()(box, wall)
)
def rcube(dim: Iterable[float], radius: float, segments: int = 32):
"""
creates a cube with rounded corners
.. image:: img/round_box.png
Args:
dim: cube dimensions to base box on.
radius: box radius.
segments: circular object segments
"""
corners = []
for dx in [radius, dim[0] - radius]:
for dy in [radius, dim[1] - radius]:
for dz in [radius, dim[2] - radius]:
corners.append(
scad.translate([dx, dy, dz])((
scad.sphere(r=radius)
))
)
return scad.hull()(*corners)
def round_flat_box(x: float, y: float, z: float, r: float, segments: int = 32):
"""creates a round box with radius r and flat top/bottom
.. image:: img/round_flat_box.png
"""
box = scad.cube([x - 2 * r, y - 2 * r, z- 2 * r])
wall = scad.cylinder(r=r, h=2*r, segments=SEGMENTS)
return scad.translate([r, r, 0])(
scad.minkowski()(box, wall)
)
def pill(ps: List[List[float]], r, segments: int = 36):
"""
create a pill around given points
"""
if len(ps) < 2:
raise Exception("requires 2 or more points")
s = scad.sphere(r=r, segments=segments)
spheres = []
# log.info(ps)
for p in ps:
spheres.append(
scad.translate(p)(s)
)
return scad.hull()(*spheres)