manim 0.18.0.post0__py3-none-any.whl → 0.19.0__py3-none-any.whl

manim/mobject/graphing/number_line.py

@@ -2,15 +2,18 @@
 
 from __future__ import annotations
 
+from manim.mobject.mobject import Mobject
 from manim.mobject.opengl.opengl_vectorized_mobject import OpenGLVMobject
 
 __all__ = ["NumberLine", "UnitInterval"]
 
 
-from typing import TYPE_CHECKING, Callable, Iterable, Sequence
+from collections.abc import Iterable, Sequence
+from typing import TYPE_CHECKING, Callable
 
 if TYPE_CHECKING:
     from manim.mobject.geometry.tips import ArrowTip
+    from manim.typing import Point3DLike
 
 import numpy as np
 
@@ -274,7 +277,8 @@ class NumberLine(Line):
 
     def add_ticks(self):
         """Adds ticks to the number line. Ticks can be accessed after creation
-        via ``self.ticks``."""
+        via ``self.ticks``.
+        """
         ticks = VGroup()
         elongated_tick_size = self.tick_size * self.longer_tick_multiple
         elongated_tick_offsets = self.numbers_with_elongated_ticks - self.x_min
@@ -343,6 +347,7 @@ class NumberLine(Line):
     def number_to_point(self, number: float | np.ndarray) -> np.ndarray:
         """Accepts a value along the number line and returns a point with
         respect to the scene.
+        Equivalent to `NumberLine @ number`
 
         Parameters
         ----------
@@ -363,7 +368,9 @@ class NumberLine(Line):
             array([0., 0., 0.])
             >>> number_line.number_to_point(1)
             array([1., 0., 0.])
-            >>> number_line.number_to_point([1,2,3])
+            >>> number_line @ 1
+            array([1., 0., 0.])
+            >>> number_line.number_to_point([1, 2, 3])
             array([[1., 0., 0.],
                    [2., 0., 0.],
                    [3., 0., 0.]])
@@ -395,11 +402,11 @@ class NumberLine(Line):
 
             >>> from manim import NumberLine
             >>> number_line = NumberLine()
-            >>> number_line.point_to_number((0,0,0))
-            0.0
-            >>> number_line.point_to_number((1,0,0))
-            1.0
-            >>> number_line.point_to_number([[0.5,0,0],[1,0,0],[1.5,0,0]])
+            >>> number_line.point_to_number((0, 0, 0))
+            np.float64(0.0)
+            >>> number_line.point_to_number((1, 0, 0))
+            np.float64(1.0)
+            >>> number_line.point_to_number([[0.5, 0, 0], [1, 0, 0], [1.5, 0, 0]])
             array([0.5, 1. , 1.5])
 
         """
@@ -475,7 +482,7 @@ class NumberLine(Line):
         num_mob.next_to(self.number_to_point(x), direction=direction, buff=buff)
         if x < 0 and self.label_direction[0] == 0:
             # Align without the minus sign
-            num_mob.shift(num_mob[0].get_width() * LEFT / 2)
+            num_mob.shift(num_mob[0].width * LEFT / 2)
         return num_mob
 
     def get_number_mobjects(self, *numbers, **kwargs) -> VGroup:
@@ -574,7 +581,6 @@ class NumberLine(Line):
         AttributeError
             If the label does not have a ``font_size`` attribute, an ``AttributeError`` is raised.
         """
-
         direction = self.label_direction if direction is None else direction
         buff = self.line_to_number_buff if buff is None else buff
         font_size = self.font_size if font_size is None else font_size
@@ -641,6 +647,14 @@ class NumberLine(Line):
             return 0
         return len(step.split(".")[-1])
 
+    def __matmul__(self, other: float):
+        return self.n2p(other)
+
+    def __rmatmul__(self, other: Point3DLike | Mobject):
+        if isinstance(other, Mobject):
+            other = other.get_center()
+        return self.p2n(other)
+
 
 class UnitInterval(NumberLine):
     def __init__(

manim/mobject/graphing/probability.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 __all__ = ["SampleSpace", "BarChart"]
 
 
-from typing import Iterable, MutableSequence, Sequence
+from collections.abc import Iterable, MutableSequence, Sequence
 
 import numpy as np
 
@@ -319,7 +319,6 @@ class BarChart(Axes):
         Primarily used when the bars are initialized with ``self._add_bars``
         or updated via ``self.change_bar_values``.
         """
-
         self.bars.set_color_by_gradient(*self.bar_colors)
 
     def _add_x_axis_labels(self):
@@ -329,7 +328,6 @@ class BarChart(Axes):
 
         UP for negative values and DOWN for positive values.
         """
-
         val_range = np.arange(
             0.5, len(self.bar_names), 1
         )  # 0.5 shifted so that labels are centered, not on ticks
@@ -339,10 +337,7 @@ class BarChart(Axes):
         for i, (value, bar_name) in enumerate(zip(val_range, self.bar_names)):
             # to accommodate negative bars, the label may need to be
             # below or above the x_axis depending on the value of the bar
-            if self.values[i] < 0:
-                direction = UP
-            else:
-                direction = DOWN
+            direction = UP if self.values[i] < 0 else DOWN
             bar_name_label = self.x_axis.label_constructor(bar_name)
 
             bar_name_label.font_size = self.x_axis.font_size
@@ -372,7 +367,6 @@ class BarChart(Axes):
         Rectangle
             A positioned rectangle representing a bar on the chart.
         """
-
         # bar measurements relative to the axis
 
         # distance from between the y-axis and the top of the bar
@@ -435,7 +429,6 @@ class BarChart(Axes):
 
                     self.add(chart, c_bar_lbls)
         """
-
         bar_labels = VGroup()
         for bar, value in zip(self.bars, self.values):
             bar_lbl = label_constructor(str(value))
@@ -483,7 +476,6 @@ class BarChart(Axes):
                     chart.change_bar_values(list(reversed(values)))
                     self.add(chart.get_bar_labels(font_size=24))
         """
-
         for i, (bar, value) in enumerate(zip(self.bars, values)):
             chart_val = self.values[i]
 

manim/mobject/graphing/scale.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
 
 import math
-from typing import TYPE_CHECKING, Any, Iterable
+from collections.abc import Iterable
+from typing import TYPE_CHECKING, Any
 
 import numpy as np
 
@@ -89,7 +90,6 @@ class LinearBase(_ScaleBase):
         scale_factor
             The slope of the linear function, by default 1.0
         """
-
         super().__init__()
         self.scale_factor = scale_factor
 
@@ -145,7 +145,9 @@ class LogBase(_ScaleBase):
         """Inverse of ``function``. The value must be greater than 0"""
         if isinstance(value, np.ndarray):
             condition = value.any() <= 0
-            func = lambda value, base: np.log(value) / np.log(base)
+
+            def func(value, base):
+                return np.log(value) / np.log(base)
         else:
             condition = value <= 0
             func = math.log
@@ -174,12 +176,11 @@ class LogBase(_ScaleBase):
         base_config
             Additional arguments to be passed to :class:`~.Integer`.
         """
-
         # uses `format` syntax to control the number of decimal places.
         tex_labels = [
             Integer(
                 self.base,
-                unit="^{%s}" % (f"{self.inverse_function(i):.{unit_decimal_places}f}"),
+                unit="^{%s}" % (f"{self.inverse_function(i):.{unit_decimal_places}f}"),  # noqa: UP031
                 **base_config,
             )
             for i in val_range

manim/mobject/matrix.py

@@ -15,8 +15,8 @@ Examples
             m2 = DecimalMatrix(
                 [[3.456, 2.122], [33.2244, 12.33]],
                 element_to_mobject_config={"num_decimal_places": 2},
-                left_bracket="\\{",
-                right_bracket="\\}")
+                left_bracket=r"\{",
+                right_bracket=r"\}")
             m3 = MobjectMatrix(
                 [[Circle().scale(0.3), Square().scale(0.3)],
                 [MathTex("\\pi").scale(2), Star().scale(0.3)]],
@@ -40,7 +40,7 @@ __all__ = [
 
 
 import itertools as it
-from typing import Iterable, Sequence
+from collections.abc import Iterable, Sequence
 
 import numpy as np
 
@@ -72,7 +72,7 @@ def matrix_to_mobject(matrix):
 
 
 class Matrix(VMobject, metaclass=ConvertToOpenGL):
-    """A mobject that displays a matrix on the screen.
+    r"""A mobject that displays a matrix on the screen.
 
     Parameters
     ----------
@@ -118,22 +118,22 @@ class Matrix(VMobject, metaclass=ConvertToOpenGL):
 
         class MatrixExamples(Scene):
             def construct(self):
-                m0 = Matrix([[2, "\\pi"], [-1, 1]])
+                m0 = Matrix([[2, r"\pi"], [-1, 1]])
                 m1 = Matrix([[2, 0, 4], [-1, 1, 5]],
                     v_buff=1.3,
                     h_buff=0.8,
                     bracket_h_buff=SMALL_BUFF,
                     bracket_v_buff=SMALL_BUFF,
-                    left_bracket="\\{",
-                    right_bracket="\\}")
+                    left_bracket=r"\{",
+                    right_bracket=r"\}")
                 m1.add(SurroundingRectangle(m1.get_columns()[1]))
                 m2 = Matrix([[2, 1], [-1, 3]],
                     element_alignment_corner=UL,
                     left_bracket="(",
                     right_bracket=")")
                 m3 = Matrix([[2, 1], [-1, 3]],
-                    left_bracket="\\\\langle",
-                    right_bracket="\\\\rangle")
+                    left_bracket=r"\langle",
+                    right_bracket=r"\rangle")
                 m4 = Matrix([[2, 1], [-1, 3]],
                 ).set_column_colors(RED, GREEN)
                 m5 = Matrix([[2, 1], [-1, 3]],
@@ -241,7 +241,6 @@ class Matrix(VMobject, metaclass=ConvertToOpenGL):
         :class:`Matrix`
             The current matrix object (self).
         """
-
         # Height per row of LaTeX array with default settings
         BRACKET_HEIGHT = 0.5977
 
@@ -280,7 +279,7 @@ class Matrix(VMobject, metaclass=ConvertToOpenGL):
         return self
 
     def get_columns(self):
-        """Return columns of the matrix as VGroups.
+        r"""Return columns of the matrix as VGroups.
 
         Returns
         --------
@@ -295,11 +294,10 @@ class Matrix(VMobject, metaclass=ConvertToOpenGL):
 
             class GetColumnsExample(Scene):
                 def construct(self):
-                    m0 = Matrix([["\\pi", 3], [1, 5]])
+                    m0 = Matrix([[r"\pi", 3], [1, 5]])
                     m0.add(SurroundingRectangle(m0.get_columns()[1]))
                     self.add(m0)
         """
-
         return VGroup(
             *(
                 VGroup(*(row[i] for row in self.mob_matrix))
@@ -308,7 +306,7 @@ class Matrix(VMobject, metaclass=ConvertToOpenGL):
         )
 
     def set_column_colors(self, *colors: str):
-        """Set individual colors for each columns of the matrix.
+        r"""Set individual colors for each columns of the matrix.
 
         Parameters
         ----------
@@ -338,7 +336,7 @@ class Matrix(VMobject, metaclass=ConvertToOpenGL):
         return self
 
     def get_rows(self):
-        """Return rows of the matrix as VGroups.
+        r"""Return rows of the matrix as VGroups.
 
         Returns
         --------
@@ -360,7 +358,7 @@ class Matrix(VMobject, metaclass=ConvertToOpenGL):
         return VGroup(*(VGroup(*row) for row in self.mob_matrix))
 
     def set_row_colors(self, *colors: str):
-        """Set individual colors for each row of the matrix.
+        r"""Set individual colors for each row of the matrix.
 
         Parameters
         ----------
@@ -438,7 +436,7 @@ class Matrix(VMobject, metaclass=ConvertToOpenGL):
         return self.elements
 
     def get_brackets(self):
-        """Return the bracket mobjects.
+        r"""Return the bracket mobjects.
 
         Returns
         --------
@@ -464,7 +462,7 @@ class Matrix(VMobject, metaclass=ConvertToOpenGL):
 
 
 class DecimalMatrix(Matrix):
-    """A mobject that displays a matrix with decimal entries on the screen.
+    r"""A mobject that displays a matrix with decimal entries on the screen.
 
     Examples
     --------
@@ -544,7 +542,7 @@ class IntegerMatrix(Matrix):
 
 
 class MobjectMatrix(Matrix):
-    """A mobject that displays a matrix of mobject entries on the screen.
+    r"""A mobject that displays a matrix of mobject entries on the screen.
 
     Examples
     --------