Source code for sleap.gui.dialogs.export_clip

"""
Dialog for exporting clip; shows message depending on available encoder.
"""

from sleap.gui.dialogs.formbuilder import FormBuilderModalDialog


class ExportClipDialog(FormBuilderModalDialog):
    def __init__(self):
        from sleap.io.videowriter import VideoWriter

        super().__init__(form_name="labeled_clip_form")

        can_use_skvideo = VideoWriter.can_use_skvideo()

        if can_use_skvideo:
            message = (
                "<i><b>MP4</b> file will be encoded using "
                "system ffmpeg via scikit-video (preferred option).</i>"
            )
        else:
            message = (
                "<i>Unable to use ffpmeg via scikit-video. "
                "<b>AVI</b> file will be encoding using OpenCV.</i>"
            )

        self.add_message(message)

        self.setWindowTitle("Export Clip Options")