:vj d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlm Z ddl
m Z ddl m
Z
mZmZmZmZmZ ddlmZ ddlmZ G d d
Zd Z G d d
Zedk( rvd dlZd dl Z g Z e e j> dkD r4e j> d Z e e j> dkD re j> dd Z ndZ ee e du de Z! ejD d y) N)Image)HTML )Path) SetObject SetTransformDelete SetProperty SetAnimation CaptureImage)MeshPhongMaterial) start_zmq_server_as_subprocessc ^ e Zd Z ej Zd Zd Zd Zd Z d Z
d Z d Z d Z
y )
ViewerWindowc |r!t || \ | _ | _ | _ nd | _ || _ | j |s%| j | _ | j t
d t
| j y )N)zmq_url server_argsz:You can open the visualizer by visiting the following URL:)r server_procr web_url connect_zmqrequest_web_urlprint)selfr start_serverr s c/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/meshcat/visualizer.py__init__zViewerWindow.__init__ su ;Y [<: 8D dlDL $D "DL //1DL
JK
dll c | j j t j | _ | j j | j y N)contextsocketzmqREQ
zmq_socketconnectr r s r r zViewerWindow.connect_zmq' s2 ,,--cgg6 -r c | j j d | j j j d }|S )Ns urlutf-8r$ sendrecvdecode)r responses r r z ViewerWindow.request_web_url+ s6 V$??'')009r c H t j | j d | S )N )new)
webbrowseropenr r& s r r2 zViewerWindow.open0 s !, r c | j j d | j j j d S )Ns waitr( r) r& s r waitzViewerWindow.wait4 s2 W%##%,,W55r c |j }| j j |d j d |d j d t j
| g | j j
y )Ntyper( path)lowerr$ send_multipartencodeumsgpackpackbr+ )r commandcmd_datas r r* zViewerWindow.send8 sh ==? && V # #G , V # #G , NN8 $(
r c | j j d | j j j d S )z(Get the static HTML from the ZMQ server.s get_scener( r) r& s r get_scenezViewerWindow.get_sceneA s2 \*##%,,W55r c ^ t j }| j j |d j d dj d t j | g | j j }t j t j | }|S )Nr6 r( ) r r8 r$ r9 r: r; r< r+ r r2 ioBytesIO)r r> img_bytesimgs r get_imagezViewerWindow.get_imageG s >'') && V # #G , IIg NN8 $(
OO((* jjI./
r N)__name__
__module__ __qualname__r" Contextr r r r r2 r4 r* r@ rG r r r r s7 ckkmG *.
6 6 r r c F | j dd j dd S )N&z&"z")replace)xs r
srcdoc_escaperR S s 99S' " * *3 99r c e Zd ZddgZddg fdZed Zd Zd Zd Z dd Z
dd
Z d Z dd Z
ej d
fdZd ZddZd Zd Zd Zd Zd Zy)
Visualizerwindowr7 Nc b |t ||d u | | _ n|| _ t d | _ y )N)r r r )meshcat)r rU r r7 )r r rU r s r r zVisualizer.__init__Z s0 >&wgQUodopDK DK & r c , t | }||_ |S )N)rU )rT r7 )rU r7 viss r view_intozVisualizer.view_intoa s '
r c : | j j | S r )rU r2 r& s r r2 zVisualizer.openg s r c . | j j S r )rU r r& s r urlzVisualizer.urlk s {{"""r c 6 | j j S )zB
Block until a browser is connected to the server
)rU r4 r& s r r4 zVisualizer.waitn s {{ !!r c V t dj | j | S )z
Render the visualizer in a jupyter notebook or jupyterlab cell.
For this to work, it should be the very last command in the given jupyter
cell.
z
)r] height)r formatr] r r` s r jupyter_cellzVisualizer.jupyter_cellt s, 488:f5 7 7r c h t dj t | j | S )a
Render a static snapshot of the visualizer in a jupyter notebook or
jupyterlab cell. The resulting snapshot of the visualizer will still be an
interactive 3D scene, but it won't be affected by any future `set_transform`
or `set_object` calls.
Note: this method should work well even when your jupyter kernel is running
on a different machine or inside a container.
z
)srcdocr` )r ra rR static_htmlrb s r
render_staticzVisualizer.render_static s6
F-(8(8(:;FF K M Mr c t t j | j | j j | S r )rT rZ rU r7 append)r r7 s r __getitem__zVisualizer.__getitem__ s( ##DKK1A1A$1GHHr c b | j j t ||| j S r )rU r* r r7 )r geometrymaterials r
set_objectzVisualizer.set_object s$ {{ (Hdii HIIr c ` | j j t || j S r )rU r* r r7 )r matrixs r
set_transformzVisualizer.set_transform s" {{ VTYY ?@@r c b | j j t ||| j S r )rU r* r
r7 )r keyvalues r set_propertyzVisualizer.set_property s$ {{ C BCCr c P | j j t ||| S )N)play repetitions)rU r* r )r animationrx ry s r
set_animationzVisualizer.set_animation s {{ YT{ [\\r c 6 | j j S )z
Save an image)rU rG r& s r rG zVisualizer.get_image s {{$$&&r c ^ | j j t | j S r )rU r* r r7 r& s r deletezVisualizer.delete s {{ tyy 122r c 8 | j j y r )rU closer& s r r zVisualizer.close s r c 6 | j j S )z
Generate and save a static HTML file that standalone encompasses the visualizer and contents.
Ask the server for the scene (since the server knows it), and pack it all into an
HTML blob for future use.
)rU r@ r& s r rf zVisualizer.static_html s {{$$&&r c P dj | j | j S )Nz,)rU r7 )ra rU r7 r& s r __repr__zVisualizer.__repr__ s# =DDDKK^b^g^gDhhr )i r )Tr )rH rI rJ __slots__r staticmethodrZ r2 r] r4 rc rg rj rn npeyerr rv r{ rG r~ r rf r rL r r rT rT W s 6"I #Db '
#" 7M IJ $*266!9 AD]'3 'ir rT __main__r/ Td )#r1 r; numpyr r" rC PILr IPython.displayr r7 r commandsr r r r
r r rl r
servers.zmqserverr r rR rT rH timesysargslenargvr rU sleeprL r r r s
^ ^ ' =A AH:]i ]i@ z
D
388}q((1+ sxx=1 88AB