JS-08 Debugging

requirements.txt

@@ -1,7 +1,10 @@
-PyQt5 >= 5.13
+PySide6~=6.3.0
 python-vlc
-PyQtGraph
-pyechart
-influxdb
-opencv-python
-pandas
+opencv-contrib-python
+pandas~=1.3.4
+scipy~=1.7.2
+psutil~=5.9.0
+pyinstaller
+opencv-python~=4.6.0.66
+numpy~=1.21.3
+cryptography~=37.0.2

run.py

@@ -6,6 +6,44 @@
 #     [email protected] (Seong Min Chae)
 #     [email protected] (Jongjin Oh)
 
-from src.__main__ import main
+import os
+import sys
+import time
+import multiprocessing as mp
+from multiprocessing import Queue, Process
 
-main()
+from PySide6.QtWidgets import QApplication, QMessageBox
+from PySide6.QtGui import QGuiApplication
+
+from src.video_thread_mp import producer
+from src.clock import clockclock
+from src.js08 import JS08MainWindow
+from src.model import JS08Settings
+
+
+if __name__ == '__main__':
+    print(f'Start time: {time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}')
+
+    mp.freeze_support()
+    q = Queue()
+    _q = Queue()
+
+    _producer = producer
+
+    p = Process(name='clock', target=clockclock, args=(q,), daemon=True)
+    _p = Process(name='producer', target=_producer, args=(_q,), daemon=True)
+
+    p.start()
+    _p.start()
+
+    os.makedirs(f'{JS08Settings.get("data_csv_path")}', exist_ok=True)
+    os.makedirs(f'{JS08Settings.get("target_csv_path")}', exist_ok=True)
+    os.makedirs(f'{JS08Settings.get("image_save_path")}', exist_ok=True)
+
+    app = QApplication(sys.argv)
+    screen_size = QGuiApplication.screens()[0].geometry()
+    width, height = screen_size.width(), screen_size.height()
+    if width > 1920 or height > 1080:
+        QMessageBox.warning(None, 'Warning', 'JS08 is based on FHD screen.')
+    window = JS08MainWindow(q, _q)
+    sys.exit(app.exec())

src/auto_file_delete.py

@@ -0,0 +1,76 @@
+#!/usr/bin/env python3
+#
+# Copyright 2021-2022 9th grade 5th class.
+#
+# Authors:
+#     [email protected]
+
+import os
+import psutil
+import shutil
+
+from PySide6.QtWidgets import (QDialog, QApplication, QMessageBox)
+
+from model import JS08Settings
+from resources.auto_file_delete import Ui_Form
+
+
+def byte_transform(bytes, to, bsize=1024):
+    """
+    Unit conversion of byte received from shutil
+
+    :return: Capacity of the selected unit (int)
+    """
+    unit = {'KB': 1, 'MB': 2, 'GB': 3, 'TB': 4}
+    r = float(bytes)
+    for i in range(unit[to]):
+        r = r / bsize
+    return int(r)
+
+
+def delete_select_date(path: str, folder: list):
+    """
+    Delete the list containing the folder name
+
+    :param path: Path to proceed with a auto-delete
+    :param folder: Data older than the date selected as the calendarWidget
+    """
+
+    # for i in range(len(folder)):
+    #     a = os.path.join(path, str(folder[i]))
+    #     shutil.rmtree(a)
+    #     print(f'{a} delete complete.')
+    a = os.path.join(path, str(folder[0]))
+    print(f'{a} delete complete.')
+
+
+def check_file_date(path: str):
+    is_old = []
+
+    for f in os.listdir(path):
+        is_old.append(int(f))
+    delete_select_date(path, is_old)
+
+
+def FileAutoDelete():
+    save_disk = JS08Settings.get('image_save_path')[:2]
+
+    total, used, free = shutil.disk_usage(save_disk)
+    if JS08Settings.get('afd'):
+        if byte_transform(free, 'GB') <= 20:
+            check_disk()
+
+
+def check_disk():
+    check_file_date(os.path.join(JS08Settings.get('image_save_path'), 'vista',
+                                 JS08Settings.get('front_camera_name')))
+    print('-' * 10)
+    check_file_date(os.path.join(JS08Settings.get('image_save_path'), 'vista',
+                                 JS08Settings.get('rear_camera_name')))
+
+
+if __name__ == "__main__":
+    import sys
+
+    FileAutoDelete()
+    # sys.exit(app.exec())

src/cal_ext_coef.py

@@ -0,0 +1,99 @@
+#!/usr/bin/env python3
+#
+# Copyright 2021-2022 Sijung Co., Ltd.
+#
+# Authors:
+#     [email protected] (Seong Min Chae)
+#     [email protected] (Jongjin Oh)
+
+import itertools
+import os
+
+import time
+import numpy as np
+import pandas as pd
+import traceback
+
+from scipy.optimize import curve_fit
+from model import JS08Settings
+from save_log import log
+
+
+class Coef:
+
+    def __init__(self):
+        self.hanhwa_dist = []
+        self.hanhwa_x = []
+        self.hanhwa_r = []
+        self.hanhwa_g = []
+        self.hanhwa_b = []
+
+    def select_max_rgb(self, r, g, b):
+        c_list = [r, g, b]
+        c_index = c_list.index(max(c_list))
+
+        if c_index == 0:
+            select_color = 'red'
+        elif c_index == 1:
+            select_color = 'green'
+        else:
+            select_color = 'blue'
+
+        return select_color
+
+    def cal_curve(self, hanhwa: pd.DataFrame):
+        hanhwa = hanhwa.sort_values(by=['distance'])
+        self.hanhwa_dist = hanhwa[['distance']].squeeze().to_numpy()
+        self.hanhwa_x = np.linspace(self.hanhwa_dist[0], self.hanhwa_dist[-1], 100, endpoint=True)
+        self.hanhwa_x.sort()
+        self.hanhwa_r = hanhwa[['r']].squeeze().to_numpy()
+        self.hanhwa_g = hanhwa[['g']].squeeze().to_numpy()
+        self.hanhwa_b = hanhwa[['b']].squeeze().to_numpy()
+
+        r1_init = self.hanhwa_r[0] * 0.7
+        g1_init = self.hanhwa_g[0] * 0.7
+
+        b1_init = self.hanhwa_b[0] * 0.7
+        r2_init = self.hanhwa_r[-1] * 1.3
+        g2_init = self.hanhwa_g[-1] * 1.3
+        b2_init = self.hanhwa_b[-1] * 1.3
+
+        select_color = self.select_max_rgb(r2_init, g2_init, b2_init)
+
+        r_ext_init = [r1_init, r2_init, 1]
+        g_ext_init = [g1_init, g2_init, 1]
+        b_ext_init = [b1_init, b2_init, 1]
+
+        try:
+            hanhwa_opt_r, hanhwa_cov_r = curve_fit(self.func, self.hanhwa_dist, self.hanhwa_r, p0=r_ext_init, maxfev=5000)
+            hanhwa_opt_g, hanhwa_cov_g = curve_fit(self.func, self.hanhwa_dist, self.hanhwa_g, p0=g_ext_init, maxfev=5000)
+            hanhwa_opt_b, hanhwa_cov_b = curve_fit(self.func, self.hanhwa_dist, self.hanhwa_b, p0=b_ext_init, maxfev=5000)
+            JS08Settings.set('maxfev_flag', False)
+
+        except RuntimeError as e:
+            # print(f'[{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}] - {e}')
+            # JS08Settings.set('maxfev_flag', True)
+            # JS08Settings.set('maxfev_count', JS08Settings.get('maxfev_count') + 1)
+            log(JS08Settings.get('current_id'), 'maxfev Error')
+            return
+
+        list1 = []
+        list2 = []
+        list3 = []
+
+        list1.append(hanhwa_opt_r[0])
+        list1.append(hanhwa_opt_g[0])
+        list1.append(hanhwa_opt_b[0])
+
+        list2.append(hanhwa_opt_r[1])
+        list2.append(hanhwa_opt_g[1])
+        list2.append(hanhwa_opt_b[1])
+
+        list3.append(hanhwa_opt_r[2])
+        list3.append(hanhwa_opt_g[2])
+        list3.append(hanhwa_opt_b[2])
+
+        return list1, list2, list3, select_color
+
+    def func(self, x, c1, c2, a):
+        return c2 + (c1 - c2) * np.exp(-a * x)