Added ability to remove non-svg LaTeX files

manim/_config/utils.py

@@ -309,6 +309,7 @@ class MyScene(Scene):
         "write_to_movie",
         "zero_pad",
         "force_window",
+        "no_latex_cleanup",
     }
 
     def __init__(self) -> None:
@@ -580,6 +581,7 @@ def digest_parser(self, parser: configparser.ConfigParser) -> ManimConfig:
             "use_projection_stroke_shaders",
             "enable_wireframe",
             "force_window",
+            "no_latex_cleanup",
         ]:
             setattr(self, key, parser["CLI"].getboolean(key, fallback=False))
 
@@ -756,6 +758,7 @@ def digest_args(self, args: argparse.Namespace) -> ManimConfig:
             "enable_wireframe",
             "force_window",
             "dry_run",
+            "no_latex_cleanup",
         ]:
             if hasattr(args, key):
                 attr = getattr(args, key)
@@ -960,6 +963,12 @@ def digest_file(self, filename: str | os.PathLike) -> ManimConfig:
         doc="Set to force window when using the opengl renderer",
     )
 
+    no_latex_cleanup = property(
+        lambda self: self._d["no_latex_cleanup"],
+        lambda self, val: self._set_boolean("no_latex_cleanup", val),
+        doc="Deletes .aux, .dvi, and .log files produced by Tex and MathTex.",
+    )
+
     @property
     def verbosity(self):
         """Logger verbosity; "DEBUG", "INFO", "WARNING", "ERROR", or "CRITICAL" (-v)."""

manim/cli/render/global_options.py

@@ -102,4 +102,10 @@ def validate_gui_location(ctx, param, value):
         help="Renders animations without outputting image or video files and disables the window",
         default=False,
     ),
+    option(
+        "--no_latex_cleanup",
+        is_flag=True,
+        help="Deletes .aux, .dvi, and .log files produced by Tex and MathTex.",
+        default=False,
+    ),
 )

manim/scene/scene.py

@@ -209,6 +209,9 @@ def __deepcopy__(self, clone_from_id):
                 result.mobject_updater_lists.append((mobject_clone, cloned_updaters))
         return result
 
+    def delete_old_latex(self) -> int:
+        return self.renderer.file_writer.delete_tex_files(self.__class__.__name__)
+
     def render(self, preview: bool = False):
         """
         Renders this Scene.
@@ -232,12 +235,15 @@ def render(self, preview: bool = False):
         # We have to reset these settings in case of multiple renders.
         self.renderer.scene_finished(self)
 
+        deleted_latex = self.delete_old_latex() if not config["no_latex_cleanup"] else 0
         # Show info only if animations are rendered or to get image
         if (
             self.renderer.num_plays
             or config["format"] == "png"
             or config["save_last_frame"]
         ):
+            # Maybe add this logging output to tests?
+            # logger.info(f"{deleted_latex} non-essential LaTeX files deleted.{' To keep these LaTeX files, try passing in `--no_latex_cleanup`' if deleted_latex else ''}")
             logger.info(
                 f"Rendered {str(self)}\nPlayed {self.renderer.num_plays} animations",
             )

manim/scene/scene_file_writer.py

@@ -92,7 +92,7 @@ def __init__(self, renderer, scene_name, **kwargs):
                 "or set path to the ffmpeg executable under the ffmpeg header in Manim's configuration."
             )
 
-    def init_output_directories(self, scene_name):
+    def init_output_directories(self, scene_name: str):
         """Initialise output directories.
 
         Notes
@@ -172,6 +172,48 @@ def init_output_directories(self, scene_name):
                     scene_name=scene_name, module_name=module_name, log_dir=log_dir
                 )
 
+    def delete_tex_files(
+        self, scene_name: str, additional_endings: tuple[str] = ()
+    ) -> int:
+        """Deletes the .dvi, .aux, and .log files produced when using `Tex` or `MathTex`
+
+        Parameters:
+        -----------
+        scene_name
+            Name of class inheriting from `Scene`
+        additional_endings
+            Additional Endings to remove in Tex folder
+
+        Returns:
+        --------
+        :class:`int`
+            How many files were deleted
+        """
+        # ideally placed in self.init_output_directories, but when running tests that method isn't called?
+        # resulting in an AttributeError. Instead, just get directory here.
+        try:
+            tex_files_directory = guarantee_existence(
+                config.get_dir(
+                    "tex_dir",
+                    module_name=config.get_dir("input_file").stem
+                    if config["input_file"]
+                    else "",
+                    scene_name=scene_name,
+                )
+            )
+
+            file_endings = (".dvi", ".aux", ".log", *additional_endings)
+            files_deleted = 0
+            for file_name in tex_files_directory.iterdir():
+                file = tex_files_directory / file_name
+                if any(file.suffix == s for s in file_endings):
+                    file.unlink()
+                    files_deleted += 1
+            return files_deleted
+        except Exception as e:
+            logger.info("An error occured while trying to clean up LaTeX files")
+            return 0
+
     def finish_last_section(self) -> None:
         """Delete current section if it is empty."""
         if len(self.sections) and self.sections[-1].is_empty():