/*
Copyright (c) 2019 Flairieve
Copyright (c) 2020-2022 cobrapitz
Copyright (c) 2022 Péter Magyar

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software && associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, && to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice && this permission notice shall be included in all
copies || substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE && NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS || COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES || OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT || OTHERWISE, ARISING FROM,
OUT OF || IN CONNECTION WITH THE SOFTWARE || THE USE || OTHER DEALINGS IN THE
SOFTWARE.
*/

#include "paint_action.h"

#include "../paint_canvas.h"
#include "../paint_canvas_layer.h"
#include "core/object/object.h"

Dictionary PaintAction::get_action_data_undo() {
	return action_data_undo;
}
void PaintAction::set_action_data_undo(const Dictionary &val) {
	action_data_undo = val;
}

Dictionary PaintAction::get_action_data_redo() {
	return action_data_redo;
}
void PaintAction::set_action_data_redo(const Dictionary &val) {
	action_data_redo = val;
}

Dictionary PaintAction::get_action_data_preview() {
	return action_data_preview;
}
void PaintAction::set_action_data_preview(const Dictionary &val) {
	action_data_preview = val;
}

PoolVector2iArray PaintAction::get_undo_cells() {
	return undo_cells;
}
void PaintAction::set_undo_cells(const PoolVector2iArray &val) {
	undo_cells = val;
}

PoolColorArray PaintAction::get_undo_colors() {
	return undo_colors;
}
void PaintAction::set_undo_colors(const PoolColorArray &val) {
	undo_colors = val;
}

PoolVector2iArray PaintAction::get_redo_cells() {
	return redo_cells;
}
void PaintAction::set_redo_cells(const PoolVector2iArray &val) {
	redo_cells = val;
}

PoolColorArray PaintAction::get_redo_colors() {
	return redo_colors;
}
void PaintAction::set_redo_colors(const PoolColorArray &val) {
	redo_colors = val;
}

PoolVector2iArray PaintAction::get_preview_cells() {
	return preview_cells;
}
void PaintAction::set_preview_cells(const PoolVector2iArray &val) {
	preview_cells = val;
}

PoolColorArray PaintAction::get_preview_colors() {
	return preview_colors;
}
void PaintAction::set_preview_colors(const PoolColorArray &val) {
	preview_colors = val;
}

Ref<PaintCanvasLayer> PaintAction::get_layer() {
	return layer;
}
void PaintAction::set_layer(const Ref<PaintCanvasLayer> &val) {
	layer = val;
}

Dictionary PaintAction::get_action_data() {
	return action_data;
}
void PaintAction::set_action_data(const Dictionary &val) {
	action_data = val;
}

void PaintAction::do_action(PaintCanvas *canvas, const Array &data) {
	layer = canvas->get_active_layer();
}
void PaintAction::commit_action(PaintCanvas *canvas) {
}

void PaintAction::undo_action(PaintCanvas *canvas) {
	for (int idx = 0; idx < undo_cells.size(); ++idx) {
		canvas->_set_pixel_v(layer, undo_cells[idx], undo_colors[idx]);
	}
}
void PaintAction::redo_action(PaintCanvas *canvas) {
	for (int idx = 0; idx < redo_cells.size(); ++idx) {
		canvas->_set_pixel_v(layer, redo_cells[idx], redo_colors[idx]);
	}
}

bool PaintAction::can_commit() {
	return !redo_cells.empty();
}

PoolVector2iArray PaintAction::get_x_sym_points(const int canvas_width, const Vector2i &pixel) {
	int p = canvas_width - pixel.x;

	PoolVector2iArray points;
	points.append(pixel);
	points.append(Vector2i(p - 1, pixel.y));

	return points;
}
PoolVector2iArray PaintAction::get_y_sym_points(const int canvas_height, const Vector2i &pixel) {
	int p = canvas_height - pixel.y;

	PoolVector2iArray points;
	points.append(pixel);
	points.append(Vector2i(pixel.x, p - 1));

	return points;
}
PoolVector2iArray PaintAction::get_xy_sym_points(const int canvas_width, const int canvas_height, const Vector2i &pixel) {
	PoolVector2iArray all_points;
	PoolVector2iArray xpoints = get_x_sym_points(canvas_width, pixel);

	all_points.append_array(get_y_sym_points(canvas_height, xpoints[0]));
	all_points.append_array(get_y_sym_points(canvas_height, xpoints[1]));

	PoolVector2iArray points;
	for (int i = 0; i < all_points.size(); ++i) { //point in all_points:
		Vector2i point = all_points[i];
		bool found = false;

		for (int j = 0; j < points.size(); ++j) {
			if (points[j] == point) {
				found = true;
				break;
			}
		}

		if (!found) {
			points.append(point);
		}
	}

	return points;
}

PoolVector2iArray PaintAction::get_points(PaintCanvas *canvas, const Vector2i &pixel) {
	PoolVector2iArray points;

	if (canvas->symmetry_x && canvas->symmetry_y) {
		PoolVector2iArray sym_points = get_xy_sym_points(canvas->get_canvas_width(), canvas->get_canvas_height(), pixel);

		for (int i = 0; i < sym_points.size(); ++i) {
			Vector2i point = sym_points[i];

			if (undo_cells.contains(point) || !canvas->validate_pixel_v(point)) {
				continue;
			}

			if (canvas->is_alpha_locked() && canvas->get_pixel_v(pixel) == Color(0, 0, 0, 0)) {
				continue;
			}

			points.append(point);
		}
	} else if (canvas->symmetry_y) {
		PoolVector2iArray sym_points = get_y_sym_points(canvas->get_canvas_height(), pixel);

		for (int i = 0; i < sym_points.size(); ++i) {
			Vector2i point = sym_points[i];

			if (undo_cells.contains(point) || !canvas->validate_pixel_v(point)) {
				continue;
			}

			if (canvas->is_alpha_locked() && canvas->get_pixel_v(pixel) == Color(0, 0, 0, 0)) {
				continue;
			}

			points.append(point);
		}
	} else if (canvas->symmetry_x) {
		PoolVector2iArray sym_points = get_x_sym_points(canvas->get_canvas_width(), pixel);

		for (int i = 0; i < sym_points.size(); ++i) {
			Vector2i point = sym_points[i];

			if (undo_cells.contains(point) || !canvas->validate_pixel_v(point)) {
				continue;
			}

			if (canvas->is_alpha_locked() && canvas->get_pixel_v(pixel) == Color(0, 0, 0, 0)) {
				continue;
			}

			points.append(point);
		}
	} else {
		if (undo_cells.contains(pixel) || !canvas->validate_pixel_v(pixel)) {
			//empty
			return points;
		}

		if (canvas->is_alpha_locked() && canvas->get_pixel_v(pixel) == Color(0, 0, 0, 0)) {
			//empty
			return points;
		}

		points.append(pixel);
	}

	return points;
}

void PaintAction::draw_points(PaintCanvas *canvas, const PoolVector2iArray &point_arr, const PoolColorArray &color_arr) {
	for (int i = 0; i < point_arr.size(); ++i) {
		Vector2i pixel = point_arr[i];

		Color col = color_arr[i];

		if (canvas->is_alpha_locked() && col.a < 0.00001) {
			continue;
		}

		undo_cells.append(pixel);
		undo_colors.append(col);

		canvas->set_pixel_v(pixel, col);

		redo_cells.append(pixel);
		redo_colors.append(col);
	}
}

PaintAction::PaintAction() {
}

PaintAction::~PaintAction() {
}

void PaintAction::_bind_methods() {
	ClassDB::bind_method(D_METHOD("get_action_data_undo"), &PaintAction::get_action_data_undo);
	ClassDB::bind_method(D_METHOD("set_action_data_undo", "value"), &PaintAction::set_action_data_undo);
	ADD_PROPERTY(PropertyInfo(Variant::DICTIONARY, "action_data_undo"), "set_action_data_undo", "get_action_data_undo");

	ClassDB::bind_method(D_METHOD("get_action_data_redo"), &PaintAction::get_action_data_redo);
	ClassDB::bind_method(D_METHOD("set_action_data_redo", "value"), &PaintAction::set_action_data_redo);
	ADD_PROPERTY(PropertyInfo(Variant::DICTIONARY, "action_data_redo"), "set_action_data_redo", "get_action_data_redo");

	ClassDB::bind_method(D_METHOD("get_action_data_preview"), &PaintAction::get_action_data_preview);
	ClassDB::bind_method(D_METHOD("set_action_data_preview", "value"), &PaintAction::set_action_data_preview);
	ADD_PROPERTY(PropertyInfo(Variant::DICTIONARY, "action_data_preview"), "set_action_data_preview", "get_action_data_preview");

	ClassDB::bind_method(D_METHOD("get_undo_cells"), &PaintAction::get_undo_cells);
	ClassDB::bind_method(D_METHOD("set_undo_cells", "value"), &PaintAction::set_undo_cells);
	ADD_PROPERTY(PropertyInfo(Variant::POOL_VECTOR2I_ARRAY, "undo_cells"), "set_undo_cells", "get_undo_cells");

	ClassDB::bind_method(D_METHOD("get_undo_colors"), &PaintAction::get_undo_colors);
	ClassDB::bind_method(D_METHOD("set_undo_colors", "value"), &PaintAction::set_undo_colors);
	ADD_PROPERTY(PropertyInfo(Variant::POOL_COLOR_ARRAY, "undo_colors"), "set_undo_colors", "get_undo_colors");

	ClassDB::bind_method(D_METHOD("get_redo_cells"), &PaintAction::get_redo_cells);
	ClassDB::bind_method(D_METHOD("set_redo_cells", "value"), &PaintAction::set_redo_cells);
	ADD_PROPERTY(PropertyInfo(Variant::POOL_VECTOR2I_ARRAY, "redo_cells"), "set_redo_cells", "get_redo_cells");

	ClassDB::bind_method(D_METHOD("get_redo_colors"), &PaintAction::get_redo_colors);
	ClassDB::bind_method(D_METHOD("set_redo_colors", "value"), &PaintAction::set_redo_colors);
	ADD_PROPERTY(PropertyInfo(Variant::POOL_COLOR_ARRAY, "redo_colors"), "set_redo_colors", "get_redo_colors");

	ClassDB::bind_method(D_METHOD("get_preview_cells"), &PaintAction::get_preview_cells);
	ClassDB::bind_method(D_METHOD("set_preview_cells", "value"), &PaintAction::set_preview_cells);
	ADD_PROPERTY(PropertyInfo(Variant::POOL_VECTOR2I_ARRAY, "preview_cells"), "set_preview_cells", "get_preview_cells");

	ClassDB::bind_method(D_METHOD("get_preview_colors"), &PaintAction::get_preview_colors);
	ClassDB::bind_method(D_METHOD("set_preview_colors", "value"), &PaintAction::set_preview_colors);
	ADD_PROPERTY(PropertyInfo(Variant::POOL_COLOR_ARRAY, "preview_colors"), "set_preview_colors", "get_preview_colors");

	//ClassDB::bind_method(D_METHOD("get_layer"), &PaintAction::get_layer);
	//ClassDB::bind_method(D_METHOD("set_layer", "value"), &PaintAction::set_layer);
	//ADD_PROPERTY(PropertyInfo(Variant::OBJECT, "layer", PROPERTY_HINT_RESOURCE_TYPE, "PaintCanvasLayer", 0), "set_layer", "get_layer");

	ClassDB::bind_method(D_METHOD("get_action_data"), &PaintAction::get_action_data);
	ClassDB::bind_method(D_METHOD("set_action_data", "value"), &PaintAction::set_action_data);
	ADD_PROPERTY(PropertyInfo(Variant::DICTIONARY, "action_data"), "set_action_data", "get_action_data");
}