-i'FdZddlZGddZGddZdS)z:Classes for easy interpolation of trajectories and curves.Nc eZdZdZddZdZdS) 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|_ |_ dSN?) zipnparrayttssleftrightminmaxtmintmax)selfr r ttssr rs _/var/www/html/speakWrite/venv/lib/python3.11/site-packages/moviepy/video/tools/interpolators.py__init__zInterpolator.__init__*sk  $ZFB $ $  "2wwB 4999cZtj||j|j|j|jS)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$yDGTWdiDDDr)NNNNN)__name__ __module__ __qualname____doc__rrrrrrsG!!F0000 E E E E ErrceZdZdZdZdZdZdZdZddZ d Z e d Z e d Z e d Zd S) 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||_|dSr)r r r xxyyupdate_interpolators)rr r%r&s rrzTrajectory.__init__[sL $(2,,(2,, !!#####rcztj||||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-xTWWQZZ0111rcHt|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 $'47Q;888rcHt|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 $'47DGaK888rct|j|j|_t|j|j|_dS)zEUpdates the internal X and Y position interpolators for the instance.N)rr r%r)r&r*)rs rr'zTrajectory.update_interpolatorss0tw00tw00rFcPt|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*C&DDQ$'147DGDDDrc tj|tjt|ddddS)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_filesS  HT$((t(,,-- . .       rcrtj|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? jT222U B#(T/2r222rc t|}tjd|D}tj||ddd|gdzdS)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. c xg|]7}tjt|d8S)Tr9)r r r?r7).0rs r z(Trajectory.save_list..s6HHHQ$quuu"7"788HHHrr:r;)zt(ms)r-r0)r<r=headerN)lenr hstackr>join)trajsr@NrFs r save_listzTrajectory.save_listsw JJiHH%HHHII  99Q!4!4!4455       rctj|dj}|jd}dtj||dz DS)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;rCrcbg|],}td|dzdz |d|d-S)rrr4r5)r r%r&)r#)rJas rrKz(Trajectory.load_list..sL    #!*t+!1 > > >   r)r rDrEshapesplit)r@rFNliness r load_listzTrajectory.load_listsW jT22241  Xc6A:..    rN)F)rrrr rrr.r1r'r7rA staticmethodrGrRr[r!rrr#r#@s4$$$ 2 2 2999"999"111 E E E E   33\3&  \ ,  \   rr#)r numpyr rr#r!rrr^s@@7E7E7E7E7E7E7E7Etn n n n n n n n n n r