Merge branch 'pr/49'

Author: PgBiel

manim/mobject/matrix.py

@@ -63,6 +63,8 @@ class Matrix(VMobject):
         "element_to_mobject": TexMobject,
         "element_to_mobject_config": {},
         "element_alignment_corner": DR,
+        "left_bracket": "\\big[",
+        "right_bracket": "\\big]",
     }
 
     def __init__(self, matrix, **kwargs):
@@ -76,7 +78,7 @@ def __init__(self, matrix, **kwargs):
         self.organize_mob_matrix(mob_matrix)
         self.elements = VGroup(*mob_matrix.flatten())
         self.add(self.elements)
-        self.add_brackets()
+        self.add_brackets(self.left_bracket,self.right_bracket)
         self.center()
         self.mob_matrix = mob_matrix
         if self.add_background_rectangles_to_entries:
@@ -100,8 +102,8 @@ def organize_mob_matrix(self, matrix):
                 )
         return self
 
-    def add_brackets(self):
-        bracket_pair = TexMobject("\\big[", "\\big]")
+    def add_brackets(self, left = "\\big[", right = "\\big]"):
+        bracket_pair = TexMobject(left, right)
         bracket_pair.scale(2)
         bracket_pair.stretch_to_fit_height(
             self.get_height() + 2 * self.bracket_v_buff
@@ -125,6 +127,37 @@ def set_column_colors(self, *colors):
             column.set_color(color)
         return self
 
+    def get_rows(self):
+        """Return rows of the matrix as VGroups
+        
+        Returns
+        --------
+        List[:class:`~.VGroup`]
+            Each VGroup contains a row of the matrix.
+        """
+        return VGroup(*[
+            VGroup(*self.mob_matrix[i, :])
+            for i in range(self.mob_matrix.shape[1])
+        ])
+
+    def set_row_colors(self, *colors):
+        """Set individual colors for each row of the matrix
+        
+        Parameters
+        ----------
+        colors : :class:`str`
+            The list of colors; each color specified corresponds to a row.
+        
+        Returns
+        -------
+        :class:`Matrix`
+            The current matrix object (self).
+        """
+        rows = self.get_rows()
+        for color, row in zip(colors, rows):
+            row.set_color(color)
+        return self
+
     def add_background_to_entries(self):
         for mob in self.get_entries():
             mob.add_background_rectangle()