l j0(6dZddlmZddlmZmZddlZddlm Z ddlm Z m Z m Z mZmZddlmZddlmZdd lmZer dd lmZmZmZd d lmZgd ZGddeeZGddeZGddeZGddeZ GddeZ!dS)z0Boolean operations for two-dimensional mobjects.) annotations) TYPE_CHECKINGAnyN)Path)PathVerb difference intersectionunionxor)config)ConvertToOpenGL)VMobject)Point2DLike_Array Point3D_ArrayPoint3DLike_Array) RendererType)Union Intersection Difference Exclusionc.eZdZdZ ddd Zdd ZddZdS) _BooleanOpszThis class contains some helper functions which helps to convert to and from skia objects and manim objects (:class:`~.VMobject`). points%Point2DLike_Array | Point3DLike_Arrayz_dimfloatreturnrct|}t|D]0\}}t|dkrtj||||<1tj|S)a3Converts an iterable with coordinates in 2D to 3D by adding :attr:`z_dim` as the Z coordinate. Parameters ---------- points An iterable of points. z_dim Default value for the Z coordinate. Returns ------- Point3D_Array A list of the points converted to 3D. Example ------- >>> a = _BooleanOps() >>> p = [(1, 2), (3, 4)] >>> a._convert_2d_to_3d_array(p) array([[1., 2., 0.], [3., 4., 0.]]) )list enumeratelennpappendasarray)selfrrlist_of_pointsipoints ]/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/mobject/geometry/boolean_ops.py_convert_2d_to_3d_arrayz#_BooleanOps._convert_2d_to_3d_arrays`8f!.11 < >wGG  U2A2Y''#(22KCRDK1BQB1"RaR&111112271:wr{KK!JJLLL !_ 2 2 2;;FCCH# ! ! DDWMM  U2A2Y''',<>>>hm++44V<<$$U1X....hn,,$$%78888hm++55f==B66r2>>>> ; ; ;<<<r.N)r)rrrrrr)r/rrr0)rDr0rr)__name__ __module__ __qualname____doc__r-rMrar.r,rrsf * * * * *D////b######r.r) metaclassc$eZdZdZd fd ZxZS) raUnion of two or more :class:`~.VMobject` s. This returns the common region of the :class:`~VMobject` s. Parameters ---------- vmobjects The :class:`~.VMobject` s to find the union of. Raises ------ ValueError If less than 2 :class:`~.VMobject` s are passed. Example ------- .. manim:: UnionExample :save_last_frame: class UnionExample(Scene): def construct(self): sq = Square(color=RED, fill_opacity=1) sq.move_to([-2, 0, 0]) cr = Circle(color=BLUE, fill_opacity=1) cr.move_to([-1.3, 0.7, 0]) un = Union(sq, cr, color=GREEN, fill_opacity=1) un.move_to([1.5, 0.3, 0]) self.add(sq, cr, un) vmobjectsrkwargsrrNonec&t|dkrtdtjdi|fd|D}t }t |||dS)Nr!z%At least 2 mobjects needed for Union.c:g|]}|Srf)rM).0r/r(s r, z"Union.__init__..s4   >FD / / 9 9   r.rf)r$ ValueErrorsuper__init__r0r getPenra)r(rirjpathsoutpen __class__s` r,rrzUnion.__init__s y>>A  DEE E""6"""    JS    eV]]__%%% ++F33333r.rirrjrrrkrbrcrdrerr __classcell__rvs@r,rrsG< 4 4 4 4 4 4 4 4 4 4r.rc$eZdZdZd fd ZxZS) raSubtracts one :class:`~.VMobject` from another one. Parameters ---------- subject The 1st :class:`~.VMobject`. clip The 2nd :class:`~.VMobject` Example ------- .. manim:: DifferenceExample :save_last_frame: class DifferenceExample(Scene): def construct(self): sq = Square(color=RED, fill_opacity=1) sq.move_to([-2, 0, 0]) cr = Circle(color=BLUE, fill_opacity=1) cr.move_to([-1.3, 0.7, 0]) un = Difference(sq, cr, color=GREEN, fill_opacity=1) un.move_to([1.5, 0, 0]) self.add(sq, cr, un) subjectrcliprjrrrkc tjdi|t}t||g||g|||dSNrf)rqrrr0rrMrsrar(r|r}rjrurvs r,rrzDifference.__init__s""6"""  0 0 9 9 :  0 0 6 6 7 MMOO   ++F33333r.r|rr}rrjrrrkrxrzs@r,rrsG44444444444r.rc$eZdZdZd fd ZxZS) ra3Find the intersection of two :class:`~.VMobject` s. This keeps the parts covered by both :class:`~.VMobject` s. Parameters ---------- vmobjects The :class:`~.VMobject` to find the intersection. Raises ------ ValueError If less the 2 :class:`~.VMobject` are passed. Example ------- .. manim:: IntersectionExample :save_last_frame: class IntersectionExample(Scene): def construct(self): sq = Square(color=RED, fill_opacity=1) sq.move_to([-2, 0, 0]) cr = Circle(color=BLUE, fill_opacity=1) cr.move_to([-1.3, 0.7, 0]) un = Intersection(sq, cr, color=GREEN, fill_opacity=1) un.move_to([1.5, 0, 0]) self.add(sq, cr, un) rirrjrrrkcRt|dkrtdtjdi|t }t ||dg||dg||}tdt|D]P}t }t |g|||g||}Q| |dS)Nr!z,At least 2 mobjects needed for Intersection.rr2rf) r$rprqrrr0r rMrsrangera)r(rirjru new_outpen_irvs r,rrzIntersection.__init__s& y>>A  KLL L""6"""  0 01 > > ?  0 01 > > ? MMOO    3y>>**  B!J 44Yr]CCD!!##     FF ++F33333r.rwrxrzs@r,rrsG<4444444444r.rc$eZdZdZd fd ZxZS) raFind the XOR between two :class:`~.VMobject`. This creates a new :class:`~.VMobject` consisting of the region covered by exactly one of them. Parameters ---------- subject The 1st :class:`~.VMobject`. clip The 2nd :class:`~.VMobject` Example ------- .. manim:: IntersectionExample :save_last_frame: class IntersectionExample(Scene): def construct(self): sq = Square(color=RED, fill_opacity=1) sq.move_to([-2, 0, 0]) cr = Circle(color=BLUE, fill_opacity=1) cr.move_to([-1.3, 0.7, 0]) un = Exclusion(sq, cr, color=GREEN, fill_opacity=1) un.move_to([1.5, 0.4, 0]) self.add(sq, cr, un) r|rr}rjrrrkc tjdi|t}t||g||g|||dSr)rqrrr0r rMrsrars r,rrzExclusion.__init__;s""6"""  0 0 9 9 :  0 0 6 6 7 MMOO   ++F33333r.rrxrzs@r,rrsG84444444444r.r)"re __future__rtypingrrnumpyr%pathopsrr0rrr r r manimr )manim.mobject.opengl.opengl_compatibilityr &manim.mobject.types.vectorized_mobjectr manim.typingrrr constantsr__all__rrrrrrfr.r,rs66""""""%%%%%%%%$$$$$$BBBBBBBBBBBBBBEEEEEE;;;;;;QPPPPPPPPPP%%%%%% > > >|||||(o||||~(4(4(4(4(4K(4(4(4V#4#4#4#4#4#4#4#4L4444444444;444444n%4%4%4%4%4 %4%4%4%4%4r.