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

Added get_part_by_tex method

Browse source
This commit is contained in:
Grant Sanderson 2017-03-10 16:55:23 -08:00
parent 862b6796b7
commit c2178fd45b
1 changed files with 26 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
mobject/tex_mobject.py
View file

@ -110,14 +110,33 @@ class TexMobject(SVGMobject):
self.submobjects = new_submobjects self.submobjects = new_submobjects
return self return self
def highlight_by_tex(self, tex, color, substring = True): def get_part_by_tex(self, tex, substring = True, return_all_parts = False):
def test(tex1, tex2):
return tex1 == tex2 or (substring and tex1 in tex2)
if not hasattr(self, "expression_parts"): if not hasattr(self, "expression_parts"):
if tex == self.get_tex_string(): if test(tex, self.get_tex_string()):
self.highlight(color)
return self return self
for submob, part_tex in zip(self.split(), self.expression_parts): return None
if (part_tex == tex) or (substring and tex in part_tex): all_parts = [
submob.highlight(color) submob
for submob, part_tex in zip(self.split(), self.expression_parts)
if test(tex, part_tex)
]
if return_all_parts:
return all_parts
else:
if len(all_parts) > 0:
return all_parts[0]
else:
return None
def get_parts_by_tex(self, tex, **kwargs):
return self.get_part_by_tex(tex, return_all_parts = True, **kwargs)
def highlight_by_tex(self, tex, color, **kwargs):
parts_to_color = self.get_parts_by_tex(tex, **kwargs)
for part in parts_to_color:
part.highlight(color)
return self return self
def organize_submobjects_left_to_right(self): def organize_submobjects_left_to_right(self):