renderer.cpython-312.pyc•7.96 kB
�
J��gD � �| � d dl Z d dlZd dlZd dlmZmZmZ d dlmZm Z m
Z
ej e � Z
G d� d� Zy)� N)�Dict�Any�Optional)�Image� ImageDraw� ImageFontc � � e Zd ZdZd� Zddedeeeef defd�Z ddedeeeef deeef fd�Z
dd ed
edefd �Z d eeef d edefd
�Z y)�Rendererz�
Handles rendering of OpenSCAD models to preview images.
Implements multi-angle views and fallback rendering when headless mode fails.
c �* � || _ ddddd�| _ y)z�
Initialize the renderer.
Args:
openscad_wrapper: Instance of OpenSCADWrapper for generating previews
z0,0,0,0,0,0,50z0,0,0,90,0,0,50z0,0,0,0,90,0,50z20,20,20,55,0,25,100)�front�top�right� perspectiveN)�openscad_wrapper�
camera_angles)�selfr s �>/home/ubuntu/openscad-mcp-server/src/visualization/renderer.py�__init__zRenderer.__init__ s$ � � !1�� � &�$�&�1�
��� N� scad_file�
parameters�returnc �d � | j j ||| j d d�� }t j j |� r$t j j
|� dkD r|S | j |� S # t $ r�}t j dt |� � �� t j j |� j d� d }t j j | j j |� d�� }| j |� cY d}~S d}~ww xY w) a
Generate a preview image for a SCAD file.
Args:
scad_file: Path to the SCAD file
parameters: Optional parameters to override in the SCAD file
Returns:
Path to the generated preview image
r �800,600��camera_position�
image_sizer zError generating preview: �.�.pngN)r �generate_previewr �os�path�exists�getsize�_create_placeholder_image� Exception�logger�error�str�basename�split�join� preview_dir)r r r � preview_file�e�model_ids r r zRenderer.generate_preview s � � @� �0�0�A�A��� $� 2� 2�=� A�$� B� �L� �w�w�~�~�l�+������ �0M�PQ�0Q�#�#� �5�5�l�C�C��� @� �L�L�5�c�!�f�X�>� ?��w�w�'�'� �2�8�8��=�a�@�H� �7�7�<�<��(=�(=�(I�(I�h�Z�W[�K\�]�L��1�1�,�?� ?�� @�s% �A.B �1B � D/� BD*�$D/�*D/c
� � i }t j j |� j d� d }| j j � D ]� \ }}t j j
| j j |� d|� d�� } | j j |||d�� }t j j |� r(t j j |� dkD r|||< n| j ||� ||< �� |S # t $ rD}t j d|� dt! |� � �� | j ||� ||< Y d }~��d }~ww xY w)
a,
Generate preview images from multiple angles.
Args:
scad_file: Path to the SCAD file
parameters: Optional parameters to override in the SCAD file
Returns:
Dictionary mapping angle names to preview image paths
r r �_r r r zError generating �
preview: N)r! r" r* r+ r �itemsr, r r- r r# r$ r% r& r' r( r) ) r r r �previewsr0 �
angle_namer r. r/ s r � generate_multi_angle_previewsz&Renderer.generate_multi_angle_previewsA sH � � ���7�7�#�#�I�.�4�4�S�9�!�<��+/�+=�+=�+C�+C�+E� '�J�� �7�7�<�<��%�%�1�1��*�A�j�\��.� �L�
`� #� 4� 4� E� E� � �$3� (� F� � � �7�7�>�>�,�/�B�G�G�O�O�L�4Q�TU�4U�+7�H�Z�(� ,0�+I�+I�,�Xb�+c�H�Z�(��) ,F�4 ��� �
`�� � � 0�� �J�s�1�v�h� O�P�'+�'E�'E�l�T^�'_���$��
`�s �A;D� E�9E�E� output_pathr6 c �| � t j ddd�� }t j |� }|j ddd�� |j d d
|� �d�� |j |� t j d |� �� |S # t $ r-}t j d t |� � �� |cY d
}~S d
}~ww xY w)a3
Create a placeholder image when OpenSCAD rendering fails.
Args:
output_path: Path to save the placeholder image
angle_name: Name of the camera angle for the placeholder
Returns:
Path to the created placeholder image
�RGB�� �X ��� r? r? ��color)� i zPreview not available)r r r )�fill)rB i@ zView: zCreated placeholder image: z"Error creating placeholder image: N) r �newr �Draw�text�saver' �infor& r( r) )r r8 r6 �img�drawr/ s r r% z"Renderer._create_placeholder_imagek s� � � ��)�)�E�:�_�E�C� �>�>�#�&�D�
�I�I�j�$9��I� K� �I�I�j�F�:�,�"7�i�I� H�
�H�H�[� !� �K�K�5�k�]�C� D� � ��� � �L�L�=�c�!�f�X�F� G� � �� �s �BB � B;�"B6�0B;�6B;r5 c
�p � t j ddd�� }ddddd �}|j � D ]f \ }}||v s� t j j |� s�+ t j |� }|j d� }|j ||| � �h |j |� t j d
|� �� |S # t $ r.}t j d
|� d t |� � �� Y d }~��d }~ww xY w# t $ r-}t j dt |� � �� |cY d }~S d }~ww xY w)a3
Create a composite image from multiple angle previews.
Args:
previews: Dictionary mapping angle names to preview image paths
output_path: Path to save the composite image
Returns:
Path to the created composite image
r: )i@ i� r> r@ )r r )r< r )r r= r; )r r r
r zError processing r3 NzCreated composite preview: z"Error creating composite preview: )r rD r4 r! r" r# �open�resize�paster& r' r( r) rG rH ) r r5 r8 rI � positionsr6 � preview_path� angle_imgr/ s r �create_composite_previewz!Renderer.create_composite_preview� s! � � ��)�)�E�<��G�C� &�!� �#� �I� -5�N�N�,<�(�
�L� ��*�r�w�w�~�~�l�/K�Y�$)�J�J�|�$<� �$-�$4�$4�Z�$@� �� � �)�Y�z�-B�C� -=�
�H�H�[� !� �K�K�5�k�]�C� D� � ��
%� Y� � � �'8�� �J�s�ST�v�h�%W�X�X��Y�� � � �L�L�=�c�!�f�X�F� G� � �� �sL �9C? � C? � ;C�,C? � C<�$C7�2C? �7C<�<C? �? D5�"D0�*D5�0D5)N)r )
�__name__�
__module__� __qualname__�__doc__r r) r r r r r7 r% rR � r r r
r
s� � ��
�" @�#� @�8�D��c��N�;S� @�_b� @�B(�s� (��QU�VY�[^�V^�Q_�H`� (�lp�qt�vy�qy�lz� (�T �S� �c� �^a� �:) ��c�3�h� � ) �c� ) �VY� ) r r
)r! �
subprocess�logging�typingr r r �PILr r r � getLoggerrS r' r
rW r r �<module>r] s8 �� � � � &� &� +� +� �� � �8� $��h � h r