public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-09-01 00:48:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixed braces on Matrix

Browse source
This commit is contained in:
Grant Sanderson 2020-08-30 16:00:59 -07:00
parent 35a2ade954
commit 821857afa8
1 changed files with 13 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
manimlib/mobject/matrix.py
View file

@ -71,7 +71,7 @@ class Matrix(VMobject):
or mobjects or mobjects
""" """
VMobject.__init__(self, **kwargs) VMobject.__init__(self, **kwargs)
matrix = np.array(matrix, ndmin=1) matrix = self.matrix = np.array(matrix, ndmin=2)
mob_matrix = self.matrix_to_mob_matrix(matrix) mob_matrix = self.matrix_to_mob_matrix(matrix)
self.organize_mob_matrix(mob_matrix) self.organize_mob_matrix(mob_matrix)
self.elements = VGroup(*mob_matrix.flatten()) self.elements = VGroup(*mob_matrix.flatten())
@ -101,12 +101,19 @@ class Matrix(VMobject):
return self return self
def add_brackets(self): def add_brackets(self):
bracket_pair = TexMobject("\\big[", "\\big]") height = self.matrix.shape[0]
bracket_pair.scale(2) bracket_pair = TexMobject("".join([
bracket_pair.stretch_to_fit_height( "\\left[",
self.get_height() + 2 * self.bracket_v_buff "\\begin{array}{c}",
*height * ["\\quad \\\\"],
"\\end{array}"
"\\right]",
]))[0]
bracket_pair.set_height(
self.get_height() + 1 * self.bracket_v_buff
) )
l_bracket, r_bracket = bracket_pair.split() l_bracket = bracket_pair[:len(bracket_pair) // 2]
r_bracket = bracket_pair[len(bracket_pair) // 2:]
l_bracket.next_to(self, LEFT, self.bracket_h_buff) l_bracket.next_to(self, LEFT, self.bracket_h_buff)
r_bracket.next_to(self, RIGHT, self.bracket_h_buff) r_bracket.next_to(self, RIGHT, self.bracket_h_buff)
self.add(l_bracket, r_bracket) self.add(l_bracket, r_bracket)