calvin.erfmann/TagStudio
1
0
Fork 0
mirror of https://github.com/TagStudioDev/TagStudio.git synced 2026-01-31 23:29:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add rough check for invalid video codecs

Browse Source
This commit is contained in:
Travis Abendshien
2024-06-08 10:47:19 -07:00
parent 3144440365
commit d339f868a9
1 changed files with 20 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
tagstudio/src/qt/widgets/thumb_renderer.py
View File

@@ -5,6 +5,7 @@
import logging
import math
import sys
import cv2
import rawpy
import numpy
@@ -171,11 +172,17 @@ class ThumbRenderer(QObject):
# Videos =======================================================
elif _filepath.suffix.lower() in VIDEO_TYPES:
video = cv2.VideoCapture(str(_filepath))
frame_count = video.get(cv2.CAP_PROP_FRAME_COUNT)
if frame_count <= 0:
raise cv2.error("File is invalid or has 0 frames")
video.set(cv2.CAP_PROP_POS_FRAMES, frame_count // 2)
video = cv2.VideoCapture(str(_filepath), cv2.CAP_FFMPEG)
# Stupid check to try and tell if the codec can be read.
# TODO: Find a way to intercept the native FFMPEG errors.
h = int(video.get(cv2.CAP_PROP_FOURCC))
codec = h.to_bytes(4, byteorder=sys.byteorder).decode()
logging.info(f"{codec} - {h} - {video.getBackendName()}")
if h != 22:
video.set(
cv2.CAP_PROP_POS_FRAMES,
(video.get(cv2.CAP_PROP_FRAME_COUNT) // 2),
)
success, frame = video.read()
if not success:
# Depending on the video format, compression, and frame
@@ -183,8 +190,14 @@ class ThumbRenderer(QObject):
# must be pulled from the earliest available frame.
video.set(cv2.CAP_PROP_POS_FRAMES, 0)
success, frame = video.read()
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
image = Image.fromarray(frame)
if not success:
# Depending on the video format, compression, and frame
# count, seeking halfway does not work and the thumb
# must be pulled from the earliest available frame.
video.set(cv2.CAP_PROP_POS_FRAMES, 0)
success, frame = video.read()
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
image = Image.fromarray(frame)
# Plain Text ===================================================
elif _filepath.suffix.lower() in PLAINTEXT_TYPES: