exiv2-utils.cpp 24.8 KB
Newer Older
Paolo Bacchilega's avatar
Paolo Bacchilega committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
/* -*- Mode: CPP; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */

/*
 *  GThumb
 *
 *  Copyright (C) 2008-2009 Free Software Foundation, Inc.
 *
 *  This program 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 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program 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 this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Street #330, Boston, MA 02111-1307, USA.
 */

#include <config.h>
24
#include <glib.h>
25 26
#define GDK_PIXBUF_ENABLE_BACKEND
#include <gdk-pixbuf/gdk-pixbuf-io.h>
Paolo Bacchilega's avatar
Paolo Bacchilega committed
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
#include <exiv2/basicio.hpp>
#include <exiv2/error.hpp>
#include <exiv2/image.hpp>
#include <exiv2/exif.hpp>
#include <iostream>
#include <string>
#include <sstream>
#include <vector>
#include <exiv2/xmp.hpp>
#include <gthumb.h>
#include "exiv2-utils.h"

using namespace std;


/* Some bits of information may be contained in more than one metadata tag.
   The arrays below define the valid tags for a particular piece of
   information, in decreasing order of preference (best one first) */

const char *_DATE_TAG_NAMES[] = {
	"Exif::Image::DateTime",
	"Xmp::exif::DateTime",
	"Exif::Photo::DateTimeOriginal",
	"Xmp::exif::DateTimeOriginal",
	"Exif::Photo::DateTimeDigitized",
	"Xmp::exif::DateTimeDigitized",
	"Xmp::xmp::CreateDate",
	"Xmp::photoshop::DateCreated",
	"Xmp::xmp::ModifyDate",
	"Xmp::xmp::MetadataDate",
	NULL
};

60 61 62 63 64 65 66 67 68 69
const char *_ORIGINAL_DATE_TAG_NAMES[] = {
	"Exif::Photo::DateTimeOriginal",
	"Xmp::exif::DateTimeOriginal",
	"Exif::Photo::DateTimeDigitized",
	"Xmp::exif::DateTimeDigitized",
	"Xmp::xmp::CreateDate",
	"Xmp::photoshop::DateCreated",
	NULL
};

Paolo Bacchilega's avatar
Paolo Bacchilega committed
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
const char *_EXPOSURE_TIME_TAG_NAMES[] = {
	"Exif::Photo::ExposureTime",
	"Xmp::exif::ExposureTime",
	"Exif::Photo::ShutterSpeedValue",
	"Xmp::exif::ShutterSpeedValue",
	NULL
};

const char *_EXPOSURE_MODE_TAG_NAMES[] = {
	"Exif::Photo::ExposureMode",
	"Xmp::exif::ExposureMode",
	NULL
};

const char *_ISOSPEED_TAG_NAMES[] = {
	"Exif::Photo::ISOSpeedRatings",
	"Xmp::exif::ISOSpeedRatings",
	NULL
};

const char *_APERTURE_TAG_NAMES[] = {
	"Exif::Photo::ApertureValue",
	"Xmp::exif::ApertureValue",
	"Exif::Photo::FNumber",
	"Xmp::exif::FNumber",
	NULL
};

const char *_FOCAL_LENGTH_TAG_NAMES[] = {
	"Exif::Photo::FocalLength",
	"Xmp::exif::FocalLength",
	NULL
};

const char *_SHUTTER_SPEED_TAG_NAMES[] = {
	"Exif::Photo::ShutterSpeedValue",
	"Xmp::exif::ShutterSpeedValue",
	NULL
};

const char *_MAKE_TAG_NAMES[] = {
	"Exif::Image::Make",
	"Xmp::tiff::Make",
	NULL
};

const char *_MODEL_TAG_NAMES[] = {
	"Exif::Image::Model",
	"Xmp::tiff::Model",
	NULL
};

const char *_FLASH_TAG_NAMES[] = {
	"Exif::Photo::Flash",
	"Xmp::exif::Flash",
	NULL
};

const char *_ORIENTATION_TAG_NAMES[] = {
	"Exif::Image::Orientation",
	"Xmp::tiff::Orientation",
	NULL
};

134
const char *_DESCRIPTION_TAG_NAMES[] = {
Paolo Bacchilega's avatar
Paolo Bacchilega committed
135 136 137
	"Exif::Photo::UserComment",
	"Exif::Image::ImageDescription",
	"Xmp::dc::description",
138
	"Xmp::tiff::ImageDescription",
Paolo Bacchilega's avatar
Paolo Bacchilega committed
139 140 141 142 143
	"Iptc::Application2::Caption",
	"Iptc::Application2::Headline",
	NULL
};

144 145 146 147 148
const char *_TITLE_TAG_NAMES[] = {
	"Xmp::dc::title",
	NULL
};

Paolo Bacchilega's avatar
Paolo Bacchilega committed
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
const char *_LOCATION_TAG_NAMES[] = {
	"Xmp::iptc::Location",
	"Iptc::Application2::LocationName",
	NULL
};

const char *_KEYWORDS_TAG_NAMES[] = {
	"Xmp::dc::subject",
	"Xmp::iptc::Keywords",
	"Iptc::Application2::Keywords",
	NULL
};


inline static char *
exiv2_key_from_attribute (const char *attribute)
{
	return _g_replace (attribute, "::", ".");
}


inline static char *
exiv2_key_to_attribute (const char *key)
{
	return _g_replace (key, ".", "::");
}


177 178 179 180 181 182 183 184 185 186 187 188 189 190
static gboolean
attribute_is_date (const char *key)
{
	int i;

	for (i = 0; _DATE_TAG_NAMES[i] != NULL; i++) {
		if (strcmp (_DATE_TAG_NAMES[i], key) == 0)
			return TRUE;
	}

	return FALSE;
}


191
static void
Paolo Bacchilega's avatar
Paolo Bacchilega committed
192 193 194 195 196
set_file_info (GFileInfo  *info,
	       const char *key,
	       const char *description,
	       const char *formatted_value,
	       const char *raw_value,
197 198
	       const char *category,
	       const char *type_name)
Paolo Bacchilega's avatar
Paolo Bacchilega committed
199 200 201 202 203 204 205
{
	char            *attribute;
	GthMetadataInfo *metadata_info;
	GthMetadata     *metadata;
	char            *description_utf8;
	char            *formatted_value_utf8;

Paolo Bacchilega's avatar
Paolo Bacchilega committed
206 207
	/*if (_g_utf8_all_spaces (formatted_value))
		return;*/
Paolo Bacchilega's avatar
Paolo Bacchilega committed
208 209 210

	attribute = exiv2_key_to_attribute (key);
	description_utf8 = g_locale_to_utf8 (description, -1, NULL, NULL, NULL);
211 212 213 214 215 216 217 218
	if (attribute_is_date (attribute)) {
		GTimeVal time_;

		if (_g_time_val_from_exif_date (raw_value, &time_))
			formatted_value_utf8 = _g_time_val_strftime (&time_, "%x %X");
		else
			formatted_value_utf8 = g_locale_to_utf8 (formatted_value, -1, NULL, NULL, NULL);
	}
219 220 221 222 223 224 225 226 227
	else {
		const char *formatted_clean;

		if (strncmp (formatted_value, "lang=", 5) == 0)
			formatted_clean = strchr (formatted_value, ' ') + 1;
		else
			formatted_clean = formatted_value;
		formatted_value_utf8 = g_locale_to_utf8 (formatted_clean, -1, NULL, NULL, NULL);
	}
Paolo Bacchilega's avatar
Paolo Bacchilega committed
228 229 230 231 232 233

	metadata_info = gth_main_get_metadata_info (attribute);
	if ((metadata_info == NULL) && (category != NULL)) {
		GthMetadataInfo info;

		info.id = attribute;
234
		info.type = (type_name != NULL) ? g_strdup (type_name) : NULL;
Paolo Bacchilega's avatar
Paolo Bacchilega committed
235 236 237 238 239 240 241
		info.display_name = description_utf8;
		info.category = category;
		info.sort_order = 500;
		info.flags = GTH_METADATA_ALLOW_IN_PROPERTIES_VIEW;
		metadata_info = gth_main_register_metadata_info (&info);
	}

Paolo Bacchilega's avatar
Paolo Bacchilega committed
242 243 244
	if ((metadata_info != NULL) && (metadata_info->type == NULL) && (type_name != NULL))
		metadata_info->type = g_strdup (type_name);

245
	if ((metadata_info != NULL) && (metadata_info->display_name == NULL) && (description_utf8 != NULL))
Paolo Bacchilega's avatar
Paolo Bacchilega committed
246 247 248
		metadata_info->display_name = g_strdup (description_utf8);

	metadata = gth_metadata_new ();
Paolo Bacchilega's avatar
Paolo Bacchilega committed
249 250 251 252 253 254 255
	g_object_set (metadata,
		      "id", key,
		      "description", description_utf8,
		      "formatted", formatted_value_utf8,
		      "raw", raw_value,
		      "value-type", type_name,
		      NULL);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275
	g_file_info_set_attribute_object (info, attribute, G_OBJECT (metadata));

	g_object_unref (metadata);
	g_free (formatted_value_utf8);
	g_free (description_utf8);
	g_free (attribute);
}


static void
set_attribute_from_tagset (GFileInfo  *info,
			   const char *attribute,
			   const char *tagset[])
{
	GObject *metadata;
	int      i;
	char    *key;
	char    *description;
	char    *formatted_value;
	char    *raw_value;
Paolo Bacchilega's avatar
Paolo Bacchilega committed
276
	char    *type_name;
Paolo Bacchilega's avatar
Paolo Bacchilega committed
277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292

	metadata = NULL;
	for (i = 0; tagset[i] != NULL; i++) {
		metadata = g_file_info_get_attribute_object (info, tagset[i]);
		if (metadata != NULL)
			break;
	}

	if (metadata == NULL)
		return;

	g_object_get (metadata,
		      "id", &key,
		      "description", &description,
		      "formatted", &formatted_value,
		      "raw", &raw_value,
Paolo Bacchilega's avatar
Paolo Bacchilega committed
293
		      "value-type", &type_name,
Paolo Bacchilega's avatar
Paolo Bacchilega committed
294
		      NULL);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
295 296 297 298 299 300 301
	set_file_info (info,
		       attribute,
		       description,
		       formatted_value,
		       raw_value,
		       NULL,
		       type_name);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
302 303 304 305
}


static void
306 307 308
set_string_list_attribute_from_tagset (GFileInfo  *info,
				       const char *attribute,
				       const char *tagset[])
Paolo Bacchilega's avatar
Paolo Bacchilega committed
309
{
310 311 312 313 314
	GObject        *metadata;
	int             i;
	char           *raw;
	char          **keywords;
	GthStringList  *string_list;
Paolo Bacchilega's avatar
Paolo Bacchilega committed
315

316 317 318 319 320 321 322 323 324 325 326
	metadata = NULL;
	for (i = 0; tagset[i] != NULL; i++) {
		metadata = g_file_info_get_attribute_object (info, tagset[i]);
		if (metadata != NULL)
			break;
	}

	if (metadata == NULL)
		return;

	g_object_get (metadata, "raw", &raw, NULL);
327
	keywords = g_strsplit (raw, ", ", -1);
328 329 330 331 332 333 334
	string_list = gth_string_list_new_from_strv (keywords);
	g_file_info_set_attribute_object (info, attribute, G_OBJECT (string_list));

	g_strfreev (keywords);
	g_free (raw);
}

335

336 337 338
static void
set_attributes_from_tagsets (GFileInfo *info)
{
339
	set_attribute_from_tagset (info, "general::datetime", _DATE_TAG_NAMES);
340
	set_attribute_from_tagset (info, "general::description", _DESCRIPTION_TAG_NAMES);
341
	set_attribute_from_tagset (info, "general::title", _TITLE_TAG_NAMES);
342 343
	set_attribute_from_tagset (info, "general::location", _LOCATION_TAG_NAMES);
	set_string_list_attribute_from_tagset (info, "general::tags", _KEYWORDS_TAG_NAMES);
344
	set_attribute_from_tagset (info, "Embedded::Photo::DateTimeOriginal", _ORIGINAL_DATE_TAG_NAMES);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
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
	set_attribute_from_tagset (info, "Embedded::Image::Orientation", _ORIENTATION_TAG_NAMES);
}


static const char *
get_exif_default_category (const Exiv2::Exifdatum &md)
{
	if (Exiv2::ExifTags::isMakerIfd(md.ifdId()))
		return "Exif::MakerNotes";

	switch (md.ifdId()) {
	case Exiv2::ifd1Id:
		return "Exif::Thumbnail";
	case Exiv2::gpsIfdId:
		return "Exif::GPS";
	case Exiv2::iopIfdId:
		return "Exif::Versions";
	default:
		break;
	}

	return "Exif::Other";
}


370 371 372 373 374 375 376 377 378 379
static void
exiv2_read_metadata (Exiv2::Image::AutoPtr  image,
		     GFileInfo             *info)
{
	image->readMetadata();

	Exiv2::ExifData &exifData = image->exifData();
	if (! exifData.empty()) {
		Exiv2::ExifData::const_iterator end = exifData.end();
		for (Exiv2::ExifData::const_iterator md = exifData.begin(); md != end; ++md) {
Paolo Bacchilega's avatar
Paolo Bacchilega committed
380 381
			stringstream raw_value;
			raw_value << md->value();
382

Paolo Bacchilega's avatar
Paolo Bacchilega committed
383 384 385 386
			stringstream description;
			if (! md->tagLabel().empty())
				description << md->tagLabel();
			else if (md->ifdId () > Exiv2::ifd1Id)
387
				// Must be a MakerNote - include group name
Paolo Bacchilega's avatar
Paolo Bacchilega committed
388 389
				description << md->groupName() << "." << md->tagName();
			else
390
				// Normal exif tag - just use tag name
Paolo Bacchilega's avatar
Paolo Bacchilega committed
391
				description << md->tagName();
392 393 394

			set_file_info (info,
				       md->key().c_str(),
Paolo Bacchilega's avatar
Paolo Bacchilega committed
395 396 397
				       description.str().c_str(),
				       md->print().c_str(),
				       raw_value.str().c_str(),
398 399
				       get_exif_default_category (*md),
				       md->typeName());
400 401 402 403 404 405 406
		}
	}

	Exiv2::IptcData &iptcData = image->iptcData();
	if (! iptcData.empty()) {
		Exiv2::IptcData::iterator end = iptcData.end();
		for (Exiv2::IptcData::iterator md = iptcData.begin(); md != end; ++md) {
Paolo Bacchilega's avatar
Paolo Bacchilega committed
407 408
			stringstream raw_value;
			raw_value << md->value();
409

Paolo Bacchilega's avatar
Paolo Bacchilega committed
410 411 412 413 414
			stringstream description;
			if (! md->tagLabel().empty())
				description << md->tagLabel();
			else
				description << md->tagName();
415 416 417

			set_file_info (info,
				       md->key().c_str(),
Paolo Bacchilega's avatar
Paolo Bacchilega committed
418 419 420
				       description.str().c_str(),
				       md->print().c_str(),
				       raw_value.str().c_str(),
421 422
				       "Iptc",
				       md->typeName());
423 424 425 426 427 428 429
		}
	}

	Exiv2::XmpData &xmpData = image->xmpData();
	if (! xmpData.empty()) {
		Exiv2::XmpData::iterator end = xmpData.end();
		for (Exiv2::XmpData::iterator md = xmpData.begin(); md != end; ++md) {
Paolo Bacchilega's avatar
Paolo Bacchilega committed
430 431
			stringstream raw_value;
			raw_value << md->value();
432

Paolo Bacchilega's avatar
Paolo Bacchilega committed
433 434 435 436 437
			stringstream description;
			if (! md->tagLabel().empty())
				description << md->tagLabel();
			else
				description << md->groupName() << "." << md->tagName();
438 439 440

			set_file_info (info,
				       md->key().c_str(),
Paolo Bacchilega's avatar
Paolo Bacchilega committed
441 442 443
				       description.str().c_str(),
				       md->print().c_str(),
				       raw_value.str().c_str(),
444 445
				       "Xmp::Embedded",
				       md->typeName());
446 447 448 449 450 451 452
		}
	}

	set_attributes_from_tagsets (info);
}


Paolo Bacchilega's avatar
Paolo Bacchilega committed
453
/*
454
 * exiv2_read_metadata_from_file
Paolo Bacchilega's avatar
Paolo Bacchilega committed
455 456 457 458 459 460
 * reads metadata from image files
 * code relies heavily on example1 from the exiv2 website
 * http://www.exiv2.org/example1.html
 */
extern "C"
gboolean
461 462 463
exiv2_read_metadata_from_file (GFile      *file,
			       GFileInfo  *info,
			       GError    **error)
Paolo Bacchilega's avatar
Paolo Bacchilega committed
464 465 466 467 468
{
	try {
		char *path;

		path = g_file_get_path (file);
469 470 471 472 473 474
		if (path == NULL) {
			if (error != NULL)
				*error = g_error_new_literal (G_IO_ERROR, G_IO_ERROR_FAILED, _("Invalid file format"));
			return FALSE;
		}

Paolo Bacchilega's avatar
Paolo Bacchilega committed
475 476 477 478
		Exiv2::Image::AutoPtr image = Exiv2::ImageFactory::open(path);
		g_free (path);

		if (image.get() == 0) {
479 480
			if (error != NULL)
				*error = g_error_new_literal (G_IO_ERROR, G_IO_ERROR_FAILED, _("Invalid file format"));
Paolo Bacchilega's avatar
Paolo Bacchilega committed
481 482 483
			return FALSE;
		}

484 485
		exiv2_read_metadata (image, info);
	}
486
	catch (Exiv2::AnyError& e) {
487
		if (error != NULL)
488
			*error = g_error_new_literal (G_IO_ERROR, G_IO_ERROR_FAILED, e.what());
489 490
		return FALSE;
	}
Paolo Bacchilega's avatar
Paolo Bacchilega committed
491

492 493
	return TRUE;
}
Paolo Bacchilega's avatar
Paolo Bacchilega committed
494 495


496 497 498 499 500 501 502 503 504
extern "C"
gboolean
exiv2_read_metadata_from_buffer (void       *buffer,
				 gsize       buffer_size,
				 GFileInfo  *info,
				 GError    **error)
{
	try {
		Exiv2::Image::AutoPtr image = Exiv2::ImageFactory::open ((Exiv2::byte*) buffer, buffer_size);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
505

506 507 508 509
		if (image.get() == 0) {
			if (error != NULL)
				*error = g_error_new_literal (G_IO_ERROR, G_IO_ERROR_FAILED, _("Invalid file format"));
			return FALSE;
Paolo Bacchilega's avatar
Paolo Bacchilega committed
510 511
		}

512
		exiv2_read_metadata (image, info);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
513
	}
514
	catch (Exiv2::AnyError& e) {
515
		if (error != NULL)
516
			*error = g_error_new_literal (G_IO_ERROR, G_IO_ERROR_FAILED, e.what());
Paolo Bacchilega's avatar
Paolo Bacchilega committed
517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532
		return FALSE;
	}

	return TRUE;
}


extern "C"
gboolean
exiv2_read_sidecar (GFile     *file,
		    GFileInfo *info)
{
	try {
		char *path;

		path = g_file_get_path (file);
533 534 535
		if (path == NULL)
			return FALSE;

Paolo Bacchilega's avatar
Paolo Bacchilega committed
536 537 538 539 540 541 542 543 544 545 546 547 548
		Exiv2::DataBuf buf = Exiv2::readFile(path);
		g_free (path);

		std::string xmpPacket;
		xmpPacket.assign(reinterpret_cast<char*>(buf.pData_), buf.size_);
		Exiv2::XmpData xmpData;

		if (0 != Exiv2::XmpParser::decode(xmpData, xmpPacket))
			return FALSE;

		if (! xmpData.empty()) {
			Exiv2::XmpData::iterator end = xmpData.end();
			for (Exiv2::XmpData::iterator md = xmpData.begin(); md != end; ++md) {
Paolo Bacchilega's avatar
Paolo Bacchilega committed
549 550
				stringstream raw_value;
				raw_value << md->value();
Paolo Bacchilega's avatar
Paolo Bacchilega committed
551

Paolo Bacchilega's avatar
Paolo Bacchilega committed
552 553 554 555 556
				stringstream description;
				if (! md->tagLabel().empty())
					description << md->tagLabel();
				else
					description << md->groupName() << "." << md->tagName();
Paolo Bacchilega's avatar
Paolo Bacchilega committed
557 558 559

				set_file_info (info,
					       md->key().c_str(),
Paolo Bacchilega's avatar
Paolo Bacchilega committed
560 561 562
					       description.str().c_str(),
					       md->print().c_str(),
					       raw_value.str().c_str(),
563 564
					       "Xmp::Sidecar",
					       md->typeName());
Paolo Bacchilega's avatar
Paolo Bacchilega committed
565 566 567 568 569 570 571
			}
		}
		Exiv2::XmpParser::terminate();

		set_attributes_from_tagsets (info);
	}
	catch (Exiv2::AnyError& e) {
572
		std::cerr << "Caught Exiv2 exception '" << e << "'\n";
Paolo Bacchilega's avatar
Paolo Bacchilega committed
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
		return FALSE;
	}

	return TRUE;
}


static void
mandatory_int (Exiv2::ExifData &checkdata,
	       const char      *tag,
	       int              value)
{
	Exiv2::ExifKey key = Exiv2::ExifKey(tag);
	if (checkdata.findKey(key) == checkdata.end())
		checkdata[tag] = value;
}


static void
mandatory_string (Exiv2::ExifData &checkdata,
		  const char      *tag,
		  const char      *value)
{
	Exiv2::ExifKey key = Exiv2::ExifKey(tag);
	if (checkdata.findKey(key) == checkdata.end())
		checkdata[tag] = value;
}


Paolo Bacchilega's avatar
Paolo Bacchilega committed
602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623
const char *
gth_main_get_metadata_type (GthMetadata *metadata,
			    const char  *key)
{
	const char      *value_type;
	GthMetadataInfo *metadatum_info;

	value_type = gth_metadata_get_value_type (metadata);
	if (g_strcmp0 (value_type, "Undefined") == 0)
			value_type = NULL;

	if (value_type != NULL)
		return value_type;

	metadatum_info = gth_main_get_metadata_info (key);
	if (metadatum_info != NULL)
		value_type = metadatum_info->type;

	return value_type;
}


624 625 626 627
static Exiv2::DataBuf
exiv2_write_metadata_private (Exiv2::Image::AutoPtr  image,
			      GFileInfo             *info,
			      GdkPixbuf             *pixbuf)
Paolo Bacchilega's avatar
Paolo Bacchilega committed
628
{
629 630
	char **attributes;
	int    i;
Paolo Bacchilega's avatar
Paolo Bacchilega committed
631

632
	image->clearMetadata();
Paolo Bacchilega's avatar
Paolo Bacchilega committed
633

634
	// EXIF Data
Paolo Bacchilega's avatar
Paolo Bacchilega committed
635

636
	Exiv2::ExifData ed;
637 638 639 640
	attributes = g_file_info_list_attributes (info, "Exif");
	for (i = 0; attributes[i] != NULL; i++) {
		GthMetadata *metadatum;
		char *key;
Paolo Bacchilega's avatar
Paolo Bacchilega committed
641

642 643
		metadatum = (GthMetadata *) g_file_info_get_attribute_object (info, attributes[i]);
		key = exiv2_key_from_attribute (attributes[i]);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
644

645
		try {
646 647 648
			/* If the metadatum has no value yet, a new empty value
			 * is created. The type is taken from Exiv2's tag
			 * lookup tables. If the tag is not found in the table,
Paolo Bacchilega's avatar
Paolo Bacchilega committed
649 650
			 * the type defaults to ASCII.
			 * We always create the metadatum explicitly if the
651 652
			 * type is available to avoid type errors.
			 * See bug #610389 for more details.  The original
Paolo Bacchilega's avatar
Paolo Bacchilega committed
653
			 * explanation is here:
654 655 656
			 * http://uk.groups.yahoo.com/group/exiv2/message/1472
			 */

Paolo Bacchilega's avatar
Paolo Bacchilega committed
657 658 659 660 661 662
			const char *raw_value = gth_metadata_get_raw (metadatum);
			const char *value_type = gth_main_get_metadata_type (metadatum, key);

			if ((raw_value != NULL) && (strcmp (raw_value, "") != 0) &&  (value_type != NULL)) {
				Exiv2::Value::AutoPtr value = Exiv2::Value::create (Exiv2::TypeInfo::typeId (value_type));
				value->read (raw_value);
663 664 665
				Exiv2::ExifKey exif_key(key);
				ed.add (exif_key, value.get());
			}
666
		}
667
		catch (Exiv2::AnyError& e) {
668
			/* we don't care about invalid key errors */
669
			g_warning ("%s", e.what());
670
		}
Paolo Bacchilega's avatar
Paolo Bacchilega committed
671

672 673 674
		g_free (key);
	}
	g_strfreev (attributes);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
675

676
	// Mandatory tags - add if not already present
Paolo Bacchilega's avatar
Paolo Bacchilega committed
677

678 679 680 681 682 683 684 685
	mandatory_int (ed, "Exif.Image.XResolution", 72);
	mandatory_int (ed, "Exif.Image.YResolution", 72);
	mandatory_int (ed, "Exif.Image.ResolutionUnit", 2);
	mandatory_int (ed, "Exif.Image.YCbCrPositioning", 1);
	mandatory_int (ed, "Exif.Photo.ColorSpace", 1);
	mandatory_string (ed, "Exif.Photo.ExifVersion", "48 50 50 49");
	mandatory_string (ed, "Exif.Photo.ComponentsConfiguration", "1 2 3 0");
	mandatory_string (ed, "Exif.Photo.FlashpixVersion", "48 49 48 48");
Paolo Bacchilega's avatar
Paolo Bacchilega committed
686

687
	// Overwrite the software tag
Paolo Bacchilega's avatar
Paolo Bacchilega committed
688

689
	ed["Exif.Image.Software"] = PACKAGE " " VERSION;
Paolo Bacchilega's avatar
Paolo Bacchilega committed
690

691
	// Update the dimension tags with actual image values
Paolo Bacchilega's avatar
Paolo Bacchilega committed
692

693 694
	int width = 0;
	int height = 0;
Paolo Bacchilega's avatar
Paolo Bacchilega committed
695

696 697
	if (pixbuf != NULL) {
		width = gdk_pixbuf_get_width (pixbuf);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
698 699 700
		if (width > 0)
			ed["Exif.Photo.PixelXDimension"] = width;

701
		height = gdk_pixbuf_get_height (pixbuf);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
702 703
		if (height > 0)
			ed["Exif.Photo.PixelYDimension"] = height;
704

705
		ed["Exif.Image.Orientation"] = 1;
706
	}
Paolo Bacchilega's avatar
Paolo Bacchilega committed
707

708 709
	// Update the thumbnail

710
	Exiv2::ExifThumb thumb(ed);
711 712 713 714 715 716 717 718 719 720 721 722 723
	if ((pixbuf != NULL) && (width > 0) && (height > 0)) {
		GdkPixbuf *thumb_pixbuf;
		char      *buffer;
		gsize      buffer_size;

		scale_keeping_ratio (&width, &height, 128, 128, FALSE);
		thumb_pixbuf = _gdk_pixbuf_scale_simple_safe (pixbuf, width, height, GDK_INTERP_BILINEAR);
		if (gdk_pixbuf_save_to_buffer (thumb_pixbuf, &buffer, &buffer_size, "jpeg", NULL, NULL)) {
			thumb.setJpegThumbnail ((Exiv2::byte*) buffer, buffer_size);
			ed["Exif.Thumbnail.XResolution"] = 72;
			ed["Exif.Thumbnail.YResolution"] = 72;
			ed["Exif.Thumbnail.ResolutionUnit"] =  2;
			g_free (buffer);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
724 725 726 727
		}
		else
			thumb.erase();

728 729 730 731
		g_object_unref (thumb_pixbuf);
	}
	else
		thumb.erase();
Paolo Bacchilega's avatar
Paolo Bacchilega committed
732

733
	// Update the DateTime tag
Paolo Bacchilega's avatar
Paolo Bacchilega committed
734

735 736 737 738 739 740 741
	if (g_file_info_get_attribute_object (info, "Exif::Image::DateTime") == NULL) {
		GTimeVal current_time;
		g_get_current_time (&current_time);
		char *date_time = _g_time_val_to_exif_date (&current_time);
		ed["Exif.Image.DateTime"] = date_time;
		g_free (date_time);
	}
Paolo Bacchilega's avatar
Paolo Bacchilega committed
742

743
	// IPTC Data
Paolo Bacchilega's avatar
Paolo Bacchilega committed
744

745
	Exiv2::IptcData id;
746 747 748 749
	attributes = g_file_info_list_attributes (info, "Iptc");
	for (i = 0; attributes[i] != NULL; i++) {
		GthMetadata *metadatum = (GthMetadata *) g_file_info_get_attribute_object (info, attributes[i]);
		char *key = exiv2_key_from_attribute (attributes[i]);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
750

751
		try {
Paolo Bacchilega's avatar
Paolo Bacchilega committed
752 753 754 755 756 757 758
			const char *raw_value = gth_metadata_get_raw (metadatum);
			const char *value_type = gth_main_get_metadata_type (metadatum, key);

			if ((raw_value != NULL) && (strcmp (raw_value, "") != 0) &&  (value_type != NULL)) {
				/* See the exif data code above for an explanation. */
				Exiv2::Value::AutoPtr value = Exiv2::Value::create (Exiv2::TypeInfo::typeId (value_type));
				value->read (raw_value);
759 760 761
				Exiv2::IptcKey iptc_key(key);
				id.add (iptc_key, value.get());
			}
762
		}
763
		catch (Exiv2::AnyError& e) {
764
			/* we don't care about invalid key errors */
765
			g_warning ("%s", e.what());
Paolo Bacchilega's avatar
Paolo Bacchilega committed
766 767
		}

768 769 770
		g_free (key);
	}
	g_strfreev (attributes);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
771

772
	// XMP Data
Paolo Bacchilega's avatar
Paolo Bacchilega committed
773

774
	Exiv2::XmpData xd;
775 776 777 778
	attributes = g_file_info_list_attributes (info, "Xmp");
	for (i = 0; attributes[i] != NULL; i++) {
		GthMetadata *metadatum = (GthMetadata *) g_file_info_get_attribute_object (info, attributes[i]);
		char *key = exiv2_key_from_attribute (attributes[i]);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
779

780 781 782 783 784 785
		// Remove existing tags of the same type.
		// Seems to be needed for storing category keywords.
		// Not exactly sure why!
		Exiv2::XmpData::iterator iter = xd.findKey (Exiv2::XmpKey (key));
		if (iter != xd.end ())
			xd.erase (iter);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
786

787
		try {
Paolo Bacchilega's avatar
Paolo Bacchilega committed
788 789
			const char *raw_value = gth_metadata_get_raw (metadatum);
			const char *value_type = gth_main_get_metadata_type (metadatum, key);
790

Paolo Bacchilega's avatar
Paolo Bacchilega committed
791
			if ((raw_value != NULL) && (strcmp (raw_value, "") != 0) &&  (value_type != NULL)) {
792
				/* See the exif data code above for an explanation. */
Paolo Bacchilega's avatar
Paolo Bacchilega committed
793 794 795 796
				Exiv2::Value::AutoPtr value = Exiv2::Value::create (Exiv2::TypeInfo::typeId (value_type));
				value->read (raw_value);
				Exiv2::XmpKey xmp_key(key);
				xd.add (xmp_key, value.get());
797
			}
Paolo Bacchilega's avatar
Paolo Bacchilega committed
798
		}
799
		catch (Exiv2::AnyError& e) {
800
			/* we don't care about invalid key errors */
801
			g_warning ("%s", e.what());
802 803 804 805 806 807
		}

		g_free (key);
	}
	g_strfreev (attributes);

808 809 810 811 812 813 814 815 816 817
	try {
		image->setExifData(ed);
		image->setIptcData(id);
		image->setXmpData(xd);
		image->writeMetadata();
	}
	catch (Exiv2::AnyError& e) {
		g_warning ("%s", e.what());
	}

818 819 820 821 822 823 824
	Exiv2::BasicIo &io = image->io();
	io.open();

	return io.read(io.size());
}


825 826
extern "C"
gboolean
827
exiv2_supports_writes (const char *mime_type)
828
{
829 830 831 832 833
	return (g_content_type_equals (mime_type, "image/jpeg")
#if HAVE_EXIV2_020
		|| g_content_type_equals (mime_type, "image/tiff")
#endif
		|| g_content_type_equals (mime_type, "image/png"));
834 835 836
}


837 838 839 840
extern "C"
gboolean
exiv2_write_metadata (SavePixbufData *data)
{
841
	if (exiv2_supports_writes (data->mime_type)) {
842 843 844
		try {
			Exiv2::Image::AutoPtr image = Exiv2::ImageFactory::open ((Exiv2::byte*) data->buffer, data->buffer_size);
			g_assert (image.get() != 0);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
845

846
			Exiv2::DataBuf buf = exiv2_write_metadata_private (image, data->file_data->info, data->pixbuf);
Paolo Bacchilega's avatar
Paolo Bacchilega committed
847

848 849 850 851 852 853 854
			g_free (data->buffer);
			data->buffer = g_memdup (buf.pData_, buf.size_);
			data->buffer_size = buf.size_;
		}
		catch (Exiv2::AnyError& e) {
			if (data->error != NULL)
				*data->error = g_error_new_literal (G_IO_ERROR, G_IO_ERROR_FAILED, e.what());
Paolo Bacchilega's avatar
Paolo Bacchilega committed
855
			g_warning ("%s\n", e.what());
856 857
			return FALSE;
		}
Paolo Bacchilega's avatar
Paolo Bacchilega committed
858
	}
859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881

	return TRUE;
}


extern "C"
gboolean
exiv2_write_metadata_to_buffer (void      **buffer,
				gsize      *buffer_size,
				GFileInfo  *info,
				GdkPixbuf  *pixbuf,
				GError    **error)
{
	try {
		Exiv2::Image::AutoPtr image = Exiv2::ImageFactory::open ((Exiv2::byte*) *buffer, *buffer_size);
		g_assert (image.get() != 0);

		Exiv2::DataBuf buf = exiv2_write_metadata_private (image, info, pixbuf);

		g_free (*buffer);
		*buffer = g_memdup (buf.pData_, buf.size_);
		*buffer_size = buf.size_;
	}
882
	catch (Exiv2::AnyError& e) {
883
		if (error != NULL)
884
			*error = g_error_new_literal (G_IO_ERROR, G_IO_ERROR_FAILED, e.what());
885 886 887 888
		return FALSE;
	}

	return TRUE;
Paolo Bacchilega's avatar
Paolo Bacchilega committed
889
}
890 891


892 893 894
#define MAX_RATIO_ERROR_TOLERANCE 0.01


895 896 897
GdkPixbuf *
exiv2_generate_thumbnail (const char *uri,
			  const char *mime_type,
898
			  int         requested_size)
899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919
{
	GdkPixbuf *pixbuf = NULL;

	if (! _g_content_type_is_a (mime_type, "image/jpeg")
	    && ! _g_content_type_is_a (mime_type, "image/tiff"))
	{
		return NULL;
	}

	try {
		char *path;

		path = g_filename_from_uri (uri, NULL, NULL);
		if (path != NULL) {
			Exiv2::Image::AutoPtr image = Exiv2::ImageFactory::open (path);
			image->readMetadata ();
			Exiv2::ExifThumbC exifThumb (image->exifData ());
			Exiv2::DataBuf thumb = exifThumb.copy ();

			if (thumb.pData_ != NULL) {
				Exiv2::ExifData &ed = image->exifData();
920

921
				long orientation = ed["Exif.Image.Orientation"].toLong();
922 923 924
				long image_width = ed["Exif.Photo.PixelXDimension"].toLong();
				long image_height = ed["Exif.Photo.PixelYDimension"].toLong();

925
				if ((orientation == 1) && (image_width > 0) && (image_height > 0)) {
926 927 928 929 930 931 932 933 934
					GInputStream *stream = g_memory_input_stream_new_from_data (thumb.pData_, thumb.size_, NULL);
					pixbuf = gdk_pixbuf_new_from_stream (stream, NULL, NULL);

					if (pixbuf != NULL) {
						/* Heuristic to find out-of-date thumbnails: the thumbnail and image aspect ratios must be equal */

						double image_ratio = (((double) image_width) / image_height);
						double thumbnail_ratio = (((double) gdk_pixbuf_get_width (pixbuf)) / gdk_pixbuf_get_height (pixbuf));
						double ratio_delta = (image_ratio > thumbnail_ratio) ? (image_ratio - thumbnail_ratio) : (thumbnail_ratio - image_ratio);
935

936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955
						if (ratio_delta > MAX_RATIO_ERROR_TOLERANCE) {
							g_object_unref (pixbuf);
							pixbuf = NULL;
						}
						else {
							/* Save the original image size in the pixbuf options */

							char *s = g_strdup_printf ("%ld", image_width);
							gdk_pixbuf_set_option (pixbuf, "tEXt::Thumb::Image::Width", s);
							g_object_set_data (G_OBJECT (pixbuf), "gnome-original-width", GINT_TO_POINTER ((int) image_width));
							g_free (s);

							s = g_strdup_printf ("%ld", image_height);
							gdk_pixbuf_set_option (pixbuf, "tEXt::Thumb::Image::Height", s);
							g_object_set_data (G_OBJECT (pixbuf), "gnome-original-height", GINT_TO_POINTER ((int) image_height));
							g_free (s);

							/* Set the orientation option to correctly rotate the thumbnail
							 * in gnome_desktop_thumbnail_factory_generate_thumbnail() */

956 957 958 959
							long  orientation = ed["Exif.Image.Orientation"].toLong();
							char *orientation_s = g_strdup_printf ("%ld", orientation);
							gdk_pixbuf_set_option (pixbuf, "orientation", orientation_s);
							g_free (orientation_s);
960 961 962 963
						}
					}

					g_object_unref (stream);
964 965 966 967 968 969 970 971 972 973 974
				}
			}

			g_free (path);
		}
	}
	catch (Exiv2::AnyError& e) {
	}

	return pixbuf;
}