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

Directly cache svg file instead of tex, and don't store superfluous files associated with tex rendering

Browse source
This commit is contained in:
Grant Sanderson 2021-01-03 11:29:09 -08:00
parent 16ea2fd9c1
commit e5c79cbb90
1 changed files with 21 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
manimlib/utils/tex_file_writing.py
View file

@ -44,19 +44,29 @@ def tex_hash(tex_file_content):
def tex_to_svg_file(tex_file_content):
directory = get_tex_dir()
tex_file = generate_tex_file(tex_file_content, directory)
dvi_file = tex_to_dvi(tex_file)
return dvi_to_svg(dvi_file)
svg_file = os.path.join(
get_tex_dir(), tex_hash(tex_file_content) + ".svg"
)
if not os.path.exists(svg_file):
# If svg doesn't exist, create it
tex_to_svg(tex_file_content, svg_file)
return svg_file
def generate_tex_file(tex_file_content, directory):
file_name = f"{tex_hash(tex_file_content)}.tex"
path = os.path.join(directory, file_name)
if not os.path.exists(path):
with open(path, "w", encoding="utf-8") as outfile:
outfile.write(tex_file_content)
return path
def tex_to_svg(tex_file_content, svg_file):
tex_file = svg_file.replace(".svg", ".tex")
with open(tex_file, "w", encoding="utf-8") as outfile:
outfile.write(tex_file_content)
svg_file = dvi_to_svg(tex_to_dvi(tex_file))
# Cleanup superfluous documents
tex_dir, name = os.path.split(svg_file)
stem, end = name.split(".")
for file in filter(lambda s: s.startswith(stem), os.listdir(tex_dir)):
if not file.endswith(end):
os.remove(os.path.join(tex_dir, file))
return svg_file
def tex_to_dvi(tex_file):