mirror of
https://github.com/3b1b/manim.git
synced 2025-04-13 09:47:07 +00:00
103 lines
2.9 KiB
Python
103 lines
2.9 KiB
Python
from __future__ import annotations
|
|
|
|
from colour import Color
|
|
|
|
from manimlib.constants import BLACK, RED, YELLOW
|
|
from manimlib.constants import DL, DOWN, DR, LEFT, RIGHT, UL, UR
|
|
from manimlib.constants import SMALL_BUFF
|
|
from manimlib.mobject.geometry import Line
|
|
from manimlib.mobject.geometry import Rectangle
|
|
from manimlib.mobject.types.vectorized_mobject import VGroup
|
|
from manimlib.mobject.types.vectorized_mobject import VMobject
|
|
from manimlib.utils.config_ops import digest_config
|
|
from manimlib.utils.customization import get_customization
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import Union
|
|
|
|
from manimlib.mobject.mobject import Mobject
|
|
|
|
ManimColor = Union[str, Color]
|
|
|
|
|
|
class SurroundingRectangle(Rectangle):
|
|
CONFIG = {
|
|
"color": YELLOW,
|
|
"buff": SMALL_BUFF,
|
|
}
|
|
|
|
def __init__(self, mobject: Mobject, **kwargs):
|
|
digest_config(self, kwargs)
|
|
kwargs["width"] = mobject.get_width() + 2 * self.buff
|
|
kwargs["height"] = mobject.get_height() + 2 * self.buff
|
|
Rectangle.__init__(self, **kwargs)
|
|
self.move_to(mobject)
|
|
|
|
|
|
class BackgroundRectangle(SurroundingRectangle):
|
|
CONFIG = {
|
|
"stroke_width": 0,
|
|
"stroke_opacity": 0,
|
|
"fill_opacity": 0.75,
|
|
"buff": 0
|
|
}
|
|
|
|
def __init__(self, mobject: Mobject, color: ManimColor = None, **kwargs):
|
|
if color is None:
|
|
color = get_customization()['style']['background_color']
|
|
SurroundingRectangle.__init__(self, mobject, color=color, **kwargs)
|
|
self.original_fill_opacity = self.fill_opacity
|
|
|
|
def pointwise_become_partial(self, mobject: Mobject, a: float, b: float):
|
|
self.set_fill(opacity=b * self.original_fill_opacity)
|
|
return self
|
|
|
|
def set_style_data(
|
|
self,
|
|
stroke_color: ManimColor | None = None,
|
|
stroke_width: float | None = None,
|
|
fill_color: ManimColor | None = None,
|
|
fill_opacity: float | None = None,
|
|
family: bool = True
|
|
):
|
|
# Unchangeable style, except for fill_opacity
|
|
VMobject.set_style_data(
|
|
self,
|
|
stroke_color=BLACK,
|
|
stroke_width=0,
|
|
fill_color=BLACK,
|
|
fill_opacity=fill_opacity
|
|
)
|
|
return self
|
|
|
|
def get_fill_color(self) -> Color:
|
|
return Color(self.color)
|
|
|
|
|
|
class Cross(VGroup):
|
|
CONFIG = {
|
|
"stroke_color": RED,
|
|
"stroke_width": [0, 6, 0],
|
|
}
|
|
|
|
def __init__(self, mobject: Mobject, **kwargs):
|
|
super().__init__(
|
|
Line(UL, DR),
|
|
Line(UR, DL),
|
|
)
|
|
self.insert_n_curves(20)
|
|
self.replace(mobject, stretch=True)
|
|
self.set_stroke(self.stroke_color, width=self.stroke_width)
|
|
|
|
|
|
class Underline(Line):
|
|
CONFIG = {
|
|
"buff": SMALL_BUFF,
|
|
}
|
|
|
|
def __init__(self, mobject: Mobject, **kwargs):
|
|
super().__init__(LEFT, RIGHT, **kwargs)
|
|
self.match_width(mobject)
|
|
self.next_to(mobject, DOWN, buff=self.buff)
|