Have MTex.__getitem__ call MTex.select_parts

This commit is contained in:
Grant Sanderson
2022-12-20 22:37:23 -08:00
parent 4b73140435
commit 6277e28373

View File

@ -242,6 +242,12 @@ class MTex(StringMobject):
result.append(self[start:end])
return VGroup(*result)
def __getitem__(self, value: int | slice | str | re.Pattern) -> VMobject:
# TODO, maybe move this functionality up to StringMobject
if isinstance(value, (str, re.Pattern)):
return self.select_parts(value)
return super().__getitem__(value)
def get_tex(self) -> str:
return self.get_string()