code_generator.cpython-312.pyc•9.53 kB
�
���g�% � �p � d dl Z d dlZd dlZd dlmZmZmZmZmZ ej e
� Z G d� d� Zy)� N)�Dict�Any�List�Tuple�Optionalc � � e Zd ZdZdedefd�Zdedeeef defd�Zd edeeef defd
�Z de
eeef defd�Zdeeef deeef fd
�Zdedeeef defd�Z
y)�OpenSCADCodeGeneratorz�
Generates OpenSCAD code from natural language descriptions and parameters.
Implements translation of requirements to OpenSCAD primitives and modules.
�scad_templates_dir�
output_dirc � � || _ || _ t j |d�� ddddddd d
ddd
ddd�
| _ dddddddddddddd�
| _ y) z�
Initialize the code generator.
Args:
scad_templates_dir: Directory containing SCAD template files
output_dir: Directory to store generated SCAD files
T)�exist_ok�parametric_cube�parametric_sphere�parametric_cylinder�
hollow_box�rounded_box�rounded_container�tube�cone�wedge�rounded_cylinder�torus�hexagonal_prism�text_3d)
�cube�sphere�cylinder�boxr � containerr r r r r r �text�width�depth�height�radius� thickness�segments�center�inner_radius�outer_radius�
corner_radiusr �size�font)
r! r"