l j|UddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z ddlmZmZddlmZddlmZmZmZmZmZmZmZmZmZmZddlZddlZ ddl!mZ"ddl#m$Z$ddl%m&Z&m'Z'dd l(Tdd l)m*Z*dd l+m,Z,dd l-m.Z.m/Z/m0Z0m1Z1dd l2m3Z3m4Z4ddl5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;ddlZ>ddl?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGddlHmIZIddlJmKZKmLZLmMZMerjddlNmOZOddl+mPZPddl-mQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYe deZge[fZ\de]d<e dge[fZ^de]d<e^e\zZ_de]d<edZ`ede jaZbd3d!Zcgd"Zde jeZfde]d#< e"jgefZhde]d$<Gd%dZiGd&d'eiZjGd(d)eiZkGd*d+Zled,e d-ZmGd.d/eZnd4d2ZodS)5) annotationsN)CallableIterableIteratorSequence) partialmethodwraps)ceil) TYPE_CHECKINGAnyClassVarNeverProtocolSelf TypeAliasTypeVarcastoverload)override)configlogger)*)MethodWithArgs)get_colormap_code)Point3D Point3D_Array Point3DLikePoint3DLike_Array)integer_interpolate interpolate)WHITE ManimColorParsableManimColorcolor_gradient color_to_rgb rgb_to_hex)_Data _Uniforms)batch_by_property list_updatelistify make_even resize_arrayresize_preserving_orderresize_with_interpolation uniq_chain) straight_path)angle_between_vectors normalizerotation_matrix_transpose) Animation ShaderWrapper) FloatRGB_ArrayFloatRGBA_Array ManimFloatMappingFunctionMatrixMNMultiMappingFunction PathFuncTypeVector3D Vector3DLike OpenGLMobjectr_TimeBasedUpdater_NonTimeBasedUpdater_Updater_T_T_np)boundfunc(Callable[[OpenGLMobject], OpenGLMobject]returnc>tdfd }|S)NselfrArJct|D]!}||"|SN) get_familyrefresh_shader_wrapper_id)rLmobrHs ^/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/mobject/opengl/opengl_mobject.pywrapperz'affects_shader_info_id..wrapperasC??$$ , ,C DIII  ) ) + + + + )rLrArJrA)r )rHrSs` rRaffects_shader_info_idrU^s: 4[[[ NrT)rA OpenGLGroup OpenGLPoint_AnimationBuilder _ShaderDType _ShaderDatac~ eZdZUdZded<dejdfgZded<dZd ed <e Z d ed <e Z d ed <e Z ded<e Zded<e Zded<e Zded<e Zded<e Zded<eddddejddddddddfdd0Zdd5Zdd8Zedfd: Zedd<Zedd=ZddAZddBZddDZ ddEZ!eddFZ"ddGZ#ddHZ$ddIZ%ddJZ&ddMZ'ddOZ(e)ddQZ*e)ddRZ+e+j,ddTZ+e)ddUZ-e-j,ddVZ-e)ddWZ.e.j,ddXZ.e/fdd\Z0dd^Z1ddaZ2ddcZ3ddeZ4ddfZ5ddhZ6de7dfddoZ8ddpZ9ddqZ:ddrZ;ddtZ<dduZ=ddvZ>ddwZ? dddzZ@eAfdd}ZBddZCddZDddZEddZFddZGdddZHddZIddddZJ dĐddZKddddZLddZMddZNeOdfddZPddeAe7dddddf ddZQ dʐddZRddZSddfddZTdĐddZUdĐddZVdĐddZWddZXdĐddZYdĐddZZddZ[ddZ\dԐddZ]ddZ^dd„Z_ddĄZ`ddńZa dؐddɄZbddʄZcddd˄Zddd̄Zeddd̈́Zfdېdd΄ZgddτZhddфZide7fddԄZjddքZkelfddلZmeldfddڄZneofddۄZpddބZqdd߄ZrddZsddZtddZuddZveOewdfddZxddZyezfddZ{e|ewzezfddZ}e|ezfddZ~eOee7ddejgdfddZddZddZddZ dĐddZddZddZddZ dĐddZeZ dĐddZeZ dĐddZeZe7fddZe7fdd Ze7fdd Ze7fdd ZdddZe7ejgdfddZ dddZddeAfddZddZ dddZ dddZ dddZ ddd Zdd!Zdd"Zdd%Zdd&Zd d d)Zdd*Zdd d+Zdd,Zdd d-Z d dd0Zdd1Zdd2Zdd3Zdd4Zdd5Zdd6Zdd7Zdd8Zdd9Zdd:Zdd;Zdd<Zdd=Zdd>Zdd?Zdd@ZddAZddBZddCZe7fddEZe7fddFZe7fddGZe7fddHZddIZddJZddLZddNZddOZddQZddRZÐddSZĐddTZŐddUZƐddVZǐddWZe7fddXZe7fddYZe7fddZZe7fdd[Ze7fdd\Z͐dd^Zeϐdd_ZАdd`ZѐddaZҐddbZӐddcZԐdddZՐddeZeצfddjZؐddmZ d d!drZڐd"dtZېd#duZܐddvZeސddwZeސddxZeސddyZeސddzZeސdd{Zeސdd|Zd$dZd%dZ d&d'dZddZd(dZd)dZd*dZd+dZd,dZd-dZddZd.dZd/dZe)d0dZej,d1dZddZxZS(2rAaCMathematical Object: base class for objects that can be displayed on screen. Attributes ---------- submobjects : List[:class:`OpenGLMobject`] The contained objects. points : :class:`numpy.ndarray` The points of the objects. .. seealso:: :class:`~.OpenGLVMobject` zClassVar[Callable[..., None]]_original__init__point)zAClassVar[Sequence[tuple[str, type[np.generic], tuple[int, ...]]]] shader_dtypez ClassVar[str] shader_folderz_Data[Point3D_Array]points bounding_boxz_Data[FloatRGBA_Array]rgbasr(is_fixed_in_frameis_fixed_orientationz%_Uniforms[tuple[float, float, float]]fixed_orientation_centerglossshadowr^NFTcolor1ParsableManimColor | Sequence[ParsableManimColor]opacityfloatdimintrender_primitive texture_pathsdict[str, str] | None depth_testboollisten_to_events model_matrixMatrixMN | None should_rendername str | Nonekwargsr c .| |jjn||_t|di|_t|di|_||_||_||_||_ ||_ ||_ ||_ t| |_t| |_d|_| |_g|_g|_d|_|g|_t-|_d|_| t3jd|_n| |_|||t?j ||_!|"d|_#|j r|$| |_%dS)NdatauniformsrrrT)& __class____name__r{getattrrrrnrprhrirrrsrurorerfrgrw _submobjectsparentsparentfamilysetlocked_data_keysneeds_new_bounding_boxnpeyerx init_data init_updaters init_pointsr"parserl init_colorsshader_indicesapply_depth_testrz)rLrlrnrprhrirrrsrurerfrwrxrzr{r}s rR__init__zOpenGLMobject.__init__s259L00d 18vr1J1J >E *b? ?  18   %54A *!&'8!9!9$)*>$?$?!(1%'713,. ,0 ,06 *-%%,0#  *,&))D   ,D    4>4DU4K4K  48 ? $  ! ! # # ##0rT submobjectsIterable[OpenGLMobject]rJrc8||tS)aCheck that all submobjects are actually instances of :class:`OpenGLMobject`, and that none of them is ``self`` (an :class:`OpenGLMobject` cannot contain itself). This is an auxiliary function called when adding OpenGLMobjects to the :attr:`submobjects` list. This function is intended to be overridden by subclasses such as :class:`OpenGLVMobject`, which should assert that only other OpenGLVMobjects may be added into it. Parameters ---------- submobjects The list containing values to validate. Returns ------- :class:`OpenGLMobject` The OpenGLMobject itself. Raises ------ TypeError If any of the values in `submobjects` is not an :class:`OpenGLMobject`. ValueError If there was an attempt to add an :class:`OpenGLMobject` as its own submobject. )"_assert_valid_submobjects_internalrA)rLrs rR_assert_valid_submobjectsz'OpenGLMobject._assert_valid_submobjectss>66{MRRRrT mob_classtype[OpenGLMobject]c pt|D]\}}t||sdd|jdt|jd|d|dt|jd }t|tr|dz }t |||ur(t dt|jd |d |S) NzOnly values of type z can be added as submobjects of z, but the value z (at index z ) is of type .z4 You can try adding this value into a Group instead.z Cannot add z% as a submobject of itself (at index z).) enumerate isinstancertyperA TypeError ValueError)rLrrisubmob error_messages rRrz0OpenGLMobject._assert_valid_submobjects_internals#;//  IAvfi00 /09+=00)-d)<0000*+00F||,000fm44!NM ...~~ .$t**"5..()...  rTNonec Rtjdi||j|_dSN)super__init_subclass__rr\)clsr}rs rRrzOpenGLMobject.__init_subclass__s0!!++F+++ # rTstrc|jjSrN)rrrLs rR__str__zOpenGLMobject.__str__"s ~&&rTc*t|jSrN)rr{rs rR__repr__zOpenGLMobject.__repr__&s49~~rTotherrobjectctSrNNotImplementedrLrs rR__sub__zOpenGLMobject.__sub__*rTctSrNrrs rR__isub__zOpenGLMobject.__isub__-rrTmobjectctSrNrrLrs rR__add__zOpenGLMobject.__add__0rrTctSrNrrs rR__iadd__zOpenGLMobject.__iadd__3rrTc T|rt|jfi||_dS|j|_dS)a%Sets the default values of keyword arguments. If this method is called without any additional keyword arguments, the original default values of the initialization method of this class are restored. Parameters ---------- kwargs Passing any keyword argument will update the default values of the keyword arguments of the initialization function of this class. Examples -------- :: >>> from manim import Square, GREEN >>> Square.set_default(color=GREEN, fill_opacity=0.25) >>> s = Square(); s.color, s.fill_opacity (ManimColor('#83C167'), 0.25) >>> Square.set_default() >>> s = Square(); s.color, s.fill_opacity (ManimColor('#FFFFFF'), 0.0) .. manim:: ChangedDefaultTextcolor :save_last_frame: config.background_color = WHITE class ChangedDefaultTextcolor(Scene): def construct(self): Text.set_default(color=BLACK) self.add(Text("Changing default values is easy!")) # we revert the colour back to the default to prevent a bug in the docs. Text.set_default(color=WHITE) N)rrr\)rr}s rR set_defaultzOpenGLMobject.set_default6s:V  1)@@@@CLLL0CLLLrTctjd|_tjd|_tjd|_dS)zInitializes the ``points``, ``bounding_box`` and ``rgbas`` attributes and groups them into self.data. Subclasses can inherit and overwrite this method to extend `self.data`. rr^)r^r^)rjrN)rzerosrbrcrdrs rRrzOpenGLMobject.init_datahs<hv&& HV,,Xf%% rTcF||j|jdS)zCInitializes the colors. Gets called upon creation N) set_colorrlrnrs rRrzOpenGLMobject.init_colorsps" tz4<00000rTcdS)zInitializes :attr:`points` and therefore the shape. Gets called upon creation. This is an empty method that can be implemented by subclasses. Nrrs rRrzOpenGLMobject.init_pointsws  rTc \|D]\}}t||||S)aSets attributes. Mainly to be used along with :attr:`animate` to animate setting attributes. Examples -------- :: >>> mob = OpenGLMobject() >>> mob.set(foo=0) OpenGLMobject >>> mob.foo 0 Parameters ---------- **kwargs The attributes and corresponding values to set. Returns ------- :class:`OpenGLMobject` ``self`` )itemssetattr)rLr}attrvalues rRrzOpenGLMobject.sets98"<<>> ' 'KD% D$ & & & & rTrdict[str, Any]cT|D]$}|||j|<%|SrN)copyr)rLrkeys rRset_datazOpenGLMobject.set_datas1 . .C!#Y^^--DIcNN rTrc0|D]}|||j|<|SrNr)rLrrs rR set_uniformszOpenGLMobject.set_uniformss) / /C!)#DM#   rT_AnimationBuilder | Selfc t|S)a- Used to animate the application of a method. .. warning:: Passing multiple animations for the same :class:`OpenGLMobject` in one call to :meth:`~.Scene.play` is discouraged and will most likely not work properly. Instead of writing an animation like :: self.play( my_mobject.animate.shift(RIGHT), my_mobject.animate.rotate(PI) ) make use of method chaining for ``animate``, meaning:: self.play(my_mobject.animate.shift(RIGHT).rotate(PI)) Keyword arguments that can be passed to :meth:`.Scene.play` can be passed directly after accessing ``.animate``, like so:: self.play(my_mobject.animate(rate_func=linear).shift(RIGHT)) This is especially useful when animating simultaneous ``.animate`` calls that you want to behave differently:: self.play( mobject1.animate(run_time=2).rotate(PI), mobject2.animate(rate_func=there_and_back).shift(RIGHT), ) .. seealso:: :func:`override_animate` Examples -------- .. manim:: AnimateExample class AnimateExample(Scene): def construct(self): s = Square() self.play(Create(s)) self.play(s.animate.shift(RIGHT)) self.play(s.animate.scale(2)) self.play(s.animate.rotate(PI / 2)) self.play(Uncreate(s)) .. manim:: AnimateChainExample class AnimateChainExample(Scene): def construct(self): s = Square() self.play(Create(s)) self.play(s.animate.shift(RIGHT).scale(2).rotate(PI / 2)) self.play(Uncreate(s)) .. manim:: AnimateWithArgsExample class AnimateWithArgsExample(Scene): def construct(self): s = Square() c = Circle() VGroup(s, c).arrange(RIGHT, buff=2) self.add(s, c) self.play( s.animate(run_time=2).rotate(PI / 2), c.animate(rate_func=there_and_back).shift(RIGHT), ) .. warning:: ``.animate`` will interpolate the :class:`~.OpenGLMobject` between its points prior to ``.animate`` and its points after applying ``.animate`` to it. This may result in unexpected behavior when attempting to interpolate along paths, or rotations. If you want animations to consider the points between, consider using :class:`~.ValueTracker` with updaters instead. )rXrs rRanimatezOpenGLMobject.animatesp!&&&rTc,|dS)aThe width of the mobject. Returns ------- :class:`float` Examples -------- .. manim:: WidthExample class WidthExample(Scene): def construct(self): decimal = DecimalNumber().to_edge(UP) rect = Rectangle(color=BLUE) rect_copy = rect.copy().set_stroke(GRAY, opacity=0.5) decimal.add_updater(lambda d: d.set_value(rect.width)) self.add(rect_copy, rect, decimal) self.play(rect.animate.set(width=7)) self.wait() See also -------- :meth:`length_over_dim` rlength_over_dimrs rRwidthzOpenGLMobject.width<##A&&&rTrc6||dddS)NrFstretchrescale_to_fitrLrs rRrzOpenGLMobject.width&# E1e44444rTc,|dS)aThe height of the mobject. Returns ------- :class:`float` Examples -------- .. manim:: HeightExample class HeightExample(Scene): def construct(self): decimal = DecimalNumber().to_edge(UP) rect = Rectangle(color=BLUE) rect_copy = rect.copy().set_stroke(GRAY, opacity=0.5) decimal.add_updater(lambda d: d.set_value(rect.height)) self.add(rect_copy, rect, decimal) self.play(rect.animate.set(height=5)) self.wait() See also -------- :meth:`length_over_dim` rjrrs rRheightzOpenGLMobject.height*rrTc6||dddS)NrjFrrrs rRrzOpenGLMobject.heightJrrTc,|dS)zThe depth of the mobject. Returns ------- :class:`float` See also -------- :meth:`length_over_dim` rrs rRdepthzOpenGLMobject.depthNs##A&&&rTc6||dddS)NrFrrrs rRrzOpenGLMobject.depth^rrT new_length resize_func-Callable[[Point3D_Array, int], Point3D_Array]c|t|jkr||j||_||SrN)lenrbrefresh_bounding_box)rLrrs rR resize_pointszOpenGLMobject.resize_pointsbsE T[)) ) )%+dk:>>DK !!### rTrc,t|t|jkr ||jdd<nMt|tjr||_ntj||_||SrN)rrbrrndarrayrarrayr)rLrbs rR set_pointszOpenGLMobject.set_pointsls{ v;;#dk** * *#DKNN  + + + ++--DKK(6**DK !!### rTrH2Callable[[npt.NDArray[_T_np]], npt.NDArray[_T_np]]c |D]*}t|||t||+|SrN)get_array_attrsrr)rLrHrs rRapply_over_attr_arraysz$OpenGLMobject.apply_over_attr_arraysvsM((** ; ;D D$WT4%8%8 9 9 : : : : rT Iterable[str]cdgS)Nrbrrs rRrzOpenGLMobject.get_array_attrs}s zrT new_pointscntj|j|g|_||SrN)rvstackrbr)rLrs rR append_pointszOpenGLMobject.append_pointss1ij 9::  !!### rTc|D]*}|jD] }|j|ddd|j|<!+|S)N)rOr)rLrQrs rRreverse_pointszOpenGLMobject.reverse_pointssT??$$ 4 4Cx 4 4 # ddd 3  4 rTrc,|dS)amGet coordinates of the middle of the path that forms the :class:`~.OpenGLMobject`. Examples -------- .. manim:: AngleMidPoint :save_last_frame: class AngleMidPoint(Scene): def construct(self): line1 = Line(ORIGIN, 2*RIGHT) line2 = Line(ORIGIN, 2*RIGHT).rotate_about_origin(80*DEGREES) a = Angle(line1, line2, radius=1.5, other_angle=False) d = Dot(a.get_midpoint()).set_color(RED) self.add(line1, line2, a, d) self.wait() ?point_from_proportionrs rR get_midpointzOpenGLMobject.get_midpoints*))#...rTr= about_pointPoint3DLike | None about_edgeVector3DLike | Noneworks_on_bounding_boxc||||}|D]}g}|r||j|r'|||D]+}||||dd<|||z |z|dd<,|s|dn|jD]}||S)NT recurse_down)get_bounding_box_pointrO has_pointsappendrbget_bounding_boxrr) rLrHr rrrQarrsarrrs rRapply_points_functionz#OpenGLMobject.apply_points_functions4  :#955jAAK??$$ C CC(*D~~ ( CJ'''$ 4 C0022333 C C&!T#YYCFF!T# "344{BCFF  C % .  % %4 % 8 8 8 8, . .++---- rTc:||j|S)a]Edit points, positions, and submobjects to be identical to another :class:`~.OpenGLMobject`, while keeping the style unchanged. Examples -------- .. manim:: MatchPointsScene class MatchPointsScene(Scene): def construct(self): circ = Circle(fill_color=RED, fill_opacity=0.8) square = Square(fill_color=BLUE, fill_opacity=0.2) self.add(circ) self.wait(0.5) self.play(circ.animate.match_points(square)) self.wait(0.5) )rrbrs rR match_pointszOpenGLMobject.match_pointss" ''' rTc8tjd|_|SNr)remptyrbrs rR clear_pointszOpenGLMobject.clear_pointsshv&&  rTc*t|jSrN)rrbrs rRget_num_pointszOpenGLMobject.get_num_pointss4;rTrc~|jr0tjd|DS|jS)Ncg|] }|j Sr)rb.0sms rR z0OpenGLMobject.get_all_points..sDDDBbiDDDrT)rrrrOrbrs rRget_all_pointszOpenGLMobject.get_all_pointss>   9DD$//2C2CDDDEE E; rTc2|dkSNr)r#rs rRrzOpenGLMobject.has_pointss""$$q((rTc^|jr ||_d|_|jSNF)rcompute_bounding_boxrcrs rRrzOpenGLMobject.get_bounding_boxs2  & 0 $ 9 9 ; ;D */D '  rTcptj|jgd|ddD}t |dkrtjd|jfS|d}|d}||zdz }tj |||gS)Nc3fK|],}||V-dSrN)rrr'rQs rR z5OpenGLMobject.compute_bounding_box..sR~~''((**rTrjrr^r) rrrbrOrrrpminmaxr)rL all_pointsminsmaxsmidss rRr/z"OpenGLMobject.compute_bounding_boxsY  #004     z??a  8QM** *>>!$$D>>!$$D4K1$D8T4.// /rTr recurse_upc||D] }d|_ |r|jD]}||SNT)rOrrr)rLrr:rQrs rRrz"OpenGLMobject.refresh_bounding_boxs]??<00 . .C)-C & &  ., . .++---- rTrbuffc|}|d|z }|d|z}||ko||k}|S)Nrr)rall)rLr]r=bbr7r8rvs rRis_point_touchingzOpenGLMobject.is_point_touching s^ " " $ $!ut|!ut|TM&&((Betm-@-@-B-B rT int | slicect|tr>|}|||S||SrN)rsliceget_group_classsplit __getitem__)rLr GroupClasss rRrHzOpenGLMobject.__getitem__se eU # # @--//J:tzz||77>>? ?zz||''...rTIterator[OpenGLMobject]cDt|SrN)iterrGrs rR__iter__zOpenGLMobject.__iter__sDJJLL!!!rTcDt|SrN)rrGrs rR__len__zOpenGLMobject.__len__s4::<<   rTSequence[OpenGLMobject]c|jSrN)rrs rRrGzOpenGLMobject.split"s rTcd|jD}|gt||_|||jD]}||S)Nc3>K|]}|VdSrNrOr&s rRr3z0OpenGLMobject.assemble_family..&s*CCB CCCCCCrT)rr0rrefresh_has_updater_statusrrassemble_family)rL sub_familiesrs rRrVzOpenGLMobject.assemble_family%s{CC$2BCCC 8j,78  ''))) !!###l % %F  " " $ $ $ $ rTrecursec:|rt|dr|jS|gS)Nr)hasattrr)rLrXs rRrOzOpenGLMobject.get_family.s*  wtX.. ; 6MrTc>d|DS)Nc:g|]}||Sr)r)r'ms rRr)z.5s%???a ????rTrTrs rRfamily_members_with_pointsz(OpenGLMobject.family_members_with_points4s ??4??,,????rT) update_parentmobjectsr_c|r*tdks Jd|d_|tfdDrt jdD]H}||jvr|j|||jvr|j|I| |S)aAdd mobjects as submobjects. The mobjects are added to :attr:`submobjects`. Subclasses of mobject may implement ``+`` and ``+=`` dunder methods. Parameters ---------- mobjects The mobjects to add. Returns ------- :class:`OpenGLMobject` ``self`` Raises ------ :class:`ValueError` When a mobject tries to add itself. :class:`TypeError` When trying to add an object that is not an instance of :class:`OpenGLMobject`. Notes ----- A mobject cannot contain itself, and it cannot contain a submobject more than once. If the parent mobject is displayed, the newly-added submobjects will also be displayed (i.e. they are automatically added to the parent Scene). See Also -------- :meth:`remove` :meth:`add_to_back` Examples -------- :: >>> outer = OpenGLMobject() >>> inner = OpenGLMobject() >>> outer = outer.add(inner) Duplicates are not added again:: >>> outer = outer.add(inner) >>> len(outer.submobjects) 1 Only OpenGLMobjects can be added:: >>> outer.add(3) Traceback (most recent call last): ... TypeError: Only values of type OpenGLMobject can be added as submobjects of OpenGLMobject, but the value 3 (at index 0) is of type int. Adding an object to itself raises an error:: >>> outer.add(outer) Traceback (most recent call last): ... ValueError: Cannot add OpenGLMobject as a submobject of itself (at index 0). rjzCan't set multiple parents.rc3JK|]}|dkVdS)rjN)count)r'elemr`s rRr3z$OpenGLMobject.add..s4==Dx~~d##a'======rTzgAttempted adding some Mobject as a child more than once, this is not possible. Repetitions are ignored.) rrranyrwarningrrrrVrLr_r`rs ` rRaddzOpenGLMobject.add7sD  &x==A%%%'D%%%!%HQK  &&x000 ====H=== = =  NA     - -Gd... ''0007?**&&t,,,  rTindexc|r||_||g||jvr|j||||jvr|j|||S)aInserts a mobject at a specific position into self.submobjects Effectively just calls ``self.submobjects.insert(index, mobject)``, where ``self.submobjects`` is a list. Highly adapted from ``OpenGLMobject.add``. Parameters ---------- index The index at which mobject The mobject to be inserted. update_parent Whether or not to set ``mobject.parent`` to ``self``. )rrrinsertrrrV)rLrirr_s rRrkzOpenGLMobject.inserts&  "!GN &&y111 $* * *   # #E7 3 3 3 w & & O " "4 ( ( (  rTc|r*t|dks Jdd|d_|D]H}||jvr|j|||jvr|j|I||S)aRemove :attr:`submobjects`. The mobjects are removed from :attr:`submobjects`, if they exist. Subclasses of mobject may implement ``-`` and ``-=`` dunder methods. Parameters ---------- mobjects The mobjects to remove. Returns ------- :class:`OpenGLMobject` ``self`` See Also -------- :meth:`add` rjzCan't remove multiple parents.Nr)rrrremoverrVrgs rRrmzOpenGLMobject.removes,  &x==A%%%'G%%%!%HQK  - -G$*** ''000w&&&&t,,,  rTcd||t||j|_|S)aAdd all passed mobjects to the back of the submobjects. If :attr:`submobjects` already contains the given mobjects, they just get moved to the back instead. Parameters ---------- mobjects The mobjects to add. Returns ------- :class:`OpenGLMobject` ``self`` .. note:: Technically, this is done by adding (or moving) the mobjects to the head of :attr:`submobjects`. The head of this list is rendered first, which places the corresponding mobjects behind the subsequent list members. Raises ------ :class:`ValueError` When a mobject tries to add itself. :class:`TypeError` When trying to add an object that is not an instance of :class:`OpenGLMobject`. Notes ----- A mobject cannot contain itself, and it cannot contain a submobject more than once. If the parent mobject is displayed, the newly-added submobjects will also be displayed (i.e. they are automatically added to the parent Scene). See Also -------- :meth:`remove` :meth:`add` )rr*r)rLr`s rR add_to_backzOpenGLMobject.add_to_backs3Z &&x000&x1ABB rT new_submobc||g|j|}||jvr|j|||j|<||SrN)rrrrmrV)rLrirp old_submobs rRreplace_submobjectz OpenGLMobject.replace_submobjectsn && |444%e, :% % %   % %d + + +",  rT directionr@centerc t|jdd|jdddD]\}}|j||fi||r||S)aSorts :class:`~.OpenGLMobject` next to each other on screen. Examples -------- .. manim:: Example :save_last_frame: class Example(Scene): def construct(self): s1 = Square() s2 = Square() s3 = Square() s4 = Square() x = OpenGLVGroup(s1, s2, s3, s4).set_x(0).arrange(buff=1.0) self.add(x) NrrjTstrict)ziprnext_toru)rLrtrur}m1m2s rRarrangezOpenGLMobject.arrange sw.$*3B3/1A!""1EdSSS 0 0FB BJr9 / / / / / /   KKMMM rTrdrows int | Nonecolsfloat | tuple[float, float]cell_alignmentrow_alignmentscol_alignments row_heightsSequence[float | None] | None col_widths flow_orderc |"#$ddlm} |j$|} d=d } | ||d| ||d0.t t jt$t t$z t t$z zt$krtdt|tr|d}|d}n|x}}d>fd }||tttddt}||t ttddt}fdfdfdfdfd fd!fd"fd#d$}| |vr2d%d&|D}td'|d(|| "t$r3t&d?d+}t&d@d.}t&dAd1}dAd2}||}||}t)}$|gzt$z z"$fd3t-D##fd4t-D}#fd5t-D}dBd8}|||d9}|||d:}d;\}}t-D]}d<}t-D]}#|||urw||||z} | |tz|tzz|||ztz|||ztzz}!#|||!| ||||zz }||||zz }|| |S)Ca Arrange submobjects in a grid. Parameters ---------- rows The number of rows in the grid. cols The number of columns in the grid. buff The gap between grid cells. To specify a different buffer in the horizontal and vertical directions, a tuple of two values can be given - ``(row, col)``. cell_alignment The way each submobject is aligned in its grid cell. row_alignments The vertical alignment for each row (top to bottom). Accepts the following characters: ``"u"`` - up, ``"c"`` - center, ``"d"`` - down. col_alignments The horizontal alignment for each column (left to right). Accepts the following characters ``"l"`` - left, ``"c"`` - center, ``"r"`` - right. row_heights Defines a list of heights for certain rows (top to bottom). If the list contains ``None``, the corresponding row will fit its height automatically based on the highest element in that row. col_widths Defines a list of widths for certain columns (left to right). If the list contains ``None``, the corresponding column will fit its width automatically based on the widest element in that column. flow_order The order in which submobjects fill the grid. Can be one of the following values: "rd", "dr", "ld", "dl", "ru", "ur", "lu", "ul". ("rd" -> fill rightwards then downwards) Returns ------- OpenGLMobject The mobject. NOTES ----- If only one of ``cols`` and ``rows`` is set implicitly, the other one will be chosen big enough to fit all submobjects. If neither is set, they will be chosen to be about the same, tending towards ``cols`` > ``rows`` (simply because videos are wider than they are high). If both ``cell_alignment`` and ``row_alignments`` / ``col_alignments`` are defined, the latter has higher priority. Raises ------ ValueError If ``rows`` and ``cols`` are too small to fit all submobjects. ValueError If :code:`cols`, :code:`col_alignments` and :code:`col_widths` or :code:`rows`, :code:`row_alignments` and :code:`row_heights` have mismatching sizes. Examples -------- .. manim:: ExampleBoxes :save_last_frame: class ExampleBoxes(Scene): def construct(self): boxes=VGroup(*[Square() for s in range(0,6)]) boxes.arrange_in_grid(rows=2, buff=0.1) self.add(boxes) .. manim:: ArrangeInGrid :save_last_frame: class ArrangeInGrid(Scene): def construct(self): #Add some numbered boxes: np.random.seed(3) boxes = VGroup(*[ Rectangle(WHITE, np.random.random()+.5, np.random.random()+.5).add(Text(str(i+1)).scale(0.5)) for i in range(22) ]) self.add(boxes) boxes.arrange_in_grid( buff=(0.25,0.5), col_alignments="lccccr", row_alignments="uccd", col_widths=[2, *[None]*4, 2], flow_order="dr" ) r)Linenumr alignmentsr|sizesrr{rrJrqc ||S|t|S|t|Std|d|d|d|dkrdndd ) Nz+At least one of the following parameters: 'zs', 'z_alignments' or '_colwidthsheightsz', must not be None)rr)rrrr{s rR init_sizez0OpenGLMobject.arrange_in_grid..init_sizes  %:& 5zz!#d######'+u}}88)### rTrrowNz0Too few rows and columns to fit all submobjetcs.rjstr_alignmentsmappingdict[str, Vector3D]rtr?Sequence[Vector3D]c|tt|zg|zSt||krt|dfd|DS)Nz#_alignments has a mismatching size.c g|] }| Srr)r'letterrs rRr)zJOpenGLMobject.arrange_in_grid..init_alignments..sAAAGFOAAArT)rr?rr)rrrr{rtrs ` rRinit_alignmentsz6OpenGLMobject.arrange_in_grid..init_alignmentssm%X~ 'ABBCcII>""c)) D!M!M!MNNNAAAA.AAA ArT)ucd)lrrc |z dz |zzSNrjrrrrs rRz/OpenGLMobject.arrange_in_grid..sq1 D8rTc,|z dz |z dz zzSrrrrrrs rRrz/OpenGLMobject.arrange_in_grid..s!q1 A0EErTc||zzSrNrrs rRrz/OpenGLMobject.arrange_in_grid..sq1t8|rTc ||z dz zzSrrrs rRrz/OpenGLMobject.arrange_in_grid..sqD1HqLD#88rTc |z dz z|zSrrrs rRrz/OpenGLMobject.arrange_in_grid..sq1 4q8rTc,|z dz z|z dz zSrrrs rRrz/OpenGLMobject.arrange_in_grid..s!q1 4q1 ErTc|z|zSrNrrrrs rRrz/OpenGLMobject.arrange_in_grid..sq4x!|rTc |z|z dz zSrrrs rRrz/OpenGLMobject.arrange_in_grid..sq4x4!8a<8rT)drdlurulr~ldrulu,cg|]}d|d S)"r)r'rs rRr)z1OpenGLMobject.arrange_in_grid..s )G)G)G*c***)G)G)GrTz0flow_order must be one of the following values: r maybe_listrcdSrNrrs rRreversez.OpenGLMobject.arrange_in_grid..reverses363rT Sequence[_T]list[_T]cdSrNrrs rRrz.OpenGLMobject.arrange_in_grid..reverses?BsrTSequence[_T] | Nonelist[_T] | NonecdSrNrrs rRrz.OpenGLMobject.arrange_in_grid..reversesMPSrTcT|%t|}||SdSrN)listrrs rRrz.OpenGLMobject.arrange_in_grid..reverses2%!*-- ""$$$!!4rTcLg|]fdtD S)c4g|]}|Srr)r'rflow_order_funcmobsrs rRr)z.. s*BBBooa++,BBBrT)range)r'rrrrs @rRr)z1OpenGLMobject.arrange_in_grid.. s;XXXqBBBBBBeDkkBBBXXXrTcdg|]+tfdtD,S)c3>K|]}|jVdSrN)r)r'rgridrs rRr3z;OpenGLMobject.arrange_in_grid... s.77aQ !777777rTr5r)r'rrrs @rRr)z1OpenGLMobject.arrange_in_grid.. sM   <=C777775;;777 7 7   rTcdg|]+tfdtD,S)c3>K|]}|jVdSrN)r)r'rrrs rRr3z;OpenGLMobject.arrange_in_grid...s.66QQ  666666rTr)r'rrrs @rRr)z1OpenGLMobject.arrange_in_grid..sM   ;.init_sizes..s?"'q)6HQKrT)rrr)rrrr{rs` ` @rR init_sizesz1OpenGLMobject.arrange_in_grid..init_sizessz } 5zzS   D!B!B!BCCCs rTrr)rkrkrk) rrrr|rrr{rrJrq) rr|rrqrrr{rrtr?rJr)rrrJr)rrrJr)rrrJr) rrrrqrrr{rrJr)manim.mobject.geometry.linerrr get_centerr rsqrtrrrtupleUPORIGINDOWNRIGHTLEFTjoinr rrAextendrmove_to)%rLrrr=rrrrrrr}r start_posrbuff_xbuff_yrrow_alignments_seqcol_alignments_seqmappervalid_flow_ordersr placeholdermeasured_heigthsmeasured_widthsrrrxyrr alignmentlinerrrs% `` ` @@@rRarrange_in_gridzOpenGLMobject.arrange_in_grid(sL 544444$$&&OO%%     &y~z5AAy~{EBB >>XXXXXXERVKKXXX     AFt        @Ed        *[$0@-PPJo|LL1t % %AA4[[ ( (71:[00 21 58J18M MI4E AF*VAY%/1wqz>R2GGDGAJ&&tY777VAY'' f$ $AA Y rTn_rowsn_colsr float | NonerVc |||z}|j||fi|||||S)zi Returns a new mobject containing multiple copies of this one arranged in a grid ) duplicater set_height)rLrrrr}rs rRget_gridzOpenGLMobject.get_grid;sS~~fvo..VV66v666   OOF # # # rTncdfdt|DS)zIReturns an :class:`~.OpenGLGroup` containing ``n`` copies of the mobject.c8g|]}Srr)r'rrLs rRr)z+OpenGLMobject.duplicate..Ns!'F'F'F 'F'F'FrT)rFr)rLrs` rRrzOpenGLMobject.duplicateLs7%t##%%'F'F'F'FU1XX'F'F'FGGrTc|dSr,r)ps rRrzOpenGLMobject.Rs adrTpoint_to_num_funcCallable[[Point3DLike], float] submob_func%Callable[[OpenGLMobject], Any] | Nonec||j|n|jfd|S)zOSorts the list of :attr:`submobjects` by a function defined by ``submob_func``.N)rc>|SrN)r)r]rs rRrz$OpenGLMobject.sort..Ys0A0A!,,..0Q0QrT)rsort)rLrrs ` rRrzOpenGLMobject.sortPsQ  "   ! !k ! 2 2 2 2   ! !&Q&Q&Q&Q ! R R R rTc|r |jD]}|dtj|j||S)aShuffles the order of :attr:`submobjects` Examples -------- .. manim:: ShuffleSubmobjectsExample class ShuffleSubmobjectsExample(Scene): def construct(self): s= OpenGLVGroup(*[Dot().shift(i*0.1*RIGHT) for i in range(-20,20)]) s2= s.copy() s2.shuffle() s2.shift(DOWN) self.play(Write(s), Write(s2)) TrX)rshufflerandomrVrLrXrs rRrzOpenGLMobject.shuffle\s_  -* - -t,,,,t'(((  rT recursivec|r |jD]}|d|j||S)aQInverts the list of :attr:`submobjects`. Parameters ---------- recursive If ``True``, all submobject lists of this mobject's family are inverted. Examples -------- .. manim:: InvertSumobjectsExample class InvertSumobjectsExample(Scene): def construct(self): s = VGroup(*[Dot().shift(i*0.1*RIGHT) for i in range(-20,20)]) s2 = s.copy() s2.invert() s2.shift(DOWN) self.play(Write(s), Write(s2)) T)r )rinvertrrV)rLr rs rRr zOpenGLMobject.invertssa*  .* . .  ----   """  rTshallowcF|s|S|j}g|_tj|}||_t|j|_|jD])}|j||j|<*t|j|_g|_|jd|jD|||j |_ | }t|j D]\}}t|tr+||vr'||ur#t!|||t|t"jr#t!||||S)aCreate and return an identical copy of the :class:`OpenGLMobject` including all :attr:`submobjects`. Returns ------- :class:`OpenGLMobject` The copy. Parameters ---------- shallow Controls whether a shallow copy is returned. Note ---- The clone is initially not visible in the Scene, even if the original was. c3>K|]}|VdSrNrr&s rRr3z%OpenGLMobject.copy..s*@@27799@@@@@@rT)deepcopyrrdictrrrrhmatch_updatersrrOr__dict__rrrArrr)rLrr copy_mobjectrrrrs rRrzOpenGLMobject.copys$ #==?? " , y   OO 9 ; ;C%)Ys^%8%8%:%:L c " "!%T] 3 3 #%   @@t/?@@@AA##D))).2.I +"" 3 3 5 566 : :KD%5-00 :VOO%% dEJJLL999%,, : dEJJLL999rTcX|j}g|_tj|}||_|SrN)rrr)rLrresults rRrzOpenGLMobject.deepcopys+, t$$  rT use_deepcopycd|_|r||_n||_|jSrN)targetrrrLrs rRgenerate_targetzOpenGLMobject.generate_targets9,0  &--//DKK))++DK{rTct|drd|_|r||_n||_|S)zfSave the current state (position, color & size). Can be restored with :meth:`~.OpenGLMobject.restore`. saved_stateN)rZrrrrs rR save_statezOpenGLMobject.save_statesL 4 ' ' :59D   +#}}D  #yy{{D  rTct|dr|jtd||j|S)zURestores the state that was previously saved with :meth:`~.OpenGLMobject.save_state`.rNz&Trying to restore without having saved)rZr Exceptionbecomers rRrestorezOpenGLMobject.restoresFt]++ Ft/?/GDEE E D$%%% rTc>g|_g|_d|_d|_dSr.)time_based_updatersnon_time_updaters has_updatersupdating_suspendedrs rRrzOpenGLMobject.init_updaterss(>@ ?A"'(-rTrdtc|jr2|js+|jD]}||||jD] }|||r |jD]}||||SrN)r'r(r%r&rupdate)rLr)rXtime_based_updaternon_time_updaterrs rRr+zOpenGLMobject.updates   'T%< '&*&> - -"""4,,,,$($: ' '   &&&&  +* + + b'**** rTSequence[_TimeBasedUpdater]c|jSrN)r%rs rRget_time_based_updatersz%OpenGLMobject.get_time_based_updaterss ''rTc2t|jdkSr,)rr%rs rRhas_time_based_updaterz$OpenGLMobject.has_time_based_updaters4+,,q00rTSequence[_Updater]cXtd|jtd|jzSNlist[_Updater])rr%r&rs rR get_updaterszOpenGLMobject.get_updaterss3$d&>??$ d4C C   rTcpttjd|DS)Nc3>K|]}|VdSrNr7r&s rRr3z4OpenGLMobject.get_family_updaters.. s,MMRr00MMMMMMrT)ritchainrOrs rRget_family_updatersz!OpenGLMobject.get_family_updaters s0BHMM4??;L;LMMMNOOOrTupdate_functionrD call_updatercBdtj|jvr|j}n|j}|$t d||n$t d|||||r| |S)Nr)r6) inspect signature parametersr%r&rrrkrUr+)rLr>rir? updater_lists rR add_updaterzOpenGLMobject.add_updater s 7$_55@ @ @3LL1L = !< 0 0 7 7 H H H H !< 0 0 7 7 O O O '')))   KKMMM rTc|j|jfD]/}td|}||vr||||v0||Sr5)r%r&rrmrU)rLr>rDs rRremove_updaterzOpenGLMobject.remove_updater#sp!5t7MN 5 5L 0,??L!\11##O444"\11 ''))) rTcg|_g|_||r|jD]}||SrN)r%r&rUrclear_updatersr s rRrIzOpenGLMobject.clear_updaters+sX#% !# '')))  (* ( (%%'''' rTc||D]}|||SrN)rIr7rE)rLrupdaters rRrzOpenGLMobject.match_updaters4sK ++-- & &G   W % % % % rTcVd|_|r|jD]}|||Sr<)r(rsuspend_updatingr s rRrMzOpenGLMobject.suspend_updating:s@"&  1* 1 1''0000 rTcd|_|r|jD]}|||jD]}|dd|r|d||S)NF)rXr?r)r)rX)r(rresume_updatingrr+)rLrXr?rrs rRrOzOpenGLMobject.resume_updatingAs"'  0* 0 0&&w////l F FF  " "5u " E E E E  / KK1gK . . . rTcftd|D|_|S)Nc3>K|]}|VdSrNr:r2s rRr3z;OpenGLMobject.refresh_has_updater_status..Ms.PPs 0 0 2 2PPPPPPrT)rerOr'rs rRrUz(OpenGLMobject.refresh_has_updater_statusLs0PPdoo>O>OPPPPP rTvectorc>|fddd|S)Nc|zSrNr)rbrRs rRrz%OpenGLMobject.shift..Ts 6F?rTT)rrr)rLrRs `rRshiftzOpenGLMobject.shiftRs9 "" * * * *"& #    rT scale_factor_kwargsc @|fd||d|S)uScale the size by a factor. Default behavior is to scale about the center of the mobject. The argument about_edge can be a vector, indicating which side of the mobject to scale about, e.g., mob.scale(about_edge = RIGHT) scales about mob.get_right(). Otherwise, if about_point is given a value, scaling is done with respect to that point. Parameters ---------- scale_factor The scaling factor :math:`\alpha`. If :math:`0 < |\alpha| < 1`, the mobject will shrink, and for :math:`|\alpha| > 1` it will grow. Furthermore, if :math:`\alpha < 0`, the mobject is also flipped. kwargs Additional keyword arguments passed to :meth:`apply_points_function`. Returns ------- OpenGLMobject The scaled mobject. Examples -------- .. manim:: MobjectScaleExample :save_last_frame: class MobjectScaleExample(Scene): def construct(self): f1 = Text("F") f2 = Text("F").scale(2) f3 = Text("F").scale(0.5) f4 = Text("F").scale(-1) vgroup = VGroup(f1, f2, f3, f4).arrange(6 * RIGHT) self.add(vgroup) See also -------- :meth:`move_to` c|zSrNr)rbrWs rRrz%OpenGLMobject.scale..s <&0rTT)r rrrU)rLrWr rrXs ` rRscalezOpenGLMobject.scaleZs?j "" 0 0 0 0#!"& #    rTfactorc 8dfd }|j|fddi||S)NrbrrJc0|ddfxxzcc<|SrNr)rbrpr\s rRrHz#OpenGLMobject.stretch..funcs& 111c6NNNf $NNNMrTrTrbrrJrrU)rLr\rpr}rHs `` rRrzOpenGLMobject.stretchsN        #"4NNtNvNNN rTangleaxisc<|||tSNr )rotater)rLr`ras rRrotate_about_originz!OpenGLMobject.rotate_about_origins{{5$F{;;;rTc Nt|||jfdfd|i||S)z;Rotates the :class:`~.OpenGLMobject` about a certain point.c.tj|SrN)rdot)rb rot_matrix_Ts rRrz&OpenGLMobject.rotate..s26&,77rTr )r4r)rLr`rar r}rjs @rRrezOpenGLMobject.rotatesU1== "" 7 7 7 7  #     rTc 0|jtdz |fi|S)aFlips/Mirrors an mobject about its center. Examples -------- .. manim:: FlipExample :save_last_frame: class FlipExample(Scene): def construct(self): s= Line(LEFT, RIGHT+UP).shift(4*LEFT) self.add(s) s2= s.copy().flip() self.add(s2) r)reTAU)rLrar}s rRflipzOpenGLMobject.flips$"t{37D33F333rTfunctionr;c jt|dkr t|d<dfd }|j|fi||S)Nrr rbrrJc4tjd|}|Sr)rapply_along_axis)rbrrns rRmulti_mapping_functionz.multi_mapping_functions$&$7!V$L$LFMrTr_)rrr)rLrnr}rrs ` rRapply_functionzOpenGLMobject.apply_functions_ v;;!  $*F= !       #"#9DDVDDD rTcf||||SrN)rr)rLrns rRapply_function_to_positionz(OpenGLMobject.apply_function_to_positions- XXdoo//00111 rTcD|jD]}|||SrN)rru)rLrnrs rR&apply_function_to_submobject_positionsz4OpenGLMobject.apply_function_to_submobject_positionss0& 8 8F  - -h 7 7 7 7 rTmatrixr<c d|vrd|vr t|d<tj|jtj|}|d|jdd|jdf<|jfdfi||S)Nr rrrjc8tj|jSrN)rriT)rb full_matrixs rRrz,OpenGLMobject.apply_matrix..s26&+-88rT)rridentityrprshaper)rLrxr}r|s @rR apply_matrixzOpenGLMobject.apply_matrixs  ' 'l&.H.H$*F= !k$(++ &!!.R3_funcs@GAq!!'!Q--00J8Z_joqABB BrT)r]rrJr)rs)rLrnr}rs ` rRapply_complex_functionz$OpenGLMobject.apply_complex_functionsD: C C C C C C #t"755f555rTc |j|jS|jg}|}|j-||jj|j}|j-tjt t|SrN)rrxrrlinalg multi_dotrreversed)rLmodel_matricescurrent_objects rRhierarchical_model_matrixz'OpenGLMobject.hierarchical_model_matrix s ; $ $+,#/  ! !."7"D E E E+2N#/y""4(@(@#A#ABBBrT? wag_factorc |D]}tj|jtj|}|t |z}|t |z}||z}||jtj|t|dftj |d|j fz|Sr) r^rrirb transposer4r5rreshaperrrp)rLrtrarrQalphass rRwagzOpenGLMobject.wags 2244  CVCJ T(:(:;;F c&kk !F c&kk !FZ'F NN &NNCKK#344HY''//CG ==     rTcV|| |S)z-Moves the mobject to the center of the Scene.)rVrrs rRruzOpenGLMobject.center+s& DOO%%%&&& rTc<tj|tdtddfz}||}||z |tj|zz }|t tj|z}|||S)zn Direction just needs to be a vector pointing towards side or corner in the 2d plane. frame_x_radiusframe_y_radiusr)rsignrrasarrayabsrV)rLrtr= target_pointpoint_to_align shift_vals rRalign_on_borderzOpenGLMobject.align_on_border0swy)) # $ # $ -  44Y?? >1D2:i;P;P4PP BGI$6$6 7 77  9 rTcornerc.|||SrNr)rLrr=s rR to_cornerzOpenGLMobject.to_cornerDs ##FD111rTedgec.|||SrNr)rLrr=s rRto_edgezOpenGLMobject.to_edgeKs ##D$///rT)rjrjrjmobject_or_pointOpenGLMobject | Point3DLike aligned_edgesubmobject_to_alignOpenGLMobject | Noneindex_of_submobject_to_align coor_maskcjtj|}tj|} t|tr(|} | | |} n| } | | |z} n|} ||} n | ||} n|} | | |z }|| |z ||zz|z|S)a=Move this :class:`~.OpenGLMobject` next to another's :class:`~.OpenGLMobject` or coordinate. Examples -------- .. manim:: GeometricShapes :save_last_frame: class GeometricShapes(Scene): def construct(self): d = Dot() c = Circle() s = Square() t = Triangle() d.next_to(c, RIGHT) s.next_to(c, LEFT) t.next_to(c, DOWN) self.add(d, c, s, t) )rrrrArrV)rLrrtr=rrrr np_directionnp_aligned_edgerQtarget_alignerralignerrs rRrzzOpenGLMobject.next_toRs<z),, *\22 & 6 6 ,"C+7!$%A!B!$)@@,.LL,L  *)GG ) 578GGG 77,8VWW L>1D<4GG9TUUU rTc tdtdg}ttttfD]}t jt j|}|dt}|||z }| |}t j |||kr| |||S)Nrrr=)r=) rrrrrrargmaxrgetDEFAULT_MOBJECT_TO_EDGE_BUFFERget_edge_centerrir)rLr} space_lengthsvectrpr=max_val edge_centers rRshift_onto_screenzOpenGLMobject.shift_onto_screens # $ # $& dE) . .D)BF4LL))C **V-KLLD#C(4/G..t44Kvk4((722 T --- rTcv|dtjkrdS|dtjkrdS|dtjkrdSt t|dtj kS)NrTrj) get_leftrr get_right get_bottomrrroget_toprs rR is_off_screenzOpenGLMobject.is_off_screens ==??1  5 5 54 >>  A !6 6 64 ??  Q &"7 7 74E4<<>>!,--1F0FFFrTc2||||Srcr)rLr\rpr]s rRstretch_about_pointz!OpenGLMobject.stretch_about_points||FCU|;;;rTlengthrc ||}|dkr|S|r|j||z |fi|n|j||z fi||Sr,)rrr[)rLrrprr} old_lengths rRrzOpenGLMobject.rescale_to_fitsu))#.. ??K  6 DL*,c < >> from manim import * >>> import numpy as np >>> sq = Square() >>> sq.height np.float64(2.0) >>> sq.stretch_to_fit_width(5) Square >>> sq.width np.float64(5.0) >>> sq.height np.float64(2.0) rrTr)rLrr}s rRstretch_to_fit_widthz"OpenGLMobject.stretch_to_fit_widths&0#t"5!DDTDVDDDrTc $|j|dfddi|S)z^Stretches the :class:`~.OpenGLMobject` to fit a height, not keeping width/height proportional.rjrTr)rLrr}s rRstretch_to_fit_heightz#OpenGLMobject.stretch_to_fit_heights$"t"61EEdEfEEErTrc $|j|dfddi|S)z]Stretches the :class:`~.OpenGLMobject` to fit a depth, not keeping width/height proportional.rjrTr)rLrr}s rRstretch_to_fit_depthz"OpenGLMobject.stretch_to_fit_depths$"t"5!DDTDVDDDrTc $|j|dfd|i|S)a*Scales the :class:`~.OpenGLMobject` to fit a width while keeping height/depth proportional. Returns ------- :class:`OpenGLMobject` ``self`` Examples -------- :: >>> from manim import * >>> import numpy as np >>> sq = Square() >>> sq.height np.float64(2.0) >>> sq.scale_to_fit_width(5) Square >>> sq.width np.float64(5.0) >>> sq.height np.float64(5.0) rrr)rLrrr}s rR set_widthzOpenGLMobject.set_widths&:#t"5!GGWGGGGrTc $|j|dfd|i|S)z[Scales the :class:`~.OpenGLMobject` to fit a height while keeping width/depth proportional.rjrr)rLrrr}s rRrzOpenGLMobject.set_heights&#t"61HHgHHHHrTc $|j|dfd|i|S)z[Scales the :class:`~.OpenGLMobject` to fit a depth while keeping width/height proportional.rrr)rLrrr}s rR set_depthzOpenGLMobject.set_depths&#t"5!GGWGGGGrTc|||}tj|j}||z ||<|||SrN) get_coordrrrprV)rLrrprtcurr shift_vects rR set_coordzOpenGLMobject.set_coord sL~~c9--Xdh'' $, 3 : rTrc0||d|S)zPSet x value of the center of the :class:`~.OpenGLMobject` (``int`` or ``float``)rr)rLrrts rRset_xzOpenGLMobject.set_x~~aI...rTrc0||d|S)zPSet y value of the center of the :class:`~.OpenGLMobject` (``int`` or ``float``)rjr)rLrrts rRset_yzOpenGLMobject.set_yrrTrc0||d|S)zPSet z value of the center of the :class:`~.OpenGLMobject` (``int`` or ``float``)rr)rLrrts rRset_zzOpenGLMobject.set_zrrT?c f|j|fi||jD]}|d|z |SNr)r[r)rLr\r}rs rRspace_out_submobjectsz#OpenGLMobject.space_out_submobjects!sL 6$$V$$$& ' 'F LLv & & & & rTpoint_or_mobjectPoint3DLike | OpenGLMobjectct|tr||}n|}||}|||z |z|S)zBMove center of the :class:`~.OpenGLMobject` to certain coordinate.)rrArrV)rLrrrrrs rRrzOpenGLMobject.move_to'se & 6 6 &%<<\JJFF%F44\BB F^+y8999 rT dim_to_matchc|s|js|d|S|rCt|jD]-}||||d.n+||||d|||z |S)NrTrF) r#rr[rrprrrVr)rLrrrrs rRreplacezOpenGLMobject.replace7s %%'' 0C  JJqMMMK  48__ Q Q##G$;$;A$>$>4#PPPP Q   '' 55     7%%''$//*;*;;<<< rTc||||||}|||z|z |SrN)rrr[)rLrrrr=rs rRsurroundzOpenGLMobject.surroundLsM WlG444((66 FTMV+,,, rTstartendc|\}}||z }tj|dkrtdtj|tj|z }tjtj||dkr"ttj||nt}| ttj|tj|z || t|||||||z |S)Nrz(Cannot position endpoints of closed looprd)r ra)get_start_and_endrr?r!rrnormcrossr3OUTr[rorer2rV)rLrr curr_startcurr_end curr_vect target_vectras rRput_start_and_end_onz"OpenGLMobject.put_start_and_end_onXsG#5577 Hz) 6)q. ! ! HFGG Ghsmmbhuoo5 y~~bhy+>>??1DD bhy+66 7 7 7 ")..-- y0I0II J J"     !)[ 9 9"    5:%&&& rT8ParsableManimColor | Iterable[ParsableManimColor] | Nonefloat | Iterable[float] | Nonec V|+tjdt|D}|t|}|}|{||D]e}t ||jvr |j|ntjdt||j|<||j|ddddf<f|{|y||D]c}t ||jvr |j|ntjdt||j|<||j|dddf<d|l|jtjdtt||ddiD}||D]}| |j|<|S)Nc,g|]}t|Sr)r%)r'rs rRr)z0OpenGLMobject.set_rgba_array..ys,U,U,U\!__,U,U,UrT)rjr^r^c g|] \}}g|| Srr)r'rgbos rRr)z0OpenGLMobject.set_rgba_array..s$WWWvsA3WWWrTrxT) rrr+rOr-rr rryr,r) rLrlrnr{rXrgbs opacitiesrQrds rRset_rgba_arrayzOpenGLMobject.set_rgba_arrayqs  #%8,U,Ugenn,U,U,U#V#VD  ((I  w// - -!-&*ch&6&6CHTNNBHVt|jdddfS)Nrr^)r&rdrs rR get_colorzOpenGLMobject.get_colors$*QU+,,,rTc |jd}|Sr)rd)rLrAs rR get_opacityzOpenGLMobject.get_opacitysJt$ rTcolorsr#c|j|SrN)!set_submobject_colors_by_gradient)rLrs rRset_color_by_gradientz#OpenGLMobject.set_color_by_gradients5t5v>>rTc&t|dkrtdt|dkr |j|S|j}t |t|}t ||dD]\}}|||S)NrzNeed at least one colorrjTrw)rr!rrr$ry)rLrr new_colorsrQrls rRrz/OpenGLMobject.set_submobject_colors_by_gradients v;;!  566 6 [[A  !4>6* *#FCII66 dJt<<< ! !JC MM%  rTr darknessc6|d|z |S)Nrr)r )rLrrXs rRfadezOpenGLMobject.fadesh@@@rTc|jSrN)rhrs rR get_glosszOpenGLMobject.get_glosss zrTcD||D] }||_ |SrN)rOrh)rLrhrXrQs rR set_glosszOpenGLMobject.set_glosss+??7++  CCII rTc|jSrN)rirs rR get_shadowzOpenGLMobject.get_shadows {rTcD||D] }||_ |SrN)rOri)rLrirXrQs rR set_shadowzOpenGLMobject.set_shadows+??7++  CCJJ rT?ParsableManimColor | Nonec hddlm}||f||d||_||j|S)aAdd a BackgroundRectangle as submobject. The BackgroundRectangle is added behind other submobjects. This can be used to increase the mobjects visibility in front of a noisy background. Parameters ---------- color The color of the BackgroundRectangle opacity The opacity of the BackgroundRectangle kwargs Additional keyword arguments passed to the BackgroundRectangle constructor Returns ------- :class:`OpenGLMobject` ``self`` See Also -------- :meth:`add_to_back` :class:`~.BackgroundRectangle` r)BackgroundRectangle)rl fill_opacity)%manim.mobject.geometry.shape_matchersr#background_rectanglero)rLrlrnr}r#s rRadd_background_rectanglez&OpenGLMobject.add_background_rectangleshF NMMMMM9L9L :  : :  : : ! 2333 rTc 4|jD]}|jdi||Sr)rr')rLr} submobjects rR'add_background_rectangle_to_submobjectsz5OpenGLMobject.add_background_rectangle_to_submobjects$ s5* : :J /J / 9 9& 9 9 9 9 rTc N|D]}|jdi||Sr)r^r')rLr}rQs rR6add_background_rectangle_to_family_members_with_pointszDOpenGLMobject.add_background_rectangle_to_family_members_with_points) s@2244 3 3C (C ( 2 26 2 2 2 2 rTc|tj|dzttjfdt dDS)Nrjc8g|]}||Srr)r'rr@indicess rRr)z8OpenGLMobject.get_bounding_box_point..5 s&===qGAJ*===rTr^)rrrastyperqrr)rLrtr@r/s @@rRrz$OpenGLMobject.get_bounding_box_point2 se  " " $ $79%%)11#66x=====E!HH===>>>rTc,||S)z+Get edge coordinates for certain direction.rrLrts rRrzOpenGLMobject.get_edge_center7 **9555rTc,||S)z-Get corner coordinates for certain direction.r2r3s rR get_cornerzOpenGLMobject.get_corner; r4rTc6|dS)zGet center coordinates.rj)rrs rRrzOpenGLMobject.get_center? s$$&&q))rTcP|dSr,)r*meanrs rRget_center_of_massz OpenGLMobject.get_center_of_massC s"""$$))!,,,rTcl|}||z }tj|d}|tj|dt|dfz}tjtj ||}||S)Nrj)rar^) r*rrrrrepeatrrrri)rLrtr6boundary_directionsnormsris rRget_boundary_pointz OpenGLMobject.get_boundary_pointF s((** (4??+<+<< 2;;ry22::CJJ?KKK "&!4i@@AA%  rTc.|\}}}||z }tj|}||tjtjtj||tjt||dkz zS)Nr)outwhere)rrrr5r true_dividerr)rLrt_dlrur corner_vectrs rR!get_continuous_bounding_box_pointz/OpenGLMobject.get_continuous_bounding_box_pointN s//11VR6k z),,  rv F \!2!233'A-    ( (   rTc6|tS)zBGet top coordinates of a box bounding the :class:`~.OpenGLMobject`)rrrs rRrzOpenGLMobject.get_top] ##B'''rTc6|tS)zEGet bottom coordinates of a box bounding the :class:`~.OpenGLMobject`)rrrs rRrzOpenGLMobject.get_bottoma ##D)))rTc6|tS)zDGet right coordinates of a box bounding the :class:`~.OpenGLMobject`)rrrs rRrzOpenGLMobject.get_righte s##E***rTc6|tS)zCGet left coordinates of a box bounding the :class:`~.OpenGLMobject`)rrrs rRrzOpenGLMobject.get_lefti rJrTc6|tS)zGGet zenith coordinates of a box bounding a 3D :class:`~.OpenGLMobject`.)rrrs rR get_zenithzOpenGLMobject.get_zenithm s##C(((rTc6|tS)z\Get nadir (opposite the zenith) coordinates of a box bounding a 3D :class:`~.OpenGLMobject`.)rINrs rR get_nadirzOpenGLMobject.get_nadirq rHrTcv|}t|d|dz |}|S)Nrr)rr)rLrpr@rAs rRrzOpenGLMobject.length_over_dimu s6  " " $ $AA,-- rTc,|dS)z!Returns the width of the mobject.rrrs rR get_widthzOpenGLMobject.get_widthz ##A&&&rTc,|dS)z"Returns the height of the mobject.rjrrs rR get_heightzOpenGLMobject.get_height~ rUrTc,|dS)z!Returns the depth of the mobject.rrrs rR get_depthzOpenGLMobject.get_depth rUrTr:c8|||S)z6Meant to generalize ``get_x``, ``get_y`` and ``get_z``r2)rLrprts rRrzOpenGLMobject.get_coord s**955c::rTc.|d|S)zOReturns x coordinate of the center of the :class:`~.OpenGLMobject` as ``float``rrr3s rRget_xzOpenGLMobject.get_x ~~a+++rTc.|d|S)zOReturns y coordinate of the center of the :class:`~.OpenGLMobject` as ``float``rjr\r3s rRget_yzOpenGLMobject.get_y r^rTc.|d|S)zOReturns z coordinate of the center of the :class:`~.OpenGLMobject` as ``float``rr\r3s rRget_zzOpenGLMobject.get_z r^rTch|tj|jdS)zWReturns the point, where the stroke that surrounds the :class:`~.OpenGLMobject` starts.rthrow_error_if_no_pointsrrrbrs rR get_startzOpenGLMobject.get_start s* %%'''x A'''rTch|tj|jdS)zUReturns the point, where the stroke that surrounds the :class:`~.OpenGLMobject` ends.rrdrs rRget_endzOpenGLMobject.get_end s* %%'''x B(((rTtuple[Point3D, Point3D]cR||fS)z=Returns starting and ending point of a stroke as a ``tuple``.)rfrhrs rRrzOpenGLMobject.get_start_and_end s~~//rTalphac|j}tdt|dz |\}}t||||dz|S)Nrrj)rbrrr )rLrkrbrsubalphas rRr z#OpenGLMobject.point_from_proportion sF)!S[[1_eDD 86!9fQUmX>>>rTc,||S)z&Abbreviation for point_from_proportionr )rLrks rRpfpzOpenGLMobject.pfp s))%000rTn_piecesc g_tjdd|dz}t fdt |dd|dddDS)Nrrjc3pK|]0\}}||V1dSrN)rpointwise_become_partial)r'a1a2rLtemplates rRr3z+OpenGLMobject.get_pieces.. sUB 88r2FFrTrTrw)rrrlinspacerVry)rLrprrvs` @rR get_pieceszOpenGLMobject.get_pieces s99;;!Q8a<00!&"+vabbz$GGG  rTcLt|d|}|S)N z_index_group)rr)rLrzs rRget_z_index_reference_pointz)OpenGLMobject.get_z_index_reference_point s%ot<< '')))rTcP||S)zCMatch the color with the color of another :class:`~.OpenGLMobject`.)rr rs rR match_colorzOpenGLMobject.match_color s ~~g//11222rTc F|j|||fi|S)zUMatch the specified dimension with the dimension of another :class:`~.OpenGLMobject`.)rr)rLrrpr}s rRmatch_dim_sizezOpenGLMobject.match_dim_size s/#t"7#:#:3#?#?OOOOOrTc  |j|dfi|S)zCMatch the width with the width of another :class:`~.OpenGLMobject`.rrrLrr}s rR match_widthzOpenGLMobject.match_width "t"7A88888rTc  |j|dfi|S)zEMatch the height with the height of another :class:`~.OpenGLMobject`.rjrrs rR match_heightzOpenGLMobject.match_height rrTc  |j|dfi|S)zCMatch the depth with the depth of another :class:`~.OpenGLMobject`.rrrs rR match_depthzOpenGLMobject.match_depth rrTcZ||||||S)zOMatch the coordinates with the coordinates of another :class:`~.OpenGLMobject`.)rprt)rr)rLrrprts rR match_coordzOpenGLMobject.match_coord s8~~   c9 - -   rTc0||d|S)zCMatch x coord. to the x coord. of another :class:`~.OpenGLMobject`.rrrLrrts rRmatch_xzOpenGLMobject.match_x I666rTc0||d|S)zCMatch y coord. to the x coord. of another :class:`~.OpenGLMobject`.rjrrs rRmatch_yzOpenGLMobject.match_y rrTc0||d|S)zCMatch z coord. to the x coord. of another :class:`~.OpenGLMobject`.rrrs rRmatch_zzOpenGLMobject.match_z rrTct|tr||}n|}t|jD]+}||dkr|||||,|S)a* Examples: mob1.align_to(mob2, UP) moves mob1 vertically so that its top edge lines ups with mob2's top edge. mob1.align_to(mob2, alignment_vect = RIGHT) moves mob1 horizontally so that it's center is directly above/below the center of mob2 r)rrArrrpr)rLrrtr]rps rRalign_tozOpenGLMobject.align_to sw & 6 6 %$;;IFFEE$E?? ; ;C~""uSz3 ::: rTtype[OpenGLGroup]ctSrN)rVrs rRrFzOpenGLMobject.get_group_class srTctS)z+Return the base class of this mobject type.)rArrTrRget_mobject_type_classz$OpenGLMobject.get_mobject_type_class s rTcZ|||||SrN) align_family align_datars rRalign_data_and_familyz#OpenGLMobject.align_data_and_family s- '"""     rTclt||dD]\}}|||j|jzD]}|dkr |j|}|j|}t |t |kr&t |t ||j|<it |t |kr%t |t ||j|<|S)NFrwrb)ryrO align_pointsrkeysrr.)rLrmob1mob2rarr1arr2s rRrzOpenGLMobject.align_data sdoo//1C1C1E1EeTTT N NJD$   d # # #y~~''$)..*:*:: N N(??y~y~t99s4yy((%tLLL " "HC   S ! ! ! ! rTc|}g|_|d|||Sr,)rrrrh)rLrs rRpush_self_into_submobjectsz(OpenGLMobject.push_self_into_submobjects; s?}} 1  rTc |dkr|St|j}|dkr^| |g fdt |D|_|S||z}t j||z|z fdt |D}g}t|j|dD]\}}| |t d|D]X}|} | dkr| d| | Y||_|S)Nrc8g|]}Srr)r'knull_mobs rRr)z8OpenGLMobject.add_n_more_submobjects..K s!BBBA BBBrTc@g|]}|kSr)sum)r'rrepeat_indicess rRr)z8OpenGLMobject.add_n_more_submobjects..O s*JJJ.A-2244JJJrTTrwrj) rrrrrrrarangeryrrr ) rLrrr split_factors new_submobsrsfrrprrs @@rRrz$OpenGLMobject.add_n_more_submobjectsB s{ 66K4#$$ 199yy{{H   !2!2 3 4 4 4BBBBqBBBD K)F++d2v=JJJJeDkkJJJ  d. dKKK / /JFB   v & & &1b\\ / /#[[]] %%''!++**1---"":....  /' rTmobject1mobject2 path_funcr>c R|jD]~}||jvr t|j|dkr+||jvs ||jvr>|dvr|nt}||j||j|||j|dd<|jD]}|dkr0t|j||j|||j|<8t tt j|jdt j|jd||jd<|S)aaTurns this :class:`~.OpenGLMobject` into an interpolation between ``mobject1`` and ``mobject2``. Examples -------- .. manim:: DotInterpolation :save_last_frame: class DotInterpolation(Scene): def construct(self): dotR = Dot(color=DARK_GREY) dotR.shift(2 * RIGHT) dotL = Dot(color=WHITE) dotL.shift(2 * LEFT) dotMiddle = OpenGLVMobject().interpolate(dotL, dotR, alpha=0.3) self.add(dotL, dotR, dotMiddle) r)rbrcNrg)rrrr rrrr)rLrrrkrrrHs rRr zOpenGLMobject.interpolate_ sV69 T TCd+++49S>""a''(-''3hm+C+C #'A A A99{D $X]3%7s9KU S SDIcN111  =  C000%0%c*%c*&& c"" =B!23M!NOO!23M!NOO== 899 rTabc|S)z Set points in such a way as to become only part of mobject. Inputs 0 <= a < b <= 1 determine what portion of mobject to become. Returns `self` to allow method chaining. r)rLrrrs rRrsz&OpenGLMobject.pointwise_become_partial s  rTrrr match_centerc|rO||j||j||jnE|r|||r|||r|||r'| | | |t| | dD]9\}}||j||j:|d|S)a}Edit all data and submobjects to be identical to another :class:`~.OpenGLMobject` .. note:: If both match_height and match_width are ``True`` then the transformed :class:`~.OpenGLMobject` will match the height first and then the width Parameters ---------- match_height If ``True``, then the transformed :class:`~.OpenGLMobject` will match the height of the original match_width If ``True``, then the transformed :class:`~.OpenGLMobject` will match the width of the original match_depth If ``True``, then the transformed :class:`~.OpenGLMobject` will match the depth of the original match_center If ``True``, then the transformed :class:`~.OpenGLMobject` will match the center of the original stretch If ``True``, then the transformed :class:`~.OpenGLMobject` will stretch to fit the proportions of the original Examples -------- .. manim:: BecomeScene class BecomeScene(Scene): def construct(self): circ = Circle(fill_color=RED, fill_opacity=0.8) square = Square(fill_color=BLUE, fill_opacity=0.2) self.add(circ) self.wait(0.5) circ.become(square) self.wait(0.5) Trwr)rrrrrrrrrrrrryrOrrrrr) rLrrrrrrrrs rRr"zOpenGLMobject.become s_V  *  ) )$+ 6 6 6  ( ( 4 4 4  ( ( 4 4 4 4 +$$T*** *##D))) *##D)))  / OODOO-- . . . '"""DOO--w/A/A/C/CDQQQ + +HC LL " " "   S\ * * * * !!t!444 rTrch|jrdS|t||_dS)a3 To speed up some animations, particularly transformations, it can be handy to acknowledge which pieces of data won't change during the animation so that calls to interpolate can skip this, and so that it's not read into the shader_wrapper objects needlessly N)r'refresh_shader_datarr)rLrs rR lock_datazOpenGLMobject.lock_data s:    F   """ #D rTc t|||dD]\}|jjzjz}|t t fd||S)NFrwc`tjj|j|kSrN)rr?r)rrrs rRrz2OpenGLMobject.lock_matching_data.. s"BF38C=CHSM+I$J$JrT)ryrOrrrrfilter)rLrrr(rrrs @@rRlock_matching_dataz OpenGLMobject.lock_matching_data s OO      ! !    ! !      LBS 7<<>>CHMMOO3chmmooED LLJJJJJ     rTcZ|D]}t|_dSrN)rOrr)rLrQs rR unlock_datazOpenGLMobject.unlock_data s4??$$ ) )C#&55C  ) )rTcd|_|Srrers rR fix_in_framezOpenGLMobject.fix_in_frame !$ rTcnd|_t||_d|_|S)NrT)rfrrrgrurs rRfix_orientationzOpenGLMobject.fix_orientation s0$'!(-doo.?.?(@(@% rTcd|_|S)Nrkrrs rRunfix_from_framezOpenGLMobject.unfix_from_frame rrTc0d|_d|_d|_|S)NrkrF)rfrgrurs rRunfix_orientationzOpenGLMobject.unfix_orientation s$'!(1% rTcd|_|Sr<rurs rRrzOpenGLMobject.apply_depth_test' s rTcd|_|Sr.rrs rRdeactivate_depth_testz#OpenGLMobject.deactivate_depth_test, s rTold_codenew_codec`|D]}||||SrN)get_shader_wrapper_list replace_code)rLrrrSs rRreplace_shader_codez!OpenGLMobject.replace_shader_code3 s=3355 5 5G  8 4 4 4 4 rT glsl_codec2|d||S)z Takes a snippet of code and inserts it into a context which has the following variables: vec4 color, vec3 point, vec3 unit_normal. The code should change the color variable z&///// INSERT COLOR FUNCTION HERE /////)r)rLrs rRset_color_by_codezOpenGLMobject.set_color_by_code; s    !I9UUU rT@viridis glsl_snippet min_value max_valuecolormapc dD]}||d|z}t|}|d|dt|dt|dt |d |S)za Pass in a glsl expression in terms of x, y and z which returns a float. xyzzpoint.zcolor.rgb = float_to_color(z, z);)rget_colormap_listrror)rLrrrrcharrgb_lists rRset_color_by_xyz_funcz#OpenGLMobject.set_color_by_xyz_funcE s G GD'//hoFFLL%X..  B, B B% :J:J B BeT]N^N^ B Bbst|b}b} B B B    rTcR||SrN)get_shader_wrapper refresh_idrs rRrPz'OpenGLMobject.refresh_shader_wrapper_id^ s% !!,,... rT'ShaderWrapper'c ddlm}|||||j|j|j|jj |_ |j S)Nrr6) vert_data vert_indicesrrursrrra) manim.renderer.shader_wrapperr7get_shader_dataget_shader_vert_indicesget_shader_uniformsrursrrrrashader_wrapper)rLr7s rRrz OpenGLMobject.get_shader_wrapperb s?????? .;]**,,5577--//,!2.6. . . ""rTSequence['ShaderWrapper']c\tj|ggd|jDR}t |d}g}|D]a\}}|d}|s"|j|ddt|jdkr| |b|S)Nc3>K|]}|VdSrN)rr&s rRr3z8OpenGLMobject.get_shader_wrapper_list..v s. F Frb((** F F F F F FrTc*|SrN)get_id)sws rRrz7OpenGLMobject.get_shader_wrapper_list..x s rTrrj) r;r<rrr)is_valid combine_withrrr)rLshader_wrappersbatchesr wrapper_grouprrs rRrz%OpenGLMobject.get_shader_wrapper_lists s(  $ $ & & ' F FT5E F F F   $O5K5KLL(* ' . . M1*1-N!**,,  'N 'qrr): ; ;>+,,q00 n--- rTrrZdata_keyct|j|}|dkrC|t|kr0t|j|t||j|<|Sr)rrr/)rLrrd_lens rRcheck_data_alignmentz"OpenGLMobject.check_data_alignment saDIh'(( A::%3u::--"; (#E ##DIh  rTc|j}tttjt ||j}|S)N)dtype)rbrrZrrrr_)rLrrb shader_datas rRget_resized_shader_data_arrayz+OpenGLMobject.get_resized_shader_data_array s8;VDDU(V(V(VWW rTr shader_data_keych||jvrdS||||j|||<dSrN)rr r)rLr rrs rRread_data_to_shaderz!OpenGLMobject.read_data_to_shader sD t, , , F !!+x888'+y': O$$$rTc||}||dd|S)Nr]rb)rr#r)rLr s rRrzOpenGLMobject.get_shader_data s?889L9L9N9NOO    gx@@@rTc.|dSrN)rrs rRrz!OpenGLMobject.refresh_shader_data s rTc|jSrNrrs rRrz!OpenGLMobject.get_shader_uniforms s }rTSequence[int] | Nonec|jSrN)rrs rRrz%OpenGLMobject.get_shader_vert_indices s ""rTlist[OpenGLMobject]c4t|dr|jngS)Nr)rZrrs rRrzOpenGLMobject.submobjects s$+D.$A$AIt  rIrTsubmobject_listc8|j|j|j|dSrN)rmrrh)rLrs rRrzOpenGLMobject.submobjects s) T%&&/""""rTc|sBd}tjdjj}t ||dS)Nz?Cannot call OpenGLMobject.{} for a OpenGLMobject with no pointsrj)rsys _getframef_codeco_namer!format)rLmessage caller_names rRrez&OpenGLMobject.throw_error_if_no_points sW   9V -**19KGNN;7788 8  9 9rT)rlrmrnrorprqrhrorirorrrqrsrtrurvrervrfrvrwrvrxryrzrvr{r|r}r )rrrJr)rrrrrJr)r}r rJr)rJr)rrrJr)rrrJr)rJr)r}rrJr)rrrJr)rrrJr)rJrrJro)rrorJr)rrqrrrJr)rbrrJr)rHrrJr)rJr)rrrJr)rJrrJr) rHr=r rrrrrvrJr)rrArJr)rJrq)rJr)rJrv)FT)rrvr:rvrJr)r]rr=rorJrv)rrCrJrA)rJrJ)rJrP)T)rXrvrJrP)r`rAr_rvrJr)F)rirqrrAr_rvrJr)r`rArJr)rirqrprArJr)rtr@rurvr}r rJr)rrrrr=rrr@rr|rr|rrrrrrr}r rJrrN) rrqrrqrrr}r rJrV)rrqrJrV)rrrrrJr)rXrvrJr)r rvrJr)rrvrJrA)rJrA)rrvrJrA)rrvrJr)rT)r)rorXrvrJr)rJr.)rJr3r.)r>rDrirr?rvrJr)r>rDrJr)TT)rXrvr?rvrJr)rRr@rJr) rWror rrrrXrrJr)r\rorprqr}r rJr)r`rorar@rJr) r`rorar@r rr}r rJr)rar@r}r rJr)rnr;r}r rJr)rnr;rJr)rxr<r}r rJr)rnrr}r rJr)rJr<)rtr@rar@rrorJr)rtr@r=rorJr)rr@r=rorJr)rr@r=rorJr)rrrtr@r=rorr@rrrrrr@rJrr}r rJr)r\rorprqr]rrJr) rrorprqrrvr}r rJr)rror}r rJr)rror}r rJr)rror}r rJr)rrorrvr}r rJr)rrorrvr}r rJr)rrorrvr}r rJr)rrorprqrtr@rJr)rrortr@rJr)rrortr@rJr)rrortr@rJr)r)r\ror}r rJr)rrrr@rr@rJr)rF)rrArrqrrvrJr) rrArrqrrvr=rorJr)rrrrrJr)NNrdT) rlrrnrr{rrXrvrJr)rdT)rdr9r{rrXrvrJrr<)rlrrnrrXrvrJr)rnrrXrvrJr)rr#rJr)r T)rrorXrvrJr)rhrorXrvrJr)rirorXrvrJr)Nr )rlr!rnror}r rJr)rtr@rJr)rprqrJro)rprqrtr@rJr:)rtr@rJr:)rJri)rkrorJr)rprqrJrA)rrArprqr}r rJr)rrAr}r rJr)rrArprqrtr@rJr)rrArtr@rJr)rrrtr@rJr)rJr)rJr)rrqrJr) rrArrArkrorr>rJr)rrArrorrorJr)FFFFF)rrArrvrrvrrvrrvrrvrJr)rrrJr)rrArrArJr)rrrrrJr)rrrJr)rrr) rrrrorrorrrJr)rJr)rJr)rrZrrrJr)rrorJrZ)r rZrrrrrJr)rJrZ)rJr)rJr)rJr)rrrJr)r __module__ __qualname____doc____annotations__rfloat32r_rar'rbrcrdr(rerfrgrhrir!moderngl TRIANGLESrrr classmethodrrrrrrrrrrrrrrrpropertyrrsetterrrr-rrrrrrr rrrr!r#r*rrr/rMED_SMALL_BUFFrBrHrMrOrGrVrOr^rhrkrmrorsrr}rrrrrr rrrrr#rr+r0r2r7r=rErGrIrrMrOrUrVr[rrrfrerrmrsrurwrrrrrrurrrrr!DEFAULT_MOBJECT_TO_MOBJECT_BUFFERrrzrrrrrrrrscale_to_fit_widthrscale_to_fit_heightrscale_to_fit_depthrrrrrrrrrrrrr r rrrrrrrrr'r*r,rrr6rr:r?rFrrrrrNrQrrTrWrYrr]r`rbrfrhrr rorxr{r}rrrrrrrrrrF staticmethodrrrrrrrr1r rsr"rrrrUrrrrrrrrrrPrrr rrrrrrrre __classcell__rs@rRrArAss  5444 "*d#WL$&M%%%%$)577F****).L0000$)EGGE++++#,9;;....&/ikk1111 !y{{E""""! F####DI ( 2/3 "'%*"'(,"-J1J1J1J1J1XSSSSB2-----[-'''X'X/1/1/1[/1b&&&&1111    B  W'W'W'XW'r'''X'@ \555\5'''X'> ]555]5 ' ' 'X ' \555\5 FR  ////6+/*0&+ @(     ))))!!!! 0000*>B1?//// """"!!!!     @@@@CHSSSSSSlINBFK      D////b#(> ,:'-%)%)5948QQQQQn $ "HHHH =KN=A     .<88888t     ....     ((((1111    PPPP !" ,       +/)/ ;;;;;zFI<<<<< !*.  )+44444&        "6"6"6"6H C C C C#(! ,5, $d{422222"400000#(7%+4837"*"(999"5"555555n    GGGG<<<<  EEEE4FFFFEEEEHHHHH># IIIII% HHHHH#AG9?/////9?/////9?/////&,"*"(999"5"5 & 0$     6KO26 $$$$$R 437 &HL----????    AAAAA ,0,,,,,\ ???? 66666666****----!!!!     ((((****++++****))))(((( ''''''''''''=C;;;;;/5,,,,,/5,,,,,/5,,,,,(((( )))) 0000???? 1111     ****3333PPPP999999999999 KQ     IO77777IO77777IO77777#)2\  $    D#0-// 66666p     #!!"?????F * * * *())))     ! 2####""     ; ; ; ; ####JJJXJ#### 99999999rTc eZdZdfd ZxZS) rVr`rAr}r rJrcNtjdi||j|dSr)rrrh)rLr`r}rs rRrzOpenGLGroup.__init__ s5""6"""(rT)r`rAr}r rJr)rr&r'rr6r7s@rRrVrV s=rTrVceZdZeddfdfd Zedd Zedd ZddZeddZ ddZ xZ S)rWgư>locationrartificial_widthroartificial_heightr}r rJrc ||_||_tjdi|||dSr)r<r=rr set_location)rLr;r<r=r}rs rRrzOpenGLPoint.__init__ sL(8(9""6""" (#####rTc|jSrN)r<rs rRrTzOpenGLPoint.get_width s $$rTc|jSrN)r=rs rRrWzOpenGLPoint.get_height s %%rTrcftt|jdSr,)rrrbrrs rR get_locationzOpenGLPoint.get_location s$GT[^,,11333rTargsrc*|SrN)rC)rLrDr}s rRrz"OpenGLPoint.get_bounding_box_point s  """rTnew_loccd|tj|dtdS)Nr)ndminr )rrrro)rLrFs rRr?zOpenGLPoint.set_location s+ ???@@@@@rT) r;rr<ror=ror}r rJrr#r$)rDrr}r rJr)rFrrJr) rr&r'rrrrTrWrCrr?r6r7s@rRrWrW s!'"&#' $ $ $ $ $ $ $%%%X%&&&X&4444###X#AAAAAAAArTrWc.eZdZddZdd Zdd ZddZdS)rXrrArJrc||_|jd|_d|_g|_d|_i|_dSr.)rroverridden_animation is_chainingmethodscannot_pass_args anim_argsrs rRrz_AnimationBuilder.__init__ sI&-  $$&&&6:!!&-/ ',,.rTr}r rc N|jrtd||_d|_|S)NzWAnimation arguments must be passed before accessing methods and can only be passed onceT)rNrrO)rLr}s rR__call__z_AnimationBuilder.__call__ s8   i   $ rT method_namerCallable[..., Self]ctjj|tdjrsjrt dd fd }d _d _|S) N_override_animatezDMethod chaining is currently not supported for overridden animations method_argsr method_kwargsrJrcr1tdjjg|Rdji|_n1jt|||i|S)NzCallable[..., Animation]rO)rrUrrOrKrMrr)rVrWhas_overridden_animationmethodrLs rR update_targetz4_AnimationBuilder.__getattr__..update_target s' 6-D.0H--L-! --- #n - $ --)) ##N6; $V$VWWW 5}555KrTT)rVrrWrrJr)rrrrZrLrKNotImplementedErrorrN)rLrRr[rYrZs` @@rR __getattr__z_AnimationBuilder.__getattr__ s,k::#*63F#G#G   !9 d>W %V           $rT 'Animation'cddlm}|jp||j|j}|jD]\}}t||||S)Nr)_MethodAnimation)manim.animation.transformr`rKrrMrOrr)rLr`animrrs rRbuildz_AnimationBuilder.build" sp>>>>>>(X,<,//11 ' 'KD% D$ & & & & rTN)rrArJrr%)rRrrJrS)rJr^)rr&r'rrQr]rcrrTrRrXrX se / / / /    @      rTrX _Decorated).r5ceZdZddZdS)_OverrideAnimateDecorator decoratedrdrJcdSrNr)rLrgs rRrQz"_OverrideAnimateDecorator.__call__4 srTN)rgrdrJrd)rr&r'rQrrTrRrfrf2 sCCCCCCrTrfrZtypes.FunctionTypecdfd }|S)aDecorator for overriding method animations. This allows to specify a method (returning an :class:`~.Animation`) which is called when the decorated method is used with the ``.animate`` syntax for animating the application of a method. .. seealso:: :attr:`OpenGLMobject.animate` .. note:: Overridden methods cannot be combined with normal or other overridden methods using method chaining with the ``.animate`` syntax. Examples -------- .. manim:: AnimationOverrideExample class CircleWithContent(VGroup): def __init__(self, content): super().__init__() self.circle = Circle() self.content = content self.add(self.circle, content) content.move_to(self.circle.get_center()) def clear_content(self): self.remove(self.content) self.content = None @override_animate(clear_content) def _clear_content_animation(self, anim_args=None): if anim_args is None: anim_args = {} anim = Uncreate(self.content, **anim_args) self.clear_content() return anim class AnimationOverrideExample(Scene): def construct(self): t = Text("hello!") my_mobject = CircleWithContent(t) self.play(Create(my_mobject)) self.play(my_mobject.animate.clear_content()) self.wait() animation_methodrdrJc|_|SrN)rU)rkrZs rR decoratorz#override_animate..decoratork s#3 rT)rkrdrJrdr)rZrms` rRoverride_animatern7 s*h       rT)rHrIrJrI)rZrirJrf)p __future__rrrA itertoolsr;r rtypescollections.abcrrrr functoolsrr mathr typingr r r rrrrrrrr+numpyr numpy.typingnpttyping_extensionsrmanimrrmanim.constantsmanim.data_structuresrrr manim.typingrrrrmanim.utils.bezierrr manim.utils.colorr!r"r#r$r%r&manim.utils.config_opsr'r(manim.utils.iterablesr)r*r+r,r-r.r/r0manim.utils.pathsr1manim.utils.space_opsr2r3r4manim.animation.animationr5r7r8r9r:r;r<r=r>r?r@rorrBr)rCrDrEgenericrFrU__all__voidrYNDArrayrZrArVrWrXrdrfrnrrTrRrs"""""""  BBBBBBBBBBBB********                        ! 000000;;;;;; @???????43333333                    ,+++++ /333333;;;;;;                      $,_e,Df,L#MMMMM&./@&/H&IIIII.1BBHBBBB B GG2: . . .E     O N N' !!!!~\2 2222N-9N-9N-9N-9N-9N-9N-9N-9bZ- AAAAA-AAA@BBBBBBBBJW\2B)C D D D DDDDDDDD 888888rT