vj֌@*dZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl mZddlmZmZmZmZmZmZddl mZmZmZmZddlmZmZmZmZmZm Z m!Z!m"Z"m#Z#dd l$m%Z%m&Z&m'Z'dd l(m)Z)m*Z*ejVjYd d k(r dd l-m.Z.m/Z/m0Z0dZ1n dd l2m.Z.m/Z/m0Z0dZ1ddl3m4Z4m5Z5ddl2m6Z6ddl3m7Z7m8Z8m9Z9m:Z:m;Z;mda?ddddddZ@ddddddZAGddZBd'dZCdZDdddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd?dZEd ZFd!ZG d(d"ZHd#ZId$ZJ d)d%ZKd*d&ZLy)+z&Show CAD objects in Visual Studio CodeN)Enum)Logger)OcpGroup) OCP_Edges OCP_FacesOCP_Part OCP_PartGroup OCP_Vertices OcpWrapper) combined_bbget_normal_lentessellate_group to_ocpgroup) is_build123d is_cadqueryis_cadquery_assemblyis_cadquery_sketchis_toploc_locationis_topods_compoundis_topods_shape is_vector is_wrapped)ColorTimernumpy_to_buffer_json) BaseColorMap get_colormapJUPYTER_CADQUERY1) send_backend send_command send_dataTF)combined_configget_changed_config) is_pytest)CameraCollapsecheck_deprecated get_default get_defaultspreset) show show_object_show _show_object reset_showshow_all show_clearsave_screenshot none_filterobjsnamescolorsalphasotherr7r8r9progressc |jd}|jd}t||}|jddk(rtj} n |jdtj} | j|d<|jdt j } | j|d<|jd%t|djt_ |d=nr?_splashT reset_cameracollapsedefault_facecolordefault_thickedgecolordefault_vertexcolorclip clip_slider_0 clip_slider_1 clip_slider_2 clip_normal_0)rr clip_normal_1)rrKr clip_normal_2)rrrKclip_intersectionF clip_planesclip_object_colorstimeit-+cr render_mates render_joints helper_scale default_color show_parentshow_sketch_local) r7r8r9rUrVrWrXrYrZr<r)positionrotationtarget) cad_widthheight tree_widththeme)r^r_zSetting z9 cannot be set, it is determined by the VSCode panel size)r`raz! can only be set in VSCode configdebugz tessellation parameters: tessellaterender_normals deviation normal_lenbbgưgư>)xminyminzminxmaxymaxzmax)#getr#r&RESETvaluer'ROOTr percentageoc FACE_COLORTHICK_EDGE_COLOR VERTEX_COLORr*items startswithupdater+Progressrr$rlenobjects isinstancerlocis_jupyter_cadqueryprintrr r dictto_dict count_shapes)r7r8r9r<cad_objskwargsr>r?confrArBkv clip_defaultsrQcchanged_config part_group instancesr~paramsshapesmappingrgs `/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/ocp_vscode/show.py _tessellaterdsZZZ !F ::f D &t 4D xx d"|| xx = '--Dxx HMM2H~~D zz%&2f%89:EE & 'dhh':BMMJKVV  zz*+7#F+C$DEPP + ,# HH-r/B/B C * zz'(4'< =>II ( ) HH*BOO < *  v||#vzz.'AV\\'Q)^113 3TQq||F7KAqD3   88O $ 0_% 88O $ 0_% 88O $ 0_% 88O $ 0$.D ! 88O $ 0$.D ! 88O $ 0$.D ! 88' ( 4(-D$ % 88M " ."'D  88( ) 5).D% & M" Hfjj2 3F ~.1Q./ vr=! ,+- + NN4F4F~4VW **!3!3O!D NN4F4F~4VW **!3!3O!D =.2D2D]2ST$jj#^%7%78K%L#!  I( z!! "a 'Jz7I7I!7Lh,W..C#++A.J :>>1JN3 -:JJL DAq 3 3' 1  ( 1"q,C'C 1#VW %/ *  HQC@A B ]N"vzz)'<'Da F1I#&F9 zz'&6'? ,f5 vr< +%5  68VZZ5I& " 67 , *,{FJJ{34F< vr4 #   :UDtBB $F4L ffj&=&=&? HH} </ , ,6X , + $ #s=Z4Z= ZEZ6Z'#Z-=0Z9Z$-Z69[c:td|jd}|tdDcgc]}|c}}t|||||d|\}} } } } | jdd| d<n| jd | drdnd| d <| jd | d r t d | |jd |d | d <t |d dd5t r|| | | f| fcdddStt|| d| | d| fcdddScc}w#1swYyxYw)NrQrRr;darkra orbit_controlorbit trackballcontrolrbz config: exploderSzcreate data objrT)rrdata)rtypeconfigcount) r+rnrzrrrr%rr) r7r8r9r<rrrQrrrrrrs r_convertrsPHfjj2 3F.1Q./7B  8  84Ivv|Wzz&% w O $ 0'-o'>GKy zz'&6'? mV$ zz)("9-y vr,a 0 ;vv|!(DK DKc>||jvrt|ddyy)zUpdate progress indicatorrST)endflushN)rr)rmarks rryzProgress.updateFs 4;;  $Bd + rN)+)__name__ __module__ __qualname____doc__rryrrrzrz=s-! ,rrzc| |rdg|zSdSt||kr-td|dt||g|t|z zzSt||kDrtd|d||d|S|S)z.Align attributes to the length of the cad_objsNzToo few z, using defaults to fillz Too many z, trimming to length )r{rlist) attr_listlengthdefaulttagrs r align_attrsrLs")v3t3 Y&  567I'fs9~.E!FFF Y&  #3F8<=&!!rc|g}t|jDcic]\}}| ||vs||c}}Scc}}wr)rrw)dexcludesrrs rr4r4ZsB!!W]]_ T_TQ !8BSAqD_ TT Ts :::-+*c)?r7r8r9r?r<glasstoolsr`axesaxes0gridortho transparentdefault_opacity black_edgesrrrBrticks center_griduptabzoomr[ quaternionr]rArGrHrIrJrLrMrNrOrP pan_speed rotate_speed zoom_speedreangular_tolerance edge_accuracyrXdefault_edgecolorrCrDrEambient_intensitydirect_intensity metalness roughness render_edgesrdrUrVrYrZrW mate_scalerbrQ_force_in_debugc?>t|?ittdgS)aShow CAD objects in Visual Studio Code Parameters cad_objs: All cad objects that should be shown as positional parameters Keywords for show: names: List of names for the cad_objs. Needs to have the same length as cad_objs colors: List of colors for the cad_objs. Needs to have the same length as cad_objs alphas: List of alpha values for the cad_objs. Needs to have the same length as cad_objs progress: Show progress of tessellation with None is no progress indicator. (default="-+*c") for object: "-": is reference, "+": gets tessellated with Python code, "*": gets tessellated with native code, "c": from cache port: The port the viewer listens to. Typically use 'set_port(port)' instead Valid keywords to configure the viewer (**kwargs): - UI glass: Use glass mode where tree is an overlay over the cad object (default=False) tools: Show tools (default=True) tree_width: Width of the object tree (default=240) - Viewer axes: Show axes (default=False) axes0: Show axes at (0,0,0) (default=False) grid: Show grid (default=False) ortho: Use orthographic projections (default=True) transparent: Show objects transparent (default=False) default_opacity: Opacity value for transparent objects (default=0.5) black_edges: Show edges in black color (default=False) orbit_control: Mouse control use "orbit" control instead of "trackball" control (default=False) collapse: Collapse.LEAVES: collapse all single leaf nodes, Collapse.ROOT: expand root only, Collapse.ALL: collapse all nodes, Collapse.NONE: expand all nodes (default=Collapse.ROOT) ticks: Hint for the number of ticks in both directions (default=10) center_grid: Center the grid at the origin or center of mass (default=False) up: Use z-axis ('Z') or y-axis ('Y') as up direction for the camera (default="Z") explode: Turn on explode mode (default=False) zoom: Zoom factor of view (default=1.0) position: Camera position quaternion: Camera orientation as quaternion target: Camera look at target reset_camera: Camera.RESET: Reset camera position, rotation, zoom and target Camera.CENTER: Keep camera position, rotation, zoom, but look at center Camera.KEEP: Keep camera position, rotation, zoom, and target (default=Camera.RESET) clip_slider_0: Setting of clipping slider 0 (default=None) clip_slider_1: Setting of clipping slider 1 (default=None) clip_slider_2: Setting of clipping slider 2 (default=None) clip_normal_0: Setting of clipping normal 0 (default=None) clip_normal_1: Setting of clipping normal 1 (default=None) clip_normal_2: Setting of clipping normal 2 (default=None) clip_intersection: Use clipping intersection mode (default=False) clip_planes: Show clipping plane helpers (default=False) clip_object_colors: Use object color for clipping caps (default=False) pan_speed: Speed of mouse panning (default=1) rotate_speed: Speed of mouse rotate (default=1) zoom_speed: Speed of mouse zoom (default=1) - Renderer deviation: Shapes: Deviation from linear deflection value (default=0.1) angular_tolerance: Shapes: Angular deflection in radians for tessellation (default=0.2) edge_accuracy: Edges: Precision of edge discretization (default: mesh quality / 100) default_color: Default mesh color (default=(232, 176, 36)) default_edgecolor: Default color of the edges of a mesh (default=#707070) default_facecolor: Default color of the edges of a mesh (default=#ee82ee) default_thickedgecolor: Default color of the edges of a mesh (default=#ba55d3) default_vertexcolor: Default color of the edges of a mesh (default=#ba55d3) ambient_intensity: Intensity of ambient light (default=1.00) direct_intensity: Intensity of direct light (default=1.10) metalness: Metalness property of the default material (default=0.30) roughness: Roughness property of the default material (default=0.65) render_edges: Render edges (default=True) render_normals: Render normals (default=False) render_mates: Render mates for MAssemblies (default=False) render_joints: Render build123d joints (default=False) show_parent: Render parent of faces, edges or vertices as wireframe (default=False) show_sketch_local: In build123d show local sketch in addition to relocate sketch (default=True) helper_scale: Scale of rendered helpers (locations, axis, mates for MAssemblies) (default=1) - Debug debug: Show debug statements to the VS Code browser console (default=False) timeit: Show timing information from level 0-3 (default=False) r)r.r4locals)@r7r8r9r?r<rrr`rrrrrrrrrrBrrrrrrr[rr]rArGrHrIrJrLrMrNrOrPrrrrerrrXrrCrDrErrrrrrdrUrVrYrZrWrrbrQrrs@ rr,r,as | ( Bk&(ZLA BBrc |jd}|jd}|jd}|jd}|jd}|jd}|jd}|jd} |Ut|d k(sGt|d k(rE|d 4t|d ttt t frt|d d k(r td y|jD cic]\} } | | d vr| | }} } t|}|jd t|d tr |d gdz|d <td|}t|t|dd}t|tr:tt|D cgc] } t|}} dgt|z}n.t|t|dd}t|t|dd}d} t!}|*tt|D cgc] } t|} } tt|D]}||b| &t#| |dd||<|| ||||_t'||dr<||j(-t#||j(||<nt#||||<||||||||_| t#|}t+|gn|Dcgc]}|c}}t-|dd5t/|||||d|\}}| sdandaddd| tt3rfSt-|dd5t5||}dddt6r&t9dij:j<||St9di||ycc} } wcc} wcc} wcc}w#1swYxYw#1swYgxYw)Nr?rQr7r8r9rr<rrrTzshow: No CAD objects to show)rr7r8r9r?r< LAST_CALLrcolorrSoverallr;r,r:send)r?rQmodel)jcv_idrQ)rnr{r}rrtuplesetrrwr(boolr+rrrangenextrrahasattrrrzrrrr%r"rr widgetid)rrr?rQr7r8r9rr<rrr_ map_colorscolormapirtrr>s rr.r.s ::f D ZZ !F JJw E ZZ !F ZZ !F #67zz*%Hjj!23O  x=A  MQ  #x{T4,DEHQK(A- ,-LLN"DAq =    1"  f %F zz&% fVnd +$Vn-1F6N Hf %F s8}dG D*5J % %* # "s0?N"N(N-' N2N7O7OO cggggday)z&Reset the stack of objects to be shownr5N)OBJECTSrrrr0r0{sB"CGrc=@t|fittdgS)a]Incrementally show CAD objects in Visual Studio Code Parameters: obj: The CAD object to be shown Keywords for show_object: name: The name of the CAD object options: A dict of color and alpha value: {"alpha":0.5, "color": (64, 164, 223)} 0 <= alpha <= 1.0 and color is a 3-tuple of values between 0 and 255 parent: Add another object, usually the parent of e.g. edges or vertices with alpha=0.25 clear: In interactice mode, clear the stack of objects to be shown (typically used for the first object) port: The port the viewer listens to. Typically use 'set_port(port)' instead progress: Show progress of tessellation with None is no progress indicator. (default="-+*c") for object: "-": is reference, "+": gets tessellated with Python code, "*": gets tessellated with native code, "c": from cache Valid keywords to configure the viewer (**kwargs): - UI glass: Use glass mode where tree is an overlay over the cad object (default=False) tools: Show tools (default=True) tree_width: Width of the object tree (default=240) - Viewer axes: Show axes (default=False) axes0: Show axes at (0,0,0) (default=False) grid: Show grid (default=False) ortho: Use orthographic projections (default=True) transparent: Show objects transparent (default=False) default_opacity: Opacity value for transparent objects (default=0.5) black_edges: Show edges in black color (default=False) orbit_control: Mouse control use "orbit" control instead of "trackball" control (default=False) collapse: Collapse.LEAVES: collapse all single leaf nodes, Collapse.ROOT: expand root only, Collapse.ALL: collapse all nodes, Collapse.NONE: expand all nodes (default=Collapse.ROOT) ticks: Hint for the number of ticks in both directions (default=10) center_grid: Center the grid at the origin or center of mass (default=False) up: Use z-axis ('Z') or y-axis ('Y') as up direction for the camera (default="Z") zoom: Zoom factor of view (default=1.0) position: Camera position quaternion: Camera orientation as quaternion target: Camera look at target reset_camera: Camera.RESET: Reset camera position, rotation, zoom and target Camera.CENTER: Keep camera position, rotation, zoom, but look at center Camera.KEEP: Keep camera position, rotation, zoom, and target (default=Camera.RESET) clip_slider_0: Setting of clipping slider 0 (default=None) clip_slider_1: Setting of clipping slider 1 (default=None) clip_slider_2: Setting of clipping slider 2 (default=None) clip_normal_0: Setting of clipping normal 0 (default=[-1,0,0]) clip_normal_1: Setting of clipping normal 1 (default=[0,-1,0]) clip_normal_2: Setting of clipping normal 2 (default=[0,0,-1]) clip_intersection: Use clipping intersection mode (default=[False]) clip_planes: Show clipping plane helpers (default=False) clip_object_colors: Use object color for clipping caps (default=False) pan_speed: Speed of mouse panning (default=1) rotate_speed: Speed of mouse rotate (default=1) zoom_speed: Speed of mouse zoom (default=1) - Renderer deviation: Shapes: Deviation from linear deflection value (default=0.1) angular_tolerance: Shapes: Angular deflection in radians for tessellation (default=0.2) edge_accuracy: Edges: Precision of edge discretization (default: mesh quality / 100) default_color: Default mesh color (default=(232, 176, 36)) default_edgecolor: Default color of the edges of a mesh (default=(128, 128, 128)) default_facecolor: Default color of the edges of a mesh (default=#ee82ee / Violet) default_thickedgecolor: Default color of the edges of a mesh (default=#ba55d3 / MediumOrchid) default_vertexcolor: Default color of the edges of a mesh (default=#ba55d3 / MediumOrchid) ambient_intensity: Intensity of ambient light (default=1.00) direct_intensity: Intensity of direct light (default=1.10) metalness: Metalness property of the default material (default=0.30) roughness: Roughness property of the default material (default=0.65) render_edges: Render edges (default=True) render_normals: Render normals (default=False) render_mates: Render mates for MAssemblies (default=False) render_joints: Render build123d joints (default=False) show_parent: Render parent of faces, edges or vertices as wireframe (default=False) show_sketch_local: In build123d show local sketch in addition to relocate sketch (default=True) helper_scale: Scale of rendered helpers (locations, axis, mates for MAssemblies) (default=1) - Debug debug: Show debug statements to the VS Code browser console (default=False) imeit: Show timing information from level 0-3 (default=False) obj)r/r4r)=rnameoptionsparentclearr?r<rrr`rrrrrrrrrBrrrrr[rr]rArGrHrIrJrLrMrNrOrPrrrrerrrXrCrDrErrrrrrrdrUrVrYrZrWrrbrQs= rr-r-s ~  >{68eW= >>rc |jd}|jd}|jd}|jd}|jd}|jd}|jD cic]\}} | |dvr|| }}} |r t|`tdj |td j dtd j dtd j dd} d} |>t } | Ut ttd d zD]} t| ^} } n#|jd } |jdd} tdj |td j |td j | td j | ttdtd td td ||d|Scc} }w)Nr?rrrrr<)rrrrrr?r<r6r7r8r9rTralphag?)r7r8r9r?r<) rnrwr0rappendrrr{rr,)rrr?rrrrr<rrrrrrs rr/r/%s ::f D ::f D JJw E ZZ !Fjj#Gzz*%HLLN"DAq = V V 1"    v&)  &  & E E>  3ww/0145 $X 6 G$ GS) FO3 GD! HU# HU#  gx x     As:G2c"ddi}t|y)zClear the viewerrrN)r")rs rr2r2Xs  DdOrc vddl}|sda|rtdk(rdatdy|&|j}|jj }|g}g}g}|j D]^\} } t| ts| |gdzvs| jdstjd| tjd | t| d r | jt| st| tt t"t$t&j(fsk| it| t*sY| j,j.d k(r| j,j0d k(s't| t2sd t#| j,vrt| dr | j4dkDrtd| dI|t| t7|sbt| drt| dr | j8} t| dr | j:} t| dr?t=| j>stA| j>stC| j>sktE| s`tG| sUtI| sJtK| s?t| drt| dr t| dst| tLt6tNfr$|jQ| |jQ| xt| tRtTtVtXtZfr+|jQ| | | _.|jQ| t_| r=ta| g| g\} } | | _.|jQ| |jQ| t| tbr$|jQ| |jQ| Dtdt| d| date|dkDr1 tg||thjj|d|} tmstnr| Sytmrytsy#tp$r}td|Yd}~yd}~wwxYw)a Show all variables in the current scope Parameters: variables: Only show objects with names in this list of variable names, i.e. do not use all from locals() exclude: List of variable names to exclude from "show_all" classes: Only show objects which are instances of the classes in this list _visual_debug: private variable, do not use! Keywords for show_all: Valid keywords for "show_all" are the same as for "show" rNr:r,z. Skip visual debug step after a show() command)r______ih_oh_dhOutInrz^_i\d+z^_\d+_obj_Feature __future__zcad_viewer_widget.widgetareag.B}Tzinfinite face z skipped locationslocal_locationslocal_coord_systemwrappedr[ direction)r7zshow_all: Type z for name z cannot be visualized)r7rBrz show_all:):inspectrr currentframef_backf_localsrwr}rrxresearchrrcallableintfloatstrrtypes ModuleTyper __class__rrrrrrlocationrr rrrrrrrrrr rrrr rrrr r{r,r'rqr%r Exceptionr2) variablesexcludeclasses _visual_debugrr cfr|r7rrpgrresultexs rr1r1`s{(  f,  ?@  ! ! #II&& G E__& c sD !w!UUUt$yyD)5yy4(4V$)9}#UCu7G7GHI{#t$ &&*4MM,, <#v&)S-??  3 CHHtO N4&1 2  ?jeGn=sK(WS:K-Lmms01llC+' 4-ckk:-ckk:S>s#$',C+Z0[1cD%#67s# T"mY 8\Rs# T"#C( +SE$ @ Iy) T"C,s# T"%d3i[ 4&@UVY'` 7|a #! -   F{1 2 ;  # +r " " #s.P P8" P33P8c8|jtjs2tjdj }t ||z }n|}tj|}|jr|jjnd}td|d||rsd}tdD]T}|jr-|jj|kDrtd|d }ntjd V|s td y y y ) z%Save a screenshot of the current view.r screenshot)rfilename)r?FzScreenshot saved to Tg?z4Warning: Screenshot not found in 2 seconds, abortingN)rxosseppathlibPathabsoluterexistsstatst_mtimer!rrtimesleep) r&r?pollingprefix full_pathpmtimedoners rr3r3s   rvv &c"++-)*   YA!"AFFH  E,ykC$OrAxxzaffh//%7,i8 JJsO  H Ir)T)rFs,$  ##   DC8::>>$%,LLGF ' bB ?  $tdgIX     ,^ , , U                 A^CBvCrD               {_?D0f   JZJr