mirror of
				https://github.com/godotengine/godot.git
				synced 2025-11-04 07:31:16 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			342 lines
		
	
	
	
		
			9.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			342 lines
		
	
	
	
		
			9.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*************************************************************************/
 | 
						|
/*  option_button.cpp                                                    */
 | 
						|
/*************************************************************************/
 | 
						|
/*                       This file is part of:                           */
 | 
						|
/*                           GODOT ENGINE                                */
 | 
						|
/*                    http://www.godotengine.org                         */
 | 
						|
/*************************************************************************/
 | 
						|
/* Copyright (c) 2007-2016 Juan Linietsky, Ariel Manzur.                 */
 | 
						|
/*                                                                       */
 | 
						|
/* Permission is hereby granted, free of charge, to any person obtaining */
 | 
						|
/* a copy of this software and 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, and to    */
 | 
						|
/* permit persons to whom the Software is furnished to do so, subject to */
 | 
						|
/* the following conditions:                                             */
 | 
						|
/*                                                                       */
 | 
						|
/* The above copyright notice and this permission notice shall be        */
 | 
						|
/* included in all copies or 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 AND NONINFRINGEMENT.*/
 | 
						|
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY  */
 | 
						|
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,  */
 | 
						|
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE     */
 | 
						|
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                */
 | 
						|
/*************************************************************************/
 | 
						|
#include "option_button.h"
 | 
						|
#include "print_string.h"
 | 
						|
 | 
						|
 | 
						|
Size2 OptionButton::get_minimum_size() const {
 | 
						|
 | 
						|
 | 
						|
	Size2 minsize = Button::get_minimum_size();
 | 
						|
 | 
						|
	if (has_icon("arrow"))
 | 
						|
		minsize.width+=Control::get_icon("arrow")->get_width();
 | 
						|
 | 
						|
	return minsize;
 | 
						|
}
 | 
						|
 | 
						|
void OptionButton::_notification(int p_what) {
 | 
						|
 | 
						|
 | 
						|
	switch(p_what) {
 | 
						|
 | 
						|
		case NOTIFICATION_DRAW: {
 | 
						|
 | 
						|
			if (!has_icon("arrow"))
 | 
						|
				return;
 | 
						|
 | 
						|
			RID ci = get_canvas_item();
 | 
						|
			Ref<Texture> arrow = Control::get_icon("arrow");
 | 
						|
			Ref<StyleBox> normal = get_stylebox("normal" );
 | 
						|
 | 
						|
			Size2 size = get_size();
 | 
						|
 | 
						|
			Point2 ofs( size.width - arrow->get_width() - get_constant("arrow_margin"), int(Math::abs((size.height-arrow->get_height())/2)));
 | 
						|
			arrow->draw(ci,ofs);
 | 
						|
 | 
						|
		} break;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
void OptionButton::_selected(int p_which) {
 | 
						|
 | 
						|
	int selid = -1;
 | 
						|
	for (int i=0;i<popup->get_item_count();i++) {
 | 
						|
 | 
						|
		bool is_clicked = popup->get_item_ID(i)==p_which;
 | 
						|
		if (is_clicked) {
 | 
						|
			selid=i;
 | 
						|
			break;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (selid==-1 && p_which>=0 && p_which<popup->get_item_count()) {
 | 
						|
		_select(p_which,true);
 | 
						|
	} else {
 | 
						|
 | 
						|
		ERR_FAIL_COND(selid==-1);
 | 
						|
 | 
						|
		_select(selid,true);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
void OptionButton::pressed() {
 | 
						|
 | 
						|
	Size2 size=get_size();
 | 
						|
	popup->set_global_pos( get_global_pos() + Size2( 0, size.height ) );
 | 
						|
	popup->set_size( Size2( size.width, 0) );
 | 
						|
 | 
						|
	popup->popup();
 | 
						|
}
 | 
						|
 | 
						|
void OptionButton::add_icon_item(const Ref<Texture>& p_icon,const String& p_label,int p_ID) {
 | 
						|
 | 
						|
	popup->add_icon_check_item( p_icon, p_label, p_ID );
 | 
						|
	if (popup->get_item_count()==1)
 | 
						|
		select(0);
 | 
						|
}
 | 
						|
void OptionButton::add_item(const String& p_label,int p_ID) {
 | 
						|
 | 
						|
	popup->add_check_item( p_label, p_ID );
 | 
						|
	if (popup->get_item_count()==1)
 | 
						|
		select(0);
 | 
						|
}
 | 
						|
 | 
						|
void OptionButton::set_item_text(int p_idx,const String& p_text) {
 | 
						|
 | 
						|
	popup->set_item_text(p_idx,p_text);
 | 
						|
 | 
						|
}
 | 
						|
void OptionButton::set_item_icon(int p_idx,const Ref<Texture>& p_icon) {
 | 
						|
 | 
						|
	popup->set_item_icon(p_idx,p_icon);
 | 
						|
 | 
						|
}
 | 
						|
void OptionButton::set_item_ID(int p_idx,int p_ID) {
 | 
						|
 | 
						|
	popup->set_item_ID(p_idx,p_ID);
 | 
						|
}
 | 
						|
 | 
						|
void OptionButton::set_item_metadata(int p_idx,const Variant& p_metadata) {
 | 
						|
 | 
						|
	popup->set_item_metadata(p_idx,p_metadata);
 | 
						|
}
 | 
						|
 | 
						|
void OptionButton::set_item_disabled(int p_idx,bool p_disabled) {
 | 
						|
 | 
						|
	popup->set_item_disabled(p_idx,p_disabled);
 | 
						|
}
 | 
						|
 | 
						|
String OptionButton::get_item_text(int p_idx) const {
 | 
						|
 | 
						|
	return popup->get_item_text(p_idx);
 | 
						|
}
 | 
						|
 | 
						|
Ref<Texture> OptionButton::get_item_icon(int p_idx) const {
 | 
						|
 | 
						|
	return popup->get_item_icon(p_idx);
 | 
						|
}
 | 
						|
 | 
						|
int OptionButton::get_item_ID(int p_idx) const {
 | 
						|
 | 
						|
	return popup->get_item_ID(p_idx);
 | 
						|
}
 | 
						|
Variant OptionButton::get_item_metadata(int p_idx) const {
 | 
						|
 | 
						|
	return popup->get_item_metadata(p_idx);
 | 
						|
}
 | 
						|
 | 
						|
bool OptionButton::is_item_disabled(int p_idx) const {
 | 
						|
 | 
						|
	return popup->is_item_disabled(p_idx);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
int OptionButton::get_item_count() const  {
 | 
						|
 | 
						|
	return popup->get_item_count();
 | 
						|
}
 | 
						|
 | 
						|
void OptionButton::add_separator() {
 | 
						|
 | 
						|
	popup->add_separator();
 | 
						|
}
 | 
						|
 | 
						|
void OptionButton::clear() {
 | 
						|
 | 
						|
	popup->clear();
 | 
						|
	set_text("");
 | 
						|
	current=-1;
 | 
						|
}
 | 
						|
 | 
						|
void OptionButton::_select(int p_idx,bool p_emit) {
 | 
						|
 | 
						|
	if (p_idx<0)
 | 
						|
		return;
 | 
						|
	if (p_idx==current)
 | 
						|
		return;
 | 
						|
 | 
						|
	ERR_FAIL_INDEX( p_idx, popup->get_item_count() );
 | 
						|
 | 
						|
	for (int i=0;i<popup->get_item_count();i++) {
 | 
						|
 | 
						|
		popup->set_item_checked(i,i==p_idx);
 | 
						|
	}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
	current=p_idx;
 | 
						|
	set_text( popup->get_item_text( current ) );
 | 
						|
	set_icon( popup->get_item_icon( current ) );
 | 
						|
 | 
						|
	if (is_inside_tree() && p_emit)
 | 
						|
		emit_signal("item_selected",current);
 | 
						|
}
 | 
						|
 | 
						|
void OptionButton::_select_int(int p_which) {
 | 
						|
 | 
						|
	if (p_which<0 || p_which>=popup->get_item_count())
 | 
						|
		return;
 | 
						|
	_select(p_which,false);
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
void OptionButton::select(int p_idx) {
 | 
						|
 | 
						|
	_select(p_idx,false);
 | 
						|
}
 | 
						|
 | 
						|
int OptionButton::get_selected() const {
 | 
						|
 | 
						|
	return current;
 | 
						|
}
 | 
						|
 | 
						|
int OptionButton::get_selected_ID() const {
 | 
						|
 | 
						|
	int idx = get_selected();
 | 
						|
	if (idx<0)
 | 
						|
		return 0;
 | 
						|
	return get_item_ID(current);
 | 
						|
}
 | 
						|
Variant OptionButton::get_selected_metadata() const {
 | 
						|
 | 
						|
	int idx = get_selected();
 | 
						|
	if (idx<0)
 | 
						|
		return Variant();
 | 
						|
	return get_item_metadata(current);
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
void OptionButton::remove_item(int p_idx) {
 | 
						|
 | 
						|
	popup->remove_item(p_idx);
 | 
						|
}
 | 
						|
 | 
						|
Array OptionButton::_get_items() const {
 | 
						|
 | 
						|
	Array items;
 | 
						|
	for(int i=0;i<get_item_count();i++) {
 | 
						|
 | 
						|
		items.push_back(get_item_text(i));
 | 
						|
		items.push_back(get_item_icon(i));
 | 
						|
		items.push_back(is_item_disabled(i));
 | 
						|
		items.push_back(get_item_ID(i));
 | 
						|
		items.push_back(get_item_metadata(i));
 | 
						|
	}
 | 
						|
 | 
						|
	return items;
 | 
						|
 | 
						|
}
 | 
						|
void OptionButton::_set_items(const Array& p_items){
 | 
						|
 | 
						|
	ERR_FAIL_COND(p_items.size() % 5);
 | 
						|
	clear();
 | 
						|
 | 
						|
	for(int i=0;i<p_items.size();i+=5) {
 | 
						|
 | 
						|
		String text=p_items[i+0];
 | 
						|
		Ref<Texture> icon=p_items[i+1];
 | 
						|
		bool disabled=p_items[i+2];
 | 
						|
		int id=p_items[i+3];
 | 
						|
		Variant meta = p_items[i+4];
 | 
						|
 | 
						|
		int idx=get_item_count();
 | 
						|
		add_item(text,id);
 | 
						|
		set_item_icon(idx,icon);
 | 
						|
		set_item_disabled(idx,disabled);
 | 
						|
		set_item_metadata(idx,meta);
 | 
						|
	}
 | 
						|
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
void OptionButton::get_translatable_strings(List<String> *p_strings) const {
 | 
						|
 | 
						|
	return popup->get_translatable_strings(p_strings);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
void OptionButton::_bind_methods() {
 | 
						|
 | 
						|
	ObjectTypeDB::bind_method(_MD("_selected"),&OptionButton::_selected);
 | 
						|
 | 
						|
	ObjectTypeDB::bind_method(_MD("add_item","label","id"),&OptionButton::add_item,DEFVAL(-1));
 | 
						|
	ObjectTypeDB::bind_method(_MD("add_icon_item","texture:Texture","label","id"),&OptionButton::add_icon_item);
 | 
						|
	ObjectTypeDB::bind_method(_MD("set_item_text","idx","text"),&OptionButton::set_item_text);
 | 
						|
	ObjectTypeDB::bind_method(_MD("set_item_icon","idx","texture:Texture"),&OptionButton::set_item_icon);
 | 
						|
	ObjectTypeDB::bind_method(_MD("set_item_disabled","idx","disabled"),&OptionButton::set_item_disabled);
 | 
						|
	ObjectTypeDB::bind_method(_MD("set_item_ID","idx","id"),&OptionButton::set_item_ID);
 | 
						|
	ObjectTypeDB::bind_method(_MD("set_item_metadata","idx","metadata"),&OptionButton::set_item_metadata);
 | 
						|
	ObjectTypeDB::bind_method(_MD("get_item_text","idx"),&OptionButton::get_item_text);
 | 
						|
	ObjectTypeDB::bind_method(_MD("get_item_icon:Texture","idx"),&OptionButton::get_item_icon);
 | 
						|
	ObjectTypeDB::bind_method(_MD("get_item_ID","idx"),&OptionButton::get_item_ID);
 | 
						|
	ObjectTypeDB::bind_method(_MD("get_item_metadata","idx"),&OptionButton::get_item_metadata);
 | 
						|
	ObjectTypeDB::bind_method(_MD("is_item_disabled","idx"),&OptionButton::is_item_disabled);
 | 
						|
	ObjectTypeDB::bind_method(_MD("get_item_count"),&OptionButton::get_item_count);
 | 
						|
	ObjectTypeDB::bind_method(_MD("add_separator"),&OptionButton::add_separator);
 | 
						|
	ObjectTypeDB::bind_method(_MD("clear"),&OptionButton::clear);
 | 
						|
	ObjectTypeDB::bind_method(_MD("select","idx"),&OptionButton::select);
 | 
						|
	ObjectTypeDB::bind_method(_MD("get_selected"),&OptionButton::get_selected);
 | 
						|
	ObjectTypeDB::bind_method(_MD("get_selected_ID"),&OptionButton::get_selected_ID);
 | 
						|
	ObjectTypeDB::bind_method(_MD("get_selected_metadata"),&OptionButton::get_selected_metadata);
 | 
						|
	ObjectTypeDB::bind_method(_MD("remove_item","idx"),&OptionButton::remove_item);
 | 
						|
	ObjectTypeDB::bind_method(_MD("_select_int"),&OptionButton::_select_int);
 | 
						|
 | 
						|
	ObjectTypeDB::bind_method(_MD("_set_items"),&OptionButton::_set_items);
 | 
						|
	ObjectTypeDB::bind_method(_MD("_get_items"),&OptionButton::_get_items);
 | 
						|
 | 
						|
	ADD_PROPERTY( PropertyInfo(Variant::INT,"selected"), _SCS("_select_int"),_SCS("get_selected") );
 | 
						|
	ADD_PROPERTY( PropertyInfo(Variant::ARRAY,"items",PROPERTY_HINT_NONE,"",PROPERTY_USAGE_NOEDITOR), _SCS("_set_items"),_SCS("_get_items") );
 | 
						|
	ADD_SIGNAL( MethodInfo("item_selected", PropertyInfo( Variant::INT,"ID") ) );
 | 
						|
}
 | 
						|
 | 
						|
OptionButton::OptionButton() {
 | 
						|
 | 
						|
 | 
						|
	popup = memnew( PopupMenu );
 | 
						|
	popup->hide();
 | 
						|
	popup->set_as_toplevel(true);
 | 
						|
	add_child(popup);
 | 
						|
	popup->connect("item_pressed", this,"_selected");
 | 
						|
 | 
						|
	current=-1;
 | 
						|
	set_text_align(ALIGN_LEFT);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
OptionButton::~OptionButton() {
 | 
						|
 | 
						|
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
 |