public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-11-13 15:47:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add Cone

Browse source
This commit is contained in:
Grant Sanderson 2024-11-08 14:28:17 -06:00
parent 9475fcd19e
commit b84376d6fd
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
manimlib/mobject/three_dimensions.py
View file

@ -169,6 +169,20 @@ class Cylinder(Surface):
return np.array([np.cos(u), np.sin(u), v]) return np.array([np.cos(u), np.sin(u), v])
class Cone(Cylinder):
def __init__(
self,
u_range: Tuple[float, float] = (0, TAU),
v_range: Tuple[float, float] = (0, 1),
*args,
**kwargs,
):
super().__init__(u_range=u_range, v_range=v_range, *args, **kwargs)
def uv_func(self, u: float, v: float) -> np.ndarray:
return np.array([(1 - v) * np.cos(u), (1 - v) * np.sin(u), v])
class Line3D(Cylinder): class Line3D(Cylinder):
def __init__( def __init__(
self, self,