public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-09-01 00:48:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add a custom style to ArrowTip.

Browse source
This commit is contained in:
widcardw 2021-04-13 17:40:48 +08:00 • committed by GitHub
parent 6b911f5721
commit e727faaccb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
manimlib/mobject/geometry.py
View file

@ -49,6 +49,7 @@ class TipableVMobject(VMobject):
"tip_config": { "tip_config": {
"fill_opacity": 1, "fill_opacity": 1,
"stroke_width": 0, "stroke_width": 0,
"tip_look": 1, # triangle=0, inner_smooth=1, dot=2
}, },
"normal_vector": OUT, "normal_vector": OUT,
} }
@ -786,12 +787,20 @@ class ArrowTip(Triangle):
"width": DEFAULT_ARROW_TIP_WIDTH, "width": DEFAULT_ARROW_TIP_WIDTH,
"length": DEFAULT_ARROW_TIP_LENGTH, "length": DEFAULT_ARROW_TIP_LENGTH,
"angle": 0, "angle": 0,
"tip_look": 1, # triangle=0, inner_smooth=1, dot=2
} }
def __init__(self, **kwargs): def __init__(self, **kwargs):
Triangle.__init__(self, start_angle=0, **kwargs) Triangle.__init__(self, start_angle=0, **kwargs)
self.set_height(self.width) self.set_height(self.width)
self.set_width(self.length, stretch=True) self.set_width(self.length, stretch=True)
if self.tip_look == 1:
self.set_height(self.length * 0.9, stretch=True)
self.data["points"][4] += np.array([0.6 * self.length, 0, 0])
elif self.tip_look == 2:
h = self.length / 2
self.clear_points()
self.data["points"] = Dot().set_width(h).get_points()
self.rotate(self.angle) self.rotate(self.angle)
def get_base(self): def get_base(self):