Ej'8dZddlZGddZGddZy)z:Classes for easy interpolation of trajectories and curves.NceZdZdZddZdZy) InterpolatoraYPoorman's linear interpolator. Parameters ---------- tt : list, optional List of time frames for the interpolator. ss : list, optional List of values for the interpolator. ttss : list, optional Lists of time frames and their correspondients values for the interpolator. This argument can be used instead of ``tt`` and ``ss`` to instantiate the interpolator using an unique argument. left : float, optional Value to return when ``t < tt[0]``. right : float, optional Value to return when ``t > tt[-1]``. Examples -------- .. code:: python # instantiate using `tt` and `ss` interpolator = Interpolator(tt=[0, 1, 2], ss=[3, 4, 5]) # instantiate using `ttss` interpolator = Interpolator(ttss=[[0, 3], [1, 4], [2, 5]]) # [t, value] Nc| t|\}}dtj|z|_dtj|z|_||_||_t|t|c|_ |_ yN?) zipnparrayttssleftrightminmaxtmintmax)selfr r ttssr rs _/root/workspace/ytshorts/venv/lib/python3.12/site-packages/moviepy/video/tools/interpolators.py__init__zInterpolator.__init__*sc  $ZFB $ $  "2wB 49ctj||j|j|j|j S)zInterpolates ``t``. Parameters ---------- t : float Time frame for which the correspondent value will be returned. )r interpr r r rrts r__call__zInterpolator.__call__4s+yyDGGTWWdiiDDr)NNNNN)__name__ __module__ __qualname____doc__rrrrrrs!F0 ErrcleZdZdZdZdZdZdZdZd dZ dZ e d Z e d Z e d Zy ) Trajectorya=Trajectory compound by time frames and (x, y) pixels. It's designed as an interpolator, so you can get the position at a given time ``t``. You can instantiate it from a file using the methods ``from_file`` and ``load_list``. Parameters ---------- tt : list or numpy.ndarray Time frames. xx : list or numpy.ndarray X positions in the trajectory. yy : list or numpy.ndarray Y positions in the trajectory. Examples -------- >>> trajectory = Trajectory([0, .166, .333], [554, 474, 384], [100, 90, 91]) cdtj|z|_tj||_tj||_|j yr)r r r xxyyupdate_interpolators)rr r%r&s rrzTrajectory.__init__[s@ $((2,((2, !!#rcltj|j||j|gS)zInterpolates the trajectory at the given time ``t``. Parameters ---------- t : float Time for which to the corresponding position will be returned. )r r xiyirs rrzTrajectory.__call__as(xxTWWQZ011rc^t|j|j|z|jS)aAdds a value to the ``xx`` position of the trajectory. Parameters ---------- x : int Value added to ``xx`` in the trajectory. Returns ------- Trajectory : new instance with the new X position included. r#r r%r&)rxs raddxzTrajectory.addxls#$''477Q;88rc^t|j|j|j|zS)aAdds a value to the ``yy`` position of the trajectory. Parameters ---------- y : int Value added to ``yy`` in the trajectory. Returns ------- Trajectory : new instance with the new Y position included. r,)rys raddyzTrajectory.addy}s#$''477DGGaK88rct|j|j|_t|j|j|_y)zEUpdates the internal X and Y position interpolators for the instance.N)rr r%r)r&r*)rs rr'zTrajectory.update_interpolatorss.tww0tww0rcft|rdnd|jz|j|jS)zReturns all times with the X and Y values of each position. Parameters ---------- tms : bool, optional If is ``True``, the time will be returned in milliseconds. )rr r%r&)rtmss rtxyzTrajectory.txys(CDQ$''1477DGGDDrc tj|tjt|j dddy)zSaves the trajectory data in a text file. Parameters ---------- filename : str Path to the location of the new trajectory text file. Tr6%d )fmt delimiterN)r savetxtr listr7)rfilenames rto_filezTrajectory.to_files5  HHT$((t(,- .  rcvtj|d}|j\}}}td|zdz ||S)a Instantiates an object of Trajectory using a data text file. Parameters ---------- filename : str Path to the location of trajectory text file to load. Returns ------- Trajectory : new instance loaded from text file. r;r=rr4)r loadtxtTr#)r@arrr r%r&s r from_filezTrajectory.from_files: jjT2UU B#(T/2r22rc t|}tj|Dcgc]0}tjt |j d2c}}tj ||dddj|gdzycc}w)zSaves a set of trajectories into a text file. Parameters ---------- trajs : list List of trajectories to be saved. filename : str Path of the text file that will store the trajectories data. Tr9r:r;)zt(ms)r-r0)r<r=headerN)lenr hstackr r?r7r>join)trajsr@NrrFs r save_listzTrajectory.save_listsn Jii%HQ$quuu"78HI  99Q!445  Is5B c tj|dj}|jd}tj||dz Dcgc]}t d|dzdz |d|d !c}Scc}w) a'Loads a list of trajectories from a data text file. Parameters ---------- filename : str Path of the text file that stores the data of a set of trajectories. Returns ------- list : List of trajectories loaded from the file. r;rCrrr4r5)r r%r&)r rDrEshapesplitr#)r@rFNlinesas r load_listzTrajectory.load_listsv jjT2441XXc6A:.  #!*t+!1 >   s $A3N)F)rrrr rrr.r1r'r7rA staticmethodrGrOrWr!rrr#r#@sg4$ 29"9"1 E 33&  ,  rr#)r numpyr rr#r!rrrZs$@7E7Etn n r