]> git.sesse.net Git - nageru/blob - futatabi/exif_parser.cpp
Reintroduce faster DeckLink shutdown; now with a fix for the UI switcher.
[nageru] / futatabi / exif_parser.cpp
1 #include "exif_parser.h"
2
3 #include <movit/colorspace_conversion_effect.h>
4 #include <stdint.h>
5 #include <Eigen/Core>
6 #include <Eigen/LU>
7
8 using namespace Eigen;
9 using namespace movit;
10 using namespace std;
11
12 uint32_t read32be(const uint8_t *data)
13 {
14         return (uint32_t(data[0]) << 24) |
15                 (uint32_t(data[1]) << 16) |
16                 (uint32_t(data[2]) <<  8) |
17                  uint32_t(data[3]);
18 }
19
20 uint16_t read16be(const uint8_t *data)
21 {
22         return (uint16_t(data[0]) << 8) | uint16_t(data[1]);
23 }
24
25 RGBTriplet get_neutral_color(const string &exif)
26 {
27         if (exif.empty()) {
28                 return {1.0f, 1.0f, 1.0f};
29         }
30
31         const uint8_t *data = reinterpret_cast<const uint8_t *>(exif.data());
32
33         // Very rudimentary Exif parser (and probably integer-overflowable);
34         // we really only care about what Nageru sends us (MJPEGEncoder::init_jpeg_422()),
35         // but it would be nice to have a little bit of future-proofing, just in case.
36         if (exif.size() < 14 || memcmp(data, "Exif\0\0MM\0\x2a", 10) != 0) {
37                 fprintf(stderr, "WARNING: Truncated or malformed Exif header, ignoring.\n");
38                 return {1.0f, 1.0f, 1.0f};
39         }
40
41         // We only care about the first IFD.
42         uint32_t ifd_offset = read32be(data + 10);
43         ifd_offset += 6;  // Relative to the MM.
44
45         if (ifd_offset < 14 || ifd_offset >= exif.size()) {
46                 fprintf(stderr, "WARNING: Truncated or malformed Exif IFD, ignoring.\n");
47                 return {1.0f, 1.0f, 1.0f};
48         }
49
50         // Skip over number of tags (16 bits); if the white point is not the first one,
51         // we're bailing anyway.
52         if (ifd_offset + 2 > exif.size() || ifd_offset + 2 < ifd_offset) {
53                 fprintf(stderr, "WARNING: Exif IFD has no rom for number of tags, ignoring.\n");
54                 return {1.0f, 1.0f, 1.0f};
55         }
56
57         if (ifd_offset + 4 > exif.size() || ifd_offset + 4 < ifd_offset) {
58                 fprintf(stderr, "WARNING: Exif IFD has no rom for tag, ignoring.\n");
59                 return {1.0f, 1.0f, 1.0f};
60         }
61         uint16_t tag = read16be(data + ifd_offset + 2);
62         if (tag != 0x13e) {  // WhitePoint.
63                 fprintf(stderr, "WARNING: Unexpected first Exif tag, ignoring.\n");
64                 return {1.0f, 1.0f, 1.0f};
65         }
66
67         if (ifd_offset + 14 > exif.size() || ifd_offset + 14 < ifd_offset) {
68                 fprintf(stderr, "WARNING: WhitePoint Exif tag was truncated, ignoring.\n");
69                 return {1.0f, 1.0f, 1.0f};
70         }
71
72         // Just assume we're rational type and two values...
73         uint32_t white_point_offset = read32be(data + ifd_offset + 10);
74         white_point_offset += 6;  // Relative to the MM.
75
76         if (white_point_offset >= exif.size()) {
77                 fprintf(stderr, "WARNING: WhitePoint Exif tag was out of bounds, ignoring.\n");
78                 return {1.0f, 1.0f, 1.0f};
79         }
80         if (white_point_offset + 16 > exif.size()) {
81                 fprintf(stderr, "WARNING: WhitePoint Exif tag was truncated, ignoring.\n");
82                 return {1.0f, 1.0f, 1.0f};
83         }
84
85         uint32_t x_nom = read32be(data + white_point_offset);
86         uint32_t x_den = read32be(data + white_point_offset + 4);
87         uint32_t y_nom = read32be(data + white_point_offset + 8);
88         uint32_t y_den = read32be(data + white_point_offset + 12);
89
90         double x = double(x_nom) / x_den;
91         double y = double(y_nom) / y_den;
92         double z = 1.0 - x - y;
93
94         Matrix3d rgb_to_xyz_matrix = movit::ColorspaceConversionEffect::get_xyz_matrix(COLORSPACE_sRGB);
95         Vector3d rgb = rgb_to_xyz_matrix.inverse() * Vector3d(x, y, z);
96
97         return RGBTriplet(rgb[0], rgb[1], rgb[2]);
98 }
99