pandemonium_engine/modules/paint/nodes/paint_node.cpp

#include "paint_node.h"

#include "../paint_utilities.h"
#include "paint_project.h"

Vector2i PaintNode::get_size() {
	return _size;
}
void PaintNode::set_size(const Vector2i &size) {
	if (size == _size) {
		return;
	}

	_size = size;

	emit_signal("size_changed");
	_propagate_notification_resized();
	update();
}

bool PaintNode::get_draw_outline() {
	return _draw_outline;
}
void PaintNode::set_draw_outline(const bool val) {
	_draw_outline = val;

	update();
}

Ref<Image> PaintNode::save_image() {
	_propagate_notification_project_pre_save();
	Ref<Image> img = call("_save_image");
	_propagate_notification_project_post_save();

	return img;
}
Ref<Image> PaintNode::_save_image() {
	Ref<Image> image;
	image.instance();

	Vector2i size = get_size();

	if (size.x <= 0 || size.y <= 0) {
		return image;
	}

	image->create(size.x, size.y, false, Image::FORMAT_RGBA8);

	for (int i = 0; i < get_child_count(); ++i) {
		PaintNode *pn = Object::cast_to<PaintNode>(get_child(i));

		if (pn && pn->is_visible()) {
			//dont apply own transform
			save_evaluate_paint_node(pn, Transform2D(), image);
		}
	}

	save_paint_node(this, Transform2D(), image);

	return image;
}

Ref<Image> PaintNode::get_save_image() {
	return call("_get_save_image");
}
Ref<Image> PaintNode::_get_save_image() {
	return Ref<Image>();
}

void PaintNode::save_evaluate_paint_node(PaintNode *node, Transform2D transform, Ref<Image> image) {
	ERR_FAIL_COND(!node);
	ERR_FAIL_COND(!image.is_valid());

	Transform2D currtf = transform * node->get_transform();

	for (int i = 0; i < node->get_child_count(); ++i) {
		PaintNode *pn = Object::cast_to<PaintNode>(node->get_child(i));

		if (pn && pn->is_visible()) {
			save_evaluate_paint_node(pn, currtf, image);
		}
	}

	save_paint_node(node, currtf, image);
}

void PaintNode::save_paint_node(PaintNode *node, Transform2D transform, Ref<Image> image) {
	ERR_FAIL_COND(!node);
	ERR_FAIL_COND(!image.is_valid());

	Ref<Image> save_image = node->get_save_image();

	if (!save_image.is_valid() || save_image->empty()) {
		return;
	}

	Vector2i save_image_size = save_image->get_size();
	Vector2i image_size = image->get_size();

	save_image->lock();
	image->lock();

	//should return early eventually via a simple overlap check here

	for (int x = 0; x < save_image_size.x; ++x) {
		for (int y = 0; y < save_image_size.y; ++y) {
			Vector2i npos = transform.xform(Vector2(x, y));

			if (npos.x < 0 || npos.y < 0 || npos.x >= image_size.x || npos.y >= image_size.y) {
				continue;
			}

			Color sic = save_image->get_pixel(x, y);
			Color oic = image->get_pixel(npos.x, npos.y);

			image->set_pixel(npos.x, npos.y, oic.blend(sic));
		}
	}

	image->unlock();
	save_image->unlock();
}

PoolVector2iArray PaintNode::util_get_pixels_in_line(const Vector2i &from, const Vector2i &to) {
	return PaintUtilities::get_pixels_in_line(from, to);
}

int PaintNode::util_to_1d_v(const Vector2i &p, int w) {
	return PaintUtilities::to_1D_v(p, w);
}
int PaintNode::util_to_1d(int x, int y, int w) {
	return PaintUtilities::to_1D(x, y, w);
}
Vector2i PaintNode::util_to_2d(int idx, int w) {
	return PaintUtilities::to_2D(idx, w);
}

Color PaintNode::util_color_from_array(const PoolRealArray &color_array) {
	return PaintUtilities::color_from_array(color_array);
}
Color PaintNode::util_random_color() {
	return PaintUtilities::random_color();
}
Color PaintNode::util_random_color_alt() {
	return PaintUtilities::random_color_alt();
}

bool PaintNode::forward_canvas_gui_input(const Ref<InputEvent> &p_event) {
	return call("_forward_canvas_gui_input", p_event);
}
bool PaintNode::_forward_canvas_gui_input(const Ref<InputEvent> &p_event) {
	return false;
}

PaintProject *PaintNode::get_paint_project() {
	PaintNode *p = this;

	while (p) {
		PaintProject *pp = Object::cast_to<PaintProject>(p);

		if (pp) {
			return pp;
		}

		p = Object::cast_to<PaintNode>(p->get_parent());
	}

	return NULL;
}

PaintNode *PaintNode::get_parent_paint_node() {
	return Object::cast_to<PaintNode>(get_parent());
}

PaintNode *PaintNode::find_parent_paint_node() {
	Node *p = get_parent();

	while (p) {
		PaintNode *pn = Object::cast_to<PaintNode>(p);

		if (pn) {
			return pn;
		}

		p = Object::cast_to<PaintNode>(p->get_parent());
	}

	return NULL;
}

String PaintNode::get_configuration_warning() const {
	const PaintNode *p = this;

	while (p) {
		const PaintProject *pp = Object::cast_to<PaintProject>(p);

		if (pp) {
			return "";
		}

		p = Object::cast_to<PaintNode>(p->get_parent());
	}

	return "This Node should be a child of a PaintProject!";
}

void PaintNode::_propagate_notification_resized() {
	//Only send it to children
	//So if they don't resize themselves, the notification stops

	for (int i = 0; i < get_child_count(); ++i) {
		PaintNode *pn = Object::cast_to<PaintNode>(get_child(i));

		if (pn) {
			pn->notification(NOTIFICATION_PARENT_PAINT_NODE_RESIZED);
		}
	}
}

void PaintNode::_propagate_notification_project_pre_save() {
	notification(NOTIFICATION_PAINT_PROJECT_PRE_SAVE);

	for (int i = 0; i < get_child_count(); ++i) {
		PaintNode *pn = Object::cast_to<PaintNode>(get_child(i));

		if (pn) {
			pn->_propagate_notification_project_pre_save();
		}
	}
}
void PaintNode::_propagate_notification_project_post_save() {
	notification(NOTIFICATION_PAINT_PROJECT_POST_SAVE);

	for (int i = 0; i < get_child_count(); ++i) {
		PaintNode *pn = Object::cast_to<PaintNode>(get_child(i));

		if (pn) {
			pn->_propagate_notification_project_post_save();
		}
	}
}

PaintNode::PaintNode() {
	_draw_outline = true;
}

PaintNode::~PaintNode() {
}

void PaintNode::_notification(int p_what) {
	switch (p_what) {
		case NOTIFICATION_POST_ENTER_TREE: {
			if (get_size() == Vector2i(0, 0)) {
				PaintNode *pn = get_parent_paint_node();

				if (pn) {
					set_size(pn->get_size());
				}
			}
		} break;
		case NOTIFICATION_DRAW: {
			if (_draw_outline) {
				draw_line(Point2(0, 0), Point2(_size.x, 0), Color(0, 0, 0, 1));
				draw_line(Point2(0, _size.y), Point2(_size.x, _size.y), Color(0, 0, 0, 1));
				draw_line(Point2(0, 0), Point2(0, _size.y), Color(0, 0, 0, 1));
				draw_line(Point2(_size.x, 0), Point2(_size.x, _size.y), Color(0, 0, 0, 1));
			}
		} break;
		default:
			break;
	}
}

void PaintNode::_bind_methods() {
	ADD_SIGNAL(MethodInfo("size_changed"));

	ClassDB::bind_method(D_METHOD("get_size"), &PaintNode::get_size);
	ClassDB::bind_method(D_METHOD("set_size", "size"), &PaintNode::set_size);
	ADD_PROPERTY(PropertyInfo(Variant::VECTOR2I, "size"), "set_size", "get_size");

	ClassDB::bind_method(D_METHOD("get_draw_outline"), &PaintNode::get_draw_outline);
	ClassDB::bind_method(D_METHOD("set_draw_outline", "val"), &PaintNode::set_draw_outline);
	ADD_PROPERTY(PropertyInfo(Variant::BOOL, "draw_outline"), "set_draw_outline", "get_draw_outline");

	BIND_VMETHOD(MethodInfo(PropertyInfo(Variant::OBJECT, "r", PROPERTY_HINT_RESOURCE_TYPE, "Image"), "_save_image"));
	ClassDB::bind_method(D_METHOD("save_image"), &PaintNode::save_image);
	ClassDB::bind_method(D_METHOD("_save_image"), &PaintNode::_save_image);

	BIND_VMETHOD(MethodInfo(PropertyInfo(Variant::OBJECT, "r", PROPERTY_HINT_RESOURCE_TYPE, "Image"), "_get_save_image"));
	ClassDB::bind_method(D_METHOD("get_save_image"), &PaintNode::get_save_image);
	ClassDB::bind_method(D_METHOD("_get_save_image"), &PaintNode::_get_save_image);

	ClassDB::bind_method(D_METHOD("util_get_pixels_in_line", "from", "to"), &PaintNode::util_get_pixels_in_line);

	ClassDB::bind_method(D_METHOD("util_to_1d_v", "p", "w"), &PaintNode::util_to_1d_v);
	ClassDB::bind_method(D_METHOD("util_to_1d", "x", "y", "w"), &PaintNode::util_to_1d);
	ClassDB::bind_method(D_METHOD("util_to_2d", "idx", "w"), &PaintNode::util_to_2d);

	ClassDB::bind_method(D_METHOD("util_color_from_array", "color_array"), &PaintNode::util_color_from_array);
	ClassDB::bind_method(D_METHOD("util_random_color"), &PaintNode::util_random_color);
	ClassDB::bind_method(D_METHOD("util_random_color_alt"), &PaintNode::util_random_color_alt);

	BIND_VMETHOD(MethodInfo(Variant::BOOL, "_forward_canvas_gui_input", PropertyInfo(Variant::OBJECT, "event", PROPERTY_HINT_RESOURCE_TYPE, "InputEvent")));
	ClassDB::bind_method(D_METHOD("forward_canvas_gui_input", "event"), &PaintNode::forward_canvas_gui_input);
	ClassDB::bind_method(D_METHOD("_forward_canvas_gui_input", "event"), &PaintNode::_forward_canvas_gui_input);

	ClassDB::bind_method(D_METHOD("get_paint_project"), &PaintNode::get_paint_project);
	ClassDB::bind_method(D_METHOD("get_parent_paint_node"), &PaintNode::get_parent_paint_node);
	ClassDB::bind_method(D_METHOD("find_parent_paint_node"), &PaintNode::find_parent_paint_node);

	BIND_CONSTANT(NOTIFICATION_PARENT_PAINT_NODE_RESIZED);
	BIND_CONSTANT(NOTIFICATION_PAINT_PROJECT_PRE_SAVE);
	BIND_CONSTANT(NOTIFICATION_PAINT_PROJECT_POST_SAVE);
}