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

Add regex parameter

Browse source
This commit is contained in:
YishiMichael 2022-04-10 09:23:53 +08:00
parent e23f667c3d
commit 36d62ae1a3
No known key found for this signature in database
GPG key ID: EC615C0C5A86BC80
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
manimlib/mobject/svg/labelled_string.py
View file

@ -520,14 +520,17 @@ class LabelledString(_StringSVG):
]) ])
def get_parts_by_string( def get_parts_by_string(
self, substr: str, case_sensitive: bool = True, **kwargs self, substr: str,
case_sensitive: bool = True, regex: bool = False, **kwargs
) -> VGroup: ) -> VGroup:
flags = 0 flags = 0
if not case_sensitive: if not case_sensitive:
flags |= re.I flags |= re.I
pattern = substr if regex else re.escape(substr)
return VGroup(*[ return VGroup(*[
self.get_part_by_custom_span(span, **kwargs) self.get_part_by_custom_span(span, **kwargs)
for span in self.find_substr(substr, flags=flags) for span in self.find_spans(pattern, flags=flags)
if span[0] < span[1]
]) ])
def get_part_by_string( def get_part_by_string(