forked from pn2q95w37/XL
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
1.2 KiB
34 lines
1.2 KiB
2 years ago
|
from cv2 import imdecode
|
||
|
|
import numpy as np
|
||
|
|
from PyQt5.QtCore import Qt
|
||
|
|
from PyQt5.QtWidgets import QTreeView, QDockWidget, QFileSystemModel
|
||
|
|
|
||
|
|
|
||
|
|
class FileSystemTreeView(QTreeView, QDockWidget):
|
||
|
|
def __init__(self, parent=None):
|
||
|
|
super().__init__(parent=parent)
|
||
|
|
self.mainwindow = parent
|
||
|
|
self.fileSystemModel = QFileSystemModel()
|
||
|
|
self.fileSystemModel.setRootPath('.')
|
||
|
|
self.setModel(self.fileSystemModel)
|
||
|
|
# 隐藏size,date等列
|
||
|
|
self.setColumnWidth(0, 200)
|
||
|
|
self.setColumnHidden(1, True)
|
||
|
|
self.setColumnHidden(2, True)
|
||
|
|
self.setColumnHidden(3, True)
|
||
|
|
# 不显示标题栏
|
||
|
|
self.header().hide()
|
||
|
|
# 设置动画
|
||
|
|
self.setAnimated(True)
|
||
|
|
# 选中不显示虚线
|
||
|
|
self.setFocusPolicy(Qt.NoFocus)
|
||
|
|
self.doubleClicked.connect(self.select_image)
|
||
|
|
self.setMinimumWidth(200)
|
||
|
|
|
||
|
|
def select_image(self, file_index):
|
||
|
|
file_name = self.fileSystemModel.filePath(file_index)
|
||
|
|
if file_name.endswith(('.jpg', '.png', '.bmp')):
|
||
|
|
src_img = imdecode(np.fromfile(file_name, dtype=np.uint8), -1)
|
||
|
|
self.mainwindow.change_image(src_img)
|
||
|
|
|