l jPndZddlmZddgZddlZddlZddlZddl Z ddl m Z m Z ddl mZddlmZmZmZddlZddlZdd lmZdd lmZmZdd lTdd lmZdd lmZddl m!Z!ddl"m#Z#m$Z$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-erddl.mZ/ddl0m1Z1ddl2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9e:j;de:j<ej=j<e:j>ej=j>e:j?ej=j?iZ@eAj;deAj<ejBj<eAjCejBjCeAjDejBjDiZEGddZFGddZGdS)zLA camera converts the mobjects contained in a Scene into an array of pixels.) annotationsCamera"BackgroundColoredVMobjectDisplayerN)CallableIterable)reduce) TYPE_CHECKINGAnySelf)Image)configlogger)*)Mobject)PMobject)VMobject) ManimColorParsableManimColorcolor_to_int_rgba)extract_mobject_family_members)get_full_raster_image_path)list_difference_update)cross2dAbstractImageMobject)FloatRGBA_ArrayFloatRGBALike_Array ManimFloatManimInt PixelArrayPoint3D Point3D_Arrayc>eZdZdZdedddddddddddddfdd%Zdd'Zedd)Zej dd+Zedd,Z e j dd.Z dd2Z dd5Z ddd8Z dd9Z ddd=Z dddBZ dddCZ dddDZddGZddHZddJZddKZ dddRZddSZddTZddUZddWZddZZdd[Zdd^Zdd`ZddcZ dddZ!ddgZ"ddhZ# dddiZ$ dddkZ%ddlZ&ddnZ'ddpZ(ddsZ)ddyZ*dd|Z+ddZ,ddZ-ddZ.ddZ/ddZ0ddZ1ddZ2ddZ3ddZ4ddZ5ddZ6ddZ7dS)ravBase camera class. This is the object which takes care of what exactly is displayed on screen at any given moment. Parameters ---------- background_image The path to an image that should be the background image. If not set, the background is filled with :attr:`self.background_color` background What :attr:`background` is set to. By default, ``None``. pixel_height The height of the scene in pixels. pixel_width The width of the scene in pixels. kwargs Additional arguments (``background_color``, ``background_opacity``) to be set. NRGBAuint8g{Gz?Tbackground_image str | None frame_centerr! image_modestr n_channelsintpixel_array_dtypecairo_line_width_multiplefloat use_z_indexbool backgroundPixelArray | None pixel_height int | None pixel_width frame_height float | None frame_width frame_ratebackground_colorParsableManimColor | Nonebackground_opacitykwargsr returnNonec ||_||_||_||_||_||_||_||_d|_| td} | |_ | td} | |_ | td} | |_ | td} | |_ | td} | |_|%tjtd|_ntj||_|td|_n||_td|_t)j|jj|_i|_|||dS)Nr5r7r8r:r;r<r>)r'r)r*r,r.r/r1r3%background_colored_vmobject_displayerr r5r7r8r:r;rparse_background_color_background_opacitymax_allowable_normnpiinfomax rgb_max_valpixel_array_to_cairo_contextinit_backgroundresize_frame_shapereset)selfr'r)r*r,r.r/r1r3r5r7r8r:r;r<r>r?s N/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/camera/camera.py__init__zCamera.__init__Ss&!1($$!2)B&&$  2  !.1L(   /K&  !.1L(   /K&   -J$  #1;1A)*22D " "&0%56F%G%GD "  %.45I.JD $ $'9D $#)"78D$:;;?FH)  !!! memoc8d|_tj|SN)canvascopy)rPrTs rQ __deepcopy__zCamera.__deepcopy__s yrSrc|jSrV)rErPs rQr<zCamera.background_colors %%rScolorc<||_|dSrV)rErM)rPr\s rQr<zCamera.background_colors"!& rSc|jSrV)rFr[s rQr>zCamera.background_opacitys ''rSalphac<||_|dSrV)rFrM)rPr_s rQr>zCamera.background_opacitys"#(  rSmobjectrLtype[VMobject] | type[PMobject] | type[AbstractImageMobject] | type[Mobject]cddlm}t|jt|j||jtdi|_|jD]}t||r|cStd|d)aIReturn the type of mobject, if it is a type that can be rendered. If `mobject` is an instance of a class that inherits from a class that can be rendered, return the super class. For example, an instance of a Square is also an instance of VMobject, and these can be rendered. Therefore, `type_or_raise(Square())` returns True. Parameters ---------- mobject The object to take the type of. Notes ----- For a list of classes that can currently be rendered, see :meth:`display_funcs`. Returns ------- Type[:class:`~.Mobject`] The type of mobjects, if it can be rendered. Raises ------ :exc:`TypeError` When mobject is not an instance of a class that can be rendered. rc|SrV)batchpas rQz&Camera.type_or_raise..surSzDisplaying an object of class z is not supported) mobject.types.image_mobjectrr$display_multiple_vectorized_mobjectsr%display_multiple_point_cloud_mobjectsdisplay_multiple_image_mobjectsr display_funcs isinstance TypeError)rPrar_types rQ type_or_raisezCamera.type_or_raises: GFFFFF d? d@ $"F ,,   '  E'5))   QQQQRRRrS new_height new_widthc||_||_|||dS)a/This method resets the height and width of a single pixel to the passed new_height and new_width. Parameters ---------- new_height The new height of the entire scene in pixels new_width The new width of the entire scene in pixels N)r7r5rMrNrO)rPrsrts rQreset_pixel_shapezCamera.reset_pixel_shapesI%&  !!! rSrfixed_dimensionc|j}|j}|j}|j}||z }|dkr||z }n||z}||_||_dS)a Changes frame_shape to match the aspect ratio of the pixels, where fixed_dimension determines whether frame_height or frame_width remains fixed while the other changes accordingly. Parameters ---------- fixed_dimension If 0, height is scaled with respect to width else, width is scaled with respect to height. rN)r5r7r8r:)rPrwr5r7r8r: aspect_ratios rQrNzCamera.resize_frame_shapese( & ( & "\1 a  &5LL&5K(&rSc|j}|j}|jt|j}t j||j}tj |d|d|f|_ |j |j |_ dSt|j|j}tj|||jf|j |_ ||j ddddf<dS)zInitialize the background. If self.background_image is the path of an image the image is set as background; else, the default background color fills the background. Ndtype)r5r7r'rr openconvertr*rHarrayr3astyper.rr<r>zerosr,)rPheightwidthpathimagebackground_rgbas rQrMzCamera.init_backgrounds "   ,-d.CDDDJt$$,,T_==E!huoogvgvvo>DO"o44T5KLLDOOO/%'O!h0,DO%4DOAAAqqqD ! ! !rS pixel_array PixelArray | list | tuple | None Image.ImagecJ||j}tj||jS)agReturns an image from the passed pixel array, or from the current frame if the passed pixel array is none. Parameters ---------- pixel_array The pixel array from which to get an image, by default None Returns ------- PIL.Image.Image The PIL image of the array. Nmode)rr fromarrayr*rPrs rQ get_imagezCamera.get_image-s)"  *K{AAAArSFPixelArray | list | tupleconvert_from_floatsr cftj|}|rtjfdd|}|S)aConverts a pixel array from values that have floats in then to proper RGB values. Parameters ---------- pixel_array Pixel array to convert. convert_from_floats Whether or not to convert float values to ints, by default False Returns ------- np.array The new, converted pixel array. cH|jzjSrV)rKrr.)frPs rQriz,Camera.convert_pixel_array..Ws 1t//778NOOrSrd)rHrapply_along_axis)rPrrretvals` rQconvert_pixel_arrayzCamera.convert_pixel_arrayBsG$+&&  (OOOOF  rSc|||}t|dr|jj|jks ||_dS|ddddddf|jddddddf<dS)aHSets the pixel array of the camera to the passed pixel array. Parameters ---------- pixel_array The pixel array to convert and then set as the camera's pixel array. convert_from_floats Whether or not to convert float values to proper RGB values, by default False rN)rhasattrrshape)rPrrconverted_arrays rQset_pixel_arrayzCamera.set_pixel_array]s'+&>&> ,' '  D- ( ( A &/*???+:D   )8111aaa(@D QQQ111W % % %rSc<||||_dS)aPSets the background to the passed pixel_array after converting to valid RGB values. Parameters ---------- pixel_array The pixel array to set the background to. convert_from_floats Whether or not to convert floats values to proper RGB valid ones, by default False N)rr3)rPrrs rQset_backgroundzCamera.set_backgroundus 22;@STTrScoords_to_colors_func"Callable[[np.ndarray], np.ndarray]ctjd|}tj|d|}tjd||dS)a Makes a pixel array for the background by using coords_to_colors_func to determine each pixel's color. Each input pixel's color. Each input to coords_to_colors_func is an (x, y) pair in space (in ordinary space coordinates; not pixel coordinates), and each output is expected to be an RGBA array of 4 floats. Parameters ---------- coords_to_colors_func The function whose input is an (x,y) pair of coordinates and whose return values must be the colors for that point Returns ------- np.array The pixel array which can then be passed to set_background. zStarting set_backgroundrdzEnding set_backgroundT)r)rinfoget_coords_of_all_pixelsrHrr)rPrcoordsnew_backgrounds rQmake_background_from_funcz Camera.make_background_from_funcse&  -.....00,-BAvNN +,,,''D'QQQrScV|||dS)a! Sets the background to a pixel array using coords_to_colors_func to determine each pixel's color. Each input pixel's color. Each input to coords_to_colors_func is an (x, y) pair in space (in ordinary space coordinates; not pixel coordinates), and each output is expected to be an RGBA array of 4 floats. Parameters ---------- coords_to_colors_func The function whose input is an (x,y) pair of coordinates and whose return values must be the colors for that point N)rr)rPrs rQset_background_from_funczCamera.set_background_from_funcs- D::;PQQRRRRRrSr c:||j|S)zResets the camera's pixel array to that of the background Returns ------- Camera The camera object after setting the pixel array. )rr3r[s rQrOz Camera.resets T_--- rSc0||dSrV)r)rPr3s rQset_frame_to_backgroundzCamera.set_frame_to_backgrounds Z(((((rSmobjectsIterable[Mobject]include_submobjectsexcluded_mobjects list | None list[Mobject]c|r?t||jd}|r&t||j}t||}t|S)aUsed to get the list of mobjects to display with the camera. Parameters ---------- mobjects The Mobjects include_submobjects Whether or not to include the submobjects of mobjects, by default True excluded_mobjects Any mobjects to exclude, by default None Returns ------- list list of mobjects T)r1only_those_with_points)r1)rr1rlist)rPrrr all_excludeds rQget_mobjects_to_displayzCamera.get_mobjects_to_displaysr.  J5 ,'+H ! J=% $ 0    2(LIIH~~rSc |j}|j}|j}ttj|d|d|dz z k|d|d|dz zk|d|d|dz zk| d|d|dz z kg S)aChecks whether the passed mobject is in frame or not. Parameters ---------- mobject The mobject for which the checking needs to be done. Returns ------- bool True if in frame, False otherwise. rrd) r)r8r:ropor_ get_right get_bottomget_leftget_top)rPrafcfhfws rQ is_in_framezCamera.is_in_frames      F!!##A&Aa7""$$Q'"Q%"q&.8  ""1%1Q6!!!$r!urAv~5      rSc  |j|gfi|S)a=Capture mobjects by storing it in :attr:`pixel_array`. This is a single-mobject version of :meth:`capture_mobjects`. Parameters ---------- mobject Mobject to capture. kwargs Keyword arguments to be passed to :meth:`get_mobjects_to_display`. )capture_mobjects)rPrar?s rQcapture_mobjectzCamera.capture_mobjects!%t$gY99&999rSc |j|fi|}tj||jD].\}}|j|t ||j/dS)aCapture mobjects by printing them on :attr:`pixel_array`. This is the essential function that converts the contents of a Scene into an array, which is then converted to an image or video. Parameters ---------- mobjects Mobjects to capture. kwargs Keyword arguments to be passed to :meth:`get_mobjects_to_display`. Notes ----- For a list of classes that can currently be rendered, see :meth:`display_funcs`. N)ritgroupbyrrrnrr)rPrr? group_typegroups rQrzCamera.capture_mobjectssu404/CCFCC!#Hd6H!I!I J J J *D z *4;;8H I I I I J JrScairo.Context | NonecR|jt|dS)a/Returns the cached cairo context of the passed pixel array if it exists, and None if it doesn't. Parameters ---------- pixel_array The pixel array to check. Returns ------- cairo.Context The cached cairo context. N)rLgetidrs rQget_cached_cairo_contextzCamera.get_cached_cairo_context3s#044R __dKKKrSctx cairo.Contextc4||jt|<dS)zCaches the passed Pixel array into a Cairo Context Parameters ---------- pixel_array The pixel array to cache ctx The context to cache it into. N)rLr)rPrrs rQcache_cairo_contextzCamera.cache_cairo_contextCs>A)"[//:::rSc ||}|r|S|j}|j}|j}|j}|j}t j|j t j ||}t j |} | ||| t j||z dd||z |dz |d||z zz |dz |d||z zz||| | S)aReturns the cairo context for a pixel array after caching it to self.pixel_array_to_cairo_context If that array has already been cached, it returns the cached version instead. Parameters ---------- pixel_array The Pixel array to get the cairo context of. Returns ------- cairo.Context The cairo context of the pixel array. rrdr)rr7r5r:r8r)cairo ImageSurfacecreate_for_datadata FORMAT_ARGB32Contextscale set_matrixMatrixr) rPr cached_ctxpwphrrrsurfacers rQget_cairo_contextzCamera.get_cairo_contextOs& 22;??             $44          mG$$ "b  Lbr' a2a5BG,,a2a5BG,,       c222 rS vmobjectslist[VMobject]ct|dkrdStj|d}|D]4\}}|r||||||5dS)zDisplays multiple VMobjects in the pixel_array Parameters ---------- vmobjects list of VMobjects to display pixel_array The pixel array rNc*|SrVget_background_image)vms rQriz=Camera.display_multiple_vectorized_mobjects..sR=T=T=V=VrS)lenrr-display_multiple_background_colored_vmobjects1display_multiple_non_background_colored_vmobjects)rPrrbatch_image_pairsrrgs rQrkz+Camera.display_multiple_vectorized_mobjects|s y>>Q   FJy2V2VWW-  LE5 BB5+VVVVFF   rSIterable[VMobject]cf||}|D]}|||dS)aDisplays multiple VMobjects in the cairo context, as long as they don't have background colors. Parameters ---------- vmobjects list of the VMobjects pixel_array The Pixel array to add the VMobjects to. N)rdisplay_vectorized)rPrrrvmobjects rQrz8Camera.display_multiple_non_background_colored_vmobjectssI$$[11! 3 3H  # #Hc 2 2 2 2 3 3rSrrc||||||d|||||||S)aDisplays a VMobject in the cairo context Parameters ---------- vmobject The Vectorized Mobject to display ctx The cairo context to use. Returns ------- Camera The camera object Tr3)set_cairo_context_path apply_stroke apply_fill)rPrrs rQrzCamera.display_vectorizedsd ##C222 #xD999 X&&& #x((( rSc|||j}t|dkr|S|||}|D]}||}||d}|j|dd|D]0\}} } } |jg| dd| dd| ddR1| |d|dr| |S)a)Sets a path for the cairo context with the vmobject passed Parameters ---------- ctx The cairo context vmobject The VMobject Returns ------- Camera Camera object after setting cairo_context_path rNrd) transform_points_pre_displaypointsrnew_pathgen_subpaths_from_points_2d#gen_cubic_bezier_tuples_from_points new_sub_pathmove_tocurve_toconsider_points_equals_2d close_path) rPrrrsubpathssubpathquadsstart_p0p1p2p3s rQrzCamera.set_cairo_context_paths1228X_MM v;;!  K 77?? ! !G@@IIE      AJE CKrr # ##( 8 8RR 7b!f7r"1"v72A27777711'!*gbkJJ !    rSrgbasrct|dkr+|jg|dddd|ddRn|}|||}t jt jd|D}tj ddt|}t||d D]$\}}|j |g|ddd|dR%| ||S) a\Sets the color of the cairo context Parameters ---------- ctx The cairo context rgbas The RGBA array with which to color the context. vmobject The VMobject with which to set the color. Returns ------- Camera The camera object rrrdNrc3*K|]}|ddVdS)Nrdrf).0points rQ z1Camera.set_cairo_context_color..s*2Q2Q5!92Q2Q2Q2Q2Q2QrSTstrict) rset_source_rgba!get_gradient_start_and_end_pointsrrLinearGradientrchainrHlinspacezipadd_color_stop_rgba set_source) rPrr rrpatoffsetsrgbaoffsets rQset_cairo_context_colorzCamera.set_cairo_context_colors& u::?? C  >q!%R% >%(1+ > > > > >??AAF66xHHF&2Q2Q&2Q2Q2Q(RSCk!QE 33G #E74 @ @ @ G G f''FaeeFd1gFFFFF NN3    rSc|||||||S)zFills the cairo context Parameters ---------- ctx The cairo context vmobject The VMobject Returns ------- Camera The camera object. )r get_fill_rgbas fill_preserve)rPrrs rQrzCamera.apply_fillsA $$S$*=*=h*G*GRRR  rSc||}|dkr|S|||||||||jz|j|jz z|jtjkr%| t|j|j tjkr%| t|j ||S)aApplies a stroke to the VMobject in the cairo context. Parameters ---------- ctx The cairo context vmobject The VMobject background Whether or not to consider the background when applying this stroke width, by default False Returns ------- Camera The camera object with the stroke applied. rr)get_stroke_widthr get_stroke_rgbasset_line_widthr/r: joint_type LineJointTypeAUTO set_line_join LINE_JOIN_MAP cap_style CapStyleType set_line_cap CAP_STYLE_MAPstroke_preserve)rPrrr3rs rQrzCamera.apply_strokes())*55 A::K $$   ! !(z ! B B     , -$"22 4    -"4 4 4   mH,?@ A A A  !2 2 2   ]8+=> ? ? ?  rSrc,||S)aGets the RGBA array for the stroke of the passed VMobject. Parameters ---------- vmobject The VMobject background Whether or not to consider the background when getting the stroke RGBAs, by default False Returns ------- np.ndarray The RGBA array of the stroke. )r&)rPrr3s rQr&zCamera.get_stroke_rgbas9s&((444rSc*|S)zReturns the RGBA array of the fill of the passed VMobject Parameters ---------- vmobject The VMobject Returns ------- np.array The RGBA Array of the fill of the VMobject )r")rPrs rQr"zCamera.get_fill_rgbasNs&&(((rSrcF|jt||_|jS)a"Returns the background_colored_vmobject_displayer if it exists or makes one and returns it if not. Returns ------- BackgroundColoredVMobjectDisplayer Object that displays VMobjects that have the same color as the background. )rCrr[s rQ)get_background_colored_vmobject_displayerz0Camera.get_background_colored_vmobject_displayer]s+  5 =2488  699rS cvmobjectscn|}|j|}||||S)a0Displays multiple vmobjects that have the same color as the background. Parameters ---------- cvmobjects List of Colored VMobjects pixel_array The pixel array. Returns ------- Camera The camera object. )r5displayoverlay_rgba_array)rPr6r displayercvmobject_pixel_arrays rQrz4Camera.display_multiple_background_colored_vmobjectsosA"BBDD 1 1: >  -BCCC rS pmobjectsIterable[PMobject]c |D]=}|||j|j||j|>dS)zDisplays multiple PMobjects by modifying the passed pixel array. Parameters ---------- pmobjects List of PMobjects pixel_array The pixel array to modify. N)display_point_cloudrr adjusted_thickness stroke_width)rPr<rpmobjects rQrlz,Camera.display_multiple_point_cloud_mobjectss^"  H  $ $''(=>>       rSrBrrr" thicknessc*t|dkrdS|||}|||}|jd}|j|z|j}t|}|t|z} tj|g| z ||f}| |} || }|| }|j } |j } tjd| gd} | d} tj || dddf}|d}| | | z|f}|||<| | | |f|ddddf<dS)aDisplays a PMobject by modifying the pixel array suitably. TODO: Write a description for the rgbas argument. Parameters ---------- pmobject Point Cloud Mobject points The points to display in the point cloud mobject rgbas thickness The thickness of each point of the PMobject pixel_array The pixel array to modify. rNrdrr-r{)rdr)rpoints_to_pixel_coordsthickened_coordinatesrrKrr.rHrreshapeon_screen_pixelsr5r7dot)rPrBrr rCr pixel_coordsrgba_len target_lenfactoron_screen_indicesrr flattenerindicesnew_pas rQr?zCamera.display_point_clouds4 v;;!   F228VDD 11, JJ $Q'!E)11$2HII&& s5zz)%6)**22J3IJJ 11,??#$56 '(    HaWE222 %%f-- &y11!!!Q$7..''$$b2gx%899w"NNBH+=>> AAAqqqDrSimage_mobjectsIterable[AbstractImageMobject]c<|D]}|||dS)zDisplays multiple image mobjects by modifying the passed pixel_array. Parameters ---------- image_mobjects list of ImageMobjects pixel_array The pixel array to modify. N)display_image_mobject)rPrRr image_mobjects rQrmz&Camera.display_multiple_image_mobjectss:, C CM  & &}k B B B B C CrSrVr np.ndarrayc Dtj|d}tjddg|jdgd|jg|j|jgg}|||j tj }tjtd|Dtd|Dg}|z||z}td|Dtd|Df}fdd Dfd td D}tj|d } tj| } || } | | } | dkrdS|| d z d z} || d z d z}t#t%| | | z }t#t%| || z }t||}|dkrdSg}t'|dD]U\\}}\}}|||d ddd| |z| |zg|ddd||d | |z| |zgVtj|tj}|d tj} tj||}n#tjj$rYdSwxYw||tjj||j}tjtj|j|jfd}| ||d|d |d|dz|d |d zf|!||dS)aDisplay an :class:`~.ImageMobject` by changing the ``pixel_array`` suitably. Parameters ---------- image_mobject The :class:`~.ImageMobject` to display. pixel_array The pixel array to put the :class:`~.ImageMobject` in. r$rrcg|]\}}|Srfrfrxys rQ z0Camera.display_image_mobject..666DAqa666rScg|]\}}|SrfrfrZs rQr]z0Camera.display_image_mobject..r^rScg|]\}}|SrfrfrZs rQr]z0Camera.display_image_mobject.. 2221!222rScg|]\}}|SrfrfrZs rQr]z0Camera.display_image_mobject.. rarSc g|] }| Srfrf)ri target_coordss rQr]z0Camera.display_image_mobject..sCCCM!,CCCrS)rrr rdc>g|]}|dzdz|z S)rr%rf)rrdordered_verticess rQr]z0Camera.display_image_mobject..s2WWW!1q5A+.1A!1DDWWWrSr%r)axisNg?Trr{)sizemethodrresample)box)"r rget_pixel_arrayrHrrrpoints_to_subpixel_coordsrrint64minrJrangelinalgnormargmaxabsrrappendfloat64rGsolve LinAlgError transform Transform PERSPECTIVEresampling_algorithmrr5r7pasteoverlay_PIL_image)rPrVr sub_imageoriginal_coordsint_target_coords shift_vector target_sizesidesside_lengths_in_pixelslongest_side_index longest_sidelongest_side_length_in_pixels previous_side next_sideh1h2"height_from_longest_side_in_pixelshomography_matrixr[r\XYAbtransform_coefficients full_imagergres @@rQrUzCamera.display_image_mobjects3OM$A$A$C$C&QQQ (A!$I$%)"23    66 =/  *00::x66$5666766$56667   % \) 22 1222 3 22 1222 3 DCCClCCCWWWWeTUhhWWW!#A!>!>Y'=>>/0 (>?Q(R% (A - - F1A5:; -1Q67 }55 6 69V V y11 2 25R R-0R[[* - 3 3 F!-NNN I INFQFQ  $ $aAq!QQQ%G H H H  $ $aAq!QQQ%G H H H H H&bj 9 9 9  # #A & & - -bj 9 9 %'Y__Q%:%: " "y$    FF  ''?.'"7 (  _ Hd')9: ; ;     QQQ+a.0Q+a.0      {J77777s( K K! K! new_arraycX||||dS)zOverlays an RGBA array on top of the given Pixel array. Parameters ---------- pixel_array The original pixel array to modify. new_array The new pixel array to overlay. N)rr)rPrrs rQr9zCamera.overlay_rgba_arrayUs, {DNN9,E,EFFFFFrSrr ctjtj|||d|ddddf<dS)zOverlays a PIL image on the passed pixel array. Parameters ---------- pixel_array The Pixel array image The Image to overlay. r&r{N)rHrr alpha_compositer)rPrrs rQrzCamera.overlay_PIL_imagecsPH  !$.."="=u E E   AAAqqqDrScvtj||jks|Stjtjjd|}||jk}||ddf}||}tj|t|df|j dd}|j|z|z }|||<|S)aIf any of the points in the passed array are out of the viable range, they are adjusted suitably. Parameters ---------- points The points to adjust Returns ------- np.array The adjusted points. rN) rHanyrGrrsrtrepeatrGrr)rPrnormsviolator_indices violatorsviolator_normsreshaped_normsrescaleds rQadjust_out_of_range_pointsz!Camera.adjust_out_of_range_pointsrsvft6677 M#BINAv>> 4#::+QQQ./ /0  " "C$7$7#; < < LO    *Y6G#+  rScztjtj|stjd}|S)N)rr )rHallisfiniterrPrars rQrz#Camera.transform_points_pre_displays6vbk&))** &Xf%%F rSnpt.NDArray[ManimFloat]cZ|||}||jz }tjt |df}|j}|j}|j}|j}||z } |dz } ||z } |dz } | dz} |dddf| z| z|dddf<|dddf| z| z|dddf<|S)Nrdrrr) rr)rHrrr5r7r8r:) rPrarshifted_pointsresultr5r7r8r: width_mult width_add height_mult height_adds rQroz Camera.points_to_subpixel_coordss227FCC$"333v;;*++( & ( &  ;. !O "\1 !A% r %aaad+j89Dqqq!t %aaad+k9JFqqq!t  rSnpt.NDArray[ManimInt]ch|||tjSrV)rorrHrprs rQrEzCamera.points_to_pixel_coordss* --gv>>EEbhOOOrSrJc ttj|dddfdk|dddf|jk|dddfdk|dddf|jkgS)a$Returns array of pixels that are on the screen from a given array of pixel_coordinates Parameters ---------- pixel_coords The pixel coords to check. Returns ------- np.array The pixel coords on screen. Nrr)rrand_r7r5)rPrJs rQrHzCamera.on_screen_pixelssz GQQQT"a'QQQT"T%55QQQT"a'QQQT"T%66     rSctjtdtd}tj|j|j}||z }d|dz |zzS)a/Computes the adjusted stroke width for a zoomed camera. Parameters ---------- thickness The stroke width of a mobject. Returns ------- float The adjusted stroke width that reflects zooming in with the camera. r5r7r)raddr r5r7)rPrCbig_sumthis_sumrMs rQr@zCamera.adjusted_thicknesssT~ 6}8MNN&!2D4DEE8#IMV+++rSct|}tt| dzdz|dzdz}tjtt j||S)zDetermine a list of vectors used to nudge two-dimensional pixel coordinates. Parameters ---------- thickness Returns ------- np.array rdr)r-rrrrHrrproduct)rPrC_ranges rQget_thickening_nudgeszCamera.get_thickening_nudgess^ NN eYJ!Oa/a!1CDDEExRZ7788999rSc||}tjfd|Dj}|dzdfS)aQReturns thickened coordinates for a passed array of pixel coords and a thickness to thicken by. Parameters ---------- pixel_coords Pixel coordinates thickness Thickness Returns ------- np.array Array of thickened pixel coords. cg|]}|zSrfrf)rnudgerJs rQr]z0Camera.thickened_coordinates..s J J J%!5 J J JrSrd)rrHrrjrG)rPrJrCnudgesrjs ` rQrFzCamera.thickened_coordinatess`$++I66x J J J J6 J J JKK  ##TQYN333rSc,tj|j|jg}tj|j|jg}tj|j|jgdddddd}||z|z }||dz z }|dz}|S)zReturns the cartesian coordinates of each pixel. Returns ------- np.ndarray The array of cartesian coordinates. Nrrrdr)rr)rHrr:r8r7r5rP transpose)rPfull_space_dimsfull_pixel_dimsuncentered_pixel_coordsuncentered_space_coordscentered_space_coordss rQrzCamera.get_coords_of_all_pixelss(D$4d6G#HII(D$4d6G#HII#%*d.?AQ-R"S"S DbD# )Aq!    $o 5 #!8?Q;N O!6 ?$$rS)"r'r(r)r!r*r+r,r-r.r+r/r0r1r2r3r4r5r6r7r6r8r9r:r9r;r9r<r=r>r9r?r r@rA)rTr r@r)r@r)r\rr@rA)r@r0)r_r0r@rA)rarr@rb)rsr0rtr0r@rA)r)rwr-r@rAr@rArV)rrr@r)F)rrrr2r@r )rrrr2r@rA)rrr@r )rrr@rA)r@r )r3r r@rA)TN)rrrr2rrr@r)rarr@r2)rarr?r r@rA)rrr?r r@rA)rr r@r)rr rrr@rA)rr r@r)rrrr r@rA)rrrr r@rA)rrrrr@r )rrrrr@r )rrr rrrr@r )rrrrr3r2r@r )rrr3r2r@r)rrr@r)r@r)r6rrr r@r )r<r=rr r@rA) rBrrr"r rrCr0rr r@rA)rRrSrr r@rA)rVrrrWr@rA)rrWrrWr@rA)rrWrr r@rA)rrWr@rW)rarrr"r@r")rarrr"r@r)rarrr"r@r)rJrWr@r )rCr0r@r0)rCr0r@r )rJrWrCr0r@r )r@r )8__name__ __module__ __qualname____doc__ORIGINrRrYpropertyr<setterr>rrrvrNrMrrrrrrrOrrrrrrrrrkrrrr rrr&r"r5rrlr?rmrUr9rrrrorErHr@rrFrrfrSrQrr=s#.(, & !(+/ (,#'"&%)$(#'6:+/!LLLLL\&&&X&(((X(-S-S-S-S^"'''''244446?CBBBBB,SX8SXAAAAA2SX U U U U U RRRR4SSSS    ))))%))- #####J    6:::: JJJJFLLLL A A A A++++Z03333"*    DB(JO'''''T6;55555* ) ) ) )::::$6*3?3?3?3?jCCCC"n8n8n8n8` G G G G     :    4PPPP    0,,,,(::::"44440%%%%%%rScFeZdZdZddZddZ dddZddZd dZd!dZ dS)"rzAuxiliary class that handles displaying vectorized mobjects with a set background image. Parameters ---------- camera Camera object to use. camerarc||_i|_tj|j|_|dSrV)rfile_name_to_pixel_array_maprHrrreset_pixel_array)rPrs rQrRz+BackgroundColoredVMobjectDisplayer.__init__Bs> CE)8F$677      rSr@rAc&d|jddddf<dS)Nr)rr[s rQrz4BackgroundColoredVMobjectDisplayer.reset_pixel_arrayHs!!"AAArSr$background_arrayr rtr0rsrr+ctj|}||}|||f}t j|S)aResizes the pixel array representing the background. Parameters ---------- background_array The pixel new_width The new width of the background new_height The new height of the background mode The PIL image mode, by default "RGBA" Returns ------- np.array The numpy pixel array of the resized background. )r rr~resizerHr)rPrrtrsrr resized_images rQresize_background_arrayz:BackgroundColoredVMobjectDisplayer.resize_background_arrayKsK2 011 d## i%<== x &&&rSrc|jdd\}}|jddkrdnd}|||||S)aeResizes the background array to match the passed pixel array. Parameters ---------- background_array The prospective pixel array. pixel_array The pixel array whose width and height should be matched. Returns ------- np.array The resized background array. Nrdr%r$RGB)rr)rPrrrrrs rQ resize_background_array_to_matchzCBackgroundColoredVMobjectDisplayer.resize_background_array_to_matchisO"$)"1"- $*1-22vv++,>z;WWJ7A))4rSr6rr4ctj|d}d}|D]\}}||}|j}|j||t j||dzdz |jj }||}nt j ||}| |S)zDisplays the colored VMobjects. Parameters ---------- *cvmobjects The VMobjects Returns ------- np.array The pixel array with the `cvmobjects` displayed. c*|SrVr)cvs rQriz.sb>U>U>W>WrSNr0r{) rrrrrrrHrrr.maximumr) rPr6r curr_arrayrrgrrrs rQr8z*BackgroundColoredVMobjectDisplayer.displaysJz3W3WXX - % %LE5#88?? *K K I I   !K$6$6w$?$??#Ek3I!& Z I>>  " " $ $ $ $rSN)rrr)r$) rr rtr0rsr0rr+r@r )rr rr r@r )rrr@r )r6rr@r4) rrrrrRrrrrr8rfrSrQrr8s!!!! #### '''''<SSSS*>rS)Hr __future__r__all__rX itertoolsroperatorrpathlibcollections.abcrr functoolsrtypingr r r rnumpyrHPILr manim._configr rmanim.constantsmanim.mobject.mobjectr'manim.mobject.types.point_cloud_mobjectr&manim.mobject.types.vectorized_mobjectrmanim.utils.colorrrrmanim.utils.familyrmanim.utils.imagesrmanim.utils.iterablesrmanim.utils.space_opsr numpy.typingnpt!manim.mobject.types.image_mobjectr manim.typingrrrrr r!r"r)r*ROUNDLineJoinBEVELMITERr,r.LineCapBUTTSQUAREr0rrrfrSrQr sRR"""""" 9 : ........++++++++++ (((((((())))))<<<<<<;;;;;;OOOOOOOOOO======999999888888)))))) FFFFFF---  t +u})-  v%v%v%v%v%v%v%v%v'DDDDDDDDDDrS