godot/scene/gui/button_group.cpp

/*************************************************************************/
/*  button_group.cpp                                                     */
/*************************************************************************/
/*                       This file is part of:                           */
/*                           GODOT ENGINE                                */
/*                      https://godotengine.org                          */
/*************************************************************************/
/* Copyright (c) 2007-2018 Juan Linietsky, Ariel Manzur.                 */
/* Copyright (c) 2014-2018 Godot Engine contributors (cf. AUTHORS.md)    */
/*                                                                       */
/* 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 "button_group.h"
#include "base_button.h"

void ButtonGroup::_add_button(BaseButton *p_button) {

	buttons.insert(p_button);
	p_button->set_toggle_mode(true);
	p_button->set_click_on_press(true);
	p_button->connect("pressed", this, "_pressed", make_binds(p_button));
}

void ButtonGroup::_remove_button(BaseButton *p_button) {

	buttons.erase(p_button);
	p_button->disconnect("pressed", this, "_pressed");
}

void ButtonGroup::set_pressed_button(BaseButton *p_button) {

	_pressed(p_button);
}

void ButtonGroup::_pressed(Object *p_button) {

	ERR_FAIL_NULL(p_button);
	BaseButton *b = p_button->cast_to<BaseButton>();
	ERR_FAIL_COND(!b);

	for (Set<BaseButton *>::Element *E = buttons.front(); E; E = E->next()) {

		BaseButton *bb = E->get();
		bb->set_pressed(b == bb);
		if (b == bb) {
			emit_signal("button_selected", b);
		}
	}
}

Array ButtonGroup::_get_button_list() const {

	List<BaseButton *> b;
	get_button_list(&b);

	b.sort_custom<Node::Comparator>();

	Array arr;
	arr.resize(b.size());

	int idx = 0;

	for (List<BaseButton *>::Element *E = b.front(); E; E = E->next(), idx++) {

		arr[idx] = E->get();
	}

	return arr;
}

void ButtonGroup::get_button_list(List<BaseButton *> *p_buttons) const {

	for (Set<BaseButton *>::Element *E = buttons.front(); E; E = E->next()) {

		p_buttons->push_back(E->get());
	}
}

BaseButton *ButtonGroup::get_pressed_button() const {

	for (Set<BaseButton *>::Element *E = buttons.front(); E; E = E->next()) {

		if (E->get()->is_pressed())
			return E->get();
	}

	return NULL;
}

BaseButton *ButtonGroup::get_focused_button() const {

	for (Set<BaseButton *>::Element *E = buttons.front(); E; E = E->next()) {

		if (E->get()->has_focus())
			return E->get();
	}

	return NULL;
}

int ButtonGroup::get_pressed_button_index() const {
	//in tree order, this is bizarre

	ERR_FAIL_COND_V(!is_inside_tree(), 0);

	BaseButton *pressed = get_pressed_button();
	if (!pressed)
		return -1;

	List<BaseButton *> blist;
	for (Set<BaseButton *>::Element *E = buttons.front(); E; E = E->next()) {

		blist.push_back(E->get());
	}

	blist.sort_custom<Node::Comparator>();

	int idx = 0;
	for (List<BaseButton *>::Element *E = blist.front(); E; E = E->next()) {

		if (E->get() == pressed)
			return idx;

		idx++;
	}

	return -1;
}

void ButtonGroup::_bind_methods() {

	ObjectTypeDB::bind_method(_MD("get_pressed_button:BaseButton"), &ButtonGroup::get_pressed_button);
	ObjectTypeDB::bind_method(_MD("get_pressed_button_index"), &ButtonGroup::get_pressed_button_index);
	ObjectTypeDB::bind_method(_MD("get_focused_button:BaseButton"), &ButtonGroup::get_focused_button);
	ObjectTypeDB::bind_method(_MD("get_button_list"), &ButtonGroup::_get_button_list);
	ObjectTypeDB::bind_method(_MD("_pressed"), &ButtonGroup::_pressed);
	ObjectTypeDB::bind_method(_MD("set_pressed_button", "button:BaseButton"), &ButtonGroup::_pressed);

	ADD_SIGNAL(MethodInfo("button_selected", PropertyInfo(Variant::OBJECT, "button", PROPERTY_HINT_RESOURCE_TYPE, "BaseButton")));
}

ButtonGroup::ButtonGroup() :
		BoxContainer(true) {
}