godot/tools/editor/import_dock.cpp

#include "import_dock.h"

class ImportDockParameters : public Object {
	GDCLASS(ImportDockParameters,Object)
public:
	Map<StringName,Variant> values;
	List<PropertyInfo> properties;
	Ref<ResourceImporter> importer;
	Vector<String> paths;


	bool _set(const StringName& p_name, const Variant& p_value) {

		if (values.has(p_name)) {
			values[p_name]=p_value;
			return true;
		}

		return false;
	}

	bool _get(const StringName& p_name,Variant &r_ret) const {

		if (values.has(p_name)) {
			r_ret=values[p_name];
			return true;
		}

		return false;

	}
	void _get_property_list( List<PropertyInfo> *p_list) const {

		for (const List<PropertyInfo>::Element *E=properties.front();E;E=E->next()) {
			if (!importer->get_option_visibility(E->get().name,values))
				continue;
			p_list->push_back(E->get());
		}
	}

	void update() {
		_change_notify();
	}
};

void ImportDock::set_edit_path(const String& p_path) {

	Ref<ConfigFile> config;
	config.instance();
	Error err = config->load(p_path+".import");
	if (err!=OK) {
		clear();
		return;
	}


	params->importer = ResourceFormatImporter::get_singleton()->get_importer_by_name(config->get_value("remap","importer"));
	if (params->importer.is_null()) {
		clear();
		return;
	}

	List<ResourceImporter::ImportOption> options;
	params->importer->get_import_options(&options);

	params->properties.clear();
	params->values.clear();

	for (List<ResourceImporter::ImportOption>::Element *E=options.front();E;E=E->next()) {

		params->properties.push_back(E->get().option);
		if (config->has_section_key("params",E->get().option.name)) {
			params->values[E->get().option.name]=config->get_value("params",E->get().option.name);
		} else {
			params->values[E->get().option.name]=E->get().default_value;
		}
	}

	params->update();

	List<Ref<ResourceImporter> > importers;
	ResourceFormatImporter::get_singleton()->get_importers_for_extension(p_path.get_extension(),&importers);
	List<Pair<String,String> > importer_names;

	for (List<Ref<ResourceImporter> > ::Element *E=importers.front();E;E=E->next()) {
		importer_names.push_back(Pair<String,String>(E->get()->get_visible_name(),E->get()->get_importer_name()));
	}

	importer_names.sort_custom<PairSort<String,String> >();

	import_as->clear();

	for (List<Pair<String,String> >::Element *E=importer_names.front();E;E=E->next()) {
		import_as->add_item(E->get().first);
		import_as->set_item_metadata(import_as->get_item_count()-1,E->get().second);
		if (E->get().second==params->importer->get_importer_name()) {
			import_as->select(import_as->get_item_count()-1);
		}
	}

	preset->get_popup()->clear();

	if (params->importer->get_preset_count()==0) {
		preset->get_popup()->add_item(TTR("Default"));
	} else {
		for (int i=0;i<params->importer->get_preset_count();i++) {
			preset->get_popup()->add_item(params->importer->get_preset_name(i));
		}
	}

	params->paths.clear();
	params->paths.push_back(p_path);
	import->set_disabled(false);
	import_as->set_disabled(false);

	imported->set_text(p_path.get_file());
}

void ImportDock::set_edit_multiple_paths(const Vector<String>& p_paths) {

	clear();

	//use the value that is repeated the mot
	Map<String,Dictionary> value_frequency;

	for(int i=0;i<p_paths.size();i++) {

		Ref<ConfigFile> config;
		config.instance();
		Error err = config->load(p_paths[i]+".import");
		ERR_CONTINUE(err!=OK);

		if (i==0) {
			params->importer = ResourceFormatImporter::get_singleton()->get_importer_by_name(config->get_value("remap","importer"));
			if (params->importer.is_null()) {
				clear();
				return;
			}
		}

		List<String> keys;
		config->get_section_keys("params",&keys);


		for (List<String>::Element *E=keys.front();E;E=E->next()) {

			if (!value_frequency.has(E->get())) {
				value_frequency[E->get()]=Dictionary();
			}

			Variant value = config->get_value("params",E->get());

			if (value_frequency[E->get()].has(value)) {
				value_frequency[E->get()][value]=int(value_frequency[E->get()][value])+1;
			} else {
				value_frequency[E->get()][value]=1;
			}
		}

	}

	ERR_FAIL_COND(params->importer.is_null());

	List<ResourceImporter::ImportOption> options;
	params->importer->get_import_options(&options);

	params->properties.clear();
	params->values.clear();

	for (List<ResourceImporter::ImportOption>::Element *E=options.front();E;E=E->next()) {

		params->properties.push_back(E->get().option);

		if (value_frequency.has(E->get().option.name)) {

			Dictionary d = value_frequency[E->get().option.name];
			int freq=0;
			List<Variant> v;
			d.get_key_list(&v);
			Variant value;
			for (List<Variant>::Element *F=v.front();F;F=F->next()) {
				int f = d[F->get()];
				if (f>freq) {
					value=F->get();
				}
			}

			params->values[E->get().option.name]=value;
		} else {
			params->values[E->get().option.name]=E->get().default_value;
		}
	}

	params->update();

	List<Ref<ResourceImporter> > importers;
	ResourceFormatImporter::get_singleton()->get_importers_for_extension(p_paths[0].get_extension(),&importers);
	List<Pair<String,String> > importer_names;

	for (List<Ref<ResourceImporter> > ::Element *E=importers.front();E;E=E->next()) {
		importer_names.push_back(Pair<String,String>(E->get()->get_visible_name(),E->get()->get_importer_name()));
	}

	importer_names.sort_custom<PairSort<String,String> >();

	import_as->clear();

	for (List<Pair<String,String> >::Element *E=importer_names.front();E;E=E->next()) {
		import_as->add_item(E->get().first);
		import_as->set_item_metadata(import_as->get_item_count()-1,E->get().second);
		if (E->get().second==params->importer->get_importer_name()) {
			import_as->select(import_as->get_item_count()-1);
		}
	}

	preset->get_popup()->clear();

	if (params->importer->get_preset_count()==0) {
		preset->get_popup()->add_item(TTR("Default"));
	} else {
		for (int i=0;i<params->importer->get_preset_count();i++) {
			preset->get_popup()->add_item(params->importer->get_preset_name(i));
		}
	}

	params->paths=p_paths;
	import->set_disabled(false);
	import_as->set_disabled(false);

	imported->set_text(itos(p_paths.size())+TTR(" Files"));
}

void ImportDock::_preset_selected(int p_idx) {

	print_line("preset selected? "+p_idx);
	List<ResourceImporter::ImportOption> options;

	params->importer->get_import_options(&options,p_idx);

	for (List<ResourceImporter::ImportOption>::Element *E=options.front();E;E=E->next()) {

		params->values[E->get().option.name]=E->get().default_value;
	}

	params->update();

}


void ImportDock::clear() {

	imported->set_text("");
	import->set_disabled(true);
	import_as->clear();
	import_as->set_disabled(true);
	params->values.clear();
	params->properties.clear();
	params->update();
	preset->get_popup()->clear();

}

void ImportDock::_reimport() {

	for(int i=0;i<params->paths.size();i++) {

		Ref<ConfigFile> config;
		config.instance();
		Error err = config->load(params->paths[i]+".import");
		ERR_CONTINUE(err!=OK);

		config->erase_section("params");

		for (List<PropertyInfo>::Element *E=params->properties.front();E;E=E->next()) {
			config->set_value("params",E->get().name,params->values[E->get().name]);
		}

		config->save(params->paths[i]+".import");
	}

	EditorFileSystem::get_singleton()->reimport_files(params->paths);
	EditorFileSystem::get_singleton()->emit_signal("filesystem_changed"); //it changed, so force emitting the signal

}

void ImportDock::_bind_methods() {

	ClassDB::bind_method(_MD("_reimport"),&ImportDock::_reimport);
	ClassDB::bind_method(_MD("_preset_selected"),&ImportDock::_preset_selected);
}

ImportDock::ImportDock() {


	imported = memnew( LineEdit );
	imported->set_editable(false);
	add_child(imported);
	HBoxContainer *hb = memnew(HBoxContainer);
	add_margin_child(TTR("Import As:"),hb);
	import_as = memnew( OptionButton );
	hb->add_child(import_as);
	import_as->set_h_size_flags(SIZE_EXPAND_FILL);
	preset = memnew( MenuButton );
	preset->set_text(TTR("Preset.."));
	preset->get_popup()->connect("index_pressed",this,"_preset_selected");
	hb->add_child(preset);

	import_opts = memnew( PropertyEditor );
	add_child(import_opts);
	import_opts->set_v_size_flags(SIZE_EXPAND_FILL);
	import_opts->hide_top_label();
	import_opts->set_hide_script(true);

	hb = memnew( HBoxContainer );
	add_child(hb);
	import = memnew( Button );
	import->set_text(TTR("Reimport"));
	import->connect("pressed",this,"_reimport");
	hb->add_spacer();
	hb->add_child(import);
	hb->add_spacer();

	params = memnew( ImportDockParameters );
	import_opts->edit(params);

}

ImportDock::~ImportDock() {

	memdelete(params);
}