:vjddlZddlZddlZddlZddlZddlmZGdde Z Gdde Z Gdd e Z d Z d ZGd d e ZddZy)N)transformationsc(eZdZgdZddZdZdZy)AnimationTracknamejstypeframesvaluesNcd||_||_|g|_n||_|g|_y||_yNr)selfrr r r s b/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/meshcat/animation.py__init__zAnimationTrack.__init__ s6  >DK DK >DK DKctj|j|}|jj|||jj||yr )bisectr insertr )rframevalueis r set_propertyzAnimationTrack.set_propertys> MM$++u - 1e$ 1e$rctd|jzt|jtt |j Dcgc]!}|j ||j |d#c}dScc}w)N.)timer)rtypekeys)strrr rangelenr r )rrs rlowerzAnimationTrack.lowersttyy)%S-.0/!Q++a./0  0s &A7)NN__name__ __module__ __qualname__ __slots__rrr!rrrr s6I !%  rrc(eZdZgdZddZdZdZy) AnimationCliptracksfpsrNcB|i|_n||_||_||_yr r*)rr+r,rs rrzAnimationClip.__init__,s# >DK DK rc||jvrt|||j|<|j|}|j||yr )r+rr)rrpropertyr rtracks rrzAnimationClip.set_property4sA 4;; &$28V$DDKK ! H% 5%(rc|jt|j|jj Dcgc]}|j c}dScc}w)N)r,rr+)r,rrr+r r!)rts rr!zAnimationClip.lower:sKHH^+/;;+=+=+?@+?a +?@  AsA)Ndefaultr"r'rrr)r))s)I)  rr)c(eZdZddgZddZdZdZy) Animationclipsdefault_framerateNcB|i|_||_y||_||_yr )r7r8)rr7r8s rrzAnimation.__init__Es) =DJ"3DJ!2rc|jjDcgc]&\}}|j|jd(c}}Scc}}w)N)pathclip)r7itemsr!)rr;r<s rr!zAnimation.lowerLsR#jj..021ltTZZ\ZZ\ 12 22s+A c0t||j|Sr )AnimationFrameVisualizerr;)r visualizerrs rat_framezAnimation.at_frameRs'joouEEr)Nr3)r#r$r%r&rr!rAr'rrr6r6Bs-.I32 Frr6c&t|dddfS)N)list)matrixs r js_positionrFVs rr1u rcPtj|}|d|d|d|dgS)NrrCr)tfquaternion_from_matrix)rEquats r js_quaternionrLZs1 $ $V ,D GT!Wd1gtAw //rc>eZdZgdZdZdZdZdZdZdZ dZ y ) r? animationr; current_framec.||_||_||_yr rN)rrOr;rPs rrz!AnimationFrameVisualizer.__init__bs" *rc|j|jjvrAt|jj|jj|j<|jj|jS)N)r,)r;rOr7r)r8rs rget_clipz!AnimationFrameVisualizer.get_clipgsX 99DNN00 0.;@`@`.aDNN  +~~##DII..rc|j}|j|jddt||j|jddt |y)Npositionvector3 quaternion)rTrrPrFrL)rrEr<s r set_transformz&AnimationFrameVisualizer.set_transformlsM}} $,,k:{SYGZ[ $,,m]MZ`Labrc`|j}|j|j|||yr )rTrrP)rpropr rr<s rrz%AnimationFrameVisualizer.set_propertyqs'}} $,,dFEBrcvt|j|jj||jSr )r?rOr;appendrP)rr;s r __getitem__z$AnimationFrameVisualizer.__getitem__us+' 8H8H8NPTPbPbccrc|Sr r'rSs r __enter__z"AnimationFrameVisualizer.__enter__xs rcyr r')rargs r__exit__z!AnimationFrameVisualizer.__exit__{s rN) r#r$r%r&rrTrYrr^r`rcr'rrr?r?_s-6I+ / c Cd rr?c tjj|}tjj|r|st dj |t j5}tj|5}|j|dddddt|ddddd d d d g }|r|jd |j| tj|| dddt!dj ||S#1swYxYw#tj$r}t!dd}~wwxYw#1swYXxYw)z Try to convert a tar file containing a sequence of frames saved by the meshcat viewer into a single video file. This relies on having `ffmpeg` installed on your system. zjThe output path {:s} already exists. To overwrite that file, you can pass overwrite=True to this function.Nffmpegz-rz-iz%07d.pngz-vcodeclibx264z-presetslowz-crf18z-y)cwda% Could not call `ffmpeg` to convert your frames into a video. If you want to convert the frames manually, you can extract the .tar archive into a directory, cd to that directory, and run: ffmpeg -r 60 -i %07d.png \ -vcodec libx264 \ -preset slow \ -crf 18 \ output.mp4 zSaved output as {:s})osr;abspathisfile ValueErrorformattempfileTemporaryDirectorytarfileopen extractallrr] subprocess check_callCalledProcessErrorprint) tar_file_path output_path framerate overwritetmp_dirtarargses rconvert_frames_to_videors8''//+.K ww~~k"9FMMNYZ[ [  $ $ &' \\- (C NN7 #)c)nk96   KK  K    ! !$G 4 ', ' ' 45 -) (,,      ' &sC/E DAE D"D E "E5 EEE  E)z output.mp4<F)rorqos.pathrjrtrrrIobjectrr)r6rFrLr?rr'rrrsY # V > F 2FF(0  v @!r