Python-Funscript-Editor/funscript_editor/api.py at main · lookingaround256/Python-Funscript-Editor · GitHub

Repository navigation

History

69 lines (60 loc) · 2.34 KB

Raw

import sys

import os

import cv2

import funscript_editor.utils.logging as logging

from funscript_editor.ui.funscript_editor_window import FunscriptEditorWindow

from funscript_editor.ui.funscript_generator_window import FunscriptGeneratorWindow

from funscript_editor.utils.config import VERSION

from PyQt5 import QtCore, QtGui, QtWidgets

def show_editor() -> None:

""" Show the Funscript Editor Main Window """

logging.setup_logging(silent=False)

logger = logging.getLogger(__name__)

logger.info("Python Funscript Editor %s", VERSION)

logger.info("Startup Path: %s", str(os.getcwd()))

app = QtWidgets.QApplication(sys.argv)

widget = FunscriptEditorWindow()

widget.show()

ret = app.exec_()

logger.info("EXIT")

sys.exit(ret)

def generate_funscript(

video_file: str,

start_time: float,

end_time :float,

output_file: str,

include_multiaxis_options: bool = False,

no_tracking: bool = False,

enable_logging: bool = False,

stdout_logging: bool = False) -> None:

""" Generate a funscript with minimal UI

Args:

video_file (str): path to video file

start_time (float): start time in milliseconds

end_time (float): end time in milliseconds (set -1.0 to use video end)

output_file (str): path for the output file

include_multiaxis_options (bool): include options for multiaxis output

no_tracking (bool): Use previous tracking result

enable_logging (bool): enable logging

stdout_logging (bool): enable stdout logging

"""

if stdout_logging:

enable_logging = True

if enable_logging:

logging.setup_logging(silent=not stdout_logging)

else:

logging.disable_logging()

logger = logging.getLogger(__name__)

logger.info("Python Funscript Generator %s", VERSION)

logger.info("Startup Path: %s", str(os.getcwd()))

logger.info("Args: video_file=%s, start_time=%s, end_time=%s, output_file=%s", \

str(video_file), str(start_time), str(end_time), str(output_file))

app = QtWidgets.QApplication(sys.argv)

w = FunscriptGeneratorWindow(video_file, start_time, end_time, output_file, include_multiaxis_options, no_tracking)

w.setFixedHeight(5)

w.setFixedWidth(5)

w.move(1,1)

w.show()

ret = app.exec_()

logger.info("EXIT")

sys.exit(ret)

Python-Funscript-Editor/funscript_editor/api.py at main · lookingaround256/Python-Funscript-Editor · GitHub

Repository navigation

Expand file tree

Breadcrumbs

api.py

Latest commit

History

Breadcrumbs

api.py

File metadata and controls