manim 0.17.0__py3-none-any.whl → 0.19.1__py3-none-any.whl

manim/mobject/table.py

@@ -65,9 +65,7 @@ __all__ = [
 
 
 import itertools as it
-from typing import Callable, Iterable, Sequence
-
-from colour import Color
+from collections.abc import Callable, Iterable, Sequence
 
 from manim.mobject.geometry.line import Line
 from manim.mobject.geometry.polygram import Polygon
@@ -76,18 +74,54 @@ from manim.mobject.text.numbers import DecimalNumber, Integer
 from manim.mobject.text.tex_mobject import MathTex
 from manim.mobject.text.text_mobject import Paragraph
 
-from .. import config
 from ..animation.animation import Animation
 from ..animation.composition import AnimationGroup
 from ..animation.creation import Create, Write
 from ..animation.fading import FadeIn
 from ..mobject.types.vectorized_mobject import VGroup, VMobject
-from ..utils.color import BLACK, YELLOW
+from ..utils.color import BLACK, YELLOW, ManimColor, ParsableManimColor
 from .utils import get_vectorized_mobject_class
 
 
 class Table(VGroup):
-    """A mobject that displays a table on the screen.
+    r"""A mobject that displays a table on the screen.
+
+    Parameters
+    ----------
+    table
+        A 2D array or list of lists. Content of the table has to be a valid input
+        for the callable set in ``element_to_mobject``.
+    row_labels
+        List of :class:`~.VMobject` representing the labels of each row.
+    col_labels
+        List of :class:`~.VMobject` representing the labels of each column.
+    top_left_entry
+        The top-left entry of the table, can only be specified if row and
+        column labels are given.
+    v_buff
+        Vertical buffer passed to :meth:`~.Mobject.arrange_in_grid`, by default 0.8.
+    h_buff
+        Horizontal buffer passed to :meth:`~.Mobject.arrange_in_grid`, by default 1.3.
+    include_outer_lines
+        ``True`` if the table should include outer lines, by default False.
+    add_background_rectangles_to_entries
+        ``True`` if background rectangles should be added to entries, by default ``False``.
+    entries_background_color
+        Background color of entries if ``add_background_rectangles_to_entries`` is ``True``.
+    include_background_rectangle
+        ``True`` if the table should have a background rectangle, by default ``False``.
+    background_rectangle_color
+        Background color of table if ``include_background_rectangle`` is ``True``.
+    element_to_mobject
+        The :class:`~.Mobject` class applied to the table entries. by default :class:`~.Paragraph`. For common choices, see :mod:`~.text_mobject`/:mod:`~.tex_mobject`.
+    element_to_mobject_config
+        Custom configuration passed to :attr:`element_to_mobject`, by default {}.
+    arrange_in_grid_config
+        Dict passed to :meth:`~.Mobject.arrange_in_grid`, customizes the arrangement of the table.
+    line_config
+        Dict passed to :class:`~.Line`, customizes the lines of the table.
+    kwargs
+        Additional arguments to be passed to :class:`~.VGroup`.
 
     Examples
     --------
@@ -160,9 +194,9 @@ class Table(VGroup):
         h_buff: float = 1.3,
         include_outer_lines: bool = False,
         add_background_rectangles_to_entries: bool = False,
-        entries_background_color: Color = BLACK,
+        entries_background_color: ParsableManimColor = BLACK,
         include_background_rectangle: bool = False,
-        background_rectangle_color: Color = BLACK,
+        background_rectangle_color: ParsableManimColor = BLACK,
         element_to_mobject: Callable[
             [float | str | VMobject],
             VMobject,
@@ -172,45 +206,6 @@ class Table(VGroup):
         line_config: dict = {},
         **kwargs,
     ):
-        """
-        Parameters
-        ----------
-        table
-            A 2D array or list of lists. Content of the table has to be a valid input
-            for the callable set in ``element_to_mobject``.
-        row_labels
-            List of :class:`~.VMobject` representing the labels of each row.
-        col_labels
-            List of :class:`~.VMobject` representing the labels of each column.
-        top_left_entry
-            The top-left entry of the table, can only be specified if row and
-            column labels are given.
-        v_buff
-            Vertical buffer passed to :meth:`~.Mobject.arrange_in_grid`, by default 0.8.
-        h_buff
-            Horizontal buffer passed to :meth:`~.Mobject.arrange_in_grid`, by default 1.3.
-        include_outer_lines
-            ``True`` if the table should include outer lines, by default False.
-        add_background_rectangles_to_entries
-            ``True`` if background rectangles should be added to entries, by default ``False``.
-        entries_background_color
-            Background color of entries if ``add_background_rectangles_to_entries`` is ``True``.
-        include_background_rectangle
-            ``True`` if the table should have a background rectangle, by default ``False``.
-        background_rectangle_color
-            Background color of table if ``include_background_rectangle`` is ``True``.
-        element_to_mobject
-            The :class:`~.Mobject` class applied to the table entries. by default :class:`~.Paragraph`. For common choices, see :mod:`~.text_mobject`/:mod:`~.tex_mobject`.
-        element_to_mobject_config
-            Custom configuration passed to :attr:`element_to_mobject`, by default {}.
-        arrange_in_grid_config
-            Dict passed to :meth:`~.Mobject.arrange_in_grid`, customizes the arrangement of the table.
-        line_config
-            Dict passed to :class:`~.Line`, customizes the lines of the table.
-        kwargs
-            Additional arguments to be passed to :class:`~.VGroup`.
-        """
-
         self.row_labels = row_labels
         self.col_labels = col_labels
         self.top_left_entry = top_left_entry
@@ -220,9 +215,9 @@ class Table(VGroup):
         self.h_buff = h_buff
         self.include_outer_lines = include_outer_lines
         self.add_background_rectangles_to_entries = add_background_rectangles_to_entries
-        self.entries_background_color = entries_background_color
+        self.entries_background_color = ManimColor(entries_background_color)
         self.include_background_rectangle = include_background_rectangle
-        self.background_rectangle_color = background_rectangle_color
+        self.background_rectangle_color = ManimColor(background_rectangle_color)
         self.element_to_mobject = element_to_mobject
         self.element_to_mobject_config = element_to_mobject_config
         self.arrange_in_grid_config = arrange_in_grid_config
@@ -258,7 +253,7 @@ class Table(VGroup):
         self,
         table: Iterable[Iterable[float | str | VMobject]],
     ) -> list:
-        """Initilaizes the entries of ``table`` as :class:`~.VMobject`.
+        """Initializes the entries of ``table`` as :class:`~.VMobject`.
 
         Parameters
         ----------
@@ -506,7 +501,7 @@ class Table(VGroup):
         """
         return VGroup(*(VGroup(*row) for row in self.mob_table))
 
-    def set_column_colors(self, *colors: Iterable[Color]) -> Table:
+    def set_column_colors(self, *colors: Iterable[ParsableManimColor]) -> Table:
         """Set individual colors for each column of the table.
 
         Parameters
@@ -535,7 +530,7 @@ class Table(VGroup):
             column.set_color(color)
         return self
 
-    def set_row_colors(self, *colors: Iterable[Color]) -> Table:
+    def set_row_colors(self, *colors: Iterable[ParsableManimColor]) -> Table:
         """Set individual colors for each row of the table.
 
         Parameters
@@ -687,7 +682,6 @@ class Table(VGroup):
                         item.set_color(random_bright_color())
                     self.add(table)
         """
-
         return VGroup(*self.row_labels)
 
     def get_col_labels(self) -> VGroup:
@@ -716,7 +710,6 @@ class Table(VGroup):
                         item.set_color(random_bright_color())
                     self.add(table)
         """
-
         return VGroup(*self.col_labels)
 
     def get_labels(self) -> VGroup:
@@ -754,10 +747,10 @@ class Table(VGroup):
                 label_group.add(*label)
         return label_group
 
-    def add_background_to_entries(self, color: Color = BLACK) -> Table:
+    def add_background_to_entries(self, color: ParsableManimColor = BLACK) -> Table:
         """Adds a black :class:`~.BackgroundRectangle` to each entry of the table."""
         for mob in self.get_entries():
-            mob.add_background_rectangle(color=color)
+            mob.add_background_rectangle(color=ManimColor(color))
         return self
 
     def get_cell(self, pos: Sequence[int] = (1, 1), **kwargs) -> Polygon:
@@ -818,7 +811,7 @@ class Table(VGroup):
         return rec
 
     def get_highlighted_cell(
-        self, pos: Sequence[int] = (1, 1), color: Color = YELLOW, **kwargs
+        self, pos: Sequence[int] = (1, 1), color: ParsableManimColor = YELLOW, **kwargs
     ) -> BackgroundRectangle:
         """Returns a :class:`~.BackgroundRectangle` of the cell at the given position.
 
@@ -850,11 +843,11 @@ class Table(VGroup):
                     self.add(table)
         """
         cell = self.get_cell(pos)
-        bg_cell = BackgroundRectangle(cell, color=color, **kwargs)
+        bg_cell = BackgroundRectangle(cell, color=ManimColor(color), **kwargs)
         return bg_cell
 
     def add_highlighted_cell(
-        self, pos: Sequence[int] = (1, 1), color: Color = YELLOW, **kwargs
+        self, pos: Sequence[int] = (1, 1), color: ParsableManimColor = YELLOW, **kwargs
     ) -> Table:
         """Highlights one cell at a specific position on the table by adding a :class:`~.BackgroundRectangle`.
 
@@ -884,7 +877,7 @@ class Table(VGroup):
                     table.add_highlighted_cell((2,2), color=GREEN)
                     self.add(table)
         """
-        bg_cell = self.get_highlighted_cell(pos, color=color, **kwargs)
+        bg_cell = self.get_highlighted_cell(pos, color=ManimColor(color), **kwargs)
         self.add_to_back(bg_cell)
         entry = self.get_entries(pos)
         entry.background_rectangle = bg_cell
@@ -1071,7 +1064,7 @@ class MobjectTable(Table):
 
 
 class IntegerTable(Table):
-    """A specialized :class:`~.Table` mobject for use with :class:`~.Integer`.
+    r"""A specialized :class:`~.Table` mobject for use with :class:`~.Integer`.
 
     Examples
     --------
@@ -1085,14 +1078,14 @@ class IntegerTable(Table):
                     [[0,30,45,60,90],
                     [90,60,45,30,0]],
                     col_labels=[
-                        MathTex("\\\\frac{\\sqrt{0}}{2}"),
-                        MathTex("\\\\frac{\\sqrt{1}}{2}"),
-                        MathTex("\\\\frac{\\sqrt{2}}{2}"),
-                        MathTex("\\\\frac{\\sqrt{3}}{2}"),
-                        MathTex("\\\\frac{\\sqrt{4}}{2}")],
-                    row_labels=[MathTex("\\sin"), MathTex("\\cos")],
+                        MathTex(r"\frac{\sqrt{0}}{2}"),
+                        MathTex(r"\frac{\sqrt{1}}{2}"),
+                        MathTex(r"\frac{\sqrt{2}}{2}"),
+                        MathTex(r"\frac{\sqrt{3}}{2}"),
+                        MathTex(r"\frac{\sqrt{4}}{2}")],
+                    row_labels=[MathTex(r"\sin"), MathTex(r"\cos")],
                     h_buff=1,
-                    element_to_mobject_config={"unit": "^{\\circ}"})
+                    element_to_mobject_config={"unit": r"^{\circ}"})
                 self.add(t0)
     """