vjlDdZddlZddlmZddlmZd dZGddZy) zAnimation class for the viewerN) numpy_to_json) send_datacg}|d|j}|j||jD]}|jt |||S)z&Collect all paths in the assembly tree/)labelappendchildrenextend collect_paths)assemblypathresultnew_pathchilds e/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/ocp_vscode/animation.pyr r sP Fq()H MM("" mE845# Mc"eZdZdZdZdZdZy) Animationz)Class to create animations for the viewercg|_t|dxr t|d |_t|d|_|jr)t |j j |_yt||_y)Nmatesfq_namejoints) trackshasattr is_cadquery is_build123dlistobjectskeyspathsr )selfr s r__init__zAnimation.__init__&so "8W5 g i? ; $Hh7   h..3356DJ&x0DJrc^t|t|k7r td|jrS|jdj d\}}}||j vs||j dgzvrtd|d|j r |jj||||fy)a Adding a three.js animation track. Parameters ---------- path : string The path (or id) of the cad object for which this track is meant. Usually of the form `/top-level/level2/...` action : {"t", "tx", "ty", "tz", "q", "rx", "ry", "rz"} The action type: - "tx", "ty", "tz" for translations along the x, y or z-axis - "t" to add a position vector (3-dim array) to the current position of the CAD object - "rx", "ry", "rz" for rotations around x, y or z-axis - "q" to apply a quaternion to the location of the CAD object times : list of float or int An array of floats describing the points in time where CAD object (with id `path`) should be at the location defined by `action` and `values` values : list of float or int An array of same length as `times` defining the locations where the CAD objects should be according to the `action` provided. Formats: - "tx", "ty", "tz": float distance to move - "t": 3-dim tuples or lists defining the positions to move to - "rx", "ry", "rz": float angle in degrees - "q" quaternions of the form (x,y,z,w) the represent the rotation to be applied Examples -------- ``` AnimationTrack( '/bottom/left_middle/lower', # path 'rz', # action [0.0, 0.5, 1.0, 1.5, 2.0, 2.5, 3.0, 3.5, 4.0], # times (seconds) [-15.0, -15.0, -15.0, 9.7, 20.0, 9.7, -15.0, -15.0, -15.0] # angles ) AnimationTrack( 'base/link_4_6', # path 't', # action [0.0, 1.0, 2.0, 3.0, 4.0], # times (seconds) [[0.0, 0.0, 0.0], [0.0, 1.9509, 3.9049], [0.0 , -3.2974, -16.7545], [0.0 , 0.05894 , -32.0217], [0.0 , -3.2212, -13.3424]] # 3-dim positions ) ``` See also -------- - [three.js NumberKeyframeTrack](https://threejs.org/docs/index.html?q=track#api/en/animation/tracks/NumberKeyframeTrack) - [three.js QuaternionKeyframeTrack](https://threejs.org/docs/index.html?q=track#api/en/animation/tracks/QuaternionKeyframeTrack) z8Parameters 'times' and 'values' need to have same lengthrzPath 'z' does not exist in assemblyN) len ValueErrorrstrip partitionr!rrr )r"ractiontimesvaluesroot_cq_paths r add_trackzAnimation.add_track1sx u:V $WX X   #zz#88= D!W4::% bT8I)I 6$/K!LMM     D&%89rcv|jdd|id}ttjt |y)zAnimate the tracks animationspeed)datatypeconfigN)rrjsonloadsr)r"r3r4s ranimatezAnimation.animate}s- [WeDTU$**]4012rN)__name__ __module__ __qualname____doc__r#r0r9rrrr#s3 1J:X3rr)r%)r=r7ocp_tessellate.utilsrcommsrr rr>rrrAs$$$ .]3]3r