wehub-resource-sync
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
"""
|
|
Postprocessing functions for cleaning up latex equations in linear format which don't give valid LaTeX.
|
|
"""
|
|
|
|
import re
|
|
|
|
clean_exps = {
|
|
r"\\degf": "°F",
|
|
r"\\degc": "°C",
|
|
r"(\\cbrt)(\w+)": r"\\sqrt[3]{\2}",
|
|
r"(\\qdrt)(\w+)": r"\\sqrt[4]{\2}",
|
|
r"\\sfrac": r"\\frac",
|
|
r"(\\o[i]+nt)(\w+)": r"\1{\2}",
|
|
r"\\bullet(\w+)": r"\\bullet \1",
|
|
r"\\sum([a-zA-Z0-9]+)": r"\\sum{\1}",
|
|
r"\\prod([a-zA-Z0-9]+)": r"\\prod{\1}",
|
|
r"\\amalg([a-zA-Z0-9]+)": r"\\amalg{\1}",
|
|
r"\\bigcup([a-zA-Z0-9]+)": r"\\bigcup{\1}",
|
|
r"\\bigcap([a-zA-Z0-9]+)": r"\\bigcap{\1}",
|
|
r"\\bigvee([a-zA-Z0-9]+)": r"\\bigvee{\1}",
|
|
r"\\bigwedge([a-zA-Z0-9]+)": r"\\bigwedge{\1}",
|
|
r"\\lfloor([a-zA-Z0-9]+)": r"\\lfloor{\1}",
|
|
r"\\lceil([a-zA-Z0-9]+)": r"\\lceil{\1}",
|
|
r"\\lim\\below\{(.+)\}\{(.+)\}": r"\\lim_{\1}{\2}",
|
|
r"\\min\\below\{(.+)\}\{(.+)\}": r"\\min_{\1}{\2}",
|
|
r"\\max\\below\{(.+)\}\{(.+)\}": r"\\max_{\1}{\2}",
|
|
}
|
|
|
|
|
|
def clean_exp(exp):
|
|
"""
|
|
Takes in a linear expression and converts known invalid LaTeX equations to valid LaTeX
|
|
:param exp:str - An equation in invalid syntax
|
|
:return :str - A valid equation
|
|
"""
|
|
for e in clean_exps:
|
|
exp = re.sub(e, clean_exps[e], exp)
|
|
return exp
|