U a+d@sdZddlZddlmZmZddlmZddlm Z zddl Z dZ Wnd Z YnXed d gedd d Z dddZ dddZdS)a This module contains different functions for tracking objects in videos, manually or automatically. The tracking functions return results under the form: ``( txy, (fx,fy) )`` where txy is of the form [(ti, xi, yi)...] and (fx(t),fy(t)) give the position of the track for all times t (if the time t is out of the time bounds of the tracking time interval fx and fy return the position of the object at the start or at the end of the tracking time interval). N)convert_to_secondsuse_clip_fps_by_default) imdisplay) TrajectoryTFt1t2csddljjd|}|dkr<|dkr>> from moviepy.editor import VideoFileClip >>> from moviepy.video.tools.tracking import manual_tracking >>> clip = VideoFileClip("myvideo.mp4") >>> # manually indicate 3 trajectories, save them to a file >>> trajectories = manual_tracking(clip, t1=5, t2=7, fps=5, nobjects=3, savefile="track.txt") >>> # ... >>> # LATER, IN ANOTHER SCRIPT, RECOVER THESE TRAJECTORIES >>> from moviepy.video.tools.tracking import Trajectory >>> traj1, traj2, traj3 = Trajectory.load_list('track.txt') >>> # If ever you only have one object being tracked, recover it with >>> traj, = Trajectory.load_list('track.txt') rN?rcst|}g}|rjD]f}|jjkr\|jjkrHdS|jjkrt q&|jj kr&j \}}| ||f|d8}q&q|S)Nreturnr)r get_frameeventgettypeZKEYDOWNkeyZ K_BACKSLASHZK_ESCAPEKeyboardInterruptZMOUSEBUTTONDOWNZmouseZget_posappend)tZobjects_to_clickclicksr xy)clipnobjectspgscreen@/tmp/pip-unpacked-wheel-0yp4gafk/moviepy/video/tools/tracking.py gatherClicks[s     z%manual_tracking..gatherClicksr csg|] }|qSrr).0e)irr sz#manual_tracking..) ZpygameZdisplayset_modesizedurationpoprzipranger save_list)rrr fpsrZsavefilesteprZtxy_listrrttZxylistresultxysxxyyr)rr rrrrmanual_trackings41        r0c Cs|rN|rN|jdd\}}|\}}|||||||||||f}t||tj}t||j\} } |r|r||| ||| fS| | fS)zq find image pattern ``pat`` in ``pic[x +/- r, y +/- r]``. if xy is none, consider the whole picture. Nr)shapecv2Z matchTemplateZTM_CCOEFF_NORMEDnpZ unravel_indexZargmax) ZpicpatxyrhwrrmatchesZyfZxfrrr findArounds,r:c Csts td|s$t||d|}|dkr@td|jd|}|g}|ddD]$}|t||||d|dqRt|\}} t ||| S)a Tracks a given pattern (small image array) in a video clip. Returns [(x1,y1),(x2,y2)...] where xi,yi are the coordinates of the pattern in the clip on frame i. To select the frames you can either specify a list of times with ``tt`` or select a frame rate with ``fps``. This algorithm assumes that the pattern's aspect does not vary much and that the distance between two occurences of the pattern in two consecutive frames is smaller than ``radius`` (if you set ``radius`` to -1 the pattern will be searched in the whole screen at each frame). You can also provide the original position of the pattern with xy0. zTSorry, autotrack requires OpenCV for the moment. Install OpenCV (aka cv2) to use it.rNr r)r5r6) autotracking_possibleIOErrorr:r r3Zaranger$rr&r) rpatternr+r)ZradiusZxy0r-rr.r/rrr autoTracks  r@)NNNrN)NN)NNr;N)__doc__Znumpyr3Zmoviepy.decoratorsrrZ io.previewrZ interpolatorsrr2r=r0r:r@rrrrs      i