OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/utility/media_galleries/image_metadata_extractor.h" | |
6 | |
7 #include "base/bind.h" | |
8 #include "base/callback.h" | |
9 #include "base/memory/ref_counted.h" | |
10 #include "base/numerics/safe_conversions.h" | |
11 #include "base/strings/string_number_conversions.h" | |
12 #include "media/base/data_source.h" | |
13 #include "net/base/io_buffer.h" | |
14 | |
15 extern "C" { | |
16 #include <libexif/exif-data.h> | |
17 } // extern "C" | |
18 | |
19 namespace metadata { | |
20 | |
21 namespace { | |
22 | |
23 const int kMaxBufferSize = 50 * 1024 * 1024; // Arbitrary maximum of 50MB. | |
24 | |
25 void FinishGetImageBytes( | |
26 net::DrainableIOBuffer* buffer, | |
27 media::DataSource* source, | |
28 const base::Callback<void(net::DrainableIOBuffer*)>& callback, | |
29 int bytes_read) { | |
30 if (bytes_read == media::DataSource::kReadError) { | |
31 callback.Run(NULL); | |
32 return; | |
33 } | |
34 | |
35 buffer->DidConsume(bytes_read); | |
36 // Didn't get the whole file. Continue reading to get the rest. | |
37 if (buffer->BytesRemaining() > 0) { | |
38 source->Read(0, buffer->BytesRemaining(), (uint8*)buffer->data(), | |
39 base::Bind(&FinishGetImageBytes, make_scoped_refptr(buffer), | |
40 base::Unretained(source), callback)); | |
41 return; | |
42 } | |
43 | |
44 buffer->SetOffset(0); | |
45 callback.Run(make_scoped_refptr(buffer)); | |
46 } | |
47 | |
48 void GetImageBytes( | |
49 media::DataSource* source, | |
50 const base::Callback<void(net::DrainableIOBuffer*)>& callback) { | |
51 int64 size64 = 0; | |
52 if (!source->GetSize(&size64) || size64 > kMaxBufferSize) | |
53 return callback.Run(NULL); | |
54 int size = base::checked_cast<int>(size64); | |
55 | |
56 scoped_refptr<net::DrainableIOBuffer> buffer( | |
57 new net::DrainableIOBuffer(new net::IOBuffer(size), size)); | |
58 source->Read(0, buffer->BytesRemaining(), (uint8*)buffer->data(), | |
59 base::Bind(&FinishGetImageBytes, buffer, | |
60 base::Unretained(source), callback)); | |
61 } | |
62 | |
63 void ExtractInt(ExifData* data, ExifTag tag, int* result) { | |
64 DCHECK(result); | |
65 | |
66 ExifEntry* entry = exif_data_get_entry(data, tag); | |
67 | |
68 if (entry) { | |
69 ExifByteOrder order = exif_data_get_byte_order(data); | |
70 | |
71 switch (entry->format) { | |
72 case EXIF_FORMAT_SHORT: { | |
73 ExifShort v = exif_get_short(entry->data, order); | |
74 *result = base::checked_cast<int>(v); | |
75 break; | |
76 } | |
77 case EXIF_FORMAT_LONG: { | |
78 ExifLong v = exif_get_long(entry->data, order); | |
79 // Ignore values that don't fit in a signed int - probably invalid data. | |
80 if (base::IsValueInRangeForNumericType<int>(v)) | |
81 *result = base::checked_cast<int>(v); | |
82 break; | |
83 } | |
84 default: { | |
85 // Ignore all other entry formats. | |
86 } | |
87 } | |
88 } | |
89 } | |
90 | |
91 void ExtractDouble(ExifData* data, ExifTag tag, double* result) { | |
92 DCHECK(result); | |
93 | |
94 ExifEntry* entry = exif_data_get_entry(data, tag); | |
95 | |
96 if (entry) { | |
97 ExifByteOrder order = exif_data_get_byte_order(data); | |
98 | |
99 if (entry->format == EXIF_FORMAT_RATIONAL) { | |
100 ExifRational v = exif_get_rational(entry->data, order); | |
101 *result = base::checked_cast<double>(v.numerator) / | |
102 base::checked_cast<double>(v.denominator); | |
103 } | |
104 } | |
105 } | |
106 | |
107 void ExtractString(ExifData* data, ExifTag tag, std::string* result) { | |
108 DCHECK(result); | |
109 | |
110 ExifEntry* entry = exif_data_get_entry(data, tag); | |
111 | |
112 if (entry) { | |
113 char buf[1024]; | |
114 exif_entry_get_value(entry, buf, sizeof(buf)); | |
115 *result = buf; | |
116 } | |
117 } | |
118 | |
119 } // namespace | |
120 | |
121 ImageMetadataExtractor::ImageMetadataExtractor() | |
122 : extracted_(false), | |
123 width_(-1), | |
124 height_(-1), | |
125 rotation_(-1), | |
126 x_resolution_(-1), | |
127 y_resolution_(-1), | |
128 exposure_time_sec_(-1), | |
129 flash_fired_(false), | |
130 f_number_(-1), | |
131 focal_length_mm_(-1), | |
132 iso_equivalent_(-1) { | |
133 } | |
134 | |
135 ImageMetadataExtractor::~ImageMetadataExtractor() { | |
136 } | |
137 | |
138 void ImageMetadataExtractor::Extract(media::DataSource* source, | |
139 const DoneCallback& callback) { | |
140 DCHECK(!extracted_); | |
141 | |
142 GetImageBytes(source, base::Bind(&ImageMetadataExtractor::FinishExtraction, | |
143 base::Unretained(this), callback)); | |
144 | |
145 } | |
146 | |
147 int ImageMetadataExtractor::width() const { | |
148 DCHECK(extracted_); | |
149 return width_; | |
150 } | |
151 | |
152 int ImageMetadataExtractor::height() const { | |
153 DCHECK(extracted_); | |
154 return height_; | |
155 } | |
156 | |
157 int ImageMetadataExtractor::rotation() const { | |
158 DCHECK(extracted_); | |
159 return rotation_; | |
160 } | |
161 | |
162 double ImageMetadataExtractor::x_resolution() const { | |
163 DCHECK(extracted_); | |
164 return x_resolution_; | |
165 } | |
166 | |
167 double ImageMetadataExtractor::y_resolution() const { | |
168 DCHECK(extracted_); | |
169 return y_resolution_; | |
170 } | |
171 | |
172 const std::string& ImageMetadataExtractor::date() const { | |
173 DCHECK(extracted_); | |
174 return date_; | |
175 } | |
176 | |
177 const std::string& ImageMetadataExtractor::camera_make() const { | |
178 DCHECK(extracted_); | |
179 return camera_make_; | |
180 } | |
181 | |
182 const std::string& ImageMetadataExtractor::camera_model() const { | |
183 DCHECK(extracted_); | |
184 return camera_model_; | |
185 } | |
186 | |
187 double ImageMetadataExtractor::exposure_time_sec() const { | |
188 DCHECK(extracted_); | |
189 return exposure_time_sec_; | |
190 } | |
191 | |
192 bool ImageMetadataExtractor::flash_fired() const { | |
193 DCHECK(extracted_); | |
194 return flash_fired_; | |
195 } | |
196 | |
197 double ImageMetadataExtractor::f_number() const { | |
198 DCHECK(extracted_); | |
199 return f_number_; | |
200 } | |
201 | |
202 double ImageMetadataExtractor::focal_length_mm() const { | |
203 DCHECK(extracted_); | |
204 return focal_length_mm_; | |
205 } | |
206 | |
207 int ImageMetadataExtractor::iso_equivalent() const { | |
208 DCHECK(extracted_); | |
209 return iso_equivalent_; | |
210 } | |
211 | |
212 void ImageMetadataExtractor::FinishExtraction( | |
213 const DoneCallback& callback, net::DrainableIOBuffer* buffer) { | |
214 if (!buffer) { | |
215 callback.Run(false); | |
216 return; | |
217 } | |
218 | |
219 ExifData* data = exif_data_new_from_data( | |
220 (unsigned char*)buffer->data(), buffer->BytesRemaining()); | |
221 | |
222 if (!data) { | |
223 callback.Run(false); | |
224 return; | |
225 } | |
226 | |
227 ExtractInt(data, EXIF_TAG_IMAGE_WIDTH, &width_); | |
228 ExtractInt(data, EXIF_TAG_IMAGE_LENGTH, &height_); | |
229 | |
230 // We ignore the mirrored-aspect of the mirrored-orientations and just | |
231 // indicate the rotation. Mirrored-orientations are very rare. | |
232 int orientation = 0; | |
233 ExtractInt(data, EXIF_TAG_ORIENTATION, &orientation); | |
234 switch (orientation) { | |
235 case 1: | |
236 case 2: | |
237 rotation_ = 0; | |
238 break; | |
239 case 3: | |
240 case 4: | |
241 rotation_ = 180; | |
242 break; | |
243 case 5: | |
244 case 6: | |
245 rotation_ = 90; | |
246 break; | |
247 case 7: | |
248 case 8: | |
249 rotation_ = 270; | |
250 break; | |
251 } | |
252 | |
253 ExtractDouble(data, EXIF_TAG_X_RESOLUTION, &x_resolution_); | |
254 ExtractDouble(data, EXIF_TAG_Y_RESOLUTION, &y_resolution_); | |
255 | |
256 ExtractString(data, EXIF_TAG_DATE_TIME, &date_); | |
257 | |
258 ExtractString(data, EXIF_TAG_MAKE, &camera_make_); | |
259 ExtractString(data, EXIF_TAG_MODEL, &camera_model_); | |
260 ExtractDouble(data, EXIF_TAG_EXPOSURE_TIME, &exposure_time_sec_); | |
261 | |
262 int flash_value = -1; | |
263 ExtractInt(data, EXIF_TAG_FLASH, &flash_value); | |
264 if (flash_value >= 0) { | |
265 flash_fired_ = (flash_value & 0x1) != 0; | |
266 } | |
267 | |
268 ExtractDouble(data, EXIF_TAG_FNUMBER, &f_number_); | |
269 ExtractDouble(data, EXIF_TAG_FOCAL_LENGTH, &focal_length_mm_); | |
270 ExtractInt(data, EXIF_TAG_ISO_SPEED_RATINGS, &iso_equivalent_); | |
271 | |
272 exif_data_free(data); | |
273 | |
274 extracted_ = true; | |
275 | |
276 callback.Run(true); | |
277 } | |
278 | |
279 } // namespace metadata | |
OLD | NEW |