add crop canvas function

handyview/canvas_crop.py

@@ -0,0 +1,211 @@
+import glob
+import os
+import time
+from PyQt5.QtCore import QSize
+from PyQt5.QtGui import QIcon
+from PyQt5.QtWidgets import (QComboBox, QGridLayout, QGroupBox, QLabel, QLineEdit, QListWidget, QListWidgetItem,
+                             QPushButton, QScrollArea, QVBoxLayout, QWidget)
+
+from handyview.utils import ROOT_DIR, crop_images
+from handyview.widgets import HLine, HVLable, show_msg
+
+
+class CanvasCrop(QWidget):
+    """Crop canvas
+
+    TODO:
+    1. modify selected rectangular
+    2. drag and drop mode
+
+    """
+
+    def __init__(self, parent, db):
+        super(CanvasCrop, self).__init__()
+        self.parent = parent
+        self.db = db  # database
+
+        # initialize widgets and layout
+        self.init_widgets_layout()
+
+    def init_widgets_layout(self):
+        # show thumbnails
+        self.scrollArea = QScrollArea(widgetResizable=True)
+        self.thumbnails = QListWidget()
+        self.thumbnails.setIconSize(QSize(200, 150))
+        self.thumbnails.setViewMode(QListWidget.IconMode)
+        self.thumbnails.setResizeMode(QListWidget.Adjust)
+        self.scrollArea.setWidget(self.thumbnails)
+        self.thumbnails.itemSelectionChanged.connect(self.selectionChanged)
+
+        # show thumbnails for cropped images
+        self.crop_scrollArea = QScrollArea(widgetResizable=True)
+        self.crop_thumbnails = QListWidget()
+        self.crop_thumbnails.setIconSize(QSize(200, 150))
+        self.crop_thumbnails.setViewMode(QListWidget.IconMode)
+        self.crop_thumbnails.setResizeMode(QListWidget.Adjust)
+        self.crop_scrollArea.setWidget(self.crop_thumbnails)
+
+        # ---------------------------------------
+        # layouts
+        # ---------------------------------------
+        # selected rectangular
+        label_height = HVLable('Height', self, color='blue')
+        label_width = HVLable('Width', self, color='blue')
+        label_start = HVLable('Start', self, color='blue')
+        label_len = HVLable('Length', self, color='blue')
+        # get init label pos
+        self.edit_start_h = QLineEdit('0')
+        self.edit_start_w = QLineEdit('0')
+        self.edit_len_h = QLineEdit('0')
+        self.edit_len_w = QLineEdit('0')
+        # scale
+        label_scale = HVLable('Scale', self, color='blue')
+        label_ratio = HVLable('Ratio', self, color='blue')
+        label_mode = HVLable('Mode', self, color='blue')
+        self.edit_ratio = QLineEdit('2')
+        self.combo_mode = QComboBox()
+        self.combo_mode.addItems(['bicubic', 'bilinear', 'nearest'])
+
+        # config grid
+        config_grid = QGridLayout()
+        # config_grid.setSpacing(10)
+        # int row, int column, int rowSpan, int columnSpan
+        # -----  Start  Len   -----  Scale
+        # Height [   ] [   ]  Ratio  [   ]
+        # Width  [   ] [   ]  Mode   Combo
+        # row 0
+        config_grid.addWidget(label_start, 0, 1, 1, 1)
+        config_grid.addWidget(label_len, 0, 2, 1, 1)
+        config_grid.addWidget(label_scale, 0, 4, 1, 1)
+        # row 1
+        config_grid.addWidget(label_height, 1, 0, 1, 1)
+        config_grid.addWidget(self.edit_start_h, 1, 1, 1, 1)
+        config_grid.addWidget(self.edit_len_h, 1, 2, 1, 1)
+        config_grid.addWidget(label_ratio, 1, 3, 1, 1)
+        config_grid.addWidget(self.edit_ratio, 1, 4, 1, 1)
+        # row 1
+        config_grid.addWidget(label_width, 2, 0, 1, 1)
+        config_grid.addWidget(self.edit_start_w, 2, 1, 1, 1)
+        config_grid.addWidget(self.edit_len_w, 2, 2, 1, 1)
+        config_grid.addWidget(label_mode, 2, 3, 1, 1)
+        config_grid.addWidget(self.combo_mode, 2, 4, 1, 1)
+        # blank
+        config_grid.addWidget(QLabel(), 3, 0, 5, 5)
+
+        # actions
+        button_add = QPushButton('Add ALL', self)
+        button_add.clicked.connect(self.add_all_images)
+        button_selection_pos = QPushButton('Set Selection Pos', self)
+        button_selection_pos.clicked.connect(self.set_selection_pos)
+        button_crop = QPushButton('Crop', self)
+        button_crop.clicked.connect(self.crop_images)
+        button_open = QPushButton('Open Patch Folder', self)
+        button_open.clicked.connect(self.open_patch_folder)
+        button_open_history = QPushButton('Open History Info', self)
+        button_open_history.clicked.connect(self.open_history_file)
+
+        action_grid = QGridLayout()
+        action_grid.addWidget(button_add, 0, 0, 1, 1)
+        action_grid.addWidget(button_selection_pos, 1, 0, 1, 1)
+        action_grid.addWidget(button_crop, 2, 0, 1, 1)
+        action_grid.addWidget(HLine(), 3, 0, 1, 1)
+        action_grid.addWidget(button_open, 4, 0, 1, 1)
+        action_grid.addWidget(button_open_history, 5, 0, 1, 1)
+
+        config_box = QGroupBox('Config')
+        config_box.setLayout(config_grid)
+        action_box = QGroupBox('Actions')
+        action_box.setLayout(action_grid)
+
+        panel_grid = QVBoxLayout()
+        panel_grid.addWidget(config_box)
+        panel_grid.addWidget(action_box)
+        panel_grid.setSpacing(0)  # No Spacing
+        panel_grid.addStretch(1)  # No Expanding when the window expands
+
+        # ---------------------------------------
+        # main layouts
+        # ---------------------------------------
+        main_layout = QGridLayout(self)
+        # int row, int column, int rowSpan, int columnSpan
+        main_layout.addWidget(self.scrollArea, 0, 0, 20, 20)
+        main_layout.addWidget(self.crop_scrollArea, 20, 0, 20, 20)
+
+        main_layout.addLayout(panel_grid, 0, 20, 40, 5)
+
+    def selectionChanged(self):
+        print('Selected items: ', self.thumbnails.selectedItems())
+
+    def set_selection_pos(self):
+        start_h, start_w, len_h, len_w = self.db.selection_pos
+        self.edit_start_h.setText(str(start_h))
+        self.edit_start_w.setText(str(start_w))
+        self.edit_len_h.setText(str(len_h))
+        self.edit_len_w.setText(str(len_w))
+
+    def add_all_images(self):
+        self.set_selection_pos()
+        # 1. clear all the existing thumbnails
+        self.thumbnails.clear()
+        # 2. add thumbnails
+        for path in self.db.path_list[0]:
+            self.thumbnails.addItem(QListWidgetItem(QIcon(path), os.path.basename(path)))
+
+    def update_crop_images(self, patch_folder):
+        # 1. clear all the existing thumbnails
+        self.crop_thumbnails.clear()
+        # 2. add thumbnails
+        for path in sorted(glob.glob(os.path.join(patch_folder, '*'))):
+            self.crop_thumbnails.addItem(QListWidgetItem(QIcon(path), os.path.basename(path)))
+
+    def crop_images(self):
+        # 1. check all images has the same shape
+        # TODO
+        # 2. crop
+        try:
+            start_h = int(self.edit_start_h.text())
+            start_w = int(self.edit_start_w.text())
+            len_h = int(self.edit_len_h.text())
+            len_w = int(self.edit_len_w.text())
+            ratio = int(self.edit_ratio.text())
+            mode = self.combo_mode.currentText()
+        except ValueError as error:
+            show_msg(icon='Critical', title='Title', text=f'Wrong input: {error}', timeout=None)
+            return 0
+        # get patch folder
+        self.patch_folder = os.path.join(os.path.dirname(self.db.path_list[0][0]), os.pardir, 'crop_patch')
+
+        try:
+            crop_images(
+                self.db.path_list[0], [start_h, start_w, len_h, len_w],
+                self.patch_folder,
+                enlarge_ratio=ratio,
+                interpolation=mode,
+                line_width=0,
+                color='yellow',
+                rect_folder=None)
+        except Exception as error:
+            show_msg(icon='Critical', title='Title', text=f'Crop error: {error}', timeout=None)
+        else:
+            # update crop info to txt
+            self.record_crop_history(self.db.path_list[0][0], [start_h, start_w, len_h, len_w], ratio, mode)
+            # show cropped image
+            self.update_crop_images(self.patch_folder)
+
+    def record_crop_history(self, path, pos, ratio, mode):
+        pos_str = ', '.join(map(str, pos))
+        content = f'{time.strftime("%Y%m%d-%H%M%S", time.localtime())} {path} ({pos_str}) {ratio} {mode}\n'
+        try:
+            with open(os.path.join(ROOT_DIR, 'history_crop.txt'), 'a') as f:
+                f.write(content)
+        except Exception as error:
+            show_msg(icon='Warning', title='Title', text=f'Record crop history error: {error}', timeout=None)
+
+    def open_patch_folder(self):
+        os.startfile(self.patch_folder)
+
+    def open_history_file(self):
+        os.startfile(os.path.join(ROOT_DIR, 'history_crop.txt'))
+
+    def keyPressEvent(self, event):
+        pass

handyview/db.py

@@ -36,6 +36,9 @@ def __init__(self, init_path):
         self.md5_list = [[]]
         self.phash_list = [[]]
 
+        # for selection pos in crop canvas
+        self.selection_pos = [0, 0, 0, 0]
+
         self.get_init_path_list()
 
     def get_init_path_list(self):