godot/editor/plugins/curve_editor_plugin.cpp

520 lines
13 KiB
C++

#include "curve_editor_plugin.h"
#include "canvas_item_editor_plugin.h"
#include "os/keyboard.h"
#include "spatial_editor_plugin.h"
void CurveTextureEdit::_gui_input(const InputEvent &p_event) {
if (p_event.type == InputEvent::KEY && p_event.key.pressed && p_event.key.scancode == KEY_DELETE && grabbed != -1) {
points.remove(grabbed);
grabbed = -1;
update();
emit_signal("curve_changed");
accept_event();
}
if (p_event.type == InputEvent::MOUSE_BUTTON && p_event.mouse_button.button_index == 1 && p_event.mouse_button.pressed) {
update();
Ref<Font> font = get_font("font", "Label");
int font_h = font->get_height();
Vector2 size = get_size();
size.y -= font_h;
Point2 p = Vector2(p_event.mouse_button.x, p_event.mouse_button.y) / size;
p.y = CLAMP(1.0 - p.y, 0, 1) * (max - min) + min;
grabbed = -1;
grabbing = true;
for (int i = 0; i < points.size(); i++) {
Vector2 ps = p * get_size();
Vector2 pt = Vector2(points[i].offset, points[i].height) * get_size();
if (ps.distance_to(pt) < 4) {
grabbed = i;
}
}
//grab or select
if (grabbed != -1) {
return;
}
//insert
Point np;
np.offset = p.x;
np.height = p.y;
points.push_back(np);
points.sort();
for (int i = 0; i < points.size(); i++) {
if (points[i].offset == p.x && points[i].height == p.y) {
grabbed = i;
break;
}
}
emit_signal("curve_changed");
}
if (p_event.type == InputEvent::MOUSE_BUTTON && p_event.mouse_button.button_index == 1 && !p_event.mouse_button.pressed) {
if (grabbing) {
grabbing = false;
emit_signal("curve_changed");
}
update();
}
if (p_event.type == InputEvent::MOUSE_MOTION && grabbing && grabbed != -1) {
Ref<Font> font = get_font("font", "Label");
int font_h = font->get_height();
Vector2 size = get_size();
size.y -= font_h;
Point2 p = Vector2(p_event.mouse_motion.x, p_event.mouse_motion.y) / size;
p.y = CLAMP(1.0 - p.y, 0, 1) * (max - min) + min;
p.x = CLAMP(p.x, 0.0, 1.0);
bool valid = true;
for (int i = 0; i < points.size(); i++) {
if (points[i].offset == p.x && points[i].height == p.y && i != grabbed) {
valid = false;
}
}
if (!valid)
return;
points[grabbed].offset = p.x;
points[grabbed].height = p.y;
points.sort();
for (int i = 0; i < points.size(); i++) {
if (points[i].offset == p.x && points[i].height == p.y) {
grabbed = i;
break;
}
}
emit_signal("curve_changed");
update();
}
}
void CurveTextureEdit::_plot_curve(const Vector2 &p_a, const Vector2 &p_b, const Vector2 &p_c, const Vector2 &p_d) {
Ref<Font> font = get_font("font", "Label");
int font_h = font->get_height();
float geometry[4][4];
float tmp1[4][4];
float tmp2[4][4];
float deltas[4][4];
double x, dx, dx2, dx3;
double y, dy, dy2, dy3;
double d, d2, d3;
int lastx, lasty;
int newx, newy;
int ntimes;
int i, j;
int xmax = get_size().x;
int ymax = get_size().y - font_h;
int vsplits = 4;
int zero_ofs = (1.0 - (0.0 - min) / (max - min)) * ymax;
draw_line(Vector2(0, zero_ofs), Vector2(xmax, zero_ofs), Color(0.8, 0.8, 0.8, 0.15), 2.0);
for (int i = 0; i <= vsplits; i++) {
float fofs = float(i) / vsplits;
int yofs = fofs * ymax;
draw_line(Vector2(xmax, yofs), Vector2(xmax - 4, yofs), Color(0.8, 0.8, 0.8, 0.8), 2.0);
String text = rtos((1.0 - fofs) * (max - min) + min);
int ppos = text.find(".");
if (ppos != -1) {
if (text.length() > ppos + 2)
text = text.substr(0, ppos + 2);
}
int size = font->get_string_size(text).x;
int xofs = xmax - size - 4;
yofs -= font_h / 2;
if (yofs < 2) {
yofs = 2;
} else if (yofs + font_h > ymax - 2) {
yofs = ymax - font_h - 2;
}
draw_string(font, Vector2(xofs, yofs + font->get_ascent()), text, Color(0.8, 0.8, 0.8, 1));
}
/* construct the geometry matrix from the segment */
for (i = 0; i < 4; i++) {
geometry[i][2] = 0;
geometry[i][3] = 0;
}
geometry[0][0] = (p_a[0] * xmax);
geometry[1][0] = (p_b[0] * xmax);
geometry[2][0] = (p_c[0] * xmax);
geometry[3][0] = (p_d[0] * xmax);
geometry[0][1] = ((p_a[1] - min) / (max - min) * ymax);
geometry[1][1] = ((p_b[1] - min) / (max - min) * ymax);
geometry[2][1] = ((p_c[1] - min) / (max - min) * ymax);
geometry[3][1] = ((p_d[1] - min) / (max - min) * ymax);
/* subdivide the curve ntimes (1000) times */
ntimes = 4 * xmax;
/* ntimes can be adjusted to give a finer or coarser curve */
d = 1.0 / ntimes;
d2 = d * d;
d3 = d * d * d;
/* construct a temporary matrix for determining the forward differencing deltas */
tmp2[0][0] = 0;
tmp2[0][1] = 0;
tmp2[0][2] = 0;
tmp2[0][3] = 1;
tmp2[1][0] = d3;
tmp2[1][1] = d2;
tmp2[1][2] = d;
tmp2[1][3] = 0;
tmp2[2][0] = 6 * d3;
tmp2[2][1] = 2 * d2;
tmp2[2][2] = 0;
tmp2[2][3] = 0;
tmp2[3][0] = 6 * d3;
tmp2[3][1] = 0;
tmp2[3][2] = 0;
tmp2[3][3] = 0;
/* compose the basis and geometry matrices */
static const float CR_basis[4][4] = {
{ -0.5, 1.5, -1.5, 0.5 },
{ 1.0, -2.5, 2.0, -0.5 },
{ -0.5, 0.0, 0.5, 0.0 },
{ 0.0, 1.0, 0.0, 0.0 },
};
for (i = 0; i < 4; i++) {
for (j = 0; j < 4; j++) {
tmp1[i][j] = (CR_basis[i][0] * geometry[0][j] +
CR_basis[i][1] * geometry[1][j] +
CR_basis[i][2] * geometry[2][j] +
CR_basis[i][3] * geometry[3][j]);
}
}
/* compose the above results to get the deltas matrix */
for (i = 0; i < 4; i++) {
for (j = 0; j < 4; j++) {
deltas[i][j] = (tmp2[i][0] * tmp1[0][j] +
tmp2[i][1] * tmp1[1][j] +
tmp2[i][2] * tmp1[2][j] +
tmp2[i][3] * tmp1[3][j]);
}
}
/* extract the x deltas */
x = deltas[0][0];
dx = deltas[1][0];
dx2 = deltas[2][0];
dx3 = deltas[3][0];
/* extract the y deltas */
y = deltas[0][1];
dy = deltas[1][1];
dy2 = deltas[2][1];
dy3 = deltas[3][1];
lastx = CLAMP(x, 0, xmax);
lasty = CLAMP(y, 0, ymax);
/* if (fix255)
{
cd->curve[cd->outline][lastx] = lasty;
}
else
{
cd->curve_ptr[cd->outline][lastx] = lasty;
if(gb_debug) printf("bender_plot_curve xmax:%d ymax:%d\n", (int)xmax, (int)ymax);
}
*/
/* loop over the curve */
for (i = 0; i < ntimes; i++) {
/* increment the x values */
x += dx;
dx += dx2;
dx2 += dx3;
/* increment the y values */
y += dy;
dy += dy2;
dy2 += dy3;
newx = CLAMP((Math::round(x)), 0, xmax);
newy = CLAMP((Math::round(y)), 0, ymax);
/* if this point is different than the last one...then draw it */
if ((lastx != newx) || (lasty != newy)) {
#if 0
if(fix255)
{
/* use fixed array size (for the curve graph) */
cd->curve[cd->outline][newx] = newy;
}
else
{
/* use dynamic allocated curve_ptr (for the real curve) */
cd->curve_ptr[cd->outline][newx] = newy;
if(gb_debug) printf("outline: %d cX: %d cY: %d\n", (int)cd->outline, (int)newx, (int)newy);
}
#endif
draw_line(Vector2(lastx, ymax - lasty), Vector2(newx, ymax - newy), Color(0.8, 0.8, 0.8, 0.8), 2.0);
}
lastx = newx;
lasty = newy;
}
int splits = 8;
draw_line(Vector2(0, ymax - 1), Vector2(xmax, ymax - 1), Color(0.8, 0.8, 0.8, 0.3), 2.0);
for (int i = 0; i <= splits; i++) {
float fofs = float(i) / splits;
draw_line(Vector2(fofs * xmax, ymax), Vector2(fofs * xmax, ymax - 2), Color(0.8, 0.8, 0.8, 0.8), 2.0);
String text = rtos(fofs);
int size = font->get_string_size(text).x;
int ofs = fofs * xmax - size * 0.5;
if (ofs < 2) {
ofs = 2;
} else if (ofs + size > xmax - 2) {
ofs = xmax - size - 2;
}
draw_string(font, Vector2(ofs, ymax + font->get_ascent()), text, Color(0.8, 0.8, 0.8, 1));
}
}
void CurveTextureEdit::_notification(int p_what) {
if (p_what == NOTIFICATION_DRAW) {
Ref<Font> font = get_font("font", "Label");
int font_h = font->get_height();
draw_style_box(get_stylebox("bg", "Tree"), Rect2(Point2(), get_size()));
int w = get_size().x;
int h = get_size().y;
Vector2 prev = Vector2(0, 0);
Vector2 prev2 = Vector2(0, 0);
for (int i = -1; i < points.size(); i++) {
Vector2 next;
Vector2 next2;
if (i + 1 >= points.size()) {
next = Vector2(1, 0);
} else {
next = Vector2(points[i + 1].offset, points[i + 1].height);
}
if (i + 2 >= points.size()) {
next2 = Vector2(1, 0);
} else {
next2 = Vector2(points[i + 2].offset, points[i + 2].height);
}
/*if (i==-1 && prev.offset==next.offset) {
prev=next;
continue;
}*/
_plot_curve(prev2, prev, next, next2);
prev2 = prev;
prev = next;
}
Vector2 size = get_size();
size.y -= font_h;
for (int i = 0; i < points.size(); i++) {
Color col = i == grabbed ? Color(1, 0.0, 0.0, 0.9) : Color(1, 1, 1, 0.8);
float h = (points[i].height - min) / (max - min);
draw_rect(Rect2(Vector2(points[i].offset, 1.0 - h) * size - Vector2(2, 2), Vector2(5, 5)), col);
}
/* if (grabbed!=-1) {
draw_rect(Rect2(total_w+3,0,h,h),points[grabbed].color);
}
*/
if (has_focus()) {
draw_line(Vector2(-1, -1), Vector2(w + 1, -1), Color(1, 1, 1, 0.6));
draw_line(Vector2(w + 1, -1), Vector2(w + 1, h + 1), Color(1, 1, 1, 0.6));
draw_line(Vector2(w + 1, h + 1), Vector2(-1, h + 1), Color(1, 1, 1, 0.6));
draw_line(Vector2(-1, -1), Vector2(-1, h + 1), Color(1, 1, 1, 0.6));
}
}
}
Size2 CurveTextureEdit::get_minimum_size() const {
return Vector2(64, 64);
}
void CurveTextureEdit::set_range(float p_min, float p_max) {
max = p_max;
min = p_min;
update();
}
void CurveTextureEdit::set_points(const Vector<Vector2> &p_points) {
points.clear();
for (int i = 0; i < p_points.size(); i++) {
Point p;
p.offset = p_points[i].x;
p.height = p_points[i].y;
points.push_back(p);
}
points.sort();
update();
}
Vector<Vector2> CurveTextureEdit::get_points() const {
Vector<Vector2> ret;
for (int i = 0; i < points.size(); i++)
ret.push_back(Vector2(points[i].offset, points[i].height));
return ret;
}
void CurveTextureEdit::_bind_methods() {
ClassDB::bind_method(D_METHOD("_gui_input"), &CurveTextureEdit::_gui_input);
ADD_SIGNAL(MethodInfo("curve_changed"));
}
CurveTextureEdit::CurveTextureEdit() {
grabbed = -1;
grabbing = false;
max = 1;
min = 0;
set_focus_mode(FOCUS_ALL);
}
void CurveTextureEditorPlugin::_curve_settings_changed() {
if (!curve_texture_ref.is_valid())
return;
curve_editor->set_points(Variant(curve_texture_ref->get_points()));
curve_editor->set_range(curve_texture_ref->get_min(), curve_texture_ref->get_max());
}
CurveTextureEditorPlugin::CurveTextureEditorPlugin(EditorNode *p_node) {
editor = p_node;
curve_editor = memnew(CurveTextureEdit);
curve_button = editor->add_bottom_panel_item("CurveTexture", curve_editor);
curve_button->hide();
curve_editor->set_custom_minimum_size(Size2(100, 128 * EDSCALE));
curve_editor->hide();
curve_editor->connect("curve_changed", this, "curve_changed");
}
void CurveTextureEditorPlugin::edit(Object *p_object) {
if (curve_texture_ref.is_valid()) {
curve_texture_ref->disconnect("changed", this, "_curve_settings_changed");
}
CurveTexture *curve_texture = p_object->cast_to<CurveTexture>();
if (!curve_texture)
return;
curve_texture_ref = Ref<CurveTexture>(curve_texture);
curve_editor->set_points(Variant(curve_texture_ref->get_points()));
curve_editor->set_range(curve_texture_ref->get_min(), curve_texture_ref->get_max());
if (!curve_texture_ref->is_connected("changed", this, "_curve_settings_changed")) {
curve_texture_ref->connect("changed", this, "_curve_settings_changed");
}
}
bool CurveTextureEditorPlugin::handles(Object *p_object) const {
return p_object->is_class("CurveTexture");
}
void CurveTextureEditorPlugin::make_visible(bool p_visible) {
if (p_visible) {
curve_button->show();
editor->make_bottom_panel_item_visible(curve_editor);
} else {
curve_button->hide();
if (curve_editor->is_visible_in_tree())
editor->hide_bottom_panel();
}
}
void CurveTextureEditorPlugin::_curve_changed() {
if (curve_texture_ref.is_valid()) {
UndoRedo *ur = EditorNode::get_singleton()->get_undo_redo();
Vector<Vector2> points = curve_editor->get_points();
PoolVector<Vector2> ppoints = Variant(points);
ur->create_action(TTR("Modify Curve"), UndoRedo::MERGE_ENDS);
ur->add_do_method(this, "undo_redo_curve_texture", ppoints);
ur->add_undo_method(this, "undo_redo_curve_texture", curve_texture_ref->get_points());
ur->commit_action();
}
}
void CurveTextureEditorPlugin::_undo_redo_curve_texture(const PoolVector<Vector2> &points) {
curve_texture_ref->set_points(points);
curve_editor->set_points(Variant(curve_texture_ref->get_points()));
curve_editor->update();
}
CurveTextureEditorPlugin::~CurveTextureEditorPlugin() {
}
void CurveTextureEditorPlugin::_bind_methods() {
ClassDB::bind_method(D_METHOD("curve_changed"), &CurveTextureEditorPlugin::_curve_changed);
ClassDB::bind_method(D_METHOD("_curve_settings_changed"), &CurveTextureEditorPlugin::_curve_settings_changed);
ClassDB::bind_method(D_METHOD("undo_redo_curve_texture", "points"), &CurveTextureEditorPlugin::_undo_redo_curve_texture);
}