feat(cli): add resize mode option (#59)

This adds a new option that allows to control how the video is going to be scaled. By default, we now use a smooth rescaling, but fast mode with no interpolation can be used (previously default).

Author: jeertmans

manim_slides/config.py

@@ -148,3 +148,6 @@ def animation_indices_match_files(cls, values):
                 )
 
         return values
+
+
+DEFAULT_CONFIG = Config()

manim_slides/present.py

@@ -16,7 +16,7 @@
 from tqdm import tqdm
 
 from .commons import config_path_option, verbosity_option
-from .config import Config, PresentationConfig, SlideConfig, SlideType
+from .config import DEFAULT_CONFIG, Config, PresentationConfig, SlideConfig, SlideType
 from .defaults import FOLDER_PATH
 from .manim import logger
 
@@ -33,6 +33,11 @@
     "keep": Qt.KeepAspectRatio,
 }
 
+RESIZE_MODES = {
+    "fast": Qt.FastTransformation,
+    "smooth": Qt.SmoothTransformation,
+}
+
 
 @unique
 class State(IntEnum):
@@ -284,7 +289,7 @@ class Display(QThread):
     def __init__(
         self,
         presentations,
-        config: Config = Config(),
+        config: Config = DEFAULT_CONFIG,
         start_paused=False,
         skip_all=False,
         record_to=None,
@@ -517,18 +522,20 @@ class App(QWidget):
     def __init__(
         self,
         *args,
-        config: Config = Config(),
+        config: Config = DEFAULT_CONFIG,
         fullscreen: bool = False,
         resolution: Tuple[int, int] = (1980, 1080),
         hide_mouse: bool = False,
         aspect_ratio: Qt.AspectRatioMode = Qt.IgnoreAspectRatio,
+        resize_mode: Qt.TransformationMode = Qt.SmoothTransformation,
         **kwargs,
     ):
         super().__init__()
 
         self.setWindowTitle(WINDOW_NAME)
         self.display_width, self.display_height = resolution
         self.aspect_ratio = aspect_ratio
+        self.resize_mode = resize_mode
         self.hide_mouse = hide_mouse
         self.config = config
         if self.hide_mouse:
@@ -584,7 +591,9 @@ def closeAll(self):
         self.deleteLater()
 
     def resizeEvent(self, event):
-        self.pixmap = self.pixmap.scaled(self.width(), self.height(), self.aspect_ratio)
+        self.pixmap = self.pixmap.scaled(
+            self.width(), self.height(), self.aspect_ratio, self.resize_mode
+        )
         self.label.setPixmap(self.pixmap)
         self.label.resize(self.width(), self.height())
 
@@ -615,6 +624,7 @@ def convert_cv_qt(self, cv_img):
             self.width(),
             self.height(),
             self.aspect_ratio,
+            self.resize_mode,
         )
         return QPixmap.fromImage(p)
 
@@ -708,6 +718,13 @@ def _list_scenes(folder) -> List[str]:
     help="Set the aspect ratio mode to be used when rescaling video.",
     show_default=True,
 )
+@click.option(
+    "--resize-mode",
+    type=click.Choice(RESIZE_MODES.keys(), case_sensitive=False),
+    default="smooth",
+    help="Set the resize (i.e., transformation) mode to be used when rescaling video.",
+    show_default=True,
+)
 @click.help_option("-h", "--help")
 @verbosity_option
 def present(
@@ -722,6 +739,7 @@ def present(
     exit_after_last_slide,
     hide_mouse,
     aspect_ratio,
+    resize_mode,
 ) -> None:
     """
     Present SCENE(s), one at a time, in order.
@@ -813,6 +831,7 @@ def value_proc(value: str) -> List[str]:
         exit_after_last_slide=exit_after_last_slide,
         hide_mouse=hide_mouse,
         aspect_ratio=ASPECT_RATIO_MODES[aspect_ratio],
+        resize_mode=RESIZE_MODES[resize_mode],
     )
     a.show()
     sys.exit(app.exec_())