add screen color picking

This commit is contained in:
MarianoGNU 2015-12-25 11:08:02 -03:00
parent ad2387f0be
commit d1ee2ab893
2 changed files with 49 additions and 3 deletions

View File

@ -29,6 +29,7 @@
#include "color_picker.h"
#include "scene/gui/separator.h"
#include "scene/main/viewport.h"
#include "os/os.h"
#include "os/input.h"
#include "os/keyboard.h"
@ -282,8 +283,8 @@ void ColorPicker::_preset_input(const InputEvent &ev) {
_update_color();
emit_signal("color_changed", color);
} else if (ev.type == InputEvent::MOUSE_MOTION) {
const InputEventMouseButton &bev = ev.mouse_button;
int index = bev.x/(preset->get_size().x/presets.size());
const InputEventMouse &mev = ev.mouse_motion;
int index = mev.x/(preset->get_size().x/presets.size());
if (index<0 || index >= presets.size())
return;
preset->set_tooltip("Color: #"+presets[index].to_html(presets[index].a<1)+"\n"
@ -292,10 +293,46 @@ void ColorPicker::_preset_input(const InputEvent &ev) {
}
}
void ColorPicker::_screen_input(const InputEvent &ev)
{
if (ev.type==InputEvent::MOUSE_BUTTON) {
const InputEventMouseButton &bev = ev.mouse_button;
if (bev.button_index==BUTTON_LEFT&&!bev.pressed) {
emit_signal("color_changed", color);
screen->hide();
}
} else if (ev.type==InputEvent::MOUSE_MOTION) {
const InputEventMouse &mev = ev.mouse_motion;
Viewport *r=get_tree()->get_root();
if (!r->get_rect().has_point(Point2(mev.global_x,mev.global_y)))
return;
Image &img =r->get_screen_capture();
if (!img.empty())
last_capture=img;
r->queue_screen_capture();
if (!last_capture.empty())
set_color(last_capture.get_pixel(mev.global_x,mev.global_y));
}
}
void ColorPicker::_add_preset_pressed() {
add_preset(color);
}
void ColorPicker::_screen_pick_pressed()
{
Viewport *r=get_tree()->get_root();
if (!screen) {
screen=memnew( Control );
r->add_child(screen);
screen->set_area_as_parent_rect();
screen->connect("input_event",this,"_screen_input");
}
screen->raise();
screen->show();
r->queue_screen_capture();
}
void ColorPicker::_bind_methods() {
ObjectTypeDB::bind_method(_MD("set_color","color"),&ColorPicker::set_color);
@ -308,10 +345,12 @@ void ColorPicker::_bind_methods() {
ObjectTypeDB::bind_method(_MD("_value_changed"),&ColorPicker::_value_changed);
ObjectTypeDB::bind_method(_MD("_html_entered"),&ColorPicker::_html_entered);
ObjectTypeDB::bind_method(_MD("_add_preset_pressed"), &ColorPicker::_add_preset_pressed);
ObjectTypeDB::bind_method(_MD("_screen_pick_pressed"), &ColorPicker::_screen_pick_pressed);
ObjectTypeDB::bind_method(_MD("_sample_draw"),&ColorPicker::_sample_draw);
ObjectTypeDB::bind_method(_MD("_uv_input"),&ColorPicker::_uv_input);
ObjectTypeDB::bind_method(_MD("_w_input"),&ColorPicker::_w_input);
ObjectTypeDB::bind_method(_MD("_preset_input"),&ColorPicker::_preset_input);
ObjectTypeDB::bind_method(_MD("_screen_input"),&ColorPicker::_screen_input);
ADD_SIGNAL( MethodInfo("color_changed",PropertyInfo(Variant::COLOR,"color")));
}
@ -323,9 +362,12 @@ ColorPicker::ColorPicker() :
edit_alpha=true;
raw_mode_enabled=false;
changing_color=false;
screen=NULL;
HBoxContainer *hb_smpl = memnew( HBoxContainer );
btn_pick = memnew( ToolButton );
btn_pick->connect("pressed",this,"_screen_pick_pressed");
sample = memnew( TextureFrame );
sample->set_h_size_flags(SIZE_EXPAND_FILL);
sample->connect("draw",this,"_sample_draw");
@ -460,7 +502,7 @@ ColorPicker::ColorPicker() :
bt_add_preset = memnew ( Button );
bt_add_preset->set_icon(get_icon("add_preset"));
bt_add_preset->set_tooltip("Add current color as a preset");
bt_add_preset->connect("pressed", this, "_add_preset_pressed", Vector<Variant>());
bt_add_preset->connect("pressed", this, "_add_preset_pressed");
bbc->add_child(bt_add_preset);
}

View File

@ -47,6 +47,8 @@ class ColorPicker : public BoxContainer {
private:
Control *screen;
Image last_capture;
TextureFrame *uv_edit;
TextureFrame *w_edit;
TextureFrame *sample;
@ -81,7 +83,9 @@ private:
void _uv_input(const InputEvent& p_input);
void _w_input(const InputEvent& p_input);
void _preset_input(const InputEvent& p_input);
void _screen_input(const InputEvent& p_input);
void _add_preset_pressed();
void _screen_pick_pressed();
protected:
void _notification(int);