l j(|dZddlmZgdZddlmZmZddlZddl Tddl m Z ddl m Z mZdd lmZdd lmZdd lmZerdd lmZmZGd deeZGddeZGddeeZGddeZGddee ZGddeZGddee ZGddeZdS)zEA collection of tip mobjects for use with :class:`~.TipableVMobject`.) annotations)ArrowTipArrowCircleFilledTipArrowCircleTipArrowSquareTipArrowSquareFilledTipArrowTriangleTipArrowTriangleFilledTip StealthTip) TYPE_CHECKINGAnyN)*)Circle)SquareTriangle)ConvertToOpenGL)VMobject)angle_of_vector)Point3DVector3DceZdZdZddZedd Zedd Zedd ZeddZ eddZ dS)ra Base class for arrow tips. .. seealso:: :class:`ArrowTriangleTip` :class:`ArrowTriangleFilledTip` :class:`ArrowCircleTip` :class:`ArrowCircleFilledTip` :class:`ArrowSquareTip` :class:`ArrowSquareFilledTip` :class:`StealthTip` Examples -------- Cannot be used directly, only intended for inheritance:: >>> tip = ArrowTip() Traceback (most recent call last): ... NotImplementedError: Has to be implemented in inheriting subclasses. Instead, use one of the pre-defined ones, or make a custom one like this: .. manim:: CustomTipExample >>> from manim import RegularPolygon, Arrow >>> class MyCustomArrowTip(ArrowTip, RegularPolygon): ... def __init__(self, length=0.35, **kwargs): ... RegularPolygon.__init__(self, n=5, **kwargs) ... self.width = length ... self.stretch_to_fit_height(length) >>> arr = Arrow( ... np.array([-2, -2, 0]), np.array([2, 2, 0]), tip_shape=MyCustomArrowTip ... ) >>> isinstance(arr.tip, RegularPolygon) True >>> from manim import Scene, Create >>> class CustomTipExample(Scene): ... def construct(self): ... self.play(Create(arr)) Using a class inherited from :class:`ArrowTip` to get a non-filled tip is a shorthand to manually specifying the arrow tip style as follows:: >>> arrow = Arrow(np.array([0, 0, 0]), np.array([1, 1, 0]), ... tip_style={'fill_opacity': 0, 'stroke_width': 3}) The following example illustrates the usage of all of the predefined arrow tips. .. manim:: ArrowTipsShowcase :save_last_frame: class ArrowTipsShowcase(Scene): def construct(self): tip_names = [ 'Default (YELLOW)', 'ArrowTriangleTip', 'Default', 'ArrowSquareTip', 'ArrowSquareFilledTip', 'ArrowCircleTip', 'ArrowCircleFilledTip', 'StealthTip' ] big_arrows = [ Arrow(start=[-4, 3.5, 0], end=[2, 3.5, 0], color=YELLOW), Arrow(start=[-4, 2.5, 0], end=[2, 2.5, 0], tip_shape=ArrowTriangleTip), Arrow(start=[-4, 1.5, 0], end=[2, 1.5, 0]), Arrow(start=[-4, 0.5, 0], end=[2, 0.5, 0], tip_shape=ArrowSquareTip), Arrow([-4, -0.5, 0], [2, -0.5, 0], tip_shape=ArrowSquareFilledTip), Arrow([-4, -1.5, 0], [2, -1.5, 0], tip_shape=ArrowCircleTip), Arrow([-4, -2.5, 0], [2, -2.5, 0], tip_shape=ArrowCircleFilledTip), Arrow([-4, -3.5, 0], [2, -3.5, 0], tip_shape=StealthTip) ] small_arrows = ( arrow.copy().scale(0.5, scale_tips=True).next_to(arrow, RIGHT) for arrow in big_arrows ) labels = ( Text(tip_names[i], font='monospace', font_size=20, color=BLUE).next_to(big_arrows[i], LEFT) for i in range(len(big_arrows)) ) self.add(*big_arrows, *small_arrows, *labels) argsr kwargsreturnNonec td)Nz/Has to be implemented in inheriting subclasses.)NotImplementedError)selfrrs V/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/mobject/geometry/tips.py__init__zArrowTip.__init__ss!"STTTrc,|dS)aThe base point of the arrow tip. This is the point connecting to the arrow line. Examples -------- :: >>> from manim import Arrow >>> arrow = Arrow(np.array([0, 0, 0]), np.array([2, 0, 0]), buff=0) >>> arrow.tip.base.round(2) + 0. # add 0. to avoid negative 0 in output array([1.65, 0. , 0. ]) g?)point_from_proportionrs rbasez ArrowTip.basevs ))#...r!c |jd}|S)a"The tip point of the arrow tip. Examples -------- :: >>> from manim import Arrow >>> arrow = Arrow(np.array([0, 0, 0]), np.array([2, 0, 0]), buff=0) >>> arrow.tip.tip_point.round(2) + 0. array([2., 0., 0.]) r)points)r tip_points rr(zArrowTip.tip_points""[^ r!rc |j|jz S)a?The vector pointing from the base point to the tip point. Examples -------- :: >>> from manim import Arrow >>> arrow = Arrow(np.array([0, 0, 0]), np.array([2, 2, 0]), buff=0) >>> arrow.tip.vector.round(2) + 0. array([0.25, 0.25, 0. ]) )r(r%r$s rvectorzArrowTip.vectors~ ))r!floatc*t|jS)a#The angle of the arrow tip. Examples -------- :: >>> from manim import Arrow >>> arrow = Arrow(np.array([0, 0, 0]), np.array([1, 1, 0]), buff=0) >>> bool(round(arrow.tip.tip_angle, 5) == round(PI/4, 5)) True )rr*r$s r tip_anglezArrowTip.tip_anglest{+++r!cdttj|jS)aThe length of the arrow tip. Examples -------- :: >>> from manim import Arrow >>> arrow = Arrow(np.array([0, 0, 0]), np.array([1, 2, 0])) >>> round(arrow.tip.length, 3) 0.35 r+nplinalgnormr*r$s rlengthzArrowTip.lengths"RY^^DK00111r!N)rr rr rr)rr)rrrr+) __name__ __module__ __qualname____doc__r propertyr%r(r*r-r3r!rrrsQQfUUUU///X/"X& * * *X * , , ,X , 2 2 2X 2 2 2r!r) metaclasscBeZdZdZddedz efdd ZeddZdS)r z'Stealth' fighter / kite arrow shape. Naming is inspired by the corresponding `TikZ arrow shape `__.  fill_opacityr+ stroke_widthr3 start_anglerr c ||_tj|f||d||t jgdgdgdgdgdg|||jz dS)Nr@rA)r?rr)333333皙?r)rrr)rEgr)rBrr set_points_as_cornersr0arrayscaler3rr@rAr3rBrs rr zStealthTip.__init__s'  +,  JP    "" HII"NNII#OOII     6DK'(((((r!rcjttj|jdzS)zThe length of the arrow tip. In this case, the length is computed as the height of the triangle encompassing the stealth tip (otherwise, the tip is scaled too large). rFr/r$s rr3zStealthTip.lengths'RY^^DK0036777r!N) r@r+rAr+r3r+rBr+rr r4) r5r6r7r8DEFAULT_ARROW_TIP_LENGTHPIr r9r3r:r!rr r sj 014 )))))2888X888r!r c&eZdZdZddeeefddZdS)r zTriangular arrow tip.rr>r@r+rAr3widthrBrr rrc tj|f|||d|||_||||dS)N)r@rArB)rr rOstretch_to_fit_widthstretch_to_fit_height)rr@rAr3rOrBrs rr zArrowTriangleTip.__init__sp   %%#         !!&))) ""5)))))r!N)r@r+rAr+r3r+rOr+rBr+rr rrr5r6r7r8rLrMr r:r!rr r sD   0/ *******r!r c(eZdZdZ d d fd ZxZS)r zTTriangular arrow tip with filled tip. This is the default arrow tip shape. r=rr@r+rArr rrc @tjd||d|dSNrDr:superr rr@rAr __class__s rr zArrowTriangleFilledTip.__init__0 XlXXQWXXXXXr!r=rr@r+rAr+rr rrr5r6r7r8r __classcell__rZs@rr r s\ >?YYYYYYYYYYYr!r c$eZdZdZddeefdd ZdS)rzCircular arrow tip.rr>r@r+rAr3rBrr rrc v||_tj|f||d|||_||dS)NrD)rBrr rOrRrJs rr zArrowCircleTip.__init__ s]'  +,  JP     ""6*****r!N r@r+rAr+r3r+rBr+rr rrrSr:r!rrrsA 0 + + + + + + +r!rc(eZdZdZ d d fd ZxZS)rz#Circular arrow tip with filled tip.r=rr@r+rArr rrc @tjd||d|dSrVrWrYs rr zArrowCircleFilledTip.__init__3r[r!r\r]r^r`s@rrr0sX..>?YYYYYYYYYYYr!rc$eZdZdZddeefdd ZdS)rzSquare arrow tip.rr>r@r+rAr3rBrr rrc x||_tj|f|||d|||_||dS)N)r@rA side_length)rBrr rOrRrJs rr zArrowSquareTip.__init__<sc'  %%         ""6*****r!NrcrSr:r!rrr9sA 0 +++++++r!rc(eZdZdZ d d fd ZxZS)rz!Square arrow tip with filled tip.r=rr@r+rArr rrc @tjd||d|dSrVrWrYs rr zArrowSquareFilledTip.__init__Sr[r!r\r]r^r`s@rrrPsX,,>?YYYYYYYYYYYr!r) r8 __future__r__all__typingr r numpyr0manim.constantsmanim.mobject.geometry.arcrmanim.mobject.geometry.polygramrr)manim.mobject.opengl.opengl_compatibilityr&manim.mobject.types.vectorized_mobjectrmanim.utils.space_opsr manim.typingrrrr r r rrrrr:r!rrvsmLL""""""   &%%%%%%%------<<<<<<<<EEEEEE;;;;;;111111/........k2k2k2k2k2x?k2k2k2k2\(8(8(8(8(8(8(8(8V*****x***2 Y Y Y Y Y- Y Y Y+++++Xv+++&YYYYY>YYY+++++Xv+++.YYYYY>YYYYYr!