// Copyright (c) 2010, Amar Takhar <verm@aegisub.org>
//
// Permission to use, copy, modify, and distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
//
// $Id$

/// @file icon.h
/// @brief Icon for commands.
/// @ingroup command


#ifndef AGI_PRE
#include <map>

#include <wx/bitmap.h>
#include <wx/image.h>
#include <wx/mstream.h>
#endif

#include <libaegisub/log.h>

#include "icon.h"
#include "../libresrc/bitmap.h"

namespace icon {
typedef std::map<std::string, wxBitmap*> iconMap;
typedef std::pair<std::string, wxBitmap*> iconPair;

iconMap icon16;
iconMap icon24;

wxBitmap* get(std::string name, const int size) {

	// XXX: This code will go away with dynamic icon generation so I'm not
	//      concerned about it.
	if (size == 16) {
		iconMap::iterator index;

		if ((index = icon16.find(name)) != icon16.end()) {
			return index->second;
		}
		printf("icon::get NOT FOUND (%s)\n", name.c_str());

	} else if (size == 24) {
		iconMap::iterator index;

		if ((index = icon24.find(name)) != icon24.end()) {
			return index->second;
		}
		printf("icon::get NOT FOUND (%s)\n", name.c_str());
	} else {
		throw IconInvalid("Valid icon sizes are 16 or 24.");
	}


	wxBitmap *img = new wxBitmap(); // stub to silence compiler warnings.
	return img;
}


wxBitmap* getimage(const unsigned char *buff, size_t size) {
	wxMemoryInputStream mem(buff, size);
	wxImage img(mem);
	wxBitmap *bitmap = new wxBitmap(img);
    return bitmap;
}


#define INSERT_ICON(a, b) \
	icon16.insert(iconPair(a, getimage(b##_16, sizeof(b##_16)))); \
	icon24.insert(iconPair(a, getimage(b##_24, sizeof(b##_24))));


void icon_init() {
	// Seems that WX doesn't install the handlers early enough for our use.
	wxPNGHandler *handler = new wxPNGHandler();
	wxImage::AddHandler(handler);
	wxString handler_name(handler->GetName());

	LOG_D("icon/init") << "Generating 24x24, 16x16 icons";

INSERT_ICON("am/manager", automation_toolbutton)
INSERT_ICON("app/about", about_menu)
INSERT_ICON("app/exit", exit_button)
INSERT_ICON("app/language", languages_menu)
INSERT_ICON("app/log", about_menu)
INSERT_ICON("app/new_window", new_window_menu)
INSERT_ICON("app/options", options_button)
INSERT_ICON("app/updates", blank_button)
INSERT_ICON("audio/close", close_audio_menu)
INSERT_ICON("audio/open", open_audio_menu)
INSERT_ICON("audio/open/video", open_audio_from_video_menu)
INSERT_ICON("edit/line/copy", copy_button)
INSERT_ICON("edit/line/cut", cut_button)
INSERT_ICON("edit/line/delete", delete_button)
INSERT_ICON("edit/line/duplicate", blank_button)
INSERT_ICON("edit/line/duplicate/shift", blank_button)
INSERT_ICON("edit/line/join/as_karaoke", blank_button)
INSERT_ICON("edit/line/join/concatenate", blank_button)
INSERT_ICON("edit/line/join/keep_first", blank_button)
INSERT_ICON("edit/line/paste", paste_button)
INSERT_ICON("edit/line/paste/over", paste_over_button)
INSERT_ICON("edit/line/recombine", blank_button)
INSERT_ICON("edit/line/split/by_karaoke", blank_button)
INSERT_ICON("edit/line/swap", arrow_sort)
INSERT_ICON("edit/redo", redo_button)
INSERT_ICON("edit/search_replace", find_replace_menu)
INSERT_ICON("edit/undo", undo_button)
INSERT_ICON("grid/tag/cycle_hiding", toggle_tag_hiding)
INSERT_ICON("help/bugs", bugtracker_button)
INSERT_ICON("help/contents", contents_button)
INSERT_ICON("help/files", contents_button)
INSERT_ICON("help/forums", forums_button)
INSERT_ICON("help/irc", irc_button)
INSERT_ICON("help/website", website_button)
INSERT_ICON("keyframe/close", close_keyframes_menu)
INSERT_ICON("keyframe/open", open_keyframes_menu)
INSERT_ICON("keyframe/save", save_keyframes_menu)
INSERT_ICON("subtitle/attachment", attach_button)
INSERT_ICON("subtitle/find", find_button)
INSERT_ICON("subtitle/find/next", find_next_menu)
INSERT_ICON("subtitle/insert/after", blank_button)
INSERT_ICON("subtitle/insert/after/videotime", blank_button)
INSERT_ICON("subtitle/insert/before", blank_button)
INSERT_ICON("subtitle/insert/before/videotime", blank_button)
INSERT_ICON("subtitle/new", new_toolbutton)
INSERT_ICON("subtitle/open", open_toolbutton)
INSERT_ICON("subtitle/open/charset", open_with_toolbutton)
INSERT_ICON("subtitle/properties", properties_toolbutton)
INSERT_ICON("subtitle/save", save_toolbutton)
INSERT_ICON("subtitle/save/as", save_as_toolbutton)
INSERT_ICON("subtitle/select/visible", select_visible_button)
INSERT_ICON("subtitle/spellcheck", spellcheck_toolbutton)
INSERT_ICON("time/continous/end", blank_button)
INSERT_ICON("time/continous/start", blank_button)
INSERT_ICON("time/frame/current", shift_to_frame)
INSERT_ICON("time/shift", shift_times_toolbutton)
INSERT_ICON("time/snap/end_video", subend_to_video)
INSERT_ICON("time/snap/frame", shift_to_frame)
INSERT_ICON("time/snap/scene", snap_subs_to_scene)
INSERT_ICON("time/snap/start_video", substart_to_video)
INSERT_ICON("time/sort/end", blank_button)
INSERT_ICON("time/sort/start", blank_button)
INSERT_ICON("time/sort/style", blank_button)
INSERT_ICON("timecode/close", close_timecodes_menu)
INSERT_ICON("timecode/open", open_timecodes_menu)
INSERT_ICON("timecode/save", save_timecodes_menu)
INSERT_ICON("tool/assdraw", assdraw)
INSERT_ICON("tool/export", export_menu)
INSERT_ICON("tool/font_collector", font_collector_button)
INSERT_ICON("tool/line/select", select_lines_button)
INSERT_ICON("tool/resampleres", resample_toolbutton)
INSERT_ICON("tool/style/assistant", styling_toolbutton)
INSERT_ICON("tool/style/manager", style_toolbutton)
INSERT_ICON("tool/time/kanji", kara_timing_copier)
INSERT_ICON("tool/time/kanji", spellcheck_toolbutton)
INSERT_ICON("tool/time/postprocess", timing_processor_toolbutton)
INSERT_ICON("tool/translation_assistant", translation_toolbutton)
INSERT_ICON("video/close", close_video_menu)
INSERT_ICON("video/detach", detach_video_menu)
INSERT_ICON("video/details", show_video_details_menu)
INSERT_ICON("video/jump", jumpto_button)
INSERT_ICON("video/jump/end", video_to_subend)
INSERT_ICON("video/jump/start", video_to_substart)
INSERT_ICON("video/open", open_video_menu)
INSERT_ICON("video/open/dummy", use_dummy_video_menu)
INSERT_ICON("video/zoom/in", zoom_in_button)
INSERT_ICON("video/zoom/out", zoom_out_button)


	// Remove the handler to aboid "Duplicate handler" warnings from WX since
	// it will attempt to install all the handlers later on.
	wxImage::RemoveHandler(handler_name);
}
} // namespace icon