Added tex_to_color_map abilty to TexMobject config

This commit is contained in:
Grant Sanderson
2018-05-05 20:19:33 -07:00
parent fb5f1517a9
commit c2a67346f9

View File

@ -131,6 +131,7 @@ class TexMobject(SingleStringTexMobject):
CONFIG = { CONFIG = {
"arg_separator": " ", "arg_separator": " ",
"substrings_to_isolate": [], "substrings_to_isolate": [],
"tex_to_color_map": {},
} }
def __init__(self, *tex_strings, **kwargs): def __init__(self, *tex_strings, **kwargs):
@ -141,13 +142,18 @@ class TexMobject(SingleStringTexMobject):
self, self.arg_separator.join(tex_strings), **kwargs self, self.arg_separator.join(tex_strings), **kwargs
) )
self.break_up_by_substrings() self.break_up_by_substrings()
self.set_color_by_tex_to_color_map(self.tex_to_color_map)
if self.organize_left_to_right: if self.organize_left_to_right:
self.organize_submobjects_left_to_right() self.organize_submobjects_left_to_right()
def break_up_tex_strings(self, tex_strings): def break_up_tex_strings(self, tex_strings):
substrings_to_isolate = op.add(
self.substrings_to_isolate,
self.tex_to_color_map.keys()
)
split_list = split_string_list_to_isolate_substring( split_list = split_string_list_to_isolate_substring(
tex_strings, *self.substrings_to_isolate tex_strings, *substrings_to_isolate
) )
split_list = map(str.strip, split_list) split_list = map(str.strip, split_list)
split_list = filter(lambda s: s != '', split_list) split_list = filter(lambda s: s != '', split_list)