OTVision 0.5.3__py3-none-any.whl

OTVision/__init__.py

@@ -0,0 +1,30 @@
+"""
+OTVision init module
+"""
+
+# Copyright (C) 2022 OpenTrafficCam Contributors
+# <https://github.com/OpenTrafficCam
+# <team@opentrafficcam.org>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+
+# TODO: Might need to change this
+from .convert.convert import main as convert
+from .track.track import main as track
+from .transform.transform import main as transform
+
+# from .view.view import main as view
+
+__all__: list = ["track", "convert", "transform"]

OTVision/application/configure_logger.py

@@ -0,0 +1,23 @@
+import logging
+from logging import Logger
+from pathlib import Path
+
+from OTVision.config import Config
+from OTVision.helpers.log import LOGGER_NAME, log
+
+
+def logger() -> Logger:
+    return logging.getLogger(LOGGER_NAME)
+
+
+class ConfigureLogger:
+    def configure(
+        self, config: Config, log_file: Path, logfile_overwrite: bool
+    ) -> Logger:
+        log.add_console_handler(level=config.log.log_level_console)
+        log.add_file_handler(
+            log_file.expanduser(),
+            config.log.log_level_file,
+            logfile_overwrite,
+        )
+        return logging.getLogger(LOGGER_NAME)

OTVision/application/detect/get_detect_cli_args.py

@@ -0,0 +1,9 @@
+from OTVision.domain.cli import DetectCliArgs, DetectCliParser
+
+
+class GetDetectCliArgs:
+    def __init__(self, cli_parser: DetectCliParser) -> None:
+        self._cli_parser = cli_parser
+
+    def get(self) -> DetectCliArgs:
+        return self._cli_parser.parse()

OTVision/application/detect/update_detect_config_with_cli_args.py

@@ -0,0 +1,95 @@
+from OTVision.application.detect.get_detect_cli_args import GetDetectCliArgs
+from OTVision.config import Config, DetectConfig, YoloConfig, _LogConfig
+from OTVision.domain.cli import DetectCliArgs
+
+
+class UpdateDetectConfigWithCliArgs:
+    def __init__(self, get_detect_cli_args: GetDetectCliArgs) -> None:
+        self._get_detect_cli_args = get_detect_cli_args
+
+    def update(self, config: Config) -> Config:
+        cli_args = self._get_detect_cli_args.get()
+        return Config(
+            log=self._update_log_config(config, cli_args),
+            search_subdirs=config.search_subdirs,
+            default_filetype=config.default_filetype,
+            last_paths=config.last_paths,
+            convert=config.convert,
+            detect=self._update_detect_config(config.detect, cli_args),
+            track=config.track,
+            undistort=config.undistort,
+            transform=config.transform,
+            gui=config.gui,
+        )
+
+    def _update_detect_config(
+        self, detect_config: DetectConfig, cli_args: DetectCliArgs
+    ) -> DetectConfig:
+        yolo_config = YoloConfig(
+            weights=(
+                cli_args.weights
+                if cli_args.weights is not None
+                else detect_config.yolo_config.weights
+            ),
+            available_weights=detect_config.yolo_config.available_weights,
+            conf=(
+                cli_args.conf
+                if cli_args.conf is not None
+                else detect_config.yolo_config.conf
+            ),
+            iou=(
+                cli_args.iou
+                if cli_args.iou is not None
+                else detect_config.yolo_config.iou
+            ),
+            img_size=(
+                cli_args.imagesize
+                if cli_args.imagesize is not None
+                else detect_config.yolo_config.img_size
+            ),
+            chunk_size=detect_config.yolo_config.chunk_size,
+            normalized=detect_config.yolo_config.normalized,
+        )
+        return DetectConfig(
+            paths=cli_args.paths if cli_args.paths is not None else detect_config.paths,
+            yolo_config=yolo_config,
+            expected_duration=(
+                cli_args.expected_duration
+                if cli_args.expected_duration is not None
+                else detect_config.expected_duration
+            ),
+            overwrite=(
+                cli_args.overwrite
+                if cli_args.overwrite is not None
+                else detect_config.overwrite
+            ),
+            half_precision=(
+                cli_args.half
+                if cli_args.half is not None
+                else detect_config.half_precision
+            ),
+            detect_start=(
+                cli_args.detect_start
+                if cli_args.detect_start is not None
+                else detect_config.detect_start
+            ),
+            detect_end=(
+                cli_args.detect_end
+                if cli_args.detect_end is not None
+                else detect_config.detect_end
+            ),
+        )
+
+    def _update_log_config(self, config: Config, cli_args: DetectCliArgs) -> _LogConfig:
+        return _LogConfig(
+            log_level_console=(
+                cli_args.log_level_console
+                if cli_args.log_level_console is not None
+                else config.log.log_level_console
+            ),
+            log_level_file=(
+                cli_args.log_level_file
+                if cli_args.log_level_file is not None
+                else config.log.log_level_file
+            ),
+        )

OTVision/application/get_config.py

@@ -0,0 +1,25 @@
+from pathlib import Path
+
+from OTVision.config import Config, ConfigParser
+from OTVision.domain.cli import CliArgs
+
+DEFAULT_USER_CONFIG = "user_config.otvision.yaml"
+
+
+class GetConfig:
+    def __init__(self, config_parser: ConfigParser) -> None:
+        self._config_parser = config_parser
+
+    def get(self, args: CliArgs) -> Config:
+        config = self._get_config(args)
+        return config
+
+    def _get_config(self, args: CliArgs) -> Config:
+        if config_file := args.get_config_file():
+            return self._config_parser.parse(config_file)
+        else:
+            user_config_cwd = Path.cwd() / DEFAULT_USER_CONFIG
+
+            if user_config_cwd.exists():
+                return self._config_parser.parse(user_config_cwd)
+            return Config()