1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
|
#!/usr/bin/env python3
# Copyright (C) 2015-2016 Damon Lynch <damonlynch@gmail.com>
# This file is part of Rapid Photo Downloader.
#
# Rapid Photo Downloader is free software: you can redistribute it and/or
# modify it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Rapid Photo Downloader is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Rapid Photo Downloader. If not,
# see <http://www.gnu.org/licenses/>.
__author__ = 'Damon Lynch'
__copyright__ = "Copyright 2015-2016, Damon Lynch"
import sys
import logging
from urllib.request import pathname2url
import pickle
import os
from collections import namedtuple
import tempfile
from datetime import datetime
from typing import Optional, Set, Union, Tuple
import gi
gi.require_version('Gst', '1.0')
from gi.repository import Gst
from PyQt5.QtGui import QImage, QTransform
from PyQt5.QtCore import QSize, Qt, QIODevice, QBuffer
try:
import rawkit
import rawkit.options
import rawkit.raw
have_rawkit = True
except ImportError:
have_rawkit = False
from raphodo.interprocess import (LoadBalancerWorker, ThumbnailExtractorArgument,
GenerateThumbnailsResults)
from raphodo.constants import (ThumbnailSize, ExtractionTask, ExtractionProcessing,
ThumbnailCacheStatus, ThumbnailCacheDiskStatus)
from raphodo.rpdfile import RPDFile, Video, Photo, FileType
from raphodo.utilities import stdchannel_redirected, show_errors
from raphodo.filmstrip import add_filmstrip
from raphodo.cache import ThumbnailCacheSql, FdoCacheLarge, FdoCacheNormal
import raphodo.exiftool as exiftool
import atexit
have_gst = Gst.init_check(None)
def gst_version() -> str:
"""
:return: version of gstreamer, if it exists and is functioning, else ''
"""
if have_gst:
try:
return Gst.version_string().replace('GStreamer ', '')
except Exception:
pass
return ''
def libraw_version(suppress_errors: bool=True) -> str:
"""
Return version number of libraw, using rawkit
:param suppress_errors:
:return: version number if available, else ''
"""
if not have_rawkit:
return ''
import libraw.bindings
try:
return libraw.bindings.LibRaw().version
except ImportError as e:
if not suppress_errors:
raise
v = str(e)
if v.startswith('Unsupported'):
import re
v = ''.join(re.findall(r'\d+\.?', str(e)))
return v[:-1] if v.endswith('.') else v
return v
except Exception:
if not suppress_errors:
raise
return ''
if not have_rawkit:
have_functioning_rawkit = False
else:
try:
have_functioning_rawkit = bool(libraw_version(suppress_errors=False))
except Exception:
have_functioning_rawkit = False
def rawkit_version() -> str:
if have_rawkit:
if have_functioning_rawkit:
return rawkit.VERSION
else:
return '{} (not functional)'.format(rawkit.VERSION)
return ''
def get_video_frame(full_file_name: str,
offset: Optional[float]=5.0,
caps=Gst.Caps.from_string('image/png')) -> Optional[bytes]:
"""
Source: https://gist.github.com/dplanella/5563018
:param full_file_name: file and path of the video
:param offset:
:param caps:
:return: gstreamer buffer
"""
logging.debug("Using gstreamer to generate thumbnail from %s", full_file_name)
pipeline = Gst.parse_launch('playbin')
pipeline.props.uri = 'file://{}'.format(pathname2url(os.path.abspath(full_file_name)))
pipeline.props.audio_sink = Gst.ElementFactory.make('fakesink', 'fakeaudio')
pipeline.props.video_sink = Gst.ElementFactory.make('fakesink', 'fakevideo')
pipeline.set_state(Gst.State.PAUSED)
# Wait for state change to finish.
pipeline.get_state(Gst.CLOCK_TIME_NONE)
# Seek offset seconds into the video, if the video is long enough
# If video is shorter than offset, seek 0.25 seconds less than the duration,
# but always at least zero.
offset = max(
min(
pipeline.query_duration(Gst.Format.TIME)[1] - Gst.SECOND / 4, offset * Gst.SECOND
), 0
)
try:
assert pipeline.seek_simple(Gst.Format.TIME, Gst.SeekFlags.FLUSH, offset)
except AssertionError:
return None
# Wait for seek to finish.
pipeline.get_state(Gst.CLOCK_TIME_NONE)
sample = pipeline.emit('convert-sample', caps)
if sample is not None:
buffer = sample.get_buffer()
pipeline.set_state(Gst.State.NULL)
return buffer.extract_dup(0, buffer.get_size())
else:
return None
PhotoDetails = namedtuple('PhotoDetails', 'thumbnail, orientation')
def qimage_to_png_buffer(image: QImage) -> QBuffer:
"""
Save the image data in PNG format in a QBuffer, whose data can then
be extracted using the data() member function.
:param image: the image to be converted
:return: the buffer
"""
buffer = QBuffer()
buffer.open(QIODevice.WriteOnly)
# Quality 100 means uncompressed.
image.save(buffer, "PNG", quality=100)
return buffer
def crop_160x120_thumbnail(thumbnail: QImage, vertical_space: int=8) -> QImage:
"""
Remove black bands from the top and bottom of thumbnail
:param thumbnail: thumbnail to crop
:param vertical_space: how much to remove from the top and bottom
:return: cropped thumbnail
"""
return thumbnail.copy(0, vertical_space, 160, 120 - vertical_space * 2)
class ThumbnailExtractor(LoadBalancerWorker):
# Exif rotation constants
rotate_90 = '6'
rotate_180 = '3'
rotate_270 = '8'
maxStandardSize = QSize(max(ThumbnailSize.width, ThumbnailSize.height),
max(ThumbnailSize.width, ThumbnailSize.height))
def __init__(self) -> None:
self.thumbnailSizeNeeded = QSize(ThumbnailSize.width, ThumbnailSize.height)
self.thumbnail_cache = ThumbnailCacheSql()
self.fdo_cache_large = FdoCacheLarge()
self.fdo_cache_normal = FdoCacheNormal()
super().__init__('Thumbnail Extractor')
def rotate_thumb(self, thumbnail: QImage, orientation: str) -> QImage:
"""
If required return a rotated copy the thumbnail
:param thumbnail: thumbnail to rotate
:param orientation: EXIF orientation tag
:return: possibly rotated thumbnail
"""
if orientation == self.rotate_90:
thumbnail = thumbnail.transformed(QTransform().rotate(90))
elif orientation == self.rotate_270:
thumbnail = thumbnail.transformed(QTransform().rotate(270))
elif orientation == self.rotate_180:
thumbnail = thumbnail.transformed(QTransform().rotate(180))
return thumbnail
def image_large_enough(self, size: QSize) -> bool:
"""Check if image is equal or bigger than thumbnail size."""
return (size.width() >= self.thumbnailSizeNeeded.width() or
size.height() >= self.thumbnailSizeNeeded.height())
def image_large_enough_fdo(self, size: QSize) -> bool:
return size.width() >= 256 or size.height() >= 256
def _extract_256_thumb(self, rpd_file: RPDFile,
processing: Set[ExtractionProcessing],
orientation: Optional[str]) -> PhotoDetails:
thumbnail = None
previews = rpd_file.metadata.get_preview_properties()
if previews:
for preview in previews:
if self.image_large_enough_fdo(QSize(preview.get_width(), preview.get_height())) \
and preview.get_mime_type() == 'image/jpeg':
break
# At this point we have a preview that may or may not be bigger than 160x120.
# On older RAW files, no. On newer RAW files, yes.
data = rpd_file.metadata.get_preview_image(preview).get_data()
if isinstance(data, bytes):
thumbnail = QImage.fromData(data)
if thumbnail.isNull():
thumbnail = None
else:
if thumbnail.width() > 160 or thumbnail.height() > 120:
processing.add(ExtractionProcessing.resize)
return PhotoDetails(thumbnail, orientation)
def _extract_metadata(self, rpd_file: RPDFile,
processing: Set[ExtractionProcessing]) -> PhotoDetails:
thumbnail = orientation = None
try:
orientation = rpd_file.metadata.get_tag_string('Exif.Image.Orientation')
except KeyError:
pass
rpd_file.mdatatime = rpd_file.metadata.timestamp(missing=0.0)
# Not all files have an exif preview, but some do
# (typically CR2, ARW, PEF, RW2).
# If they exist, they are (almost!) always 160x120
# TODO how about thumbnail_cache_status?
if self.write_fdo_thumbnail and rpd_file.fdo_thumbnail_256 is None:
photo_details = self._extract_256_thumb(rpd_file=rpd_file, processing=processing,
orientation=orientation)
if photo_details.thumbnail is not None:
return photo_details
# if no valid preview found, fall back to the code below and make do with the best
# we can get
ep = rpd_file.metadata.get_exif_thumbnail()
if ep:
thumbnail = QImage.fromData(rpd_file.metadata.get_exif_thumbnail())
if thumbnail.isNull():
thumbnail = None
elif thumbnail.width() == 120 and thumbnail.height() == 160:
# The Samsung Pro815 can store its thumbnails this way!
# Perhaps some other obscure cameras also do this too.
# The orientation has already been applied to the thumbnail
orientation = '1'
elif thumbnail.width() > 160 or thumbnail.height() > 120:
processing.add(ExtractionProcessing.resize)
elif not rpd_file.is_jpeg():
processing.add(ExtractionProcessing.strip_bars_photo)
else:
previews = rpd_file.metadata.get_preview_properties()
if previews:
# In every RAW file I've analyzed, the smallest preview is always first
preview = previews[0]
data = rpd_file.metadata.get_preview_image(preview).get_data()
if isinstance(data, bytes):
thumbnail = QImage.fromData(data)
if thumbnail.isNull():
thumbnail = None
else:
if thumbnail.width() > 160 or thumbnail.height() > 120:
processing.add(ExtractionProcessing.resize)
if not rpd_file.is_jpeg():
processing.add(ExtractionProcessing.strip_bars_photo)
return PhotoDetails(thumbnail, orientation)
def get_disk_photo_thumb(self, rpd_file: Photo,
full_file_name: str,
processing: Set[ExtractionProcessing]) -> PhotoDetails:
"""
Get the photo's thumbnail from a file that is on disk.
Sets rpd_file's mdatatime.
:param rpd_file: file details
:param full_file_name: full name of the file from which to get the metadata
:param processing: processing extraction tasks to complete
:return: thumbnail and its orientation
"""
orientation = None
thumbnail = None
photo_details = PhotoDetails(thumbnail, orientation)
if rpd_file.load_metadata(full_file_name=full_file_name, et_process=self.exiftool_process):
photo_details = self._extract_metadata(rpd_file, processing)
thumbnail = photo_details.thumbnail
if thumbnail is not None:
return photo_details
elif rpd_file.is_raw() and have_functioning_rawkit:
try:
with rawkit.raw.Raw(filename=full_file_name) as raw:
raw.options.white_balance = rawkit.options.WhiteBalance(camera=True, auto=False)
if rpd_file.cache_full_file_name and not rpd_file.download_full_file_name:
temp_file = '{}.tiff'.format(os.path.splitext(full_file_name)[0])
cache_dir = os.path.dirname(rpd_file.cache_full_file_name)
if os.path.isdir(cache_dir):
temp_file = os.path.join(cache_dir, temp_file)
temp_dir = None
else:
temp_dir = tempfile.mkdtemp(prefix="rpd-tmp-")
temp_file = os.path.join(temp_dir, temp_file)
else:
temp_dir = tempfile.mkdtemp(prefix="rpd-tmp-")
name = os.path.basename(full_file_name)
temp_file = '{}.tiff'.format(os.path.splitext(name)[0])
temp_file = os.path.join(temp_dir, temp_file)
try:
logging.debug("Saving temporary rawkit render to %s", temp_file)
raw.save(filename=temp_file)
except Exception:
logging.exception("Rendering %s failed. Exception:",
rpd_file.full_file_name)
else:
thumbnail = QImage(temp_file)
os.remove(temp_file)
if thumbnail.isNull():
logging.debug("Qt failed to load rendered %s", rpd_file.full_file_name)
thumbnail = None
else:
logging.debug("Rendered %s using libraw", rpd_file.full_file_name)
processing.add(ExtractionProcessing.resize)
# libraw already correctly oriented the thumbnail
processing.remove(ExtractionProcessing.orient)
orientation = '1'
if temp_dir:
os.rmdir(temp_dir)
except ImportError as e:
logging.warning('Cannot use rawkit to render thumbnail for %s',
rpd_file.full_file_name)
except Exception as e:
logging.exception("Rendering thumbnail for %s not supported. Exception:",
rpd_file.full_file_name)
if thumbnail is None and rpd_file.is_loadable():
thumbnail = QImage(full_file_name)
processing.add(ExtractionProcessing.resize)
if not rpd_file.from_camera:
processing.remove(ExtractionProcessing.orient)
if thumbnail.isNull():
thumbnail = None
logging.warning(
"Unable to create a thumbnail out of the file: {}".format(full_file_name))
return PhotoDetails(thumbnail, orientation)
def get_from_buffer(self, rpd_file: Photo,
raw_bytes: bytearray,
processing: Set[ExtractionProcessing]) -> PhotoDetails:
if not rpd_file.load_metadata(raw_bytes=raw_bytes, et_process=self.exiftool_process):
# logging.warning("Extractor failed to load metadata from extract of %s", rpd_file.name)
return PhotoDetails(None, None)
else:
return self._extract_metadata(rpd_file, processing)
def get_photo_orientation(self, rpd_file: Photo,
full_file_name: Optional[str]=None,
raw_bytes: Optional[bytearray]=None) -> Optional[str]:
if rpd_file.metadata is None:
self.load_photo_metadata(rpd_file=rpd_file, full_file_name=full_file_name,
raw_bytes=raw_bytes)
if rpd_file.metadata is not None:
try:
return rpd_file.metadata.get_tag_string('Exif.Image.Orientation')
except KeyError:
pass
return None
def assign_mdatatime(self, rpd_file: Union[Photo, Video],
full_file_name: Optional[str]=None,
raw_bytes: Optional[bytearray]=None) -> None:
"""
Load the file's metadata and assign the metadata time to the rpd file
"""
if rpd_file.file_type == FileType.photo:
self.assign_photo_mdatatime(rpd_file=rpd_file, full_file_name=full_file_name,
raw_bytes=raw_bytes)
else:
self.assign_video_mdatatime(rpd_file=rpd_file, full_file_name=full_file_name)
def assign_photo_mdatatime(self, rpd_file: Photo,
full_file_name: Optional[str]=None,
raw_bytes: Optional[bytearray]=None) -> None:
"""
Load the photo's metadata and assign the metadata time to the rpd file
"""
self.load_photo_metadata(rpd_file=rpd_file, full_file_name=full_file_name,
raw_bytes=raw_bytes)
if rpd_file.metadata is not None and rpd_file.date_time() is None:
rpd_file.mdatatime = 0.0
def load_photo_metadata(self, rpd_file: Photo,
full_file_name: Optional[str]=None,
raw_bytes: Optional[bytearray]=None) -> None:
"""
Load the photo's metadata into the rpd file
"""
if raw_bytes is not None:
if rpd_file.is_jpeg_type():
rpd_file.load_metadata(app1_segment=raw_bytes,
et_process=self.exiftool_process)
else:
rpd_file.load_metadata(raw_bytes=raw_bytes,
et_process=self.exiftool_process)
else:
rpd_file.load_metadata(full_file_name=full_file_name,
et_process=self.exiftool_process)
def assign_video_mdatatime(self, rpd_file: Video, full_file_name: str) -> None:
"""
Load the video's metadata and assign the metadata time to the rpd file
"""
if rpd_file.metadata is None:
rpd_file.load_metadata(full_file_name=full_file_name, et_process=self.exiftool_process)
if rpd_file.date_time() is None:
rpd_file.mdatatime = 0.0
def get_video_rotation(self, rpd_file: Video, full_file_name: str) -> Optional[str]:
"""
Some videos have a rotation tag. If this video does, return it.
"""
if rpd_file.metadata is None:
rpd_file.load_metadata(full_file_name=full_file_name, et_process=self.exiftool_process)
orientation = rpd_file.metadata.rotation(missing=None)
if orientation == 180:
return self.rotate_180
elif orientation == 90:
return self.rotate_90
elif orientation == 270:
return self.rotate_270
return None
def check_for_stop(self, directive: bytes, content: bytes):
if directive == b'cmd':
assert content == b'STOP'
return True
return False
def extact_thumbnail(self, task: ExtractionTask,
rpd_file: Union[Photo, Video],
processing: Set[ExtractionProcessing],
data: ThumbnailExtractorArgument
) -> Tuple[Optional[QImage], Optional[str]]:
"""
Extract the thumbnail using one of a variety of methods,
depending on the file
:param task: extraction task to perform
:param rpd_file: rpd_file to work on
:param processing: processing tasks
:param data: some other processing arguments passed to this process
:return: thumbnail and its orientation, if found
"""
orientation = None
if task == ExtractionTask.load_from_exif:
thumbnail_details = self.get_disk_photo_thumb(
rpd_file, data.full_file_name_to_work_on, processing)
thumbnail = thumbnail_details.thumbnail
if thumbnail is not None:
orientation = thumbnail_details.orientation
elif task in (ExtractionTask.load_file_directly,
ExtractionTask.load_file_and_exif_directly,
ExtractionTask.load_file_directly_metadata_from_secondary):
thumbnail = QImage(data.full_file_name_to_work_on)
if task == ExtractionTask.load_file_and_exif_directly:
self.assign_photo_mdatatime(rpd_file=rpd_file,
full_file_name=data.full_file_name_to_work_on)
elif task == ExtractionTask.load_file_directly_metadata_from_secondary:
self.assign_mdatatime(rpd_file=rpd_file,
full_file_name=data.secondary_full_file_name)
if ExtractionProcessing.orient in processing:
orientation = self.get_photo_orientation(rpd_file=rpd_file,
full_file_name=data.full_file_name_to_work_on)
elif task in (ExtractionTask.load_from_bytes,
ExtractionTask.load_from_bytes_metadata_from_temp_extract):
thumbnail = QImage.fromData(data.thumbnail_bytes)
if thumbnail.width() > self.thumbnailSizeNeeded.width() or thumbnail.height()\
> self.thumbnailSizeNeeded.height():
processing.add(ExtractionProcessing.resize)
processing.remove(ExtractionProcessing.strip_bars_photo)
if data.exif_buffer and ExtractionProcessing.orient in processing:
orientation = self.get_photo_orientation(rpd_file=rpd_file,
raw_bytes=data.exif_buffer)
if task == ExtractionTask.load_from_bytes_metadata_from_temp_extract:
self.assign_mdatatime(rpd_file=rpd_file,
full_file_name=data.secondary_full_file_name)
os.remove(data.secondary_full_file_name)
rpd_file.temp_cache_full_file_chunk = ''
elif task == ExtractionTask.load_from_exif_buffer:
thumbnail_details = self.get_from_buffer(rpd_file, data.exif_buffer, processing)
thumbnail = thumbnail_details.thumbnail
if thumbnail is not None:
orientation = thumbnail_details.orientation
else:
assert task in (ExtractionTask.extract_from_file,
ExtractionTask.extract_from_file_and_load_metadata)
assert rpd_file.file_type == FileType.video
if ExtractionTask.extract_from_file_and_load_metadata:
self.assign_video_mdatatime(
rpd_file=rpd_file, full_file_name=data.full_file_name_to_work_on
)
if not have_gst:
thumbnail = None
else:
png = get_video_frame(data.full_file_name_to_work_on, 0.0)
if not png:
thumbnail = None
logging.warning(
"Could not extract video thumbnail from %s",
data.rpd_file.get_display_full_name()
)
else:
thumbnail = QImage.fromData(png)
if thumbnail.isNull():
thumbnail = None
else:
processing.add(ExtractionProcessing.add_film_strip)
orientation = self.get_video_rotation(rpd_file,
data.full_file_name_to_work_on)
if orientation is not None:
processing.add(ExtractionProcessing.orient)
processing.add(ExtractionProcessing.resize)
return thumbnail, orientation
def process_files(self):
"""
Loop continuously processing photo and video thumbnails
"""
logging.debug("{} worker started".format(self.requester.identity.decode()))
while True:
directive, content = self.requester.recv_multipart()
if self.check_for_stop(directive, content):
break
data = pickle.loads(content) # type: ThumbnailExtractorArgument
thumbnail_256 = png_data = None
task = data.task
processing = data.processing
rpd_file = data.rpd_file
logging.debug("Working on task %s for %s", task.name, rpd_file.download_name or
rpd_file.name)
self.write_fdo_thumbnail = data.write_fdo_thumbnail
try:
if rpd_file.fdo_thumbnail_256 is not None and data.write_fdo_thumbnail:
if rpd_file.thumbnail_status != ThumbnailCacheStatus.fdo_256_ready:
logging.error("Unexpected thumbnail cache status for %s: %s",
rpd_file.full_file_name, rpd_file.thumbnail_status.name)
thumbnail = thumbnail_256 = QImage.fromData(rpd_file.fdo_thumbnail_256)
orientation_unknown = False
else:
thumbnail, orientation = self.extact_thumbnail(task, rpd_file, processing, data)
if data.file_to_work_on_is_temporary:
os.remove(data.full_file_name_to_work_on)
rpd_file.temp_cache_full_file_chunk = ''
if thumbnail is not None:
if ExtractionProcessing.strip_bars_photo in processing:
thumbnail = crop_160x120_thumbnail(thumbnail)
elif ExtractionProcessing.strip_bars_video in processing:
thumbnail = crop_160x120_thumbnail(thumbnail, 15)
elif ExtractionProcessing.resize in processing:
# Resize the thumbnail before rotating
if ((orientation == '1' or orientation is None) and
thumbnail.height() > thumbnail.width()):
# Special case: pictures from some cellphones have already
# been rotated
thumbnail = thumbnail.scaled(
self.maxStandardSize,
Qt.KeepAspectRatio,
Qt.SmoothTransformation)
else:
if rpd_file.should_write_fdo() and \
self.image_large_enough_fdo(thumbnail.size()) \
and max(thumbnail.height(), thumbnail.width()) > 256:
thumbnail_256 = thumbnail.scaled(
QSize(256, 256),
Qt.KeepAspectRatio,
Qt.SmoothTransformation)
thumbnail = thumbnail_256
if data.send_thumb_to_main:
thumbnail = thumbnail.scaled(
self.thumbnailSizeNeeded,
Qt.KeepAspectRatio,
Qt.SmoothTransformation)
else:
thumbnail = None
if not thumbnail is None and thumbnail.isNull():
thumbnail = None
if orientation is not None:
if thumbnail is not None:
thumbnail = self.rotate_thumb(thumbnail, orientation)
if thumbnail_256 is not None:
thumbnail_256 = self.rotate_thumb(thumbnail_256, orientation)
if ExtractionProcessing.add_film_strip in processing:
if thumbnail is not None:
thumbnail = add_filmstrip(thumbnail)
if thumbnail_256 is not None:
thumbnail = add_filmstrip(thumbnail_256)
if thumbnail is not None:
buffer = qimage_to_png_buffer(thumbnail)
png_data = buffer.data()
orientation_unknown = (ExtractionProcessing.orient in processing and
orientation is None)
if data.send_thumb_to_main and data.use_thumbnail_cache and \
rpd_file.thumbnail_cache_status == ThumbnailCacheDiskStatus.not_found:
self.thumbnail_cache.save_thumbnail(
full_file_name=rpd_file.full_file_name,
size=rpd_file.size,
mtime=rpd_file.modification_time,
mdatatime=rpd_file.mdatatime,
generation_failed=thumbnail is None,
orientation_unknown=orientation_unknown,
thumbnail=thumbnail,
camera_model=rpd_file.camera_model)
if (thumbnail is not None or thumbnail_256 is not None) and \
rpd_file.should_write_fdo():
if self.write_fdo_thumbnail:
# The modification time of the file may have changed when the file was saved
# Ideally it shouldn't, but it does sometimes, e.g. on NTFS!
# So need to get the modification time from the saved file.
mtime = os.path.getmtime(rpd_file.download_full_file_name)
if thumbnail_256 is not None:
rpd_file.fdo_thumbnail_256_name = self.fdo_cache_large.save_thumbnail(
full_file_name=rpd_file.download_full_file_name,
size=rpd_file.size,
modification_time=mtime,
generation_failed=False,
thumbnail=thumbnail_256,
free_desktop_org=False)
thumbnail_128 = thumbnail_256.scaled(
QSize(128, 128),
Qt.KeepAspectRatio,
Qt.SmoothTransformation)
else:
thumbnail_128 = thumbnail.scaled(
QSize(128, 128),
Qt.KeepAspectRatio,
Qt.SmoothTransformation)
rpd_file.fdo_thumbnail_128_name = self.fdo_cache_normal.save_thumbnail(
full_file_name=rpd_file.download_full_file_name,
size=rpd_file.size,
modification_time=mtime,
generation_failed=False,
thumbnail=thumbnail_128,
free_desktop_org=False)
elif thumbnail_256 is not None and rpd_file.fdo_thumbnail_256 is None:
rpd_file.fdo_thumbnail_256 = qimage_to_png_buffer(thumbnail).data()
if thumbnail is not None:
if orientation_unknown:
rpd_file.thumbnail_status = ThumbnailCacheStatus.orientation_unknown
elif rpd_file.fdo_thumbnail_256 is not None:
rpd_file.thumbnail_status = ThumbnailCacheStatus.fdo_256_ready
else:
rpd_file.thumbnail_status = ThumbnailCacheStatus.ready
except SystemExit as e:
self.exiftool_process.terminate()
sys.exit(e)
except:
logging.error("Exception working on file %s", rpd_file.full_file_name)
logging.error("Task: %s", task)
logging.error("Processing tasks: %s", processing)
logging.exception("Traceback:")
# Purge metadata, as it cannot be pickled
if not data.send_thumb_to_main:
png_data = None
rpd_file.metadata = None
self.sender.send_multipart([b'0', b'data',
pickle.dumps(
GenerateThumbnailsResults(
rpd_file=rpd_file,
thumbnail_bytes=png_data),
pickle.HIGHEST_PROTOCOL)])
self.requester.send_multipart([b'', b'', b'OK'])
def do_work(self):
if False:
# exiv2 pumps out a LOT to stderr - use cautiously!
context = show_errors()
self.error_stream = sys.stderr
else:
# Redirect stderr, hiding error output from exiv2
context = stdchannel_redirected(sys.stderr, os.devnull)
self.error_stream = sys.stdout
with context:
# In some situations, using a context manager for exiftool can
# result in exiftool processes not being terminated. So let's
# handle starting and terminating it manually.
self.exiftool_process = exiftool.ExifTool()
self.exiftool_process.start()
self.process_files()
self.exit()
def cleanup_pre_stop(self) -> None:
logging.debug("Terminating thumbnail extractor ExifTool process for %s",
self.identity.decode())
self.exiftool_process.terminate()
if __name__ == "__main__":
thumbnail_extractor = ThumbnailExtractor()
|