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

Some refactors on LabelledString

Browse source
This commit is contained in:
YishiMichael 2022-04-06 23:16:59 +08:00
parent 93f8d3f1ca
commit 18963fb9fe
No known key found for this signature in database
GPG key ID: EC615C0C5A86BC80
1 changed files with 6 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -113,24 +113,14 @@ class LabelledString(_StringSVG):
def get_substr(self, span: Span) -> str:
return self.string[slice(*span)]
def handle_regex_method(func):
def wrapper(self, pattern, pos=0, endpos=9223372036854775807):
return func()(
re.compile(pattern), self.string, pos=pos, endpos=endpos
)
return wrapper
def finditer(self, pattern, **kwargs):
return re.compile(pattern).finditer(self.string, **kwargs)
@handle_regex_method
def finditer():
return re.Pattern.finditer
def search(self, pattern, **kwargs):
return re.compile(pattern).search(self.string, **kwargs)
@handle_regex_method
def search():
return re.Pattern.search
@handle_regex_method
def match():
return re.Pattern.match
def match(self, pattern, **kwargs):
return re.compile(pattern).match(self.string, **kwargs)
def find_spans(self, pattern: str) -> list[Span]:
return [match_obj.span() for match_obj in self.finditer(pattern)]