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

Prevent infinite loops from unexpected inputs

Browse source
This commit is contained in:
Michael W 2021-11-28 13:26:54 +08:00 • committed by GitHub
parent dae24891fa
commit e8ebfa312b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
manimlib/mobject/svg/mtex_mobject.py
View file

@ -167,6 +167,8 @@ class MTex(VMobject):
additional_strings_to_break_up = remove_list_redundancies([ additional_strings_to_break_up = remove_list_redundancies([
*self.isolate, *self.tex_to_color_map.keys() *self.isolate, *self.tex_to_color_map.keys()
]) ])
if "" in additional_strings_to_break_up:
additional_strings_to_break_up.remove("")
if not additional_strings_to_break_up: if not additional_strings_to_break_up:
return return
@ -291,7 +293,7 @@ class MTex(VMobject):
def get_all_isolated_substrings(self): def get_all_isolated_substrings(self):
tex_string = self.tex_string tex_string = self.tex_string
return remove_list_redundancies([ return remove_list_redundancies([
tex_string[span_tuple[0] : span_tuple[1]] tex_string[slice(*span_tuple)]
for span_tuple in self.tex_spans_dict.keys() for span_tuple in self.tex_spans_dict.keys()
]) ])
@ -348,14 +350,17 @@ class MTex(VMobject):
def slice_of_part(self, part): def slice_of_part(self, part):
# Only finds where the head and the tail of `part` is in. # Only finds where the head and the tail of `part` is in.
submobs = self.submobjects submobs = self.submobjects
submobs_len = len(submobs)
begin_mob = part[0] begin_mob = part[0]
begin_index = 0 begin_index = 0
while begin_mob not in submobs[begin_index]: while begin_mob < submobs_len and begin_mob not in submobs[begin_index]:
begin_index += 1 begin_index += 1
end_mob = part[-1] end_mob = part[-1]
end_index = len(submobs) - 1 end_index = submobs_len - 1
while end_mob not in submobs[end_index]: while end_index >= 0 and end_mob not in submobs[end_index]:
end_index -= 1 end_index -= 1
if begin_index > end_index:
raise ValueError("Unable to find part")
return slice(begin_index, end_index + 1) return slice(begin_index, end_index + 1)
def slice_of_part_by_tex(self, tex, index=0): def slice_of_part_by_tex(self, tex, index=0):