37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
# Created By: Virgil Dupras
|
|
# Created On: 2011-04-20
|
|
# Copyright 2012 Hardcoded Software (http://www.hardcoded.net)
|
|
#
|
|
# This software is licensed under the "BSD" License as described in the "LICENSE" file,
|
|
# which should be included with this package. The terms are also available at
|
|
# http://www.hardcoded.net/licenses/bsd_license
|
|
|
|
import logging
|
|
from collections import defaultdict
|
|
from itertools import combinations
|
|
|
|
from hscommon import io
|
|
from hscommon.trans import tr
|
|
|
|
from core.engine import Match
|
|
from . import exif
|
|
|
|
def getmatches(files, match_scaled, j):
|
|
timestamp2pic = defaultdict(set)
|
|
for picture in j.iter_with_progress(files, tr("Read EXIF of %d/%d pictures")):
|
|
try:
|
|
with io.open(picture.path, 'rb') as fp:
|
|
exifdata = exif.get_fields(fp)
|
|
timestamp = exifdata['DateTimeOriginal']
|
|
timestamp2pic[timestamp].add(picture)
|
|
except Exception:
|
|
logging.info("Couldn't read EXIF of picture: %s", picture.path)
|
|
if '0000:00:00 00:00:00' in timestamp2pic: # very likely false matches
|
|
del timestamp2pic['0000:00:00 00:00:00']
|
|
matches = []
|
|
for pictures in timestamp2pic.values():
|
|
for p1, p2 in combinations(pictures, 2):
|
|
if (not match_scaled) and (p1.dimensions != p2.dimensions):
|
|
continue
|
|
matches.append(Match(p1, p2, 100))
|
|
return matches |