summaryrefslogtreecommitdiffstats
path: root/core/metadata.cpp
blob: bf4bfcb904e1cf4e10e873305b9d4ae2217acd3e (plain) (blame)
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
// SPDX-License-Identifier: GPL-2.0
#include "metadata.h"
#include "exif.h"
#include "qthelper.h"
#include <QString>
#include <QFile>
#include <QDateTime>

// Fetch quint16 in big endian mode from QFile and return 0 on error.
// This is a very specialized function for parsing JPEGs, therefore we can get away with such an in-band error code.
static inline quint16 getShortBE(QFile &f)
{
	unsigned char buf[2];
	if (f.read(reinterpret_cast<char *>(buf), 2) != 2)
		return 0;
	return (buf[0] << 8) | buf[1];
}

static bool parseExif(QFile &f, struct metadata *metadata)
{
	if (getShortBE(f) != 0xffd8)
		return false;
	for (;;) {
		switch (getShortBE(f)) {
		case 0xffc0:
		case 0xffc2:
		case 0xffc4:
		case 0xffd0 ... 0xffd7:
		case 0xffdb:
		case 0xffdd:
		case 0xffe0:
		case 0xffe2 ... 0xffef:
		case 0xfffe: {
			quint16 len = getShortBE(f);
			if (len < 2)
				return false;
			f.seek(f.pos() + len - 2); // TODO: switch to QFile::skip()
			break;
		}
		case 0xffe1: {
			quint16 len = getShortBE(f);
			if (len < 2)
				return false;
			len -= 2;
			QByteArray data = f.read(len);
			if (data.size() != len)
				return false;
			easyexif::EXIFInfo exif;
			if (exif.parseFromEXIFSegment(reinterpret_cast<const unsigned char *>(data.constData()), len) != PARSE_EXIF_SUCCESS)
				return false;
			metadata->longitude.udeg = lrint(1000000.0 * exif.GeoLocation.Longitude);
			metadata->latitude.udeg = lrint(1000000.0 * exif.GeoLocation.Latitude);
			metadata->timestamp = exif.epoch();
			return true;
		}
		case 0xffda:
		case 0xffd9:
			// We expect EXIF data before any scan data
			return false;
		default:
			return false;
		}
	}
}

static bool parseMP4(QFile &, metadata *)
{
	// TODO: Implement MP4 parsing
	return false;
}

extern "C" mediatype_t get_metadata(const char *filename_in, metadata *data)
{
	data->timestamp = 0;
	data->latitude.udeg = 0;
	data->longitude.udeg = 0;

	QString filename = localFilePath(QString(filename_in));
	QFile f(filename);
	if (!f.open(QIODevice::ReadOnly))
		return MEDIATYPE_IO_ERROR;

	if (parseExif(f, data)) {
		return MEDIATYPE_PICTURE;
	} else if(parseMP4(f, data)) {
		return MEDIATYPE_VIDEO;
	} else {
		// If we couldn't parse EXIF or MP4 data, use file creation date.
		// TODO: QFileInfo::created is deprecated in newer Qt versions.
		data->timestamp = QFileInfo(filename).created().toMSecsSinceEpoch() / 1000;
		return MEDIATYPE_UNKNOWN;
	}
}

extern "C" timestamp_t picture_get_timestamp(const char *filename)
{
	struct metadata data;
	get_metadata(filename, &data);
	return data.timestamp;
}

extern "C" void picture_load_exif_data(struct picture *p)
{
	struct metadata data;
	if (get_metadata(p->filename, &data) == MEDIATYPE_IO_ERROR)
		return;
	p->longitude = data.longitude;
	p->latitude = data.latitude;
}