从图纸开始，用 AI 建出转子发动机模型

"""教学材料的可复跑构建脚本；参数、轮廓和输出分离。"""
from pathlib import Path
import argparse, json, math
from build123d import Vector, Wire, Face, Spline, extrude, export_step, import_step, export_stl, Compound, Pos

p=argparse.ArgumentParser()
p.add_argument('--materials', type=Path, default=Path(__file__).resolve().parent)
p.add_argument('--output',type=Path,required=True)
p.add_argument('--plate-depth',type=float)
a=p.parse_args(); a.output.mkdir(parents=True,exist_ok=True)
d=json.loads((a.materials/'dimensions.json').read_text())
c=json.loads((a.materials/'contours.json').read_text())
assert d['units']==c['units']=='mm'
pd=a.plate_depth if a.plate_depth is not None else d['side_plate']['depth']
assert 0<pd<40
def curve(key):
    return Wire([Spline(*[Vector(x,y,0) for x,y in c[key][::4]],periodic=True)])
def annulus(outer,inner,depth):
    return extrude(Face(Wire.make_circle(outer/2),[Wire.make_circle(inner/2)]),amount=depth)
rotor=extrude(Face(curve('rotor'),[Wire.make_circle(d['rotor']['bore_diameter']/2)]),amount=d['rotor']['depth'])
housing=extrude(Face(curve('outer'),[curve('housing')]),amount=d['housing']['depth'])
plate=extrude(Face(curve('flange'),[Wire.make_circle(d['side_plate']['bore_diameter']/2)]),amount=pd)
bearing=annulus(d['bearing_sleeve']['outer_diameter'],d['bearing_sleeve']['bore_diameter'],d['bearing_sleeve']['depth'])
parts={'rotor-blank':rotor,'trochoid-housing':housing,'side-plate-blank':plate,'bearing-sleeve':bearing}
rows=[]
for name,part in parts.items():
    export_step(part,a.output/f'{name}.step'); back=import_step(a.output/f'{name}.step')
    export_stl(part,a.output/f'{name}.stl',tolerance=.05,angular_tolerance=.1)
    dim=part.bounding_box().size
    row={'name':name,'valid':bool(part.is_valid),'solids':len(part.solids()),'readback_valid':bool(back.is_valid),'readback_solids':len(back.solids()),'depth_mm':dim.Z,'volume_mm3':part.volume,'roundtrip_volume_difference_mm3':abs(part.volume-back.volume)}
    assert row['valid'] and row['readback_valid'] and row['solids']==row['readback_solids']==1 and row['roundtrip_volume_difference_mm3']<.001
    rows.append(row)
h=d['housing']['depth']/2;r=d['rotor']['depth']/2;e=d['assembly']['eccentric_offset_x']
placements=[{'name':'housing','part':'trochoid-housing','xyz':[0,0,-h]}, {'name':'rotor','part':'rotor-blank','xyz':[e,0,-r]}, {'name':'rear-plate','part':'side-plate-blank','xyz':[0,0,-h-pd]}, {'name':'front-plate','part':'side-plate-blank','xyz':[0,0,h]}, {'name':'outer-sleeve','part':'bearing-sleeve','xyz':[0,0,h+pd]}]
assembly=Compound(children=[Pos(*i['xyz'])*parts[i['part']] for i in placements]);export_step(assembly,a.output/'core-assembly.step')
back=import_step(a.output/'core-assembly.step');assert len(back.solids())==5 and back.is_valid
report={'identity':'本次教学补绘材料的实际脚本执行；非历史智能体回复或制造验收','units':'mm','plate_depth_mm':pd,'parts':rows,'placements':placements,'assembly_readback_solids':len(back.solids()),'assembly_readback_valid':bool(back.is_valid),'scope':'四类核心实体和静态装配；未包含运动、齿轮、密封与外观装饰','passed':True}
(a.output/'result.json').write_text(json.dumps(report,ensure_ascii=False,indent=2)+'\n')
print(json.dumps(report,ensure_ascii=False,indent=2))