Frame13/ayon-core
1
0
Fork 1
mirror of https://github.com/ynput/ayon-core.git synced 2025-12-25 13:24:54 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

moved thumbnail widget to separated file

Browse source
This commit is contained in:
Jakub Trllo 2022-10-27 15:12:51 +02:00
parent b6a2b51dad
commit cc4d158c78
3 changed files with 314 additions and 201 deletions
Download patch file Download diff file Expand all files Collapse all files

2
openpype/tools/publisher/widgets/create_widget.py
View file

@ -7,10 +7,10 @@ from openpype.pipeline.create import (
TaskNotSetError,
)
from .thumbnail_widget import ThumbnailWidget
from .widgets import (
IconValuePixmapLabel,
CreateBtn,
ThumbnailWidget,
)
from .assets_widget import CreateWidgetAssetsWidget
from .tasks_widget import CreateWidgetTasksWidget

312
openpype/tools/publisher/widgets/thumbnail_widget.py Normal file
View file

@ -0,0 +1,312 @@
import os
import tempfile
import uuid
from Qt import QtWidgets, QtCore, QtGui
from openpype.lib import (
run_subprocess,
is_oiio_supported,
get_oiio_tools_path,
get_ffmpeg_tool_path,
)
from openpype.lib.transcoding import (
IMAGE_EXTENSIONS,
VIDEO_EXTENSIONS,
)
from openpype.tools.utils import (
paint_image_with_color,
)
from .icons import get_image
class ThumbnailWidget(QtWidgets.QWidget):
"""Instance thumbnail widget."""
thumbnail_created = QtCore.Signal(str)
width_ratio = 3.0
height_ratio = 2.0
border_width = 1
offset_sep = 4
def __init__(self, parent):
# Missing implementation for thumbnail
# - widget kept to make a visial offset of global attr widget offset
super(ThumbnailWidget, self).__init__(parent)
self.setAcceptDrops(True)
# TODO remove hardcoded colors
border_color = QtGui.QColor(67, 74, 86)
thumbnail_bg_color = QtGui.QColor(54, 61, 72)
default_image = get_image("thumbnail")
default_pix = paint_image_with_color(default_image, border_color)
self.border_color = border_color
self.thumbnail_bg_color = thumbnail_bg_color
self._default_pix = default_pix
self._current_pixes = None
self._cached_pix = None
self._height = None
self._width = None
self._adapted_to_size = True
self._last_width = None
self._last_height = None
self._review_extensions = set(IMAGE_EXTENSIONS) | set(VIDEO_EXTENSIONS)
def _get_filepath_from_event(self, event):
mime_data = event.mimeData()
if not mime_data.hasUrls():
return None
filepaths = []
for url in mime_data.urls():
filepath = url.toLocalFile()
if os.path.exists(filepath):
filepaths.append(filepath)
if len(filepaths) == 1:
filepath = filepaths[0]
ext = os.path.splitext(filepath)[-1]
if ext in self._review_extensions:
return filepath
return None
def dragEnterEvent(self, event):
filepath = self._get_filepath_from_event(event)
if filepath:
event.setDropAction(QtCore.Qt.CopyAction)
event.accept()
def dragLeaveEvent(self, event):
event.accept()
def dropEvent(self, event):
filepath = self._get_filepath_from_event(event)
if filepath:
output = export_thumbnail(filepath)
if output:
self.thumbnail_created.emit(output)
def set_adapted_to_hint(self, enabled):
self._adapted_to_size = enabled
if self._width is not None:
self.setMinimumHeight(0)
self._width = None
if self._height is not None:
self.setMinimumWidth(0)
self._height = None
def set_width(self, width):
if self._width == width:
return
self._adapted_to_size = False
self._width = width
self._cached_pix = None
self.setMinimumHeight(int(
(width / self.width_ratio) * self.height_ratio
))
if self._height is not None:
self.setMinimumWidth(0)
self._height = None
def set_height(self, height):
if self._height == height:
return
self._height = height
self._adapted_to_size = False
self._cached_pix = None
self.setMinimumWidth(int(
(height / self.height_ratio) * self.width_ratio
))
if self._width is not None:
self.setMinimumHeight(0)
self._width = None
def _get_current_pixes(self):
if self._current_pixes is None:
return [self._default_pix]
return self._current_pixes
def _cache_pix(self):
rect = self.rect()
rect_width = rect.width()
rect_height = rect.height()
pix_x_offset = 0
pix_y_offset = 0
expected_height = int(
(rect_width / self.width_ratio) * self.height_ratio
)
if expected_height > rect_height:
expected_height = rect_height
expected_width = int(
(rect_height / self.height_ratio) * self.width_ratio
)
pix_x_offset = (rect_width - expected_width) / 2
else:
expected_width = rect_width
pix_y_offset = (rect_height - expected_height) / 2
pixes_to_draw = self._get_current_pixes()
max_pix = 3
if len(pixes_to_draw) > max_pix:
pixes_to_draw = pixes_to_draw[:-max_pix]
pixes_len = len(pixes_to_draw)
width_offset, height_offset = self._get_pix_offset_size(
expected_width, expected_height, pixes_len
)
pix_width = expected_width - width_offset
pix_height = expected_height - height_offset
full_border_width = 2 * self.border_width
pix_bg_brush = QtGui.QBrush(self.thumbnail_bg_color)
pix_pen = QtGui.QPen()
pix_pen.setWidth(self.border_width)
pix_pen.setColor(self.border_color)
backgrounded_images = []
for src_pix in pixes_to_draw:
scaled_pix = src_pix.scaled(
pix_width - full_border_width,
pix_height - full_border_width,
QtCore.Qt.KeepAspectRatio,
QtCore.Qt.SmoothTransformation
)
pos_x = int(
(pix_width - scaled_pix.width()) / 2
) + self.border_width
pos_y = int(
(pix_height - scaled_pix.height()) / 2
) + self.border_width
new_pix = QtGui.QPixmap(pix_width, pix_height)
pix_painter = QtGui.QPainter()
pix_painter.begin(new_pix)
pix_painter.setBrush(pix_bg_brush)
pix_painter.setPen(pix_pen)
pix_painter.drawRect(0, 0, pix_width - 1, pix_height - 1)
pix_painter.drawPixmap(pos_x, pos_y, scaled_pix)
pix_painter.end()
backgrounded_images.append(new_pix)
if pixes_len == 1:
width_offset_part = 0
height_offset_part = 0
else:
width_offset_part = int(float(width_offset) / (pixes_len - 1))
height_offset_part = int(float(height_offset) / (pixes_len - 1))
full_width_offset = width_offset + pix_x_offset
final_pix = QtGui.QPixmap(rect_width, rect_height)
final_pix.fill(QtCore.Qt.transparent)
final_painter = QtGui.QPainter()
final_painter.begin(final_pix)
for idx, pix in enumerate(backgrounded_images):
x_offset = full_width_offset - (width_offset_part * idx)
y_offset = (height_offset_part * idx) + pix_y_offset
final_painter.drawPixmap(x_offset, y_offset, pix)
final_painter.end()
self._cached_pix = final_pix
def _get_pix_offset_size(self, width, height, image_count):
if image_count == 1:
return 0, 0
part_width = width / self.offset_sep
part_height = height / self.offset_sep
return part_width, part_height
def paintEvent(self, event):
if self._cached_pix is None:
self._cache_pix()
painter = QtGui.QPainter()
painter.begin(self)
painter.drawPixmap(0, 0, self._cached_pix)
painter.end()
def _adapt_to_size(self):
if not self._adapted_to_size:
return
width = self.width()
height = self.height()
if width == self._last_width and height == self._last_height:
return
self._last_width = width
self._last_height = height
self._cached_pix = None
def resizeEvent(self, event):
super(ThumbnailWidget, self).resizeEvent(event)
self._adapt_to_size()
def showEvent(self, event):
super(ThumbnailWidget, self).showEvent(event)
self._adapt_to_size()
def _run_silent_subprocess(args):
with open(os.devnull, "w") as devnull:
run_subprocess(args, stdout=devnull, stderr=devnull)
def _convert_thumbnail_oiio(src_path, dst_path):
if not is_oiio_supported():
return None
oiio_cmd = [
get_oiio_tools_path(),
"-i", src_path,
"--subimage", "0",
"-o", dst_path
]
try:
_run_silent_subprocess(oiio_cmd)
except Exception:
return None
return dst_path
def _convert_thumbnail_ffmpeg(src_path, dst_path):
ffmpeg_cmd = [
get_ffmpeg_tool_path(),
"-y",
"-i", src_path,
dst_path
]
try:
_run_silent_subprocess(ffmpeg_cmd)
except Exception:
return None
return dst_path
def export_thumbnail(src_path):
root_dir = os.path.join(
tempfile.gettempdir(),
"publisher_thumbnails"
)
if not os.path.exists(root_dir):
os.makedirs(root_dir)
ext = os.path.splitext(src_path)[-1]
if ext not in (".jpeg", ".jpg", ".png"):
ext = ".jpeg"
filename = str(uuid.uuid4()) + ext
dst_path = os.path.join(root_dir, filename)
output_path = _convert_thumbnail_oiio(src_path, dst_path)
if not output_path:
output_path = _convert_thumbnail_ffmpeg(src_path, dst_path)
return output_path

201
openpype/tools/publisher/widgets/widgets.py
View file

@ -16,18 +16,17 @@ from openpype.tools.utils import (
PixmapLabel,
BaseClickableFrame,
set_style_property,
paint_image_with_color,
)
from openpype.style import get_objected_colors
from openpype.pipeline.create import (
SUBSET_NAME_ALLOWED_SYMBOLS,
TaskNotSetError,
)
from .thumbnail_widget import ThumbnailWidget
from .assets_widget import AssetsDialog
from .tasks_widget import TasksModel
from .icons import (
get_pixmap,
get_image,
get_icon_path
)
@ -1621,201 +1620,3 @@ class SubsetAttributesWidget(QtWidgets.QWidget):
)
self.creator_attrs_widget.set_instances_valid(all_valid)
self.publish_attrs_widget.set_instances_valid(all_valid)
class ThumbnailWidget(QtWidgets.QWidget):
"""Instance thumbnail widget."""
width_ratio = 3.0
height_ratio = 2.0
border_width = 1
offset_sep = 4
def __init__(self, parent):
# Missing implementation for thumbnail
# - widget kept to make a visial offset of global attr widget offset
super(ThumbnailWidget, self).__init__(parent)
# TODO remove hardcoded colors
border_color = QtGui.QColor(67, 74, 86)
thumbnail_bg_color = QtGui.QColor(54, 61, 72)
default_image = get_image("thumbnail")
default_pix = paint_image_with_color(default_image, border_color)
self.border_color = border_color
self.thumbnail_bg_color = thumbnail_bg_color
self._default_pix = default_pix
self._current_pixes = None
self._cached_pix = None
self._height = None
self._width = None
self._adapted_to_size = True
self._last_width = None
self._last_height = None
def set_adapted_to_hint(self, enabled):
self._adapted_to_size = enabled
if self._width is not None:
self.setMinimumHeight(0)
self._width = None
if self._height is not None:
self.setMinimumWidth(0)
self._height = None
def set_width(self, width):
if self._width == width:
return
self._adapted_to_size = False
self._width = width
self._cached_pix = None
self.setMinimumHeight(int(
(width / self.width_ratio) * self.height_ratio
))
if self._height is not None:
self.setMinimumWidth(0)
self._height = None
def set_height(self, height):
if self._height == height:
return
self._height = height
self._adapted_to_size = False
self._cached_pix = None
self.setMinimumWidth(int(
(height / self.height_ratio) * self.width_ratio
))
if self._width is not None:
self.setMinimumHeight(0)
self._width = None
def _get_current_pixes(self):
if self._current_pixes is None:
return [self._default_pix]
return self._current_pixes
def _cache_pix(self):
rect = self.rect()
rect_width = rect.width()
rect_height = rect.height()
pix_x_offset = 0
pix_y_offset = 0
expected_height = int(
(rect_width / self.width_ratio) * self.height_ratio
)
if expected_height > rect_height:
expected_height = rect_height
expected_width = int(
(rect_height / self.height_ratio) * self.width_ratio
)
pix_x_offset = (rect_width - expected_width) / 2
else:
expected_width = rect_width
pix_y_offset = (rect_height - expected_height) / 2
pixes_to_draw = self._get_current_pixes()
max_pix = 3
if len(pixes_to_draw) > max_pix:
pixes_to_draw = pixes_to_draw[:-max_pix]
pixes_len = len(pixes_to_draw)
width_offset, height_offset = self._get_pix_offset_size(
expected_width, expected_height, pixes_len
)
pix_width = expected_width - width_offset
pix_height = expected_height - height_offset
full_border_width = 2 * self.border_width
pix_bg_brush = QtGui.QBrush(self.thumbnail_bg_color)
pix_pen = QtGui.QPen()
pix_pen.setWidth(self.border_width)
pix_pen.setColor(self.border_color)
backgrounded_images = []
for src_pix in pixes_to_draw:
scaled_pix = src_pix.scaled(
pix_width - full_border_width,
pix_height - full_border_width,
QtCore.Qt.KeepAspectRatio,
QtCore.Qt.SmoothTransformation
)
pos_x = int(
(pix_width - scaled_pix.width()) / 2
) + self.border_width
pos_y = int(
(pix_height - scaled_pix.height()) / 2
) + self.border_width
new_pix = QtGui.QPixmap(pix_width, pix_height)
pix_painter = QtGui.QPainter()
pix_painter.begin(new_pix)
pix_painter.setBrush(pix_bg_brush)
pix_painter.setPen(pix_pen)
pix_painter.drawRect(0, 0, pix_width - 1, pix_height - 1)
pix_painter.drawPixmap(pos_x, pos_y, scaled_pix)
pix_painter.end()
backgrounded_images.append(new_pix)
if pixes_len == 1:
width_offset_part = 0
height_offset_part = 0
else:
width_offset_part = int(float(width_offset) / (pixes_len - 1))
height_offset_part = int(float(height_offset) / (pixes_len - 1))
full_width_offset = width_offset + pix_x_offset
final_pix = QtGui.QPixmap(rect_width, rect_height)
final_pix.fill(QtCore.Qt.transparent)
final_painter = QtGui.QPainter()
final_painter.begin(final_pix)
for idx, pix in enumerate(backgrounded_images):
x_offset = full_width_offset - (width_offset_part * idx)
y_offset = (height_offset_part * idx) + pix_y_offset
final_painter.drawPixmap(x_offset, y_offset, pix)
final_painter.end()
self._cached_pix = final_pix
def _get_pix_offset_size(self, width, height, image_count):
if image_count == 1:
return 0, 0
part_width = width / self.offset_sep
part_height = height / self.offset_sep
return part_width, part_height
def paintEvent(self, event):
if self._cached_pix is None:
self._cache_pix()
painter = QtGui.QPainter()
painter.begin(self)
painter.drawPixmap(0, 0, self._cached_pix)
painter.end()
def _adapt_to_size(self):
if not self._adapted_to_size:
return
width = self.width()
height = self.height()
if width == self._last_width and height == self._last_height:
return
self._last_width = width
self._last_height = height
self._cached_pix = None
def resizeEvent(self, event):
super(ThumbnailWidget, self).resizeEvent(event)
self._adapt_to_size()
def showEvent(self, event):
super(ThumbnailWidget, self).showEvent(event)
self._adapt_to_size()