106 lines
3.0 KiB
Python
Executable File
106 lines
3.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
toplevel file used for interactive modeling.
|
|
|
|
- `cq-editor ./cq_toplevel.py`
|
|
- then press green play button to render
|
|
- edit files externally, and press render again to refresh the view
|
|
"""
|
|
|
|
import cadquery as cq
|
|
import argparse
|
|
import logging
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
sys.path.append(os.path.join(os.getcwd(), "src"))
|
|
|
|
import case
|
|
import pinephone
|
|
import ldtek_battery
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def svg_export_options(view: str):
|
|
proj = (1, 1, 1)
|
|
if view == "front":
|
|
proj = (0.00, -0.05, 0.10)
|
|
elif view == "back":
|
|
proj = (0.00, -0.05, -0.10)
|
|
elif view == "right":
|
|
proj = (-0.10, 0.00, 0.00)
|
|
return dict(
|
|
width = 1024,
|
|
height = 1024,
|
|
marginLeft = 100,
|
|
marginTop = 10,
|
|
showAxes = False,
|
|
# projectionDir controls both the angle and the distance from the camera to the model
|
|
projectionDir = proj,
|
|
strokeWidth = 0.25,
|
|
strokeColor = (255, 0, 0),
|
|
hiddenColor = (0, 0, 255),
|
|
showHidden = True,
|
|
)
|
|
|
|
def model():
|
|
render_phone = os.environ.get("CASE_RENDER_PHONE", "") not in ("", "0")
|
|
phone = pinephone.PinePhone()
|
|
battery = ldtek_battery.LdtekBattery()
|
|
return case.case(phone, battery=battery, render_phone=render_phone)
|
|
|
|
def main():
|
|
logging.basicConfig()
|
|
logging.getLogger().setLevel(logging.INFO)
|
|
|
|
parser = argparse.ArgumentParser(description="toplevel cadquery interface")
|
|
parser.add_argument("--render-phone", action="store_true", help="include the phone model itself in the stl; useful to confirm fit visually before printing")
|
|
parser.add_argument("--export-stl")
|
|
parser.add_argument("--export-svg")
|
|
parser.add_argument("--export-tjs")
|
|
parser.add_argument("--editor", action="store_true", help="view in cq-editor")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.render_phone:
|
|
os.environ["CASE_RENDER_PHONE"] = "1"
|
|
|
|
logger.info("computing model ...")
|
|
model_ = model()
|
|
|
|
if args.export_stl:
|
|
logger.info("exporting stl to %s", args.export_stl)
|
|
cq.exporters.export(model_, args.export_stl)
|
|
|
|
if args.export_svg:
|
|
view = None
|
|
if "front" in args.export_svg:
|
|
view = "front"
|
|
elif "back" in args.export_svg:
|
|
view = "back"
|
|
elif "right" in args.export_svg:
|
|
view = "right"
|
|
logger.info("exporting svg to %s (view: %s)", args.export_svg, str(view))
|
|
cq.exporters.export(model_, args.export_svg, opt=svg_export_options(view))
|
|
|
|
if args.export_tjs:
|
|
logger.info("exporting three.js to %s", args.export_tjs)
|
|
cq.exporters.export(
|
|
model_,
|
|
args.export_tjs,
|
|
tolerance=0.01,
|
|
angularTolerance=0.1,
|
|
exportType=cq.exporters.ExportTypes.TJS,
|
|
)
|
|
|
|
if args.editor:
|
|
logger.info("launching cq-editor")
|
|
subprocess.check_call(["cq-editor", __file__])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
else:
|
|
result = model()
|