/*
 *  This file is part of RawTherapee.
 *
 *  Copyright (c) 2004-2010 Gabor Horvath <hgabor@rawtherapee.com>
 *  Copyright (c) 2011 Jean-Christophe FRISCH <natureh@free.fr>
 *
 *  RawTherapee 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 3 of the License, or
 *  (at your option) any later version.
 *
 *  RawTherapee 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 RawTherapee.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "rtimage.h"
#include "../rtengine/safegtk.h"

extern Glib::ustring argv0;
extern Options options;

std::vector<Glib::ustring> imagesPaths;
std::map<Glib::ustring,Glib::RefPtr<Gdk::Pixbuf> > pixBufMap; // List of image buffers in order to live update them on theme switch and to avoid a lot of file accesses

/*
 * RTImage is a derived class of Gtk::Image, in order to handle theme related iconsets
 */
RTImage::RTImage(Glib::ustring fileName, Glib::ustring rtlFileName) : Gtk::Image() {
	Glib::ustring mapKey;
	if (rtlFileName.length()) {
		if (get_direction() == Gtk::TEXT_DIR_RTL) {
		    mapKey = rtlFileName;
		} else {
		    mapKey = fileName;
		}
	} else {
	    mapKey = fileName;
	}
    
    std::map<Glib::ustring,Glib::RefPtr<Gdk::Pixbuf> >::iterator it;
    it = pixBufMap.find(mapKey);
    if (it != pixBufMap.end()) {
        set(it->second);
    } else {
        Glib::RefPtr<Gdk::Pixbuf> tempPixPuf = Gdk::Pixbuf::create_from_file(findIconAbsolutePath(mapKey));
        pixBufMap.insert(std::pair<Glib::ustring,Glib::RefPtr<Gdk::Pixbuf> >(mapKey, tempPixPuf));
        set(tempPixPuf);
    }
}

void RTImage::updateImages() {
    std::map<Glib::ustring,Glib::RefPtr<Gdk::Pixbuf> >::iterator it;
    for (it=pixBufMap.begin(); it!=pixBufMap.end(); ++it) {
        Glib::ustring fullPath = findIconAbsolutePath(it->first);
        it->second = Gdk::Pixbuf::create_from_file(fullPath);
    }
}

// DONE (was TODO: Maybe this could be optimized: in order to avoid looking up for an icon file in the filesystem on each popupmenu selection, maybe we could find a way to copy the image data from another RTImage)
void RTImage::changeImage(Glib::ustring &newImage) {
	clear();
    std::map<Glib::ustring,Glib::RefPtr<Gdk::Pixbuf> >::iterator it;
    it = pixBufMap.find(newImage);
    if (it != pixBufMap.end()) {
        set(it->second);
    } else {
        Glib::ustring fullPath = findIconAbsolutePath(newImage);
        Glib::RefPtr<Gdk::Pixbuf> tempPixPuf = Gdk::Pixbuf::create_from_file(fullPath);
        pixBufMap.insert(std::pair<Glib::ustring,Glib::RefPtr<Gdk::Pixbuf> >(newImage, tempPixPuf));
        set(tempPixPuf);
    }
}

Glib::ustring RTImage::findIconAbsolutePath(const Glib::ustring &iconFName) {
	Glib::ustring path;
	for (unsigned int i=0; i<imagesPaths.size(); i++) {
		path = Glib::build_filename(imagesPaths[i], iconFName);
		if (safe_file_test(path, Glib::FILE_TEST_EXISTS)) {
			return path;
		}
	}
	printf("\"%s\" not found!\n", iconFName.c_str());
	return "";
}

void RTImage::setPaths(Options &opt) {
	Glib::ustring configFilename;
	rtengine::SafeKeyFile keyFile;
	bool hasKeyFile = true;

	imagesPaths.clear();

	// system theme will use the theme set in system.iconset
	if (opt.useSystemTheme) {
		configFilename = Glib::build_filename(argv0, Glib::build_filename("themes","system.iconset"));
	}
	// Gtk theme will use the theme set in it's *.iconset fiel, if it exists
	else {
		configFilename = Glib::build_filename(argv0, Glib::build_filename("themes", Glib::ustring::format(opt.theme, ".iconset")));
	}
	try {
		if (!safe_file_test(configFilename, Glib::FILE_TEST_EXISTS) || !keyFile.load_from_file (configFilename)) {
			// ...otherwise fallback to the iconset set in default.iconset
			configFilename = Glib::build_filename(argv0, Glib::build_filename("themes", "Default.iconset"));
			if (!keyFile.load_from_file (configFilename)) {
				hasKeyFile = false;
			}
		}
	}
	catch (Glib::Error &err) {
		if (options.rtSettings.verbose)
			printf("RTImage::setPaths / Error code %d while reading values from \"%s\":\n%s\n", err.code(), configFilename.c_str(), err.what().c_str());
	}
	catch (...) {
		if (options.rtSettings.verbose)
			printf("RTImage::setPaths / Unknown exception while trying to load \"%s\"!\n", configFilename.c_str());
	}

	if (hasKeyFile && keyFile.has_group ("General")) {
		Glib::ustring iSet;

		if (keyFile.has_key ("General", "Iconset"))
			iSet = keyFile.get_string ("General", "Iconset");
		if (iSet.length()) {
			imagesPaths.push_back (Glib::build_filename(argv0, Glib::build_filename("images", Glib::build_filename(iSet, "actions"))));
			imagesPaths.push_back (Glib::build_filename(argv0, Glib::build_filename("images", iSet)));
			imagesPaths.push_back (Glib::build_filename(argv0, Glib::build_filename("images", Glib::build_filename(iSet, "devices"))));
			imagesPaths.push_back (Glib::build_filename(argv0, Glib::build_filename("images", Glib::build_filename(iSet, "places"))));
		}

		iSet.clear();
		if (keyFile.has_key ("General", "FallbackIconset"))
			iSet = keyFile.get_string ("General", "FallbackIconset");
		if (iSet.length()) {
			imagesPaths.push_back (Glib::build_filename(argv0, Glib::build_filename("images", Glib::build_filename(iSet, "actions"))));
			imagesPaths.push_back (Glib::build_filename(argv0, Glib::build_filename("images", iSet)));
			imagesPaths.push_back (Glib::build_filename(argv0, Glib::build_filename("images", Glib::build_filename(iSet, "devices"))));
			imagesPaths.push_back (Glib::build_filename(argv0, Glib::build_filename("images", Glib::build_filename(iSet, "places"))));
		}
	}
	// The images/ folder is the second fallback solution
	imagesPaths.push_back (Glib::build_filename(argv0, "images"));
}