pandemonium_engine/modules/paint/nodes/paint_project.cpp

#include "paint_project.h"

#include "../ui/paint_canvas_background.h"
#include "../ui/paint_visual_grid.h"
#include "core/config/project_settings.h"

#include "core/config/engine.h"

String PaintProject::get_save_file_name() {
	return _save_file_name;
}
void PaintProject::set_save_file_name(const String &fn) {
	_save_file_name = fn;
}

Color PaintProject::get_current_color() {
	return _current_color;
}
void PaintProject::set_current_color(const Color &color) {
	_current_color = color;

	emit_signal("current_color_changed", _current_color);
}

void PaintProject::add_preset_color(const Color &color) {
	_color_presets.push_back(color);

	emit_signal("color_presets_changed");
}
void PaintProject::remove_preset_color(const int index) {
	ERR_FAIL_INDEX(index, _color_presets.size());

	_color_presets.remove(index);

	emit_signal("color_presets_changed");
}
Color PaintProject::get_preset_color(const int index) {
	ERR_FAIL_INDEX_V(index, _color_presets.size(), Color());

	return _color_presets.get(index);
}
void PaintProject::set_preset_color(const int index, const Color &color) {
	ERR_FAIL_INDEX(index, _color_presets.size());

	_color_presets.set(index, color);

	emit_signal("color_presets_changed");
}
int PaintProject::get_preset_color_count() {
	return _color_presets.size();
}

PoolColorArray PaintProject::get_color_presets() {
	return _color_presets;
}
void PaintProject::set_color_presets(const PoolColorArray &colors) {
	_color_presets = colors;

	emit_signal("color_presets_changed");
}

void PaintProject::set_colors_as_default() {
	if (ProjectSettings::get_singleton()) {
		ProjectSettings::get_singleton()->set_setting("paint/color_presets/colors", _color_presets);
	}
}

void PaintProject::save_image_to_file() {
	ERR_FAIL_COND(_save_file_name.empty());

	Ref<Image> img = render_image();

	ERR_FAIL_COND(!img.is_valid());

	img->save_png("res://" + _save_file_name);
}

void PaintProject::add_paint_canvas_backgorund() {
	PaintCanvasBackground *bg = memnew(PaintCanvasBackground);
	add_child(bg);
	move_child(bg, 0);

	if (Engine::get_singleton()->is_editor_hint()) {
		bg->set_owner(get_tree()->get_edited_scene_root());
	}
}
void PaintProject::add_paint_visual_grid() {
	PaintVisualGrid *grid = memnew(PaintVisualGrid);
	add_child(grid);
	move_child(grid, 0);

	if (Engine::get_singleton()->is_editor_hint()) {
		grid->set_owner(get_tree()->get_edited_scene_root());
	}
}

PaintProject::PaintProject() {
	_current_color = Color(1, 1, 1, 1);
}

PaintProject::~PaintProject() {
}

void PaintProject::_notification(int p_what) {
	if (p_what == NOTIFICATION_ENTER_TREE) {
		if (_color_presets.size() == 0) {
			if (ProjectSettings::get_singleton()->has_setting("paint/color_presets/colors")) {
				_color_presets = ProjectSettings::get_singleton()->get_setting("paint/color_presets/colors");
			}
		}
	}
}

void PaintProject::_bind_methods() {
	ADD_SIGNAL(MethodInfo("current_color_changed", PropertyInfo(Variant::COLOR, "color")));
	ADD_SIGNAL(MethodInfo("color_presets_changed"));

	ClassDB::bind_method(D_METHOD("get_save_file_name"), &PaintProject::get_save_file_name);
	ClassDB::bind_method(D_METHOD("set_save_file_name", "size"), &PaintProject::set_save_file_name);
	ADD_PROPERTY(PropertyInfo(Variant::STRING, "save_file_name"), "set_save_file_name", "get_save_file_name");

	ClassDB::bind_method(D_METHOD("save_image_to_file"), &PaintProject::save_image_to_file);
	ADD_PROPERTY(PropertyInfo(Variant::NIL, "run_save_image", PROPERTY_HINT_BUTTON, "save_image_to_file"), "", "");

	ClassDB::bind_method(D_METHOD("get_current_color"), &PaintProject::get_current_color);
	ClassDB::bind_method(D_METHOD("set_current_color", "size"), &PaintProject::set_current_color);
	ADD_PROPERTY(PropertyInfo(Variant::COLOR, "current_color"), "set_current_color", "get_current_color");

	ClassDB::bind_method(D_METHOD("add_preset_color", "color"), &PaintProject::add_preset_color);
	ClassDB::bind_method(D_METHOD("remove_preset_color", "index"), &PaintProject::remove_preset_color);
	ClassDB::bind_method(D_METHOD("get_preset_color", "index"), &PaintProject::get_preset_color);
	ClassDB::bind_method(D_METHOD("set_preset_color", "index", "color"), &PaintProject::set_preset_color);
	ClassDB::bind_method(D_METHOD("get_preset_color_count"), &PaintProject::get_preset_color_count);

	ClassDB::bind_method(D_METHOD("get_color_presets"), &PaintProject::get_color_presets);
	ClassDB::bind_method(D_METHOD("set_color_presets", "colors"), &PaintProject::set_color_presets);
	ADD_PROPERTY(PropertyInfo(Variant::POOL_COLOR_ARRAY, "color_presets"), "set_color_presets", "get_color_presets");

	ClassDB::bind_method(D_METHOD("set_colors_as_default"), &PaintProject::set_colors_as_default);
	ADD_PROPERTY(PropertyInfo(Variant::NIL, "run_set_colors_as_default", PROPERTY_HINT_BUTTON, "set_colors_as_default"), "", "");

	ClassDB::bind_method(D_METHOD("add_paint_canvas_backgorund"), &PaintProject::add_paint_canvas_backgorund);
	ClassDB::bind_method(D_METHOD("add_paint_visual_grid"), &PaintProject::add_paint_visual_grid);
	ADD_PROPERTY(PropertyInfo(Variant::NIL, "run_add_paint_visual_grid", PROPERTY_HINT_BUTTON, "add_paint_visual_grid"), "", "");
	ADD_PROPERTY(PropertyInfo(Variant::NIL, "run_add_paint_canvas_backgorund", PROPERTY_HINT_BUTTON, "add_paint_canvas_backgorund"), "", "");
}