From 9dbf4e4b89ab60948476d28704a82d623c9ce29c Mon Sep 17 00:00:00 2001 From: Grant Sanderson Date: Fri, 27 Apr 2018 13:50:08 -0700 Subject: [PATCH] Added add_background_rectangle to Matrix --- mobject/matrix.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/mobject/matrix.py b/mobject/matrix.py index e65674ad..2c1fe1dd 100644 --- a/mobject/matrix.py +++ b/mobject/matrix.py @@ -86,8 +86,7 @@ class Matrix(VMobject): for mob in self.elements: mob.add_background_rectangle() if self.include_background_rectangle: - self.background_rectangle = BackgroundRectangle(self) - self.add_to_back(self.background_rectangle) + self.add_background_rectangle() def matrix_to_mob_matrix(self, matrix): return np.vectorize( @@ -117,6 +116,10 @@ class Matrix(VMobject): self.brackets = VGroup(l_bracket, r_bracket) return self + def add_background_rectangle(self): + self.background_rectangle = BackgroundRectangle(self) + self.add_to_back(self.background_rectangle) + def set_color_columns(self, *colors): for i, color in enumerate(colors): VGroup(*self.mob_matrix[:, i]).set_color(color)