k jOXddlZddlZddlmZmZmZddlZddlmZGddZ dS)N)AnyOptionalUnion)TimerceZdZdZ ddeejejfdeee ffdZ de ddfd Z de fd Zdd Zdeeeffd Zdd Z ddeedeee fddfdZddZddZdS)BaseVideoCapturea^ ``BaseVideoCapture`` is a base class to video capture Args: source (moderngl.Texture, moderngl.Framebuffer): the source of the capture framerate (int, float) : the framerate of the video, by thefault is 60 fps if the source is texture there are some requirements: - dtype = 'f1'; - components >= 3. <source frameratec||_||_d|_d|_d|_d|_d|_t|_d|_ t|jtj r|jj |_ dSdS)NFgr)_source _framerate _recording _last_time _filename_width_heightr_timer _components isinstancemodernglTexture components)selfr r s W/home/agentuser/manim-venv/lib/python3.11/site-packages/moderngl_window/capture/base.py__init__zBaseVideoCapture.__init__s|  #*/!$ %) &* gg ! dlH$4 5 5 7#|6D    7 7framereturnNc td)zq custom function called during self.save() Args: frame: frame data in bytes override this functionNotImplementedError)rrs r _dump_framezBaseVideoCapture._dump_frame-s"":;;;rc td)z custom function called during self.start_capture() must return a True if this function complete without errors r"r#rs r _start_funczBaseVideoCapture._start_func6s "":;;;rc td)z> custom function called during self.release() r"r#r's r _release_funczBaseVideoCapture._release_func>s"":;;;rc2|jj|jjfS)zJ Return a tuple of the width and the height of the source )rwidthheightr's r_get_whzBaseVideoCapture._get_whDs|!4<#666rctj|jrtj|jdSdS)z,Remove the filename of the video is it existN)ospathexistsrremover's r _remove_filezBaseVideoCapture._remove_fileJs< 7>>$. ) ) & Idn % % % % % & &rfilenamec|jrtddSt|jtjr=|jjdkrtddS|jdkrtddS|jtddS|s%tj }d|d d }||_ ||_ | \|_ |_|s%|td dS|j|jj|_d |_dS) a Start the capturing process Args: filename (str): name of the output file framerate (int, float): framerate of the video if filename is not specified it will be generated based on the datetime. zCapturing is already startedNf1z0source type: moderngl.Texture must be type `f1` z=source type: moderngl.Texture must have at least 3 componentsz-No source defined, there is nothing to recordvideo_z %Y%m%d_%H%M%Sz.mp4zCapturing failedT)rprintrrrrdtyperdatetimenowrrr.rrr(releaserstarttimer)rr5r r=s r start_capturezBaseVideoCapture.start_captureOsR ?  0 1 1 1 F dlH$4 5 5 |!T))HIII!##UVVV <  A B B B F 8#''))C77777H!#$(LLNN! T\ !!  LLNNN $ % % % F +*rc|jsdS|jdSd|jz }|jj|jz |kr|jj|_t |jtjr2|j d}| |dS|j }| |dSdS)zE Save function to call at the end of render function Ng?r8)r) rrrrr@rrr Framebufferreadr%)rdtrs rsavezBaseVideoCapture.saves  F <  F 4? " ; do - 2 2"k.DO$,(<== ( ))Q)77  ''''' ))++  ''''' 3 2rc|jrD||jt d|jd|_dS)z, Stop the recording process z Video file succesfully saved as N)rr*rstopr:rr's rr>zBaseVideoCapture.releasesY ? G    K      ET^EE F F Fr)r )r N)Nr )__name__ __module__ __qualname____doc__rrrrCintfloatrrr%boolr(r*tupler.r4rstrrArFr>rrrr s]  (*77h&(<<=7e$7777,<<<<<<sEz9J1 1111f((((2      rr) r<r0typingrrrrmoderngl_window.timers.clockrrrRrrrUs ''''''''''......ZZZZZZZZZZr