\j ddlZddlmZddlmZddlmZeje e Z ejejGddeZ ejejGddZy) N)libav)SideData) uintptr_tcheZdZdedej fdZdZdZdejfdZ dZ dZ y ) MotionVectorsframeindexctj||||i|_|jjt j tjz|_ yN) r__init___vectorsptrsizecythonsizeoflibAVMotionVector_len)selfsentinelrr s L/root/.hermes/venv/lib/python3.12/site-packages/av/sidedata/motionvectors.pyr zMotionVectors.__init__ s@$%7 HHMMV]]33E3E%FF c d|jjdt|dtjt |jj ddS)Nz)rrlenrcastrdatars r__repr__zMotionVectors.__repr__sJ)$((--9d)N6;;y$((--+PQS*TTU W rc|jSr )rrs r__len__zMotionVectors.__len__s yyrc |j|S#t$rYnwxYw||jk\r t|t t ||x}|j|<|Sr )r KeyErrorr IndexError MotionVector_cinit_bypass_sentinel)rr vectors r __getitem__zMotionVectors.__getitem__si ==' '    DII U# #(4 "D%)  u% s  c#NKt|jD] }|| yw)z Iterate over all motion vectors.N)ranger)ris r__iter__zMotionVectors.__iter__(s'tyy! Aq'M s#%cVddl}|j||jgddS)z Convert motion vectors to a NumPy structured array. Returns a NumPy array with fields corresponding to the AVMotionVector structure. rN) )sourceint32)wuint8)hr2)src_xint16)src_yr5)dst_xr5)dst_yr5)flagsuint64)motion_xr0)motion_yr0) motion_scaleuint16T)align)dtype)numpy frombufferr@)rnps r to_ndarrayzMotionVectors.to_ndarray-s; }} ((   rN) __name__ __module__ __qualname__Framerintr r r" Py_ssize_tr)r-rDrrrr sBGGfjjG   !2!2   rrceZdZdZdedej fdZdZe dZ e dZ e dZ e d Z e d Ze d Ze d Ze d Ze dZe dZy)r&z Represents a single motion vector from video frame data. Motion vectors describe the motion of a block of pixels between frames. parentr c|tur td||_tjtj t j|jj}||z|_y)Nz(cannot manually instantiate MotionVector) r' RuntimeErrorrMrrpointerrrrr)rrrMr bases rr zMotionVector.__init__SsU 1 1IJ J 39;; NN3-- . 4 %<rc d|jd|jd|jd|jd|jd|j d S)Nz)r1r3r4r6r7r8rs rr zMotionVector.__repr__\sP($&&:ZZL$**VDJJrks{ (91'9< H< < ~<%<%<%r