bklronin
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
import os, sys
|
|
os.environ["QT_QPA_PLATFORM"] = "offscreen"
|
|
sys.path.insert(0, "/Volumes/Data_drive/Programming/fluency/src")
|
|
|
|
from PySide6.QtWidgets import QApplication
|
|
from PySide6.QtGui import QPixmap, QPainter, QColor
|
|
from PySide6.QtCore import QRectF
|
|
import math
|
|
|
|
app = QApplication.instance() or QApplication([])
|
|
|
|
from fluency.geometry.base import Point2D
|
|
from fluency.geometry_occ.kernel import OCGeometryKernel
|
|
from fluency.models.data_model import Body, Component, Project, DrawingView, TechnicalDrawing
|
|
from fluency.technical_drawing import generate_drawing, render_drawing, _A3_WIDTH_MM, _A3_HEIGHT_MM
|
|
|
|
kernel = OCGeometryKernel()
|
|
# Long thin bar: 120 x 25 x 30 (matches the "wide bar" screenshot case).
|
|
points = [Point2D(0, 0), Point2D(120, 0), Point2D(120, 25), Point2D(0, 25)]
|
|
box = kernel.extrude(kernel.create_polygon(points), 30.0)
|
|
body = Body(name="Bar", geometry=box)
|
|
comp = Component(name="BarComp")
|
|
comp.bodies[body.id] = body
|
|
project = Project()
|
|
project.components[comp.id] = comp
|
|
project.active_component = comp.id
|
|
|
|
W = 2400
|
|
H = int(W * _A3_HEIGHT_MM / _A3_WIDTH_MM)
|
|
pm = QPixmap(W, H)
|
|
pm.fill(QColor(255, 255, 255))
|
|
|
|
for name, kinds in [
|
|
("four", ["front", "top", "right", "back"]),
|
|
("six", ["front", "top", "right", "left", "back", "bottom"]),
|
|
("sixiso", ["front", "top", "right", "left", "back", "bottom", "isometric"]),
|
|
]:
|
|
drawing = TechnicalDrawing(
|
|
source_kind="component", source_id=comp.id,
|
|
views=[DrawingView(kind=k) for k in kinds],
|
|
auto_dimensions=True, title=name,
|
|
)
|
|
result = generate_drawing(drawing, project, kernel)
|
|
p = QPainter(pm)
|
|
render_drawing(p, result, QRectF(0, 0, W, H))
|
|
p.end()
|
|
out = f"/tmp/drawing_{name}.png"
|
|
pm.save(out)
|
|
print(name, "saved", out, "prims", len(result.primitives), "scale",
|
|
round(result.view_transforms.get("front", (None,))[0] or 0, 3))
|