l j\JUdZddlmZddlZddlmZddlmZddlm Z ddl Z ddl ZddlmZmZddlmZmZd d lmZd d lmZd d lmZd d lmZddlmZddlm Z ddl!m"Z"m#Z#m$Z$ddl%m&Z&ddl'm(Z(m)Z)ddgZ*iZ+de,d<d!dZ-Gdde)e&Z.Gd de)e&Z/dS)"z$Mobjects generated from an SVG file.) annotationsN)Path)Any) ElementTree)configlogger) ManimColorParsableManimColor)RIGHT)$get_quadratic_approximation_of_cubic)get_full_vector_image_path)hash_obj)Circle)Line)Polygon RectangleRoundedRectangle)ConvertToOpenGL)VGroupVMobject SVGMobjectVMobjectFromSVGPathzdict[int, SVGMobject]SVG_HASH_TO_MOB_MAPxfloatyreturn np.ndarrayc0tj||dgS)N)nparray)rrs X/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/mobject/svg/svg_mobject.py_convert_point_to_3dr& s 8Q3K  cReZdZdZ dRdSfd ZdTdZedUdZdVd ZdWd"Z dXd%Z dYd'Z dZd+Z d[d/Z ed\d4Zed]d6Zd^d:Zed_d>Zed`dBZedadFZedbdJZdcdMZedddPZdVdQZxZS)era\ A vectorized mobject created from importing an SVG file. Parameters ---------- file_name The path to the SVG file. should_center Whether or not the mobject should be centered after being imported. height The target height of the mobject, set to 2 Manim units by default. If the height and width are both set to ``None``, the mobject is imported without being scaled. width The target width of the mobject, set to ``None`` by default. If the height and the width are both set to ``None``, the mobject is imported without being scaled. color The color (both fill and stroke color) of the mobject. If ``None`` (the default), the colors set in the SVG file are used. opacity The opacity (both fill and stroke opacity) of the mobject. If ``None`` (the default), the opacity set in the SVG file is used. fill_color The fill color of the mobject. If ``None`` (the default), the fill colors set in the SVG file are used. fill_opacity The fill opacity of the mobject. If ``None`` (the default), the fill opacities set in the SVG file are used. stroke_color The stroke color of the mobject. If ``None`` (the default), the stroke colors set in the SVG file are used. stroke_opacity The stroke opacity of the mobject. If ``None`` (the default), the stroke opacities set in the SVG file are used. stroke_width The stroke width of the mobject. If ``None`` (the default), the stroke width values set in the SVG file are used. svg_default A dictionary in which fallback values for unspecified properties of elements in the SVG file are defined. If ``None`` (the default), ``color``, ``opacity``, ``fill_color`` ``fill_opacity``, ``stroke_color``, and ``stroke_opacity`` are set to ``None``, and ``stroke_width`` is set to 0. path_string_config A dictionary with keyword arguments passed to :class:`.VMobjectFromSVGPath` used for importing path elements. If ``None`` (the default), no additional arguments are passed. use_svg_cache If True (default), the svg inputs (e.g. file_name, settings) will be used as a key and a copy of the created mobject will be saved using that key to be quickly retrieved if the same inputs need be processed later. For large SVGs which are used only once, this can be omitted to improve performance. kwargs Further arguments passed to the parent class. NTr file_namestr | os.PathLike | None should_centerboolheight float | NonewidthcolorParsableManimColor | Noneopacity fill_color fill_opacity stroke_colorstroke_opacity stroke_width svg_default dict | Nonepath_string_config use_svg_cachekwargsrc tjddddd||t|nd|_||_||_||_t||_||_ ||_ ||_ | |_ | |_ | |_i|_|jd|_| dddddddd} | |_| i} | |_|||||| | | |dS)N)r0r5r3r)r0r2r3r4r7r5r6)r;)r3r4r5r6r7)super__init__rr)r+ svg_height svg_widthr r0r2r3r4r5r6r7id_to_vgroup_dictr8r:init_svg_mobject set_stylemove_into_position)selfr)r+r-r/r0r2r3r4r5r6r7r8r:r;r< __class__s r%r@zSVGMobject.__init__asO$ Rt$4RR6RRR-6,Ait* &&  $((,(46   $ !D   " $ ! $"&K'  %!# "4 M::: !%%)%     !!!!!r'rNonec|rTt|j}|tvr7t|}|j||j|_dS||r|t|<dSdS)zChecks whether the SVG has already been imported and generates it if not. See also -------- :meth:`.SVGMobject.generate_mobject` N)r hash_seedrcopyaddrCgenerate_mobject)rGr;hash_valmobs r%rDzSVGMobject.init_svg_mobjects  //H...)(388::#),)>&   8,0IIKK  ) ) ) 8 8r'tuplecV|jj|j|j|jt jfS)zA unique hash representing the result of the generated mobject points. Used as keys in the ``SVG_HASH_TO_MOB_MAP`` caching dictionary. )rH__name__r8r:r)rrendererrGs r%rKzSVGMobject.hash_seeds, N #    # N O   r'c|}tj|}||}||jd|j}||tj |}| | |\}}|j |||_ |tdS)z8Parse the SVG and translate its elements to submobjects._N) get_file_pathETparsemodify_xml_tree with_namestemsuffixwriteseSVGunlinkget_mobjects_fromrMrCflipr )rG file_path element_treenew_treemodified_file_pathsvgmobjects mobject_dicts r%rNzSVGMobject.generate_mobjects&&(( x ** '' 55&00IN1W1WYEU1W1WXX)***fll-..!!###!%!7!7! !?@@ @)$.999r'rfET.ElementTreecv|}d|}fd|jD}t jdi}t j|d|}t j|d|}||t j|S)zModifies the SVG element tree to include default style information. Parameters ---------- element_tree The parsed element tree from the SVG file. )fill fill-opacitystrokestroke-opacity stroke-widthstylec$i|] \}}|v || Sr>r>).0kv style_keyss r% z.SVGMobject.modify_xml_tree..s$SSSDAq1 ??1a???r'rig) generate_config_style_dictgetrootattribitemsrYElement SubElementextendr) rGrfconfig_style_dictrootroot_style_dictnew_rootconfig_style_noderoot_style_noderzs @r%r[zSVGMobject.modify_xml_trees!;;== ##%%SSSSDK,=,=,?,?SSS:eR((M(C9JKK-(93PPt$$$~h'''r'dict[str, str]cdddddd}|j}i}|D]+\}}|D]#}|| t||||<$,|S)z)rw subelementdepths r% z/SVGMobject.get_mobjects_from..(s,UUj%!)4UUUUUUr'Nz4Exception occurred in 'get_mobjects_from'. Details: )appendrlenpoprvalues Exception isinstancer`GroupUserr SimpleLineRectrEllipserPolylineTextget_mob_from_shape_elementrMrerror)rGrirstackgroup_id_number vgroup_stack vgroup_namesvgroupselement group_namevgrP parent_nameers @r%rczSVGMobject.get_mobjects_fromsd"$13 c1X#)( #)( &,fhh%7%jj1nn"YY[[NGU'5 2L % !566  % % %@@@ 1$ %B    + + +    + + +"$GJ 'BHbf#566 V UUUUwttQSt}UUUUUU Y        :99'BBC c***+7+<::K#K044S9999 Y Y Y WTUWWXXXXXXXX YI%jj1nnNws++BBB.B#>CC r} - - &&u--CC rw ' ' &&u--CC  2:6 7 7 ))%00CC rz * * ))%00CC r{ + + **511CC rw ' ' &&u--CC NEU EE F F FC ;cnn..;J ##C/// eR- . . 85; 8  ! !#u 7 7 7 r'rPrmatrix se.Matrixctj|j|jg|j|jgg}tj|j|jdg}||| ||S)zApply SVG transformations to the converted mobject. Parameters ---------- mob The converted mobject. matrix The transformation matrix determined from the SVG transformation. r") r#r$acbdrf apply_matrixshift)rPrmatvecs r%rzSVGMobject.handle_transform[slh68,vx.BCDDh&(C011  # r'se.GraphicObjectc||j|jj|jj|jj|jj|S)zApply SVG style information to the converted mobject. Parameters ---------- mob The converted mobject. shape The parsed SVG element. )r7r5r6r3r4)rEr7rrhexrgbr2rp)rPrs r%rz!SVGMobject.apply_style_to_mobjectmsL +, </z(+     r'pathse.Pathrc&t|fi|jS)zConvert a path element to a vectorized mobject. Parameters ---------- path The parsed SVG path. )rr:)rGrs r%rzSVGMobject.path_to_mobjects#4CC4+BCCCr'linese.Linerctt|j|jt|j|jS)zConvert a line element to a vectorized mobject. Parameters ---------- line The parsed SVG line. )startend)rr&x1y1x2y2)rs r%rzSVGMobject.line_to_mobjects=&tw88$TWdg66    r'rectse.Rectrc|jdks |jdkrt|j|j}nKt |j|j|jz|jz |j}||j|t|j |jdz z|j |jdz z|S)zConvert a rectangle element to a vectorized mobject. Parameters ---------- rect The parsed SVG rectangle. r)r/r-)r/r- corner_radiusr) rxryrr/r-rstretch_to_fit_heightrr&rr)rrPs r%rzSVGMobject.rect_to_mobjects 7a<<47a<<j{CC #j{TW,tw6"gC  % %dk 2 2 2 $*q.!8$&4;QR?:R S S    r'ellipsese.Ellipse | se.Circlerct|j}|j|jkr|d|jz|t |j|j|S)zConvert an ellipse or circle element to a vectorized mobject. Parameters ---------- ellipse The parsed SVG ellipse or circle. )radiusr)rrrrrr&cxcy)rrPs r%rzSVGMobject.ellipse_to_mobjectseGJ''' : # #  % %a'*n 5 5 5 &wz7:>>??? r'polygon se.Polygonrc,d|D}t|S)zConvert a polygon element to a vectorized mobject. Parameters ---------- polygon The parsed SVG polygon. c g|] }t| Sr>r&rwpoints r% z1SVGMobject.polygon_to_mobject..sDDD5&.DDDr')r)rpointss r%rzSVGMobject.polygon_to_mobjects#EDGDDDr'polyline se.Polylinec|d|D}|}||S)zConvert a polyline element to a vectorized mobject. Parameters ---------- polyline The parsed SVG polyline. c g|] }t| Sr>rrs r%rz2SVGMobject.polyline_to_mobject..sEEE5&.EEEr')get_mobject_type_classset_points_as_corners)rGrrvmobject_classs r%rzSVGMobject.polyline_to_mobjectsEFEHEEE4466~55f===r'textse.TextcNtjdt|dS)zConvert a text element to a vectorized mobject. .. warning:: Not yet implemented. Parameters ---------- text The parsed SVG text. rN)rrr)rs r%rzSVGMobject.text_to_mobjects( @DJJ@@AAAr'c|jr||j||j|j||jdSdS)z3Scale and move the generated mobject into position.N)r-)r/)r+centerrAsetrBrUs r%rFzSVGMobject.move_into_positionsd    KKMMM ? & HHDOH , , , > % HH4>H * * * * * & %r')NTrNNNNNNNNNNT)r)r*r+r,r-r.r/r.r0r1r2r.r3r1r4r.r5r1r6r.r7r.r8r9r:r9r;r,r<r)r;r,rrI)rrQrrI)rr)rfrnrrn)rr)rirrr)rrrr)rPrrrrr)rPrrrrr)rrrr)rrrr)rrrr)rrrr)rrrr)rrrr)rrrr)rS __module__ __qualname____doc__r@rDpropertyrKrNrXr[r}rcr staticmethodrrrrrrrrrrF __classcell__rHs@r%rr$sG::|/3" "+/ $04%)26'+%)#'*.">">">">">">">"@8888(    X  ":::: ((((6$8888t2\"\&DDDD    \  \2   \     \   > > > >   \ ++++++++r') metaclasscDeZdZdZ ddfd ZddZddZddZxZS)raA vectorized mobject representing an SVG path. .. note:: The ``long_lines``, ``should_subdivide_sharp_curves``, and ``should_remove_null_curves`` keyword arguments are only respected with the OpenGL renderer. Parameters ---------- path_obj A parsed SVG path object. long_lines Whether or not straight lines in the vectorized mobject are drawn in one or two segments. should_subdivide_sharp_curves Whether or not to subdivide subcurves further in case two segments meet at an angle that is sharper than a given threshold. should_remove_null_curves Whether or not to remove subcurves of length 0. kwargs Further keyword arguments are passed to the parent class. Fpath_objr long_linesr,should_subdivide_sharp_curvesshould_remove_null_curvesr<rc |||_||_||_||_t jdi|dS)Nr>)approximate_arcs_with_quadsrrrrr?r@)rGrrrrr<rHs r%r@zVMobjectFromSVGPath.__init__sX ,,...  $-J*)B&""6"""""r'rrIc|tjdkrK|jr||jr+||dSdSdS)Nopengl)handle_commandsrrTrsubdivide_sharp_curvesr set_pointsget_points_without_null_curvesrUs r%generate_pointsz#VMobjectFromSVGPath.generate_points!s  ?h & &1 .++---- G C C E EFFFFF ' & G Gr'c.|dSN)rrUs r% init_pointszVMobjectFromSVGPath.init_points.s r'c  gddd ddd fd  |jd krd fd d  fd d! fd }nd fd d  fd d! fd }|jD]9}|j}|tjkr t |jd4|tjkr|t |j]|tjkr&t |j t |j|tj kr3t |j t |j t |j|tj krCttj z dkr | d)t%d|tjdd|_t+dkr!tj|jd|_dSdS)"NF) true_moveptr r#r,rrIc ||rdSdSr r>)r$r# curve_start last_movelast_true_moves r%move_penz5VMobjectFromSVGPath.handle_commands..move_pen7s.I"'  +!* + +r'rcp1cp2rctdzdksJt||||gz |dS)Nr*rr)rr+r,r all_pointsr)s r% add_cubicz6VMobjectFromSVGPath.handle_commands..add_cubicAsS:*a///Z///uc344  r'cpc^|||z|zdz ||z|zdz ||dSNr r>)rr1rr0r)s r%add_quadz5VMobjectFromSVGPath.handle_commands..add_quadIsD %%"*r/Q!6b3!8KSQQQ r'c^|||z|zdz ||z|zdz ||dSr3r>)rrr0r)s r%add_linez5VMobjectFromSVGPath.handle_commands..add_lineMsO EEMC/14us{S7HA6Ms r'ctdzdksJtt||||}|ddz |ddz |dSNr r)rr tolist)rr+r,r two_quadsr/r)s r%r0z6VMobjectFromSVGPath.handle_commands..add_cubicUs:*a///Z///@  im22444 im22444  r'ctdzdksJt|||gz |dSr8r.)rr1rr/r)s r%r4z5VMobjectFromSVGPath.handle_commands..add_quaddsO:*a///Z///ub#..  r'cD|||zdz ||dS)Nrr>)rrr4r)s r%r6z5VMobjectFromSVGPath.handle_commands..add_linejs1 13777 r'Tg-C6?zNot implemented: rfloat64)ndmindtyper)rr )r$r r#r,rrI) rr r+r r,r rr rrI)rr r1r rr rrI)rr rr rrI)n_points_per_curverrHr`Mover&rrQuadraticBeziercontrol CubicBeziercontrol1control2Closeabsr#linalgnormAssertionErrorr$rrreshape) rGr6segment segment_classr0r4r/r&r'r(r)s @@@@@@@r%rz#VMobjectFromSVGPath.handle_commands1s') $  :? + + + + + + + + + +  "a ' '                                            } J JG#-M''-w{;tLLLLL"'))$8'+$FGGGG""444('/:('+6 ".00 ('*:;('*:;('+6  "(**ry~~i.&@AABBVKKHY777" $%H%H%HIIIhz)DDD  z??a  *T[&99DKKK r')FFF) rrrr,rr,rr,r<rr) rSr r r r@rr!rrrs@r%rrs:!.3*/ #######$ G G G G_:_:_:_:_:_:_:_:r')rrrrrr )0r  __future__rospathlibrtypingr xml.etreerrYnumpyr# svgelementsr`manimrrmanim.utils.colorr r constantsr utils.bezierr utils.imagesrutils.iterablesr geometry.arcr geometry.linergeometry.polygramrrropengl.opengl_compatibilityrtypes.vectorized_mobjectrr__all__r__annotations__r&rrr>r'r%rcs***"""""" '''''' <<<<<<<<@@@@@@666666''''''!!!!!! DDDDDDDDDD99999977777777 . /.0////!!!!M+M+M+M+M+_M+M+M+M+`\:\:\:\:\:(o\:\:\:\:\:\:r'