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

Add surround method for Rectangles and SurroundingRectangles

Browse source
This commit is contained in:
Grant Sanderson 2023-09-04 18:46:11 -04:00
parent fa798a2018
commit 87e4a71ca3
3 changed files with 24 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
manimlib/mobject/geometry.py
View file

@ -7,7 +7,7 @@ import numpy as np
from manimlib.constants import DL, DOWN, DR, LEFT, ORIGIN, OUT, RIGHT, UL, UP, UR from manimlib.constants import DL, DOWN, DR, LEFT, ORIGIN, OUT, RIGHT, UL, UP, UR
from manimlib.constants import GREY_A, RED, WHITE, BLACK from manimlib.constants import GREY_A, RED, WHITE, BLACK
from manimlib.constants import MED_SMALL_BUFF from manimlib.constants import MED_SMALL_BUFF, SMALL_BUFF
from manimlib.constants import DEGREES, PI, TAU from manimlib.constants import DEGREES, PI, TAU
from manimlib.mobject.mobject import Mobject from manimlib.mobject.mobject import Mobject
from manimlib.mobject.types.vectorized_mobject import DashedVMobject from manimlib.mobject.types.vectorized_mobject import DashedVMobject
@ -1046,6 +1046,12 @@ class Rectangle(Polygon):
self.set_width(width, stretch=True) self.set_width(width, stretch=True)
self.set_height(height, stretch=True) self.set_height(height, stretch=True)
def surround(self, mobject, buff=SMALL_BUFF) -> Self:
target_shape = np.array(mobject.get_shape()) + 2 * buff
self.set_shape(*target_shape)
self.move_to(mobject)
return self
class Square(Rectangle): class Square(Rectangle):
def __init__(self, side_length: float = 2.0, **kwargs): def __init__(self, side_length: float = 2.0, **kwargs):

3
manimlib/mobject/mobject.py
View file

@ -1540,6 +1540,9 @@ class Mobject(object):
def get_depth(self) -> float: def get_depth(self) -> float:
return self.length_over_dim(2) return self.length_over_dim(2)
def get_shape(self) -> Tuple[float]:
return tuple(self.length_over_dim(dim) for dim in range(3))
def get_coord(self, dim: int, direction: Vect3 = ORIGIN) -> float: def get_coord(self, dim: int, direction: Vect3 = ORIGIN) -> float:
""" """
Meant to generalize get_x, get_y, get_z Meant to generalize get_x, get_y, get_z

21
manimlib/mobject/shape_matchers.py
View file

@ -27,13 +27,20 @@ class SurroundingRectangle(Rectangle):
color: ManimColor = YELLOW, color: ManimColor = YELLOW,
**kwargs **kwargs
): ):
super().__init__( super().__init__(color=color, **kwargs)
width=mobject.get_width() + 2 * buff, self.buff = buff
height=mobject.get_height() + 2 * buff, self.surround(mobject)
color=color,
**kwargs def surround(self, mobject, buff=None) -> Self:
) self.mobject = mobject
self.move_to(mobject) self.buff = buff if buff is not None else self.buff
super().surround(mobject, self.buff)
return self
def set_buff(self, buff) -> Self:
self.buff = buff
self.surround(self.mobject)
return self
class BackgroundRectangle(SurroundingRectangle): class BackgroundRectangle(SurroundingRectangle):